skywalking.c 37.0 KB
Newer Older
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
/*
  +----------------------------------------------------------------------+
  | PHP Version 7                                                        |
  +----------------------------------------------------------------------+
  | Copyright (c) 1997-2017 The PHP Group                                |
  +----------------------------------------------------------------------+
  | This source file is subject to version 3.01 of the PHP license,      |
  | that is bundled with this package in the file LICENSE, and is        |
  | available through the world-wide-web at the following url:           |
  | http://www.php.net/license/3_01.txt                                  |
  | If you did not receive a copy of the PHP license and are unable to   |
  | obtain it through the world-wide-web, please send a note to          |
  | license@php.net so we can mail you a copy immediately.               |
  +----------------------------------------------------------------------+
  | Author:                                                              |
  +----------------------------------------------------------------------+
*/

/* $Id$ */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <sys/time.h>


#include "main/SAPI.h" /* for sapi_module */
#include "zend_smart_str.h" /* for smart_str */
#include "php.h"
#include "php_ini.h"
#include "ext/standard/info.h"
#include "ext/standard/php_string.h"
H
heyanlong 已提交
33
#include "src/components.h"
34 35
#include "php_skywalking.h"
#include "ext/standard/url.h" /* for php_url */
H
heyanlong 已提交
36
#include "ext/standard/php_var.h"
37 38 39 40 41 42 43 44

#include "ext/standard/basic_functions.h"
#include "ext/standard/php_math.h"
#include <string.h>
#include "ext/json/php_json.h"
#include "ext/date/php_date.h"
#include <curl/curl.h>
#include <curl/easy.h>
S
songzhian 已提交
45 46 47 48 49 50 51 52 53
#include <unistd.h>
#include <netdb.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <pthread.h>
#include <fcntl.h>
#include <dirent.h>

H
heyanlong 已提交
54
#include <sys/un.h>
55

H
heyanlong 已提交
56
#include "b64.h"
S
songzhian 已提交
57 58

/* If you declare any globals in php_skywalking.h uncomment this:
59 60 61 62 63
*/
ZEND_DECLARE_MODULE_GLOBALS(skywalking)

/* True global resources - no need for thread safety here */
static int le_skywalking;
H
heyanlong 已提交
64 65
static int application_instance = -100000;
static int application_id = -100000;
S
songzhian 已提交
66 67
static int sky_close = 0;
static int sky_increment_id = 0;
H
heyanlong 已提交
68
const char *sock_path = "/tmp/sky_agent.sock";
S
songzhian 已提交
69

70 71 72 73
/* {{{ PHP_INI
 */
/* Remove comments and fill if you need to have entries in php.ini*/
PHP_INI_BEGIN()
H
heyanlong 已提交
74
	STD_PHP_INI_BOOLEAN("skywalking.enable",   	"0", PHP_INI_ALL, OnUpdateBool, enable, zend_skywalking_globals, skywalking_globals)
H
heyanlong 已提交
75
	STD_PHP_INI_ENTRY("skywalking.version",   	"5", PHP_INI_ALL, OnUpdateLong, version, zend_skywalking_globals, skywalking_globals)
H
report  
heyanlong 已提交
76 77 78
	STD_PHP_INI_ENTRY("skywalking.app_code", "", PHP_INI_ALL, OnUpdateString, app_code, zend_skywalking_globals, skywalking_globals)
    STD_PHP_INI_ENTRY("skywalking.log_path", "/tmp", PHP_INI_ALL, OnUpdateString, log_path, zend_skywalking_globals, skywalking_globals)
    STD_PHP_INI_ENTRY("skywalking.grpc", "127.0.0.1:11800", PHP_INI_ALL, OnUpdateString, grpc, zend_skywalking_globals, skywalking_globals)
79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97
PHP_INI_END()

/* }}} */

/* {{{ skywalking_functions[]
 *
 * Every user visible function must have an entry in skywalking_functions[].
 */
const zend_function_entry skywalking_functions[] = {
	PHP_FE_END	/* Must be the last line in skywalking_functions[] */
};
/* }}} */



const zend_function_entry class_skywalking[] = {
	PHP_FE_END
};

S
songzhian 已提交
98 99 100



101 102 103 104 105 106
/* The previous line is meant for vim and emacs, so it can correctly fold and
   unfold functions in source code. See the corresponding marks just before
   function definition, where the functions purpose is also documented. Please
   follow this convention for the convenience of others editing your code.
*/

