aboutsummaryrefslogtreecommitdiff
path: root/core/common/src/main/kotlin/trebuchet/importers/ftrace/events/EventParserState.kt
blob: 7e9d14147d174a83f1c94a504cc74ab9afc2525b (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
108
109
110
111
112
113
114
115
116
/*
 * Copyright 2018 Google Inc.
 *
 * 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
 *
 *     https://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.
 */

package trebuchet.importers.ftrace.events

import trebuchet.importers.ftrace.FtraceImporterState
import trebuchet.importers.ftrace.FtraceLine
import trebuchet.io.DataSlice
import trebuchet.model.InvalidId
import trebuchet.util.BufferReader
import trebuchet.util.MatchResult
import trebuchet.util.StringCache
import java.util.regex.Matcher
import java.util.regex.Pattern

const val FtraceLineRE = """^ *(.{1,16})-(\d+) +(?:\( *(\d+)?-*\) )?\[(\d+)] (?:[dX.][^\s]+)? *([\d.]*): *([^:]*): *(.*) *$"""

interface FtraceEventDetails {
    fun import(event: FtraceEvent, state: FtraceImporterState)
}

val NoDetails = object : FtraceEventDetails {
    override fun import(event: FtraceEvent, state: FtraceImporterState) {}
}

typealias EventDetailsParser = (state: EventParserState, slice: DataSlice) -> FtraceEventDetails?

class SharedEventParserState {
    private val patterns = mutableListOf<Pattern>()
    private val detailHandlers = hashMapOf<String, EventDetailsParser>()
    private var frozen = false

    init {
        EventRegistry.forEach { it(this) }
    }

    fun addPattern(newPattern: String) = addPattern(Pattern.compile(newPattern))

    fun addPattern(newPattern: Pattern): Int {
        if (frozen) throw IllegalStateException("State is frozen")
        val index = patterns.size
        patterns.add(newPattern)
        return index
    }

    fun createParser(): EventParserState {
        frozen = true
        return EventParserState(patterns, detailHandlers)
    }

    fun onParseDetails(funcName: String, parser: EventDetailsParser) {
        if (frozen) throw IllegalStateException("State is frozen")
        detailHandlers[funcName] = parser
    }

    inline fun onParseDetailsWithMatch(funcName: String, regex: String,
                                       crossinline result: MatchResult.() -> FtraceEventDetails?) {
        val patternMatcher = addPattern(regex)
        onParseDetails(funcName) { state, details ->
            state.ifMatches(patternMatcher, details) {
                return@onParseDetails this.result()
            }
            null
        }
    }

    inline fun onParseDetailsWithMatch(funcNames: Array<String>, regex: String,
                                       crossinline result: MatchResult.() -> FtraceEventDetails?) {
        val patternMatcher = addPattern(regex)
        funcNames.forEach { funcName ->
            onParseDetails(funcName) { state, details ->
                state.ifMatches(patternMatcher, details) {
                    return@onParseDetails this.result()
                }
                null
            }
        }
    }
}

val GlobalSharedParserState = SharedEventParserState()

fun createEventParser() = GlobalSharedParserState.createParser()

class EventParserState constructor(patterns: List<Pattern>,
                                   private val detailHandlers: Map<String, EventDetailsParser>) {
    val matchers = Array<Matcher>(patterns.size) { patterns[it].matcher("") }
    val stringCache = StringCache()
    val reader = BufferReader()

    inline fun ifMatches(matcher: Int, slice: DataSlice, result: MatchResult.() -> Unit): Boolean {
        reader.reset(slice, stringCache)
        return reader.tryMatch(matchers[matcher], result)
    }

    fun detailsForText(funcName: CharSequence, detailsSlice: DataSlice): FtraceEventDetails {
        return detailHandlers[funcName]?.invoke(this, detailsSlice) ?: NoDetails
    }

    inline fun <T> readDetails(detailsSlice: DataSlice, init: BufferReader.() -> T): T {
        return reader.read(detailsSlice, stringCache, init)
    }
}