skywalking.c 38.3 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;
G
goerzh 已提交
367
	skywalking_globals->header_version = 2;
368 369 370 371 372 373 374
}



static char *sky_json_encode(zval *parameter){

	smart_str buf = {0};
H
report  
heyanlong 已提交
375
	zend_long options = 64;
376 377 378 379 380 381 382 383 384 385
#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 已提交
386 387 388
	char *bufs = ZSTR_VAL(buf.s);
	smart_str_free(&buf);
	return bufs;
389 390
}

S
songzhian 已提交
391

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

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

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

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

}


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

H
heyanlong 已提交
418 419 420 421 422 423 424
    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 已提交
425 426 427 428 429 430 431 432 433 434
    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 已提交
435 436
}

G
goerzh 已提交
437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455
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 已提交
456 457 458 459
    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 已提交
460 461 462

    char *sw6 = (char*)emalloc(sw6_l + 1);
    bzero(sw6, sw6_l + 1);
G
fix bug  
goerzh 已提交
463 464 465 466
    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 已提交
467 468 469 470 471 472

    efree(contextCarrier);

    return sw6;
}

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

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

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

H
report  
heyanlong 已提交
491 492 493 494 495 496
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 已提交
497

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

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

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

H
report  
heyanlong 已提交
505 506
    // parent
    zval *carrier = NULL;
G
goerzh 已提交
507
    ContextCarrier *contextCarrier = NULL;
H
heyanlong 已提交
508

G
goerzh 已提交
509 510 511
    //http info
//    zval *http_method = NULL;

H
report  
heyanlong 已提交
512
    zend_bool jit_initialization = PG(auto_globals_jit);
H
heyanlong 已提交
513

H
report  
heyanlong 已提交
514 515 516 517 518
    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 已提交
519
    carrier = zend_hash_str_find(&EG(symbol_table), ZEND_STRL("_SERVER"));
G
goerzh 已提交
520 521 522 523 524 525

//    http_method = zend_hash_str_find(Z_ARRVAL_P(carrier), "REQUEST_METHOD", sizeof("REQUEST_METHOD") - 1);
//    if (http_method != NULL) {
//        add_assoc_string(&SKYWALKING_G(context), "method", Z_STRVAL_P(http_method));
//    }

G
goerzh 已提交
526 527
    if (SKYWALKING_G(header_version) == 2) {
        zval *sw6;
G
goerzh 已提交
528
        sw6 = zend_hash_str_find(Z_ARRVAL_P(carrier), "HTTP_SW6", sizeof("HTTP_SW6") - 1);
G
goerzh 已提交
529 530
        if (sw6 != NULL && Z_TYPE_P(sw6) == IS_STRING && Z_STRLEN_P(sw6) > 10) {
            add_assoc_string(&SKYWALKING_G(context), "sw6", Z_STRVAL_P(sw6));
G
goerzh 已提交
531
            contextCarrier = emalloc(sizeof(*contextCarrier));
G
goerzh 已提交
532 533 534 535 536 537

            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) {
G
goerzh 已提交
538 539 540 541 542 543 544 545
                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 已提交
546
            }
H
heyanlong 已提交
547
        }