H
heyanlong 已提交
107
void sky_curl_exec_handler(INTERNAL_FUNCTION_PARAMETERS)
108 109 110 111 112 113
{
	zval		*zid;
	if (zend_parse_parameters(ZEND_NUM_ARGS(), "r", &zid) == FAILURE) {
		return;
	}

H
heyanlong 已提交
114
	int is_send = 1;
H
heyanlong 已提交
115

H
report  
heyanlong 已提交
116 117 118 119 120
    zval function_name,curlInfo;
    zval params[1];
    ZVAL_COPY(&params[0], zid);
    ZVAL_STRING(&function_name,  "curl_getinfo");
    call_user_function(CG(function_table), NULL, &function_name, &curlInfo, 1, params);
H
memory  
heyanlong 已提交
121 122 123
    zval_dtor(&function_name);
    zval_dtor(&params[0]);

H
report  
heyanlong 已提交
124
    zval *z_url = zend_hash_str_find(Z_ARRVAL(curlInfo),  ZEND_STRL("url"));
H
heyanlong 已提交
125
    char *url_str = Z_STRVAL_P(z_url);
H
heyanlong 已提交
126 127
    if(strlen(url_str) <= 0) {
        zval_dtor(&curlInfo);
H
heyanlong 已提交
128
        is_send = 0;
H
heyanlong 已提交
129
    }
H
heyanlong 已提交
130
    php_url *url_info = NULL;
H
heyanlong 已提交
131 132 133 134 135 136
    if(is_send == 1) {
        url_info = php_url_parse(url_str);
        if(url_info->scheme == NULL || url_info->host == NULL) {
            zval_dtor(&curlInfo);
            php_url_free(url_info);
            is_send = 0;
H
heyanlong 已提交
137 138 139
        }
    }

G
goerzh 已提交
140
    char *sw = NULL;
H
heyanlong 已提交
141 142 143
    zval *spans = NULL;
    zval *last_span = NULL;
    zval *span_id = NULL;
H
heyanlong 已提交
144
    char *peer = NULL;
H
heyanlong 已提交
145 146
    ssize_t operation_name_l = 0;
    char *operation_name = NULL;
H
heyanlong 已提交
147 148 149 150 151 152
    if (is_send == 1) {
        int peer_port = 0;
        if (url_info->port) {
            peer_port = url_info->port;
        } else {
            if (strcasecmp("http", url_info->scheme) == 0) {
H
heyanlong 已提交
153
                peer_port = 80;
H
heyanlong 已提交
154
            } else {
H
heyanlong 已提交
155
                peer_port = 443;
H
heyanlong 已提交
156 157
            }
        }
H
heyanlong 已提交
158

159 160 161
        peer = (char *) emalloc(strlen(url_info->scheme) + 3 + strlen(url_info->host) + 7);
        bzero(peer, strlen(strlen(url_info->scheme) + 3 + url_info->host) + 7);
        sprintf(peer, "%s://%s:%d", url_info->scheme, url_info->host, peer_port);
G
goerzh 已提交
162

G
goerzh 已提交
163 164 165 166 167 168 169
        if (url_info->query) {
            if (url_info->path == NULL) {
                operation_name_l = snprintf(NULL, 0, "%s?%s", "/", url_info->query);
                operation_name = (char *) emalloc(operation_name_l + 1);
                bzero(operation_name, operation_name_l + 1);
                sprintf(operation_name, "%s?%s", "/", url_info->query);
            } else {
G
goerzh 已提交
170
                operation_name_l = snprintf(NULL, 0, "%s?%s", url_info->path, url_info->query);
G
goerzh 已提交
171 172
                operation_name = (char *) emalloc(operation_name_l + 1);
                bzero(operation_name, operation_name_l + 1);
G
goerzh 已提交
173
                sprintf(operation_name, "%s?%s", url_info->path, url_info->query);
G
goerzh 已提交
174 175 176 177 178 179 180 181
            }
        } else {
            if (url_info->path == NULL) {
                operation_name_l = snprintf(NULL, 0, "%s", "/");
                operation_name = (char *) emalloc(operation_name_l + 1);
                bzero(operation_name, operation_name_l + 1);
                sprintf(operation_name, "%s", "/");
            } else {
G
goerzh 已提交
182
                operation_name_l = snprintf(NULL, 0, "%s", url_info->path);
G
goerzh 已提交
183 184
                operation_name = (char *) emalloc(operation_name_l + 1);
                bzero(operation_name, operation_name_l + 1);
G
goerzh 已提交
185
                sprintf(operation_name, "%s", url_info->path);
G
goerzh 已提交
186 187
            }
        }
G
goerzh 已提交
188

H
heyanlong 已提交
189 190 191
        spans = get_spans();
        last_span = zend_hash_index_find(Z_ARRVAL_P(spans), zend_hash_num_elements(Z_ARRVAL_P(spans)) - 1);
        span_id = zend_hash_str_find(Z_ARRVAL_P(last_span), "spanId", sizeof("spanId") - 1);
192 193 194 195 196
        if (SKYWALKING_G(version) == 5) { // skywalking 5.x
            sw = generate_sw3(Z_LVAL_P(span_id) + 1, peer, operation_name);
        } else if (SKYWALKING_G(version) == 6) { // skywalking 6.x
            // todo
            sw = generate_sw6(Z_LVAL_P(span_id) + 1, peer, NULL);
G
goerzh 已提交
197
        }
H
report  
heyanlong 已提交
198 199
    }

H
heyanlong 已提交
200

G
goerzh 已提交
201
    if (sw != NULL) {
H
heyanlong 已提交
202 203 204 205 206 207 208 209 210
        zval *option = NULL;
        int is_init = 0;
        option = zend_hash_index_find(Z_ARRVAL_P(&SKYWALKING_G(curl_header)), Z_RES_HANDLE_P(zid));

        if(option == NULL) {
            option = emalloc(sizeof(zval));
            bzero(option, sizeof(zval));
            array_init(option);
            is_init = 1;
H
heyanlong 已提交
211
        }
H
heyanlong 已提交
212

G
goerzh 已提交
213
        add_next_index_string(option, sw);
H
heyanlong 已提交
214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229
        add_index_bool(&SKYWALKING_G(curl_header_send), (zend_ulong)Z_RES_HANDLE_P(zid), IS_TRUE);

        zval func;
        zval argv[3];
        zval ret;
        ZVAL_STRING(&func, "curl_setopt");

        ZVAL_COPY(&argv[0], zid);
        ZVAL_LONG(&argv[1], CURLOPT_HTTPHEADER);
        ZVAL_COPY(&argv[2], option);
        call_user_function(CG(function_table), NULL, &func, &ret, 3, argv);
        zval_dtor(&ret);
        zval_dtor(&func);
        if(is_init == 1) {
            zval_ptr_dtor(option);
            efree(option);
H
heyanlong 已提交
230
        }
H
heyanlong 已提交
231 232 233
        zval_dtor(&argv[0]);
        zval_dtor(&argv[1]);
        zval_dtor(&argv[2]);
G
goerzh 已提交
234
        efree(sw);
H
report  
heyanlong 已提交
235
    }
H
heyanlong 已提交
236 237 238 239

    zval temp;
    char *l_millisecond;
    long millisecond;
H
heyanlong 已提交
240
    if(is_send == 1) {
H
heyanlong 已提交
241

H
heyanlong 已提交
242
        array_init(&temp);
H
heyanlong 已提交
243

H
heyanlong 已提交
244 245 246 247 248 249 250 251 252
        add_assoc_long(&temp, "spanId", Z_LVAL_P(span_id) + 1);
        add_assoc_long(&temp, "parentSpanId", 0);
        l_millisecond = get_millisecond();
        millisecond = zend_atol(l_millisecond, strlen(l_millisecond));
        efree(l_millisecond);
        add_assoc_long(&temp, "startTime", millisecond);
        add_assoc_long(&temp, "spanType", 1);
        add_assoc_long(&temp, "spanLayer", 3);
        add_assoc_long(&temp, "componentId", COMPONENT_HTTPCLIENT);
H
report  
heyanlong 已提交
253
    }
H
heyanlong 已提交
254 255


H
heyanlong 已提交
256
	orig_curl_exec(INTERNAL_FUNCTION_PARAM_PASSTHRU);
H
heyanlong 已提交
257

H
heyanlong 已提交
258 259 260 261 262 263 264 265
    if (is_send == 1) {
        zval function_name_1, curlInfo_1;
        zval params_1[1];
        ZVAL_COPY(&params_1[0], zid);
        ZVAL_STRING(&function_name_1, "curl_getinfo");
        call_user_function(CG(function_table), NULL, &function_name_1, &curlInfo_1, 1, params_1);
        zval_dtor(&params_1[0]);
        zval_dtor(&function_name_1);
H
heyanlong 已提交
266

H
heyanlong 已提交
267 268 269 270
        zval *z_http_code;
        l_millisecond = get_millisecond();
        millisecond = zend_atol(l_millisecond, strlen(l_millisecond));
        efree(l_millisecond);
H
heyanlong 已提交
271

H
heyanlong 已提交
272
        z_http_code = zend_hash_str_find(Z_ARRVAL(curlInfo_1), ZEND_STRL("http_code"));
H
heyanlong 已提交
273

H
heyanlong 已提交
274
        add_assoc_long(&temp, "endTime", millisecond);
H
heyanlong 已提交
275 276


G
goerzh 已提交
277
        add_assoc_string(&temp, "operationName", operation_name);
278
        add_assoc_string(&temp, "peer", peer);
H
heyanlong 已提交
279 280
        efree(peer);
        efree(operation_name);
H
heyanlong 已提交
281

H
heyanlong 已提交
282 283 284 285 286 287 288 289 290 291 292 293 294 295
        php_url_free(url_info);

        if (Z_LVAL_P(z_http_code) != 200) {
            add_assoc_long(&temp, "isError", 1);
        } else {
            add_assoc_long(&temp, "isError", 0);
        }
        zval _refs;
        array_init(&_refs);
        add_assoc_zval(&temp, "refs", &_refs);
        zend_hash_next_index_insert(Z_ARRVAL_P(spans), &temp);
        zval_dtor(&curlInfo_1);
        zval_dtor(&curlInfo);
    }
296
}
H
heyanlong 已提交
297 298 299 300 301 302 303 304 305

