PSForever/PSF-BotServer
1
0
Fork 0
mirror of https://github.com/2revoemag/PSF-BotServer.git synced 2026-02-08 03:11:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Validate usernames on account creation like PSFPortal

Browse source
This commit is contained in:
Chord 2020-05-09 20:13:26 +02:00
parent ceb58ed39a
commit 52de77a735
1 changed files with 44 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

77
pslogin/src/main/scala/LoginSessionActor.scala
View file

@ -34,6 +34,8 @@ class LoginSessionActor extends Actor with MDCContextAware {
import scala.concurrent.ExecutionContext.Implicits.global
private case class UpdateServerList()
val usernameRegex = """[A-Za-z0-9]{3,}""".r
var sessionId : Long = 0
var leftRef : ActorRef = ActorRef.noSender
var rightRef : ActorRef = ActorRef.noSender
@ -212,46 +214,55 @@ class LoginSessionActor extends Actor with MDCContextAware {
def createNewAccount : Receive = {
case CreateNewAccount(connection, username, password, newToken) =>
log.info(s"Account $username does not exist, creating new account...")
val bcryptPassword : String = password.bcrypt(numBcryptPasses)
connection.get.inTransaction {
c => c.sendPreparedStatement(
"INSERT INTO accounts (username, passhash) VALUES(?,?) RETURNING id",
Array(username, bcryptPassword)
)
}.onComplete {
case Success(insertResult) =>
insertResult match {
case result: QueryResult =>
if (result.rows.nonEmpty) {
val accountId = result.rows.get.head(0).asInstanceOf[Int]
accountIntermediary ! StoreAccountData(newToken, new Account(accountId, username))
log.info(s"Successfully created new account for $username")
context.become(logTheLoginOccurrence)
self ! LogTheLoginOccurrence(connection, username, newToken, true, accountId)
} else {
log.error(s"No result from account create insert for $username")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
username match {
case usernameRegex(_*) => ;
val bcryptPassword : String = password.bcrypt(numBcryptPasses)
connection.get.inTransaction {
c => c.sendPreparedStatement(
"INSERT INTO accounts (username, passhash) VALUES(?,?) RETURNING id",
Array(username, bcryptPassword)
)
}.onComplete {
case Success(insertResult) =>
insertResult match {
case result: QueryResult =>
if (result.rows.nonEmpty) {
val accountId = result.rows.get.head(0).asInstanceOf[Int]
accountIntermediary ! StoreAccountData(newToken, new Account(accountId, username))
log.info(s"Successfully created new account for $username")
context.become(logTheLoginOccurrence)
self ! LogTheLoginOccurrence(connection, username, newToken, true, accountId)
} else {
log.error(s"No result from account create insert for $username")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
}
case default =>
log.error(s"Error creating new account for $username - $default")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
}
case default =>
log.error(s"Error creating new account for $username - $default")
case Failure(e : com.github.mauricio.async.db.postgresql.exceptions.GenericDatabaseException) =>
log.error(s"Error creating new account - ${e.errorMessage.message}")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
case Failure(e : java.sql.SQLException) =>
log.error(s"Error creating new account - ${e.getMessage}")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
case _ =>
failWithError(s"Something to do?")
}
case Failure(e : com.github.mauricio.async.db.postgresql.exceptions.GenericDatabaseException) =>
log.error(s"Error creating new account - ${e.errorMessage.message}")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
case Failure(e : java.sql.SQLException) =>
log.error(s"Error creating new account - ${e.getMessage}")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
case _ =>
failWithError(s"Something to do?")
log.warn(s"Account '$username' does not match regex")
context.become(finishAccountLogin)
self ! FinishAccountLogin(connection, username, newToken, false)
}
case default =>
failWithError(s"Invalid message '$default' received in createNewAccount")
}