aboutsummaryrefslogtreecommitdiff
path: root/agent/src/main/java/com/code_intelligence/jazzer/utils/Utils.kt
blob: 1b399baf1fbf9eebbfcf02205187d70e667d0986 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
// Copyright 2021 Code Intelligence GmbH
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
@file:JvmName("Utils")

package com.code_intelligence.jazzer.utils

import java.lang.reflect.Executable
import java.lang.reflect.Method
import java.nio.ByteBuffer
import java.nio.channels.FileChannel

val Class<*>.descriptor: String
    get() = when {
        isPrimitive -> {
            when (this) {
                Boolean::class.javaPrimitiveType -> "Z"
                Byte::class.javaPrimitiveType -> "B"
                Char::class.javaPrimitiveType -> "C"
                Short::class.javaPrimitiveType -> "S"
                Int::class.javaPrimitiveType -> "I"
                Long::class.javaPrimitiveType -> "J"
                Float::class.javaPrimitiveType -> "F"
                Double::class.javaPrimitiveType -> "D"
                java.lang.Void::class.javaPrimitiveType -> "V"
                else -> throw IllegalStateException("Unknown primitive type: $name")
            }
        }
        isArray -> "[${componentType.descriptor}"
        java.lang.Object::class.java.isAssignableFrom(this) -> "L${name.replace('.', '/')};"
        else -> throw IllegalArgumentException("Unknown class type: $name")
    }

val Class<*>.readableDescriptor: String
    get() = when {
        isPrimitive -> {
            when (this) {
                Boolean::class.javaPrimitiveType -> "boolean"
                Byte::class.javaPrimitiveType -> "byte"
                Char::class.javaPrimitiveType -> "char"
                Short::class.javaPrimitiveType -> "short"
                Int::class.javaPrimitiveType -> "int"
                Long::class.javaPrimitiveType -> "long"
                Float::class.javaPrimitiveType -> "float"
                Double::class.javaPrimitiveType -> "double"
                java.lang.Void::class.javaPrimitiveType -> "void"
                else -> throw IllegalStateException("Unknown primitive type: $name")
            }
        }
        isArray -> "${componentType.readableDescriptor}[]"
        java.lang.Object::class.java.isAssignableFrom(this) -> name
        else -> throw IllegalArgumentException("Unknown class type: $name")
    }

val Executable.descriptor: String
    get() = parameterTypes.joinToString(separator = "", prefix = "(", postfix = ")") { parameterType ->
        parameterType.descriptor
    } + if (this is Method) returnType.descriptor else "V"

// This does not include the return type as the parameter descriptors already uniquely identify the executable.
val Executable.readableDescriptor: String
    get() = parameterTypes.joinToString(separator = ",", prefix = "(", postfix = ")") { parameterType ->
        parameterType.readableDescriptor
    }

fun simpleFastHash(vararg strings: String): Int {
    var hash = 0
    for (string in strings) {
        for (c in string) {
            hash = hash * 11 + c.code
        }
    }
    return hash
}

/**
 * Reads the [FileChannel] to the end as a UTF-8 string.
 */
fun FileChannel.readFully(): String {
    check(size() <= Int.MAX_VALUE)
    val buffer = ByteBuffer.allocate(size().toInt())
    while (buffer.hasRemaining()) {
        when (read(buffer)) {
            0 -> throw IllegalStateException("No bytes read")
            -1 -> break
        }
    }
    return String(buffer.array())
}

/**
 * Appends [string] to the end of the [FileChannel].
 */
fun FileChannel.append(string: String) {
    position(size())
    write(ByteBuffer.wrap(string.toByteArray()))
}