void sky_curl_setopt_handler(INTERNAL_FUNCTION_PARAMETERS) {
    zval *zid, *zvalue;
    zend_long options;

    if (zend_parse_parameters(ZEND_NUM_ARGS(), "rlz", &zid, &options, &zvalue) == FAILURE) {
        return;
    }

H
heyanlong 已提交
306 307 308 309 310 311 312 313 314 315
    zval *is_send = zend_hash_index_find(Z_ARRVAL_P(&SKYWALKING_G(curl_header_send)), Z_RES_HANDLE_P(zid));
    //
    if (is_send != NULL &&  CURLOPT_HTTPHEADER == options && Z_TYPE_P(is_send) == IS_TRUE) {
        add_index_bool(&SKYWALKING_G(curl_header_send), Z_RES_HANDLE_P(zid), IS_FALSE);
        orig_curl_setopt(INTERNAL_FUNCTION_PARAM_PASSTHRU);
    } else {
        if (CURLOPT_HTTPHEADER == options && Z_TYPE_P(zvalue) == IS_ARRAY) {
            zval copy_header;
            ZVAL_DUP(&copy_header, zvalue);
            add_index_zval(&SKYWALKING_G(curl_header), Z_RES_HANDLE_P(zid), &copy_header);
H
heyanlong 已提交
316
        } else {
H
heyanlong 已提交
317
            orig_curl_setopt(INTERNAL_FUNCTION_PARAM_PASSTHRU);
H
heyanlong 已提交
318 319 320
        }
    }
}
H
heyanlong 已提交
321 322

void sky_curl_setopt_array_handler(INTERNAL_FUNCTION_PARAMETERS) {
H
heyanlong 已提交
323 324 325
    zval *zid, *arr, *entry;
    zend_ulong option;
    zend_string *string_key;
H
heyanlong 已提交
326 327 328 329 330 331

    ZEND_PARSE_PARAMETERS_START(2, 2)
            Z_PARAM_RESOURCE(zid)
            Z_PARAM_ARRAY(arr)
    ZEND_PARSE_PARAMETERS_END();

H
heyanlong 已提交
332 333 334 335 336 337
    zval *http_header = zend_hash_index_find(Z_ARRVAL_P(arr), CURLOPT_HTTPHEADER);

    if (http_header != NULL) {
        zval copy_header;
        ZVAL_DUP(&copy_header, http_header);
        add_index_zval(&SKYWALKING_G(curl_header), Z_RES_HANDLE_P(zid), &copy_header);
H
heyanlong 已提交
338
    }
H
heyanlong 已提交
339 340 341 342

    orig_curl_setopt_array(INTERNAL_FUNCTION_PARAM_PASSTHRU);
}

H
heyanlong 已提交
343 344 345 346 347 348 349
void sky_curl_close_handler(INTERNAL_FUNCTION_PARAMETERS) {
    zval *zid;

    if (zend_parse_parameters(ZEND_NUM_ARGS(), "r", &zid) == FAILURE) {
        return;
    }

H
heyanlong 已提交
350 351 352
    zval *http_header = zend_hash_index_find(Z_ARRVAL_P(&SKYWALKING_G(curl_header)), Z_RES_HANDLE_P(zid));
    if (http_header != NULL) {
        zend_hash_index_del(Z_ARRVAL_P(&SKYWALKING_G(curl_header)), Z_RES_HANDLE_P(zid));
H
heyanlong 已提交
353 354 355 356
    }

    orig_curl_close(INTERNAL_FUNCTION_PARAM_PASSTHRU);
}
357 358 359 360 361
/* {{{ php_skywalking_init_globals
 */
/* Uncomment this function if you have INI entries*/
static void php_skywalking_init_globals(zend_skywalking_globals *skywalking_globals)
{
H
report  
heyanlong 已提交
362 363
	skywalking_globals->app_code = NULL;
	skywalking_globals->log_path = NULL;
H
heyanlong 已提交
364 365
	skywalking_globals->enable = 0;
	skywalking_globals->version = 5;
H
report  
heyanlong 已提交
366
	skywalking_globals->grpc = NULL;
367 368 369 370 371 372 373
}



static char *sky_json_encode(zval *parameter){

	smart_str buf = {0};
H
report  
heyanlong 已提交
374
	zend_long options = 64;
375 376 377 378 379 380 381 382 383 384
#if PHP_VERSION_ID >= 71000
	return_code = php_json_encode(&buf, parameter, (int)options);
	if (return_code != SUCCESS) {
		smart_str_free(&buf);
		return NULL;
	}
#else
	php_json_encode(&buf, parameter, (int)options);
#endif
	smart_str_0(&buf);
H
memory  
heyanlong 已提交
385 386 387
	char *bufs = ZSTR_VAL(buf.s);
	smart_str_free(&buf);
	return bufs;
388 389
}

S
songzhian 已提交
390

H
memory  
heyanlong 已提交
391 392
static void write_log(char *text) {
    if (application_instance != -100000) {
H
report  
heyanlong 已提交
393 394 395 396
        char *log_path;
        char logFilename[100];
        char message[strlen(text) + 1];
        log_path = SKY_G(log_path);
397

H
heyanlong 已提交
398
        zend_string *_log_path, *_log_path_lower;
H
memory  
heyanlong 已提交
399 400
        _log_path = zend_string_init(log_path, strlen(log_path), 0);
        _log_path_lower = php_string_tolower(_log_path);
401

H
report  
heyanlong 已提交
402
        bzero(logFilename, 100);
H
heyanlong 已提交
403
        sprintf(logFilename, "%s/skywalking.%d-%d.log", ZSTR_VAL(_log_path_lower), get_second(), getpid());
404

H
memory  
heyanlong 已提交
405 406
        zend_string_release(_log_path);
        zend_string_release(_log_path_lower);
H
report  
heyanlong 已提交
407 408 409 410
        bzero(message, strlen(text));
        sprintf(message, "%s\n", text);
        _php_error_log_ex(3, message, strlen(message), logFilename, NULL);
    }
411 412 413 414

}


