MyFavMirrors/lc-core
1
0
Fork 0
mirror of https://github.com/librecaptcha/lc-core.git synced 2025-01-27 13:03:12 -05:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes

Browse Source 
Signed-off-by: Rahul Rudragoudar <rr83019@gmail.com>
This commit is contained in:
Rahul Rudragoudar 2019-06-20 11:45:29 +05:30
parent a63cf3976d
commit d65d050038
No known key found for this signature in database
GPG Key ID: EC2AAF721D545305
1 changed files with 17 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/scala/lc/Main.scala
View File

@ -182,6 +182,7 @@ class RateLimiter extends DBConn {
val allowance = rate val allowance = rate
def validateUser(user: Int) : Boolean = { def validateUser(user: Int) : Boolean = {
synchronized {
val allow = if(userLastActive.contains(user)){ val allow = if(userLastActive.contains(user)){
true true
} else { } else {
@ -199,6 +200,7 @@ class RateLimiter extends DBConn {
} }
allow allow
} }
}
def checkLimit(user: Int): Boolean = { def checkLimit(user: Int): Boolean = {
synchronized { synchronized {
@ -228,10 +230,9 @@ class Server(port: Int){
implicit val formats = DefaultFormats implicit val formats = DefaultFormats
host.addContext("/v1/captcha",(req, resp) => { host.addContext("/v1/captcha",(req, resp) => {
val accessToken = if(req.getHeaders().get("access-token") != null){ val accessToken = Option(req.getHeaders().get("access-token")).map(_.toInt)
req.getHeaders().get("access-token").toInt val access = accessToken.map(t => rateLimiter.validateUser(t) && rateLimiter.checkLimit(t)).getOrElse(false)
} else 0 val id = if(access){
val id = if(true == rateLimiter.validateUser(accessToken) && true == rateLimiter.checkLimit(accessToken)){
val body = req.getJson() val body = req.getJson()
val json = parse(body) val json = parse(body)
val param = json.extract[Parameters] val param = json.extract[Parameters]