G
goerzh 已提交
548 549 550 551 552 553
    } else if (SKYWALKING_G(header_version) == 1) {
        zval *sw3;
        sw3 = zend_hash_str_find(Z_ARRVAL_P(carrier), "HTTP_SW3", sizeof("HTTP_SW3") - 1);

        if (sw3 != NULL && Z_TYPE_P(sw3) == IS_STRING && Z_STRLEN_P(sw3) > 10) {
            add_assoc_string(&SKYWALKING_G(context), "sw3", Z_STRVAL_P(sw3));
G
goerzh 已提交
554
            contextCarrier = emalloc(sizeof(*contextCarrier));
G
goerzh 已提交
555 556 557 558 559 560 561

            zval temp;
            array_init(&temp);

            php_explode(zend_string_init(ZEND_STRL("|"), 0), Z_STR_P(sw3), &temp, 10);

            if (zend_array_count(Z_ARRVAL_P(&temp)) >= 8) {
G
goerzh 已提交
562 563 564 565 566 567 568 569
                ZVAL_COPY(&contextCarrier->traceSegmentId, zend_hash_index_find(Z_ARRVAL(temp), 0));
                ZVAL_COPY(&contextCarrier->spanId, zend_hash_index_find(Z_ARRVAL(temp), 1));
                ZVAL_COPY(&contextCarrier->parentServiceInstanceId, zend_hash_index_find(Z_ARRVAL(temp), 2));
                ZVAL_COPY(&contextCarrier->entryServiceInstanceId, zend_hash_index_find(Z_ARRVAL(temp), 3));
                ZVAL_COPY(&contextCarrier->peerHost, zend_hash_index_find(Z_ARRVAL(temp), 4));
                ZVAL_COPY(&contextCarrier->entryEndpointName, zend_hash_index_find(Z_ARRVAL(temp), 5));
                ZVAL_COPY(&contextCarrier->parentEndpointName, zend_hash_index_find(Z_ARRVAL(temp), 6));
                ZVAL_COPY(&contextCarrier->primaryDistributedTraceId, zend_hash_index_find(Z_ARRVAL(temp), 7));
G
goerzh 已提交
570 571 572 573 574 575 576 577 578 579
            }
        }
    }

    if (contextCarrier != NULL) {
        zval child;
        array_init(&child);
        ZVAL_LONG(&child, 1);
        zend_hash_str_update(Z_ARRVAL_P(&SKYWALKING_G(context)), "isChild", sizeof("isChild") - 1, &child);

G
goerzh 已提交
580
        add_assoc_string(&SKYWALKING_G(context), "parentTraceSegmentId", Z_STRVAL(contextCarrier->traceSegmentId));
G
goerzh 已提交
581
        add_assoc_long(&SKYWALKING_G(context), "parentSpanId",
G
goerzh 已提交
582
                       zend_atol(Z_STRVAL(contextCarrier->spanId), sizeof(Z_STRVAL(contextCarrier->spanId)) - 1));
G
goerzh 已提交
583
        add_assoc_long(&SKYWALKING_G(context), "parentApplicationInstance",
G
goerzh 已提交
584
                       zend_atol(Z_STRVAL(contextCarrier->parentServiceInstanceId), sizeof(Z_STRVAL(contextCarrier->parentServiceInstanceId)) - 1));
G
goerzh 已提交
585
        add_assoc_long(&SKYWALKING_G(context), "entryApplicationInstance",
G
goerzh 已提交
586 587 588 589 590 591 592
                       zend_atol(Z_STRVAL(contextCarrier->entryServiceInstanceId), sizeof(Z_STRVAL(contextCarrier->entryServiceInstanceId)) - 1));
        add_assoc_str(&SKYWALKING_G(context), "networkAddress", trim_sharp(&contextCarrier->peerHost));
        add_assoc_str(&SKYWALKING_G(context), "entryOperationName", trim_sharp(&contextCarrier->entryEndpointName));
        add_assoc_str(&SKYWALKING_G(context), "parentOperationName", trim_sharp(&contextCarrier->parentEndpointName));
        add_assoc_string(&SKYWALKING_G(context), "distributedTraceId", Z_STRVAL(contextCarrier->primaryDistributedTraceId));

        efree(contextCarrier);
H
heyanlong 已提交
593 594 595
    } else {
        add_assoc_long(&SKYWALKING_G(context), "parentApplicationInstance", application_instance);
        add_assoc_long(&SKYWALKING_G(context), "entryApplicationInstance", application_instance);
G
goerzh 已提交
596 597 598 599 600 601 602 603 604 605

        char *uri = get_page_request_uri();
        char *entry_operation_name = NULL;
        size_t entry_operation_size = 0;
        entry_operation_size = snprintf(NULL, 0, "#%s", uri);
        entry_operation_name = emalloc(entry_operation_size + 1);
        bzero(entry_operation_name, entry_operation_size + 1);
        sprintf(entry_operation_name, "#%s", uri);
        add_assoc_string(&SKYWALKING_G(context), "entryOperationName", entry_operation_name);

H
heyanlong 已提交
606
        add_assoc_string(&SKYWALKING_G(context), "distributedTraceId", makeTraceId);
G
goerzh 已提交
607 608
    };

H
memory  
heyanlong 已提交
609
    efree(makeTraceId);
610 611
}

H
report  
heyanlong 已提交
612 613 614 615 616 617 618

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

619 620 621 622 623 624 625 626 627 628 629 630 631 632
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 已提交
633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655
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 已提交
656 657 658
    char *uris = ZSTR_VAL(uri.s);
    smart_str_free(&uri);
    return uris;
H
heyanlong 已提交
659
}
660

G
goerzh 已提交
661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689
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;
}

690

S
songzhian 已提交
691
/**
H
report  
heyanlong 已提交
692
 * ip
S
songzhian 已提交
693 694 695 696 697 698
 * 
 * @since 2017年11月23日
 * @copyright
 * @return return_type
//  */
static char* _get_current_machine_ip(){
G
goerzh 已提交
699 700 701
    char *ip;
    ip = (char *) emalloc(sizeof(char) * 100);
    bzero(ip, 100);
S
songzhian 已提交
702

G
goerzh 已提交
703 704 705 706 707 708 709 710 711 712 713 714 715
    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 已提交
716 717 718 719

    return ip;
}