H
heyanlong 已提交
415
static char *generate_sw3(zend_long span_id, char *peer_host, char *operation_name) {
H
heyanlong 已提交
416

H
heyanlong 已提交
417 418 419 420 421 422 423
    zval *traceId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "currentTraceId", sizeof("currentTraceId") - 1);
    zval *entryApplicationInstance = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "entryApplicationInstance",
                                                        sizeof("entryApplicationInstance") - 1);
    zval *entryOperationName = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "entryOperationName",
                                                  sizeof("entryOperationName") - 1);
    zval *distributedTraceId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "distributedTraceId",
                                                  sizeof("distributedTraceId") - 1);
H
heyanlong 已提交
424 425 426 427 428 429 430 431 432 433
    ssize_t sw3_l = 0;
    sw3_l = snprintf(NULL, 0, "sw3: %s|%d|%d|%d|#%s|#%s|#%s|%s", Z_STRVAL_P(traceId), span_id,
                     application_instance, Z_LVAL_P(entryApplicationInstance), peer_host,
                     Z_STRVAL_P(entryOperationName), operation_name, Z_STRVAL_P(distributedTraceId));
    char *sw3 = (char*)emalloc(sw3_l + 1);
    bzero(sw3, sw3_l + 1);
    snprintf(sw3, sw3_l + 1, "sw3: %s|%d|%d|%d|#%s|#%s|#%s|%s", Z_STRVAL_P(traceId), span_id,
             application_instance, Z_LVAL_P(entryApplicationInstance), peer_host,
             Z_STRVAL_P(entryOperationName), operation_name, Z_STRVAL_P(distributedTraceId));
    return sw3;
H
heyanlong 已提交
434 435
}

G
goerzh 已提交
436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454
static char *generate_sw6(zend_long span_id, char *peer_host, char *operation_name) {
    zval *traceId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "currentTraceId", sizeof("currentTraceId") - 1);
    zval *entryApplicationInstance = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "entryApplicationInstance",
                                                        sizeof("entryApplicationInstance") - 1);
    zval *entryOperationName = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "entryOperationName",
                                                  sizeof("entryOperationName") - 1);
    zval *distributedTraceId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "distributedTraceId",
                                                  sizeof("distributedTraceId") - 1);

    ContextCarrier *contextCarrier;

    contextCarrier = emalloc(sizeof(*contextCarrier));
    zval_b64_encode(&contextCarrier->primaryDistributedTraceId, Z_STRVAL_P(distributedTraceId));
    zval_b64_encode(&contextCarrier->traceSegmentId, Z_STRVAL_P(traceId));
    zval_b64_encode(&contextCarrier->peerHost, peer_host);
    zval_b64_encode(&contextCarrier->entryEndpointName, Z_STRVAL_P(entryOperationName));
    zval_b64_encode(&contextCarrier->parentEndpointName, operation_name);

    ssize_t sw6_l = 0;
G
fix bug  
goerzh 已提交
455 456 457 458
    sw6_l = snprintf(NULL, 0, "sw6: 1-%s-%s-%d-%d-%d-%s-%s-%s", Z_STRVAL(contextCarrier->primaryDistributedTraceId),
                     Z_STRVAL(contextCarrier->traceSegmentId), span_id, application_instance, Z_LVAL_P(entryApplicationInstance),
                     Z_STRVAL(contextCarrier->peerHost), Z_STRVAL(contextCarrier->entryEndpointName),
                     Z_STRVAL(contextCarrier->parentEndpointName));
G
goerzh 已提交
459 460 461

    char *sw6 = (char*)emalloc(sw6_l + 1);
    bzero(sw6, sw6_l + 1);
G
fix bug  
goerzh 已提交
462 463 464 465
    snprintf(sw6, sw6_l + 1, "sw6: 1-%s-%s-%d-%d-%d-%s-%s-%s", Z_STRVAL(contextCarrier->primaryDistributedTraceId),
             Z_STRVAL(contextCarrier->traceSegmentId), span_id, application_instance, Z_LVAL_P(entryApplicationInstance),
             Z_STRVAL(contextCarrier->peerHost), Z_STRVAL(contextCarrier->entryEndpointName),
             Z_STRVAL(contextCarrier->parentEndpointName));
G
goerzh 已提交
466 467 468 469 470 471

    efree(contextCarrier);

    return sw6;
}

H
heyanlong 已提交
472 473 474 475
static zend_string *trim_sharp(zval *tmp) {
    return php_trim(Z_STR_P(tmp), "#", sizeof("#") - 1, 1);
}

G
goerzh 已提交
476 477
static void zval_b64_encode(zval *out, char *in) {
    char *enc = b64_encode(in, strlen(in));
G
goerzh 已提交
478
    zend_string *str = zend_string_init(enc, strlen(enc), 0);
G
goerzh 已提交
479 480 481 482 483 484
    ZVAL_STR(out, str);
    free(enc);
}

static void zval_b64_decode(zval *out, char *in) {
    char *dec = b64_decode(in, strlen(in));
G
goerzh 已提交
485
    zend_string *str = zend_string_init(dec, strlen(dec), 0);
G
goerzh 已提交
486 487 488 489
    ZVAL_STR(out, str);
    free(dec);
}

