Browse Source

Bail if message is too big

master
Aditya Kulkarni 5 years ago
parent
commit
7cd32fecac
  1. 1
      build.gradle
  2. 2
      src/main/kotlin/com/zecqtwallet/wormhole/Service.kt
  3. 7
      src/main/resources/log4j.properties

1
build.gradle

@ -18,6 +18,7 @@ dependencies {
compile "org.jetbrains.kotlin:kotlin-stdlib-jdk8"
compile 'io.javalin:javalin:2.6.0'
compile "org.slf4j:slf4j-log4j12:1.7.7"
compile "log4j:apache-log4j-extras:1.2.17"
compile 'com.beust:klaxon:5.0.1'
testCompile group: 'junit', name: 'junit', version: '4.12'
}

2
src/main/kotlin/com/zecqtwallet/wormhole/Service.kt

@ -35,6 +35,7 @@ fun main(args : Array<String>) {
// Limit message size to 50kb of hex encoded text
if (message.length > 2 * 50 * 1024) {
sendError(session, "Message too big")
return@onMessage
}
//println("Recieved $message")
@ -45,7 +46,6 @@ fun main(args : Array<String>) {
if (j.contains("ping")) {
// Ignore, this is a keep-alive ping
logInfo("Ping ${usermap[session]}", j)
// Just send the ping back
session.send(message)
return@onMessage

7
src/main/resources/log4j.properties

@ -1,9 +1,8 @@
log4j.rootLogger=INFO, rollingFile
log4j.appender.rollingFile=org.apache.log4j.RollingFileAppender
log4j.appender.rollingFile=org.apache.log4j.DailyRollingFileAppender
log4j.appender.rollingFile.File=application.log
log4j.appender.rollingFile.layout=org.apache.log4j.PatternLayout
log4j.appender.rollingFile.DatePattern='.'yyyy-MM-dd
log4j.appender.rollingFile.layout.ConversionPattern=%d{yyyy-MM-dd HH:mm:ss} %-5p %m%n
log4j.appender.rollingFile.MaxFileSize=10MB
log4j.appender.rollingFile.MaxBackupIndex=5
log4j.appender.rollingFile.append=true
log4j.appender.rollingFile.append=true

Loading…
Cancel
Save