MyFavMirrors/lc-core
1
0
Fork 0
mirror of https://github.com/librecaptcha/lc-core.git synced 2025-03-31 09:13:38 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add Captcha thread throttle

Browse Source 
Signed-off-by: Rahul Rudragoudar <rr83019@gmail.com>
This commit is contained in:
Rahul Rudragoudar 2019-04-06 11:30:06 +05:30
parent 4c5cdc5699
commit 37bf3b1d7f
No known key found for this signature in database
GPG Key ID: 5825856C50762765
1 changed files with 14 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -18,7 +18,7 @@ import org.json4s.jackson.Serialization.{read, write}
import java.util.concurrent._ import java.util.concurrent._
import scala.Array import scala.Array
class Captcha { class Captcha(throttle: Int) {
val con: Connection = DriverManager.getConnection("jdbc:h2:./captcha", "sa", "") val con: Connection = DriverManager.getConnection("jdbc:h2:./captcha", "sa", "")
val stmt: Statement = con.createStatement() val stmt: Statement = con.createStatement()
stmt.execute("CREATE TABLE IF NOT EXISTS challenge(token varchar, id varchar, secret varchar, provider varchar, contentType varchar, image blob)") stmt.execute("CREATE TABLE IF NOT EXISTS challenge(token varchar, id varchar, secret varchar, provider varchar, contentType varchar, image blob)")
@ -26,7 +26,6 @@ class Captcha {
val selectPstmt: PreparedStatement = con.prepareStatement("SELECT secret, provider FROM challenge WHERE token = ?") val selectPstmt: PreparedStatement = con.prepareStatement("SELECT secret, provider FROM challenge WHERE token = ?")
val imagePstmt: PreparedStatement = con.prepareStatement("SELECT image FROM challenge WHERE token = ?") val imagePstmt: PreparedStatement = con.prepareStatement("SELECT image FROM challenge WHERE token = ?")
val filters = Map("FilterChallenge" -> new FilterChallenge, val filters = Map("FilterChallenge" -> new FilterChallenge,
"FontFunCaptcha" -> new FontFunCaptcha, "FontFunCaptcha" -> new FontFunCaptcha,
"GifCaptcha" -> new GifCaptcha, "GifCaptcha" -> new GifCaptcha,
@ -70,26 +69,21 @@ class Captcha {
} }
val task = new Runnable { val task = new Runnable {
val providerMap = getProvider()
val provider = filters(providerMap)
def run(): Unit = { def run(): Unit = {
val challenge = provider.returnChallenge() val imageNum = stmt.executeQuery("SELECT COUNT(*) AS total FROM challenge")
val blob = new ByteArrayInputStream(challenge.content) var throttleIn = throttle
val token = scala.util.Random.nextInt(10000).toString if(imageNum.next())
val id = Id(token) throttleIn = (throttle-imageNum.getInt("total")) + ((10*throttle)/100).asInstanceOf[Int]
insertPstmt.setString(1, token) while(0 <= throttleIn-1){
insertPstmt.setString(2, provider.getId) getChallenge(Parameters("","","",Option(Size(0,0))))
insertPstmt.setString(3, challenge.secret) throttleIn -= 1
insertPstmt.setString(4, providerMap) }
insertPstmt.setString(5, challenge.contentType)
insertPstmt.setBlob(6, blob)
insertPstmt.executeUpdate()
} }
} }
def beginThread(delay: Int) : Unit = { def beginThread(delay: Int) : Unit = {
val ex = new ScheduledThreadPoolExecutor(1) val ex = new ScheduledThreadPoolExecutor(1)
val thread = ex.scheduleAtFixedRate(task, 1, delay, TimeUnit.SECONDS) val thread = ex.scheduleWithFixedDelay(task, 1, delay, TimeUnit.SECONDS)
} }
def getAnswer(answer: Answer): Boolean = { def getAnswer(answer: Answer): Boolean = {
@ -124,7 +118,7 @@ case class Id(id: String)
case class Answer(answer: String, id: String) case class Answer(answer: String, id: String)
class Server(port: Int){ class Server(port: Int){
val captcha = new Captcha() val captcha = new Captcha(0)
val server = new HTTPServer(port) val server = new HTTPServer(port)
val host = server.getVirtualHost(null) val host = server.getVirtualHost(null)
@ -175,11 +169,10 @@ class Server(port: Int){
object LCFramework{ object LCFramework{
def main(args: scala.Array[String]) { def main(args: scala.Array[String]) {
val captcha = new Captcha() val captcha = new Captcha(50)
val server = new Server(8888) val server = new Server(8888)
server.start() server.start()
//captcha.beginThread(2) captcha.beginThread(2)
} }
} }