H
report  
heyanlong 已提交
490 491 492 493 494 495
static void generate_context() {
    int sys_pid = getpid();
    long second = get_second();
    second = second * 10000 + sky_increment_id;
    char *makeTraceId;
    makeTraceId = (char *) emalloc(sizeof(char) * 180);
H
heyanlong 已提交
496

H
heyanlong 已提交
497
    bzero(makeTraceId, sizeof(char) * 180);
S
songzhian 已提交
498

H
report  
heyanlong 已提交
499
    sprintf(makeTraceId, "%d.%d.%ld", application_instance, sys_pid, second);
S
songzhian 已提交
500

H
report  
heyanlong 已提交
501
    add_assoc_string(&SKYWALKING_G(context), "currentTraceId", makeTraceId);
H
heyanlong 已提交
502
    add_assoc_long(&SKYWALKING_G(context), "isChild", 0);
H
heyanlong 已提交
503

H
report  
heyanlong 已提交
504 505
    // parent
    zval *carrier = NULL;
H
heyanlong 已提交
506
    zval *sw;
G
goerzh 已提交
507

H
report  
heyanlong 已提交
508
    zend_bool jit_initialization = PG(auto_globals_jit);
H
heyanlong 已提交
509

H
report  
heyanlong 已提交
510 511 512 513 514
    if (jit_initialization) {
        zend_string *server_str = zend_string_init("_SERVER", sizeof("_SERVER") - 1, 0);
        zend_is_auto_global(server_str);
        zend_string_release(server_str);
    }
H
heyanlong 已提交
515
    carrier = zend_hash_str_find(&EG(symbol_table), ZEND_STRL("_SERVER"));
G
goerzh 已提交
516

H
heyanlong 已提交
517 518
    if(SKYWALKING_G(version) == 5) {
        sw = zend_hash_str_find(Z_ARRVAL_P(carrier), "HTTP_SW3", sizeof("HTTP_SW3") - 1);
G
goerzh 已提交
519

H
heyanlong 已提交
520 521
        if (sw != NULL && Z_TYPE_P(sw) == IS_STRING && Z_STRLEN_P(sw) > 10) {
            add_assoc_string(&SKYWALKING_G(context), "sw3", Z_STRVAL_P(sw));
G
goerzh 已提交
522 523 524 525

            zval temp;
            array_init(&temp);

H
heyanlong 已提交
526 527 528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550
            php_explode(zend_string_init(ZEND_STRL("|"), 0), Z_STR_P(sw), &temp, 10);

            if(zend_array_count(Z_ARRVAL_P(&temp)) >= 8) {
                zval *sw3_0 = zend_hash_index_find(Z_ARRVAL(temp), 0);
                zval *sw3_1 = zend_hash_index_find(Z_ARRVAL(temp), 1);
                zval *sw3_2 = zend_hash_index_find(Z_ARRVAL(temp), 2);
                zval *sw3_3 = zend_hash_index_find(Z_ARRVAL(temp), 3);
                zval *sw3_4 = zend_hash_index_find(Z_ARRVAL(temp), 4);
                zval *sw3_5 = zend_hash_index_find(Z_ARRVAL(temp), 5);
                zval *sw3_6 = zend_hash_index_find(Z_ARRVAL(temp), 6);
                zval *sw3_7 = zend_hash_index_find(Z_ARRVAL(temp), 7);

                zval child;
                array_init(&child);
                ZVAL_LONG(&child, 1);
                zend_hash_str_update(Z_ARRVAL_P(&SKYWALKING_G(context)), "isChild", sizeof("isChild") - 1, &child);

                add_assoc_string(&SKYWALKING_G(context), "parentTraceSegmentId", Z_STRVAL_P(sw3_0));
                add_assoc_long(&SKYWALKING_G(context), "parentSpanId", zend_atol(Z_STRVAL_P(sw3_1), sizeof(Z_STRVAL_P(sw3_1)) - 1));
                add_assoc_long(&SKYWALKING_G(context), "parentApplicationInstance", zend_atol(Z_STRVAL_P(sw3_2), sizeof(Z_STRVAL_P(sw3_2)) - 1));
                add_assoc_long(&SKYWALKING_G(context), "entryApplicationInstance", zend_atol(Z_STRVAL_P(sw3_3), sizeof(Z_STRVAL_P(sw3_3)) - 1));
                add_assoc_str(&SKYWALKING_G(context), "networkAddress", trim_sharp(sw3_4));
                add_assoc_str(&SKYWALKING_G(context), "entryOperationName", trim_sharp(sw3_5));
                add_assoc_str(&SKYWALKING_G(context), "parentOperationName", trim_sharp(sw3_6));
                add_assoc_string(&SKYWALKING_G(context), "distributedTraceId", Z_STRVAL_P(sw3_7));
G
goerzh 已提交
551
            }
H
heyanlong 已提交
552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578
        } else {
            add_assoc_long(&SKYWALKING_G(context), "parentApplicationInstance", application_instance);
            add_assoc_long(&SKYWALKING_G(context), "entryApplicationInstance", application_instance);
            add_assoc_string(&SKYWALKING_G(context), "entryOperationName", get_page_request_uri());
            add_assoc_string(&SKYWALKING_G(context), "distributedTraceId", makeTraceId);
        }
    } else if (SKYWALKING_G(version) == 6) {
        sw = zend_hash_str_find(Z_ARRVAL_P(carrier), "HTTP_SW6", sizeof("HTTP_SW6") - 1);
        if (sw != NULL && Z_TYPE_P(sw) == IS_STRING && Z_STRLEN_P(sw) > 10) {
            add_assoc_string(&SKYWALKING_G(context), "sw6", Z_STRVAL_P(sw));
            // todo
//            contextCarrier = emalloc(sizeof(*contextCarrier));
//
//            zval temp;
//            array_init(&temp);
//            php_explode(zend_string_init(ZEND_STRL("-"), 0), Z_STR_P(sw6), &temp, 10);
//
//            if (zend_array_count(Z_ARRVAL_P(&temp)) >= 7) {
//                zval_b64_decode(&contextCarrier->primaryDistributedTraceId, Z_STRVAL_P(zend_hash_index_find(Z_ARRVAL(temp), 1)));
//                zval_b64_decode(&contextCarrier->traceSegmentId, Z_STRVAL_P(zend_hash_index_find(Z_ARRVAL(temp), 2)));
//                ZVAL_COPY(&contextCarrier->spanId, zend_hash_index_find(Z_ARRVAL(temp), 3));
//                ZVAL_COPY(&contextCarrier->parentServiceInstanceId, zend_hash_index_find(Z_ARRVAL(temp), 4));
//                ZVAL_COPY(&contextCarrier->entryServiceInstanceId, zend_hash_index_find(Z_ARRVAL(temp), 5));
//                zval_b64_decode(&contextCarrier->peerHost, Z_STRVAL_P(zend_hash_index_find(Z_ARRVAL(temp), 6)));
//                zval_b64_decode(&contextCarrier->entryEndpointName, Z_STRVAL_P(zend_hash_index_find(Z_ARRVAL(temp), 7)));
//                zval_b64_decode(&contextCarrier->parentEndpointName, Z_STRVAL_P(zend_hash_index_find(Z_ARRVAL(temp), 8)));
//            }
G
goerzh 已提交
579 580 581
        }
    }

H
memory  
heyanlong 已提交
582
    efree(makeTraceId);
583 584
}

H
report  
heyanlong 已提交
585 586 587 588 589 590 591

static long get_second() {
    struct timeval tv;
    gettimeofday(&tv,NULL);
    return tv.tv_sec;
}

592 593 594 595 596 597 598 599 600 601 602 603 604 605
static char *get_millisecond(){
	struct timeval tv;
	gettimeofday(&tv,NULL);
	char *buffer;
	buffer = (char *)emalloc(sizeof(char)*20);
	bzero(buffer, 20);
	long millisecond;
	millisecond = tv.tv_sec*1000 + tv.tv_usec/1000;
	sprintf(buffer, "%ld",  millisecond);

	return buffer;
}