H
heyanlong 已提交
720
static void request_init() {
S
songzhian 已提交
721

H
heyanlong 已提交
722
    array_init(&SKYWALKING_G(curl_header));
H
heyanlong 已提交
723
    array_init(&SKYWALKING_G(curl_header_send));
H
heyanlong 已提交
724 725
    array_init(&SKYWALKING_G(context));
	array_init(&SKYWALKING_G(UpstreamSegment));
S
songzhian 已提交
726

H
report  
heyanlong 已提交
727
    generate_context();
S
songzhian 已提交
728

H
report  
heyanlong 已提交
729
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "application_instance", application_instance);
H
heyanlong 已提交
730
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "pid", getppid());
H
report  
heyanlong 已提交
731
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "application_id", application_id);
H
heyanlong 已提交
732
    add_assoc_long(&SKYWALKING_G(UpstreamSegment), "version", SKYWALKING_G(version));
H
heyanlong 已提交
733
	SKY_ADD_ASSOC_ZVAL(&SKYWALKING_G(UpstreamSegment), "segment");
H
report  
heyanlong 已提交
734 735 736
	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 已提交
737

H
heyanlong 已提交
738 739 740 741
	zval traceSegmentObject;
	zval spans;
	array_init(&spans);
	array_init(&traceSegmentObject);
H
report  
heyanlong 已提交
742 743
	add_assoc_string(&traceSegmentObject, "traceSegmentId", Z_STRVAL_P(traceId));
	add_assoc_long(&traceSegmentObject, "isSizeLimited", 0);
S
songzhian 已提交
744

H
heyanlong 已提交
745
	zval temp;
G
goerzh 已提交
746 747
	char *peer = NULL;

H
heyanlong 已提交
748
	array_init(&temp);
G
goerzh 已提交
749
	peer = get_page_request_peer();
S
songzhian 已提交
750

H
heyanlong 已提交
751 752 753 754 755 756 757
    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 已提交
758
    add_assoc_string(&temp, "peer", (peer == NULL) ? "" : peer);
H
heyanlong 已提交
759 760 761
    add_assoc_long(&temp, "spanType", 0);
    add_assoc_long(&temp, "spanLayer", 3);
    add_assoc_long(&temp, "componentId", COMPONENT_HTTPCLIENT);
S
songzhian 已提交
762

G
goerzh 已提交
763 764 765 766 767 768 769 770 771 772
    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 已提交
773 774 775 776
    zval *isChild = zend_hash_str_find(Z_ARRVAL_P(&SKYWALKING_G(context)), "isChild", sizeof("isChild") - 1);
    // refs
    zval refs;
    array_init(&refs);
H
heyanlong 已提交
777 778 779 780 781

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

H
heyanlong 已提交
782
    if(Z_LVAL_P(isChild) == 1) {
H
memory  
heyanlong 已提交
783 784
        zval ref;
        array_init(&ref);
H
heyanlong 已提交
785 786 787 788 789 790
        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 已提交
791
        zval *parentOperationName = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "parentOperationName", sizeof("parentOperationName") - 1);
H
heyanlong 已提交
792
        zval *distributedTraceId = zend_hash_str_find(Z_ARRVAL(SKYWALKING_G(context)), "distributedTraceId", sizeof("distributedTraceId") - 1);
H
heyanlong 已提交
793 794 795 796 797 798 799
        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 已提交
800
        add_assoc_string(&ref, "parentServiceName", Z_STRVAL_P(parentOperationName));
H
heyanlong 已提交
801
        zend_hash_next_index_insert(Z_ARRVAL(refs), &ref);
H
heyanlong 已提交
802 803 804
        ZVAL_STRING(&tmpGlobalTraceIds, Z_STRVAL_P(distributedTraceId));
    } else {
        ZVAL_STRING(&tmpGlobalTraceIds, Z_STRVAL_P(traceId));
H
heyanlong 已提交
805 806 807 808

    }

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

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

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

H
report  
heyanlong 已提交
815 816
    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 已提交
817 818 819
}


H
report  
heyanlong 已提交
820
static void sky_flush_all() {
S
songzhian 已提交
821
	char *l_millisecond = get_millisecond();
H
heyanlong 已提交
822 823 824 825
	long millisecond = zend_atol(l_millisecond, strlen(l_millisecond));
	efree(l_millisecond);

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

H
heyanlong 已提交
827 828
	add_assoc_long(span, "endTime", millisecond);
	if ((SG(sapi_headers).http_response_code >= 500)) {
H
heyanlong 已提交
829
		add_assoc_long(span, "isError", 1);
H
heyanlong 已提交
830
	} else {
H
heyanlong 已提交
831
		add_assoc_long(span, "isError", 0);
H
heyanlong 已提交
832
	}
S
songzhian 已提交
833

H
heyanlong 已提交
834 835
	write_log(sky_json_encode(&SKYWALKING_G(UpstreamSegment)));
}
S
songzhian 已提交
836

