MyFavMirrors/lc-core
1
0
Fork 0
mirror of https://github.com/librecaptcha/lc-core.git synced 2025-04-19 02:05:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Added validation http end point

Browse Source 
Added Object column in h2 db
This commit is contained in:
Rahul Rudragoudar 2018-12-17 16:10:10 +05:30
parent cafc855711
commit eb8c8d272e
2 changed files with 24 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/scala/lc/FilterChallenge.scala
View File

@ -6,7 +6,7 @@ import java.awt.image.BufferedImage
import java.awt.Font import java.awt.Font
import java.awt.Color import java.awt.Color
class FilterChallenge extends ChallengeProvider { class FilterChallenge extends ChallengeProvider with Serializable{
val id = "filter" val id = "filter"
def returnChallenge(): (Image, String) = { def returnChallenge(): (Image, String) = {
val filterTypes = List(new FilterType1, new FilterType2) val filterTypes = List(new FilterType1, new FilterType2)

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

@ -24,9 +24,9 @@ trait ChallengeProvider {
class Captcha { class Captcha {
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, image blob)") stmt.execute("CREATE TABLE IF NOT EXISTS challenge(token varchar, id varchar, secret varchar, provider other, image blob)")
val insertPstmt: PreparedStatement = con.prepareStatement("INSERT INTO challenge(token, id, secret, image) VALUES (?, ?, ?, ?)") val insertPstmt: PreparedStatement = con.prepareStatement("INSERT INTO challenge(token, id, secret, provider, image) VALUES (?, ?, ?, ?, ?)")
val selectPstmt: PreparedStatement = con.prepareStatement("SELECT secret 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 = ?")
def getCaptcha(id: Id): Array[Byte] = { def getCaptcha(id: Id): Array[Byte] = {
@ -59,17 +59,19 @@ class Captcha {
insertPstmt.setString(1, token) insertPstmt.setString(1, token)
insertPstmt.setString(2, provider.id) insertPstmt.setString(2, provider.id)
insertPstmt.setString(3, secret) insertPstmt.setString(3, secret)
insertPstmt.setBlob(4, blob) insertPstmt.setObject(4, provider)
insertPstmt.setBlob(5, blob)
insertPstmt.executeUpdate() insertPstmt.executeUpdate()
id id
} }
def getAnswer(token: String, answer: String, provider: ChallengeProvider): Boolean = { def getAnswer(answer: Answer): Boolean = {
selectPstmt.setString(1, token) selectPstmt.setString(1, answer.id)
val rs: ResultSet = selectPstmt.executeQuery() val rs: ResultSet = selectPstmt.executeQuery()
rs.next() rs.next()
val secret = rs.getString("secret") val secret = rs.getString("secret")
provider.checkAnswer(secret, answer) val provider = rs.getObject("provider").asInstanceOf[ChallengeProvider]
provider.checkAnswer(secret, answer.answer)
} }
def display(): Unit = { def display(): Unit = {
@ -92,6 +94,7 @@ class Captcha {
case class Size(height: Int, width: Int) case class Size(height: Int, width: Int)
case class Parameters(level: String, media: String, input_type: String, size: Option[Size]) case class Parameters(level: String, media: String, input_type: String, size: Option[Size])
case class Id(id: String) case class Id(id: String)
case class Answer(answer: String, id: String)
object LCFramework{ object LCFramework{
def main(args: scala.Array[String]) { def main(args: scala.Array[String]) {
@ -117,12 +120,23 @@ object LCFramework{
val json = parse(body) val json = parse(body)
val id = json.extract[Id] val id = json.extract[Id]
val image = captcha.getCaptcha(id) val image = captcha.getCaptcha(id)
println(image.toString())
println(image)
resp.getHeaders().add("Content-Type","image/png") resp.getHeaders().add("Content-Type","image/png")
resp.send(200, image) resp.send(200, image)
0 0
}) })
host.addContext("/v1/answer",(req, resp) =>{
val body = req.getJson()
val json = parse(body)
val answer = json.extract[Answer]
val result = captcha.getAnswer(answer)
resp.getHeaders().add("Content-Type","application/json")
if(result){
resp.send(200,"""{"result":"True"}""")
}
resp.send(200,"""{"result":"False"}""")
0
})
server.start() server.start()
} }
} }