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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
|
/** @file
an example plugin showing off how to use versioning
@section license License
Licensed to the Apache Software Foundation (ASF) under one
or more contributor license agreements. See the NOTICE file
distributed with this work for additional information
regarding copyright ownership. The ASF licenses this file
to you 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.
*/
#include <ts/ts.h>
#define PLUGIN_NAME "hook-trace"
template <typename T, unsigned N>
static inline unsigned
countof(const T (&)[N])
{
return N;
}
static int
HttpHookTracer(TSCont contp, TSEvent event, void *edata)
{
union {
TSHttpTxn txn;
TSHttpSsn ssn;
TSHttpAltInfo alt;
void *ptr;
} ev;
ev.ptr = edata;
switch (event) {
case TS_EVENT_HTTP_SSN_START:
TSDebug(PLUGIN_NAME, "Received SSN_START on session %p", ev.ssn);
TSHttpSsnReenable(ev.ssn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_SSN_CLOSE:
TSDebug(PLUGIN_NAME, "Received SSN_CLOSE on session %p", ev.ssn);
TSHttpSsnReenable(ev.ssn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_SELECT_ALT:
TSDebug(PLUGIN_NAME, "Received SELECT_ALT on altinfo %p", ev.alt);
break;
case TS_EVENT_HTTP_READ_REQUEST_HDR:
TSDebug(PLUGIN_NAME, "Received READ_REQUEST_HDR on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_OS_DNS:
TSDebug(PLUGIN_NAME, "Received OS_DNS on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_SEND_REQUEST_HDR:
TSDebug(PLUGIN_NAME, "Received SEND_REQUEST_HDR on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_READ_CACHE_HDR:
TSDebug(PLUGIN_NAME, "Received READ_CACHE_HDR on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_READ_RESPONSE_HDR:
TSDebug(PLUGIN_NAME, "Received READ_RESPONSE_HDR on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_SEND_RESPONSE_HDR:
TSDebug(PLUGIN_NAME, "Received SEND_RESPONSE_HDR on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_TXN_START:
TSDebug(PLUGIN_NAME, "Received TXN_START on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_TXN_CLOSE:
TSDebug(PLUGIN_NAME, "Received TXN_CLOSE on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_CACHE_LOOKUP_COMPLETE:
TSDebug(PLUGIN_NAME, "Received CACHE_LOOKUP_COMPLETE on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_PRE_REMAP:
TSDebug(PLUGIN_NAME, "Received PRE_REMAP on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
case TS_EVENT_HTTP_POST_REMAP:
TSDebug(PLUGIN_NAME, "Received POST_REMAP on transaction %p", ev.txn);
TSHttpTxnReenable(ev.txn, TS_EVENT_HTTP_CONTINUE);
break;
default:
break;
}
return TS_EVENT_NONE;
}
static int
LifecycleHookTracer(TSCont contp, TSEvent event, void *edata)
{
switch (event) {
case TS_EVENT_LIFECYCLE_PORTS_INITIALIZED:
TSDebug(PLUGIN_NAME, "Received LIFECYCLE_PORTS_INITIALIZED data %p", edata);
break;
case TS_EVENT_LIFECYCLE_PORTS_READY:
TSDebug(PLUGIN_NAME, "Received LIFECYCLE_PORTS_READY data %p", edata);
break;
case TS_EVENT_LIFECYCLE_CACHE_READY:
TSDebug(PLUGIN_NAME, "Received LIFECYCLE_CACHE_READY data %p", edata);
break;
case TS_EVENT_LIFECYCLE_SERVER_SSL_CTX_INITIALIZED:
TSDebug(PLUGIN_NAME, "Received LIFECYCLE_SERVER_SSL_INITIALIZED data %p", edata);
break;
case TS_EVENT_LIFECYCLE_CLIENT_SSL_CTX_INITIALIZED:
TSDebug(PLUGIN_NAME, "Received LIFECYCLE_CLIENT_SSL_CTX_INITIALIZED data %p", edata);
break;
case TS_EVENT_LIFECYCLE_MSG:
TSDebug(PLUGIN_NAME, "Received LIFECYCLE_MSG data %p", edata);
break;
default:
TSDebug(PLUGIN_NAME, "Received unsupported lifecycle event %d data %p", event, edata);
break;
}
return TS_EVENT_NONE;
}
void
TSPluginInit(int argc, const char *argv[])
{
// clang-format off
static const TSHttpHookID http[] = {
TS_HTTP_READ_REQUEST_HDR_HOOK,
TS_HTTP_OS_DNS_HOOK,
TS_HTTP_SEND_REQUEST_HDR_HOOK,
TS_HTTP_READ_CACHE_HDR_HOOK,
TS_HTTP_READ_RESPONSE_HDR_HOOK,
TS_HTTP_SEND_RESPONSE_HDR_HOOK,
TS_HTTP_SELECT_ALT_HOOK,
TS_HTTP_TXN_START_HOOK,
TS_HTTP_TXN_CLOSE_HOOK,
TS_HTTP_SSN_START_HOOK,
TS_HTTP_SSN_CLOSE_HOOK,
TS_HTTP_CACHE_LOOKUP_COMPLETE_HOOK,
TS_HTTP_PRE_REMAP_HOOK,
TS_HTTP_POST_REMAP_HOOK,
};
static const TSLifecycleHookID lifecycle[] = {
TS_LIFECYCLE_PORTS_INITIALIZED_HOOK,
TS_LIFECYCLE_PORTS_READY_HOOK,
TS_LIFECYCLE_CACHE_READY_HOOK,
TS_LIFECYCLE_SERVER_SSL_CTX_INITIALIZED_HOOK,
TS_LIFECYCLE_CLIENT_SSL_CTX_INITIALIZED_HOOK,
TS_LIFECYCLE_MSG_HOOK,
};
// clang-format on
(void)argc; // unused
(void)argv; // unused
TSPluginRegistrationInfo info;
info.plugin_name = (char *)PLUGIN_NAME;
info.vendor_name = (char *)"Apache Software Foundation";
info.support_email = (char *)"dev@trafficserver.apache.org";
for (unsigned i = 0; i < countof(http); ++i) {
TSHttpHookAdd(http[i], TSContCreate(HttpHookTracer, TSMutexCreate()));
}
for (unsigned i = 0; i < countof(lifecycle); ++i) {
TSLifecycleHookAdd(lifecycle[i], TSContCreate(LifecycleHookTracer, TSMutexCreate()));
}
TSReleaseAssert(TSPluginRegister(&info) == TS_SUCCESS);
}
|