H
heyanlong 已提交
606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628
static char *get_page_request_uri() {
    zval *carrier = NULL;
    zval *request_uri;

    smart_str uri = {0};

    if (strcasecmp("cli", sapi_module.name) == 0) {
        smart_str_appendl(&uri, "cli", strlen("cli"));
    } else {
        zend_bool jit_initialization = PG(auto_globals_jit);

        if (jit_initialization) {
            zend_string *server_str = zend_string_init("_SERVER", sizeof("_SERVER") - 1, 0);
            zend_is_auto_global(server_str);
            zend_string_release(server_str);
        }
        carrier = zend_hash_str_find(&EG(symbol_table), ZEND_STRL("_SERVER"));

        request_uri = zend_hash_str_find(Z_ARRVAL_P(carrier), "REQUEST_URI", sizeof("REQUEST_URI") - 1);
        smart_str_appendl(&uri, Z_STRVAL_P(request_uri), strlen(Z_STRVAL_P(request_uri)));
    }

    smart_str_0(&uri);
H
memory  
heyanlong 已提交
629 630 631
    char *uris = ZSTR_VAL(uri.s);
    smart_str_free(&uri);
    return uris;
H
heyanlong 已提交
632
}
633

G
goerzh 已提交
634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662
static char *get_page_request_peer() {
    zval *carrier = NULL;
    zval *request_host = NULL;
    zval *request_port = NULL;

    char *peer = NULL;
    size_t peer_l = 0;

    zend_bool jit_initialization = PG(auto_globals_jit);

    if (jit_initialization) {
        zend_string *server_str = zend_string_init("_SERVER", sizeof("_SERVER") - 1, 0);
        zend_is_auto_global(server_str);
        zend_string_release(server_str);
    }
    carrier = zend_hash_str_find(&EG(symbol_table), ZEND_STRL("_SERVER"));

    request_host = zend_hash_str_find(Z_ARRVAL_P(carrier), "SERVER_ADDR", sizeof("SERVER_ADDR") - 1);
    request_port = zend_hash_str_find(Z_ARRVAL_P(carrier), "SERVER_PORT", sizeof("SERVER_PORT") - 1);

    if (request_host != NULL && request_port != NULL) {
        peer_l = snprintf(NULL, 0, "%s:%s", Z_STRVAL_P(request_host), Z_STRVAL_P(request_port));
        peer = emalloc(peer_l + 1);
        snprintf(peer, peer_l + 1, "%s:%s", Z_STRVAL_P(request_host), Z_STRVAL_P(request_port));
    }

    return peer;
}

663

S
songzhian 已提交
664
/**
H
report  
heyanlong 已提交
665
 * ip
S
songzhian 已提交
666 667 668 669 670 671
 * 
 * @since 2017年11月23日
 * @copyright
 * @return return_type
//  */
static char* _get_current_machine_ip(){
G
goerzh 已提交
672 673 674
    char *ip;
    ip = (char *) emalloc(sizeof(char) * 100);
    bzero(ip, 100);
S
songzhian 已提交
675

G
goerzh 已提交
676 677 678 679 680 681 682 683 684 685 686 687 688
    if (strcasecmp("cli", sapi_module.name) == 0) {
        strcpy(ip, "127.0.0.1");
    } else {
        char hname[128];
        struct hostent *hent;
        gethostname(hname, sizeof(hname));
        hent = gethostbyname(hname);
        if (hent == NULL) {
            strcpy(ip, "127.0.0.1");
        } else {
            ip = inet_ntoa(*(struct in_addr *) (hent->h_addr_list[0]));
        }
    }
S
songzhian 已提交
689 690 691 692

    return ip;
}

H
heyanlong 已提交
693
static void request_init() {
S
songzhian 已提交
694

H
heyanlong 已提交
695
    array_init(&SKYWALKING_G(curl_header));
H
heyanlong 已提交
696
    array_init(&SKYWALKING_G(curl_header_send));
H
heyanlong 已提交
697 698
    array_init(&SKYWALKING_G(context));
	array_init(&SKYWALKING_G(UpstreamSegment));
S
songzhian 已提交
699

H
report  
heyanlong 已提交
700
    generate_context();
S
songzhian 已提交
701

H
report  
heyanlong 已提交
702
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "application_instance", application_instance);
H
heyanlong 已提交
703
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "pid", getppid());
H
report  
heyanlong 已提交
704
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "application_id", application_id);
H
heyanlong 已提交
705
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "version", SKYWALKING_G(version));
H
heyanlong 已提交
706
	SKY_ADD_ASSOC_ZVAL(&SKYWALKING_G(UpstreamSegment), "segment");
H
report  
heyanlong 已提交
707 708 709
	SKY_ADD_ASSOC_ZVAL(&SKYWALKING_G(UpstreamSegment), "globalTraceIds");

	zval *traceId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "currentTraceId", sizeof("currentTraceId") - 1);
S
songzhian 已提交
710

H
heyanlong 已提交
711 712 713 714
	zval traceSegmentObject;
	zval spans;
	array_init(&spans);
	array_init(&traceSegmentObject);
H
report  
heyanlong 已提交
715 716
	add_assoc_string(&traceSegmentObject, "traceSegmentId", Z_STRVAL_P(traceId));
	add_assoc_long(&traceSegmentObject, "isSizeLimited", 0);
S
songzhian 已提交
717

H
heyanlong 已提交
718
	zval temp;
G
goerzh 已提交
719 720
	char *peer = NULL;

H
heyanlong 已提交
721
	array_init(&temp);
G
goerzh 已提交
722
	peer = get_page_request_peer();
S
songzhian 已提交
723

H
heyanlong 已提交
724 725 726 727 728 729 730
    add_assoc_long(&temp, "spanId", 0);
    add_assoc_long(&temp, "parentSpanId", -1);
    char *l_millisecond = get_millisecond();
    long millisecond = zend_atol(l_millisecond, strlen(l_millisecond));
    efree(l_millisecond);
    add_assoc_long(&temp, "startTime", millisecond);
    add_assoc_string(&temp, "operationName", get_page_request_uri());
G
goerzh 已提交
731
    add_assoc_string(&temp, "peer", (peer == NULL) ? "" : peer);
H
heyanlong 已提交
732 733 734
    add_assoc_long(&temp, "spanType", 0);
    add_assoc_long(&temp, "spanLayer", 3);
    add_assoc_long(&temp, "componentId", COMPONENT_HTTPCLIENT);
S
songzhian 已提交
735

G
goerzh 已提交
736 737 738 739 740 741 742 743 744 745
    if (peer != NULL) {
        efree(peer);
    }

//    zval *method = NULL;
//    method = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "method", sizeof("method") - 1);
//    if (method != NULL) {
//        add_assoc_string(&temp, "method", Z_STRVAL_P(method));
//    }

H
heyanlong 已提交
746 747 748 749
    zval *isChild = zend_hash_str_find(Z_ARRVAL_P(&SKYWALKING_G(context)), "isChild", sizeof("isChild") - 1);
    // refs
    zval refs;
    array_init(&refs);
H
heyanlong 已提交
750 751 752 753 754

    zval globalTraceIds;
    array_init(&globalTraceIds);
    zval tmpGlobalTraceIds;

