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

Add method to choose random captcha

Browse Source
This commit is contained in:
Rahul Rudragoudar 2019-03-26 18:44:31 +05:30
parent 9d070afa3d
commit 8e99f2713d
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -26,11 +26,18 @@ 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,
"ShadowTextCaptcha" -> new ShadowTextCaptcha "ShadowTextCaptcha" -> new ShadowTextCaptcha)
)
def getProvider(): String = {
val random = new scala.util.Random
val keys = filters.keys
val providerMap = keys.toVector(random.nextInt(keys.size))
providerMap
}
def getCaptcha(id: Id): Array[Byte] = { def getCaptcha(id: Id): Array[Byte] = {
imagePstmt.setString(1, id.id) imagePstmt.setString(1, id.id)
@ -45,7 +52,7 @@ class Captcha {
def getChallenge(param: Parameters): Id = { def getChallenge(param: Parameters): Id = {
//TODO: eval params to choose a provider //TODO: eval params to choose a provider
val providerMap = "GifCaptcha" val providerMap = getProvider()
val provider = filters(providerMap) val provider = filters(providerMap)
val challenge = provider.returnChallenge() val challenge = provider.returnChallenge()
val blob = new ByteArrayInputStream(challenge.content) val blob = new ByteArrayInputStream(challenge.content)
@ -63,7 +70,7 @@ class Captcha {
} }
val task = new Runnable { val task = new Runnable {
val providerMap = "FilterChallenge" val providerMap = getProvider()
val provider = filters(providerMap) val provider = filters(providerMap)
def run(): Unit = { def run(): Unit = {
val challenge = provider.returnChallenge() val challenge = provider.returnChallenge()