Initial version

Author: bruunand

.gitignore

@@ -0,0 +1,2 @@
+/target
+/.idea

pom.xml

@@ -0,0 +1,55 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd">
+
+    <modelVersion>4.0.0</modelVersion>
+
+    <groupId>dk.anderslangballe</groupId>
+    <artifactId>yuge</artifactId>
+    <version>1.0-SNAPSHOT</version>
+    <packaging>jar</packaging>
+
+    <name>yuge</name>
+
+    <properties>
+        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+        <kotlin.version>1.3.72</kotlin.version>
+        <kotlin.code.style>official</kotlin.code.style>
+        <junit.version>4.12</junit.version>
+    </properties>
+
+    <dependencies>
+        <dependency>
+            <groupId>org.jetbrains.kotlin</groupId>
+            <artifactId>kotlin-stdlib</artifactId>
+            <version>${kotlin.version}</version>
+        </dependency>
+
+        <dependency>
+            <groupId>org.jetbrains.kotlin</groupId>
+            <artifactId>kotlin-test-junit</artifactId>
+            <version>${kotlin.version}</version>
+            <scope>test</scope>
+        </dependency>
+    </dependencies>
+
+    <build>
+        <sourceDirectory>src/main/kotlin</sourceDirectory>
+
+        <plugins>
+            <plugin>
+                <groupId>org.jetbrains.kotlin</groupId>
+                <artifactId>kotlin-maven-plugin</artifactId>
+                <version>${kotlin.version}</version>
+                <executions>
+                    <execution>
+                        <id>compile</id>
+                        <phase>compile</phase>
+                        <goals>
+                            <goal>compile</goal>
+                        </goals>
+                    </execution>
+                </executions>
+            </plugin>
+        </plugins>
+    </build>
+</project>

samples/sample.request

@@ -0,0 +1,4 @@
+GET /test HTTP/1.1
+Accept-Language: en-us
+Accept-Encoding: gzip, deflate
+Connection: Keep-Alive

src/main/kotlin/dk/anderslangballe/Main.kt

@@ -0,0 +1,30 @@
+package dk.anderslangballe
+
+import dk.anderslangballe.attack.HttpAttack
+import dk.anderslangballe.utility.getSamplePath
+import java.io.File
+import kotlin.system.exitProcess
+
+fun main() {
+    val targetHost = System.getenv("TARGET_HOST")
+    val targetPort = System.getenv("TARGET_PORT")
+    val requestFile = System.getenv("REQUEST_FILE")
+
+    if (targetHost == null || targetPort == null) {
+        println("Target host and port must be declared")
+
+        exitProcess(1)
+    }
+
+    if (requestFile == null || !File(getSamplePath(requestFile)).exists()) {
+        println("Request file must be declared and exist")
+
+        exitProcess(1)
+    }
+
+    val attack = HttpAttack(
+        targetHost, Integer.parseInt(targetPort),
+        getSamplePath(requestFile)
+    )
+    attack.start()
+}

src/main/kotlin/dk/anderslangballe/attack/HttpAttack.kt

@@ -0,0 +1,37 @@
+package dk.anderslangballe.attack
+
+import java.io.File
+import java.util.concurrent.CountDownLatch
+import kotlin.concurrent.thread
+
+class HttpAttack(private val targetHost: String, private val targetPort: Int, private val sampleFile: String) {
+    private val numberOfRequests = 1000
+
+    private val launchLatch = CountDownLatch(numberOfRequests)
+    private val completeLatch = CountDownLatch(numberOfRequests)
+
+    private fun getRequestContent(): String {
+        val content = File(sampleFile).readText(Charsets.UTF_8)
+
+        // Ensure connection stays alive
+        return content.replace("Connection: Close", "Connection: Keep-Alive")
+    }
+
+    fun start() {
+        val requestContent = getRequestContent()
+
+        for (i in 1..numberOfRequests) {
+            thread {
+                HttpRequest(targetHost, targetPort, false, launchLatch, completeLatch, requestContent).start()
+            }
+        }
+
+        launchLatch.await()
+
+        val startTime = System.currentTimeMillis()
+
+        completeLatch.await()
+
+        println("RPS: ${numberOfRequests / ((System.currentTimeMillis() - startTime) / 1000F)}")
+    }
+}

src/main/kotlin/dk/anderslangballe/attack/HttpRequest.kt

@@ -0,0 +1,52 @@
+package dk.anderslangballe.attack
+
+import dk.anderslangballe.ssl.getSecureSocketFactory
+import java.io.OutputStream
+import java.net.Socket
+import java.util.concurrent.CountDownLatch
+
+class HttpRequest(private val targetHost: String, private val targetPort: Int, private val secureConnection: Boolean,
+                  private val launchLatch: CountDownLatch, private val completeLatch: CountDownLatch,
+                  private val requestContent: String) {
+    private val withheldBytes = 1
+
+    fun start() {
+        val socket: Socket = getSocket()
+        val outputStream = socket.getOutputStream()
+        val requestBytes = getRequestBytes()
+
+        writeInitial(requestBytes, outputStream)
+
+        launchLatch.await()
+
+        writeFinal(requestBytes, outputStream)
+    }
+
+    private fun getRequestBytes(): ByteArray = requestContent.toByteArray(Charsets.UTF_8)
+
+    private fun writeFinal(requestBytes: ByteArray, outputStream: OutputStream) {
+        outputStream.write(requestBytes, requestBytes.size - 1, withheldBytes)
+        outputStream.close()
+
+        completeLatch.countDown()
+    }
+
+    private fun writeInitial(requestBytes: ByteArray, outputStream: OutputStream) {
+        outputStream.write(requestBytes, 0, requestBytes.size - withheldBytes)
+        outputStream.flush()
+
+        launchLatch.countDown()
+    }
+
+    private fun getSocket(): Socket {
+        val socket: Socket = if (secureConnection) {
+            getSecureSocketFactory().createSocket(targetHost, targetPort)
+        } else {
+            Socket(targetHost, targetPort)
+        }
+
+        socket.tcpNoDelay = true
+
+        return socket
+    }
+}

src/main/kotlin/dk/anderslangballe/ssl/TrustManager.kt

@@ -0,0 +1,23 @@
+package dk.anderslangballe.ssl
+
+import java.security.cert.X509Certificate
+import javax.net.ssl.SSLContext
+import javax.net.ssl.SSLSocketFactory
+import javax.net.ssl.X509TrustManager
+
+private class TrustManager : X509TrustManager {
+    override fun getAcceptedIssuers(): Array<X509Certificate>? {
+        return null
+    }
+
+    override fun checkClientTrusted(chain: Array<X509Certificate>, authType: String) {}
+
+    override fun checkServerTrusted(chain: Array<X509Certificate>, authType: String) {}
+}
+
+fun getSecureSocketFactory(): SSLSocketFactory {
+    val context = SSLContext.getInstance("TLS")
+    context.init(null, arrayOf<javax.net.ssl.TrustManager>(TrustManager()), null)
+
+    return context.socketFactory
+}

src/main/kotlin/dk/anderslangballe/utility/PathUtility.kt

@@ -0,0 +1,7 @@
+package dk.anderslangballe.utility
+
+import java.nio.file.Paths
+
+fun getSamplePath(sampleName: String): String {
+    return Paths.get("samples", sampleName).toString()
+}