H
heyanlong 已提交
755
    if(Z_LVAL_P(isChild) == 1) {
H
memory  
heyanlong 已提交
756 757
        zval ref;
        array_init(&ref);
H
heyanlong 已提交
758 759 760 761 762 763
        zval *parentTraceSegmentId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "parentTraceSegmentId", sizeof("parentTraceSegmentId") - 1);
        zval *parentSpanId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "parentSpanId", sizeof("parentSpanId") - 1);
        zval *parentApplicationInstance = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "parentApplicationInstance", sizeof("parentApplicationInstance") - 1);
        zval *entryApplicationInstance = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "entryApplicationInstance", sizeof("entryApplicationInstance") - 1);
        zval *entryOperationName = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "entryOperationName", sizeof("entryOperationName") - 1);
        zval *networkAddress = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "networkAddress", sizeof("networkAddress") - 1);
H
heyanlong 已提交
764
        zval *parentOperationName = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "parentOperationName", sizeof("parentOperationName") - 1);
H
heyanlong 已提交
765
        zval *distributedTraceId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "distributedTraceId", sizeof("distributedTraceId") - 1);
H
heyanlong 已提交
766 767 768 769 770 771 772
        add_assoc_long(&ref, "type", 0);
        add_assoc_string(&ref, "parentTraceSegmentId", Z_STRVAL_P(parentTraceSegmentId));
        add_assoc_long(&ref, "parentSpanId", Z_LVAL_P(parentSpanId));
        add_assoc_long(&ref, "parentApplicationInstanceId", Z_LVAL_P(parentApplicationInstance));
        add_assoc_string(&ref, "networkAddress", Z_STRVAL_P(networkAddress));
        add_assoc_long(&ref, "entryApplicationInstanceId", Z_LVAL_P(entryApplicationInstance));
        add_assoc_string(&ref, "entryServiceName", Z_STRVAL_P(entryOperationName));
H
heyanlong 已提交
773
        add_assoc_string(&ref, "parentServiceName", Z_STRVAL_P(parentOperationName));
H
heyanlong 已提交
774
        zend_hash_next_index_insert(Z_ARRVAL(refs), &ref);
H
heyanlong 已提交
775 776 777
        ZVAL_STRING(&tmpGlobalTraceIds, Z_STRVAL_P(distributedTraceId));
    } else {
        ZVAL_STRING(&tmpGlobalTraceIds, Z_STRVAL_P(traceId));
H
heyanlong 已提交
778 779 780 781

    }

    zend_hash_str_add(Z_ARRVAL(temp), "refs", sizeof("refs") - 1, &refs);
H
heyanlong 已提交
782
    zend_hash_next_index_insert(Z_ARRVAL(spans), &temp);
S
songzhian 已提交
783

H
heyanlong 已提交
784
    add_assoc_zval(&traceSegmentObject, "spans", &spans);
S
songzhian 已提交
785

H
report  
heyanlong 已提交
786
    zend_hash_next_index_insert(Z_ARRVAL(globalTraceIds), &tmpGlobalTraceIds);
S
songzhian 已提交
787

H
report  
heyanlong 已提交
788 789
    zend_hash_str_update(Z_ARRVAL(SKYWALKING_G(UpstreamSegment)), "segment", sizeof("segment") - 1, &traceSegmentObject);
    zend_hash_str_update(Z_ARRVAL(SKYWALKING_G(UpstreamSegment)), "globalTraceIds", sizeof("globalTraceIds") - 1, &globalTraceIds);
S
songzhian 已提交
790 791 792
}


H
report  
heyanlong 已提交
793
static void sky_flush_all() {
S
songzhian 已提交
794
	char *l_millisecond = get_millisecond();
H
heyanlong 已提交
795 796 797 798
	long millisecond = zend_atol(l_millisecond, strlen(l_millisecond));
	efree(l_millisecond);

	zval *span = get_first_span();
S
songzhian 已提交
799

H
heyanlong 已提交
800 801
	add_assoc_long(span, "endTime", millisecond);
	if ((SG(sapi_headers).http_response_code >= 500)) {
H
heyanlong 已提交
802
		add_assoc_long(span, "isError", 1);
H
heyanlong 已提交
803
	} else {
H
heyanlong 已提交
804
		add_assoc_long(span, "isError", 0);
H
heyanlong 已提交
805
	}
S
songzhian 已提交
806

H
heyanlong 已提交
807 808
	write_log(sky_json_encode(&SKYWALKING_G(UpstreamSegment)));
}
S
songzhian 已提交
809

H
heyanlong 已提交
810 811 812 813 814 815
static zval *get_first_span() {
	zval *segment = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(UpstreamSegment)), "segment", sizeof("segment") - 1);
	zval *spans = zend_hash_str_find(Z_ARRVAL_P(segment), "spans", sizeof("spans") - 1);
	zval *span = zend_hash_index_find(Z_ARRVAL_P(spans), 0);
	return span;
}
S
songzhian 已提交
816

H
heyanlong 已提交
817 818 819 820
static zval *get_spans() {
	zval *segment = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(UpstreamSegment)), "segment", sizeof("segment") - 1);
	zval *spans = zend_hash_str_find(Z_ARRVAL_P(segment), "spans", sizeof("spans") - 1);
	return spans;
S
songzhian 已提交
821 822 823
}


H
heyanlong 已提交
824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864
static int sky_register() {
    if (application_id == -100000 || application_instance == -100000) {
        struct sockaddr_un un;
        un.sun_family = AF_UNIX;
        strcpy(un.sun_path, sock_path);
        int fd;
        char message[4096];
        char return_message[4096];

        fd = socket(AF_UNIX, SOCK_STREAM, 0);
        if (fd >= 0) {
            int conn = connect(fd, (struct sockaddr *) &un, sizeof(un));

            if (conn >= 0) {
                bzero(message, sizeof(message));
                sprintf(message, "0{\"app_code\":\"%s\",\"pid\":%d}\n", SKYWALKING_G(app_code), getppid());
                write(fd, message, strlen(message));

                bzero(return_message, sizeof(return_message));
                read(fd, return_message, sizeof(return_message));

                application_id = atoi(return_message);

                if (application_id != -100000) {
                    bzero(message, sizeof(message));
                    sprintf(message, "1{\"application_id\":\"%d\",\"pid\":%d}\n", application_id, getppid());
                    write(fd, message, strlen(message));

                    bzero(return_message, sizeof(return_message));
                    read(fd, return_message, sizeof(return_message));

                    application_instance = atoi(return_message);
                }
            }

            close(fd);
        }
    }
    return 0;
}

S
songzhian 已提交
865

866 867
/* {{{ PHP_MINIT_FUNCTION
 */