H
heyanlong 已提交
837 838 839 840 841 842
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 已提交
843

H
heyanlong 已提交
844 845 846 847
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 已提交
848 849 850
}


H
heyanlong 已提交
851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891
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 已提交
892

893 894
/* {{{ PHP_MINIT_FUNCTION
 */
H
heyanlong 已提交
895
PHP_MINIT_FUNCTION (skywalking) {
H
docs  
heyanlong 已提交
896
	ZEND_INIT_MODULE_GLOBALS(skywalking, php_skywalking_init_globals, NULL);
S
songzhian 已提交
897
	//data_register_hashtable();
898 899 900
	REGISTER_INI_ENTRIES();
	/* If you have INI entries, uncomment these lines
	*/
H
report  
heyanlong 已提交
901
	if (SKYWALKING_G(enable)) {
902
        if (strcasecmp("cli", sapi_module.name) == 0) {
H
heyanlong 已提交
903
            return SUCCESS;
904 905
        }

H
heyanlong 已提交
906
		// bind curl
907
		zend_function *old_function;
H
heyanlong 已提交
908
		if ((old_function = zend_hash_str_find_ptr(CG(function_table), "curl_exec", sizeof("curl_exec") - 1)) != NULL) {
909
			orig_curl_exec = old_function->internal_function.handler;
H
heyanlong 已提交
910
			old_function->internal_function.handler = sky_curl_exec_handler;
911
		}
H
heyanlong 已提交
912 913 914 915
        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 已提交
916 917 918 919
        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 已提交
920 921 922 923
        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;
        }
924 925 926 927 928 929 930 931 932 933 934
	}

	return SUCCESS;
}
/* }}} */

/* {{{ PHP_MSHUTDOWN_FUNCTION
 */
PHP_MSHUTDOWN_FUNCTION(skywalking)
{
	UNREGISTER_INI_ENTRIES();
H
memory  
heyanlong 已提交
935
    return SUCCESS;
936 937 938 939
}
/* }}} */

/* Remove if there's nothing to do at request start */
S
songzhian 已提交
940
/* {{{ PHP_RINIT_FUNCTION7
941 942 943
 */
PHP_RINIT_FUNCTION(skywalking)
{
S
songzhian 已提交
944

945 946 947
#if defined(COMPILE_DL_SKYWALKING) && defined(ZTS)
	ZEND_TSRMLS_CACHE_UPDATE();
#endif
H
report  
heyanlong 已提交
948
	if (SKYWALKING_G(enable)) {
H
heyanlong 已提交
949 950 951
	    sky_register();
        if(application_id == -100000 || application_instance == -100000) {
            sky_close = 1;
H
heyanlong 已提交
952 953
            return SUCCESS;
        }
H
report  
heyanlong 已提交
954 955 956 957
		sky_increment_id++;
		if (sky_increment_id >= 9999) {
			sky_increment_id = 0;
		}
H
memory  
heyanlong 已提交
958
        request_init();
H
report  
heyanlong 已提交
959
	}
960 961 962 963 964 965 966 967 968
	return SUCCESS;
}
/* }}} */

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

H
report  
heyanlong 已提交
970
	if(SKYWALKING_G(enable)){
H
memory  
heyanlong 已提交
971 972 973
        if (sky_close == 1) {
            return SUCCESS;
        }
S
songzhian 已提交
974
		sky_flush_all();
H
memory  
heyanlong 已提交
975
        zval_dtor(&SKYWALKING_G(context));
H
heyanlong 已提交
976
        zval_dtor(&SKYWALKING_G(curl_header));
H
heyanlong 已提交
977
        zval_dtor(&SKYWALKING_G(curl_header_send));
H
memory  
heyanlong 已提交
978
        zval_dtor(&SKYWALKING_G(UpstreamSegment));
979 980 981 982 983 984 985 986 987
	}
	return SUCCESS;
}
/* }}} */

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

989
	php_info_print_table_start();
H
docs  
heyanlong 已提交
990 991 992 993 994
    if (sky_close) {
        php_info_print_table_header(2, "skywalking Support", "disabled (register fail)");
    } else {
        php_info_print_table_header(2, "skywalking Support", "enabled");
    }
995 996 997 998 999 1000 1001 1002

    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);

1003 1004 1005
	php_info_print_table_end();

	DISPLAY_INI_ENTRIES();
1006
    efree(instance_id);
1007 1008 1009
}
/* }}} */

H
memory  
heyanlong 已提交
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021
/* {{{ 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}
};
1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053

/* {{{ 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
 */