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
|
/** @file
A brief file description
@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 "Protocol.h"
#include "TxnSM.h"
#include "tscore/ink_defs.h"
#include <math.h>
/* global variable */
TSTextLogObject protocol_plugin_log;
/* static variable */
static TSAction pending_action;
static int accept_port;
static int server_port;
/* Functions only seen in this file, should be static. */
static void protocol_init(int accept_port, int server_port);
static int accept_handler(TSCont contp, TSEvent event, void *edata);
/* When the handle is called, the net_vc is returned. */
static int
accept_handler(TSCont contp, TSEvent event, void *edata)
{
TSCont txn_sm;
TSMutex pmutex;
switch (event) {
case TS_EVENT_NET_ACCEPT:
/* Create a new mutex for the TxnSM, which is going
to handle the incoming request. */
pmutex = TSMutexCreate();
txn_sm = TxnSMCreate(pmutex, (TSVConn)edata, server_port);
/* This is no reason for not grabbing the lock.
So skip the routine which handle LockTry failure case. */
TSMutexLockTry(pmutex); // TODO: why should it not check if we got the lock??
TSContCall(txn_sm, 0, NULL);
TSMutexUnlock(pmutex);
break;
default:
/* Something wrong with the network, if there are any
pending NetAccept, cancel them. */
if (pending_action && !TSActionDone(pending_action)) {
TSActionCancel(pending_action);
}
TSContDestroy(contp);
break;
}
return TS_EVENT_NONE;
}
static void
protocol_init(int accept_port, int server_port ATS_UNUSED)
{
TSCont contp;
int ret_val;
/* create customized log */
ret_val = TSTextLogObjectCreate(PLUGIN_NAME, TS_LOG_MODE_ADD_TIMESTAMP, &protocol_plugin_log);
if (ret_val != TS_SUCCESS) {
TSError("[%s] Failed to create log", PLUGIN_NAME);
}
/* format of the log entries, for caching_status, 1 for HIT and 0 for MISS */
ret_val = TSTextLogObjectWrite(protocol_plugin_log, "timestamp filename servername caching_status\n\n");
if (ret_val != TS_SUCCESS) {
TSError("[%s] Failed to write into log", PLUGIN_NAME);
}
contp = TSContCreate(accept_handler, TSMutexCreate());
/* Accept network traffic from the accept_port.
When there are requests coming in, contp's handler
should be called, in this case, contp's handler
is accept_event, see AcceptSM.c */
pending_action = TSNetAccept(contp, accept_port, -1, 1);
}
void
TSPluginInit(int argc, const char *argv[])
{
TSPluginRegistrationInfo info;
char *end;
int tmp;
info.plugin_name = PLUGIN_NAME;
info.vendor_name = "Apache Software Foundation";
info.support_email = "dev@trafficserver.apache.org";
if (TSPluginRegister(&info) != TS_SUCCESS) {
TSError("[%s] Plugin registration failed", PLUGIN_NAME);
goto error;
}
/* default value */
accept_port = 4666;
server_port = 4666;
if (argc < 3) {
TSDebug(PLUGIN_NAME, "Usage: protocol.so <accept_port> <server_port>. Using default ports accept=%d server=%d", accept_port,
server_port);
} else {
tmp = strtol(argv[1], &end, 10);
if (*end == '\0') {
accept_port = tmp;
TSDebug(PLUGIN_NAME, "using accept_port %d", accept_port);
} else {
TSError("[%s] Wrong argument for accept_port, using default port %d", PLUGIN_NAME, accept_port);
}
tmp = strtol(argv[2], &end, 10);
if (*end == '\0') {
server_port = tmp;
TSDebug(PLUGIN_NAME, "using server_port %d", server_port);
} else {
TSError("[%s] Wrong argument for server_port, using default port %d", PLUGIN_NAME, server_port);
}
}
protocol_init(accept_port, server_port);
error:
TSError("[%s] Plugin not initialized", PLUGIN_NAME);
}
|