H
heyanlong 已提交
868
PHP_MINIT_FUNCTION (skywalking) {
H
docs  
heyanlong 已提交
869
	ZEND_INIT_MODULE_GLOBALS(skywalking, php_skywalking_init_globals, NULL);
S
songzhian 已提交
870
	//data_register_hashtable();
871 872 873
	REGISTER_INI_ENTRIES();
	/* If you have INI entries, uncomment these lines
	*/
H
report  
heyanlong 已提交
874
	if (SKYWALKING_G(enable)) {
875
        if (strcasecmp("cli", sapi_module.name) == 0) {
H
heyanlong 已提交
876
            return SUCCESS;
877 878
        }

H
heyanlong 已提交
879
		// bind curl
880
		zend_function *old_function;
H
heyanlong 已提交
881
		if ((old_function = zend_hash_str_find_ptr(CG(function_table), "curl_exec", sizeof("curl_exec") - 1)) != NULL) {
882
			orig_curl_exec = old_function->internal_function.handler;
H
heyanlong 已提交
883
			old_function->internal_function.handler = sky_curl_exec_handler;
884
		}
H
heyanlong 已提交
885 886 887 888
        if ((old_function = zend_hash_str_find_ptr(CG(function_table), "curl_setopt", sizeof("curl_setopt")-1)) != NULL) {
            orig_curl_setopt = old_function->internal_function.handler;
            old_function->internal_function.handler = sky_curl_setopt_handler;
        }
H
heyanlong 已提交
889 890 891 892
        if ((old_function = zend_hash_str_find_ptr(CG(function_table), "curl_setopt_array", sizeof("curl_setopt_array")-1)) != NULL) {
            orig_curl_setopt_array = old_function->internal_function.handler;
            old_function->internal_function.handler = sky_curl_setopt_array_handler;
        }
H
heyanlong 已提交
893 894 895 896
        if ((old_function = zend_hash_str_find_ptr(CG(function_table), "curl_close", sizeof("curl_close")-1)) != NULL) {
            orig_curl_close = old_function->internal_function.handler;
            old_function->internal_function.handler = sky_curl_close_handler;
        }
897 898 899 900 901 902 903 904 905 906 907
	}

	return SUCCESS;
}
/* }}} */

/* {{{ PHP_MSHUTDOWN_FUNCTION
 */
PHP_MSHUTDOWN_FUNCTION(skywalking)
{
	UNREGISTER_INI_ENTRIES();
H
memory  
heyanlong 已提交
908
    return SUCCESS;
909 910 911 912
}
/* }}} */

/* Remove if there's nothing to do at request start */
S
songzhian 已提交
913
/* {{{ PHP_RINIT_FUNCTION7
914 915 916
 */
PHP_RINIT_FUNCTION(skywalking)
{
S
songzhian 已提交
917

918 919 920
#if defined(COMPILE_DL_SKYWALKING) && defined(ZTS)
	ZEND_TSRMLS_CACHE_UPDATE();
#endif
H
report  
heyanlong 已提交
921
	if (SKYWALKING_G(enable)) {
H
heyanlong 已提交
922 923 924
	    sky_register();
        if(application_id == -100000 || application_instance == -100000) {
            sky_close = 1;
H
heyanlong 已提交
925 926
            return SUCCESS;
        }
H
report  
heyanlong 已提交
927 928 929 930
		sky_increment_id++;
		if (sky_increment_id >= 9999) {
			sky_increment_id = 0;
		}
H
memory  
heyanlong 已提交
931
        request_init();
H
report  
heyanlong 已提交
932
	}
933 934 935 936 937 938 939 940 941
	return SUCCESS;
}
/* }}} */

/* Remove if there's nothing to do at request end */
/* {{{ PHP_RSHUTDOWN_FUNCTION
 */
PHP_RSHUTDOWN_FUNCTION(skywalking)
{
H
heyanlong 已提交
942

H
report  
heyanlong 已提交
943
	if(SKYWALKING_G(enable)){
H
memory  
heyanlong 已提交
944 945 946
        if (sky_close == 1) {
            return SUCCESS;
        }
S
songzhian 已提交
947
		sky_flush_all();
H
memory  
heyanlong 已提交
948
        zval_dtor(&SKYWALKING_G(context));
H
heyanlong 已提交
949
        zval_dtor(&SKYWALKING_G(curl_header));
H
heyanlong 已提交
950
        zval_dtor(&SKYWALKING_G(curl_header_send));
H
memory  
heyanlong 已提交
951
        zval_dtor(&SKYWALKING_G(UpstreamSegment));
952 953 954 955 956 957 958 959 960
	}
	return SUCCESS;
}
/* }}} */

/* {{{ PHP_MINFO_FUNCTION
 */
PHP_MINFO_FUNCTION(skywalking)
{
S
songzhian 已提交
961

962
	php_info_print_table_start();
H
docs  
heyanlong 已提交
963 964 965 966 967
    if (sky_close) {
        php_info_print_table_header(2, "skywalking Support", "disabled (register fail)");
    } else {
        php_info_print_table_header(2, "skywalking Support", "enabled");
    }
968 969 970 971 972 973 974 975

    size_t instance_size = 0;
    instance_size = snprintf(NULL, 0, "%d", application_instance);
    char *instance_id = (char *)emalloc(instance_size + 1);
    bzero(instance_id, instance_size + 1);
    snprintf(instance_id, instance_size+1, "%d", application_instance);
    php_info_print_table_header(2, "Instance ID", instance_id);

976 977 978
	php_info_print_table_end();

	DISPLAY_INI_ENTRIES();
979
    efree(instance_id);
980 981 982
}
/* }}} */

H
memory  
heyanlong 已提交
983 984 985 986 987 988 989 990 991 992 993 994
/* {{{ PHP_MINFO_FUNCTION
 */
PHP_GINIT_FUNCTION(skywalking)
{
    memset(skywalking_globals, 0, sizeof(*skywalking_globals));
}
/* }}} */

zend_module_dep skywalking_deps[] = {
        ZEND_MOD_REQUIRED("json")
        {NULL, NULL, NULL}
};
995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026

/* {{{ skywalking_module_entry
 */
zend_module_entry skywalking_module_entry = {
	STANDARD_MODULE_HEADER,
	"skywalking",
	skywalking_functions,
	PHP_MINIT(skywalking),
	PHP_MSHUTDOWN(skywalking),
	PHP_RINIT(skywalking),		/* Replace with NULL if there's nothing to do at request start */
	PHP_RSHUTDOWN(skywalking),	/* Replace with NULL if there's nothing to do at request end */
	PHP_MINFO(skywalking),
	PHP_SKYWALKING_VERSION,
	STANDARD_MODULE_PROPERTIES
};
/* }}} */

#ifdef COMPILE_DL_SKYWALKING
#ifdef ZTS
ZEND_TSRMLS_CACHE_DEFINE()
#endif
ZEND_GET_MODULE(skywalking)
#endif

/*
 * Local variables:
 * tab-width: 4
 * c-basic-offset: 4
 * End:
 * vim600: noet sw=4 ts=4 fdm=marker
 * vim<600: noet sw=4 ts=4
 */