1*804Svbart@nginx.com 2*804Svbart@nginx.com /* 3*804Svbart@nginx.com * Copyright (C) Max Romanov 4*804Svbart@nginx.com * Copyright (C) NGINX, Inc. 5*804Svbart@nginx.com */ 6*804Svbart@nginx.com 7*804Svbart@nginx.com #include <nxt_main.h> 8*804Svbart@nginx.com #include <nxt_router.h> 9*804Svbart@nginx.com #include <nxt_unit.h> 10*804Svbart@nginx.com 11*804Svbart@nginx.com 12*804Svbart@nginx.com static nxt_int_t nxt_external_init(nxt_task_t *task, 13*804Svbart@nginx.com nxt_common_app_conf_t *conf); 14*804Svbart@nginx.com 15*804Svbart@nginx.com 16*804Svbart@nginx.com nxt_app_module_t nxt_external_module = { 17*804Svbart@nginx.com 0, 18*804Svbart@nginx.com NULL, 19*804Svbart@nginx.com nxt_string("external"), 20*804Svbart@nginx.com "*", 21*804Svbart@nginx.com nxt_external_init, 22*804Svbart@nginx.com }; 23*804Svbart@nginx.com 24*804Svbart@nginx.com 25*804Svbart@nginx.com extern char **environ; 26*804Svbart@nginx.com 27*804Svbart@nginx.com 28*804Svbart@nginx.com nxt_inline nxt_int_t 29*804Svbart@nginx.com nxt_external_fd_no_cloexec(nxt_task_t *task, nxt_socket_t fd) 30*804Svbart@nginx.com { 31*804Svbart@nginx.com int res, flags; 32*804Svbart@nginx.com 33*804Svbart@nginx.com if (fd == -1) { 34*804Svbart@nginx.com return NXT_OK; 35*804Svbart@nginx.com } 36*804Svbart@nginx.com 37*804Svbart@nginx.com flags = fcntl(fd, F_GETFD); 38*804Svbart@nginx.com 39*804Svbart@nginx.com if (nxt_slow_path(flags == -1)) { 40*804Svbart@nginx.com nxt_alert(task, "fcntl(%d, F_GETFD) failed %E", fd, nxt_errno); 41*804Svbart@nginx.com return NXT_ERROR; 42*804Svbart@nginx.com } 43*804Svbart@nginx.com 44*804Svbart@nginx.com flags &= ~FD_CLOEXEC; 45*804Svbart@nginx.com 46*804Svbart@nginx.com res = fcntl(fd, F_SETFD, flags); 47*804Svbart@nginx.com 48*804Svbart@nginx.com if (nxt_slow_path(res == -1)) { 49*804Svbart@nginx.com nxt_alert(task, "fcntl(%d, F_SETFD) failed %E", fd, nxt_errno); 50*804Svbart@nginx.com return NXT_ERROR; 51*804Svbart@nginx.com } 52*804Svbart@nginx.com 53*804Svbart@nginx.com nxt_fd_blocking(task, fd); 54*804Svbart@nginx.com 55*804Svbart@nginx.com return NXT_OK; 56*804Svbart@nginx.com } 57*804Svbart@nginx.com 58*804Svbart@nginx.com 59*804Svbart@nginx.com static nxt_int_t 60*804Svbart@nginx.com nxt_external_init(nxt_task_t *task, nxt_common_app_conf_t *conf) 61*804Svbart@nginx.com { 62*804Svbart@nginx.com char **argv; 63*804Svbart@nginx.com u_char buf[256]; 64*804Svbart@nginx.com u_char *p, *end; 65*804Svbart@nginx.com uint32_t index; 66*804Svbart@nginx.com size_t size; 67*804Svbart@nginx.com nxt_str_t str; 68*804Svbart@nginx.com nxt_int_t rc; 69*804Svbart@nginx.com nxt_uint_t i, argc; 70*804Svbart@nginx.com nxt_port_t *my_port, *main_port; 71*804Svbart@nginx.com nxt_runtime_t *rt; 72*804Svbart@nginx.com nxt_conf_value_t *value; 73*804Svbart@nginx.com nxt_external_app_conf_t *c; 74*804Svbart@nginx.com 75*804Svbart@nginx.com rt = task->thread->runtime; 76*804Svbart@nginx.com 77*804Svbart@nginx.com main_port = rt->port_by_type[NXT_PROCESS_MAIN]; 78*804Svbart@nginx.com my_port = nxt_runtime_port_find(rt, nxt_pid, 0); 79*804Svbart@nginx.com 80*804Svbart@nginx.com if (nxt_slow_path(main_port == NULL || my_port == NULL)) { 81*804Svbart@nginx.com return NXT_ERROR; 82*804Svbart@nginx.com } 83*804Svbart@nginx.com 84*804Svbart@nginx.com rc = nxt_external_fd_no_cloexec(task, main_port->pair[1]); 85*804Svbart@nginx.com if (nxt_slow_path(rc != NXT_OK)) { 86*804Svbart@nginx.com return NXT_ERROR; 87*804Svbart@nginx.com } 88*804Svbart@nginx.com 89*804Svbart@nginx.com rc = nxt_external_fd_no_cloexec(task, my_port->pair[0]); 90*804Svbart@nginx.com if (nxt_slow_path(rc != NXT_OK)) { 91*804Svbart@nginx.com return NXT_ERROR; 92*804Svbart@nginx.com } 93*804Svbart@nginx.com 94*804Svbart@nginx.com end = buf + sizeof(buf); 95*804Svbart@nginx.com 96*804Svbart@nginx.com p = nxt_sprintf(buf, end, 97*804Svbart@nginx.com "%s;%uD;" 98*804Svbart@nginx.com "%PI,%ud,%d;" 99*804Svbart@nginx.com "%PI,%ud,%d;" 100*804Svbart@nginx.com "%d,%Z", 101*804Svbart@nginx.com NXT_VERSION, my_port->process->init->stream, 102*804Svbart@nginx.com main_port->pid, main_port->id, main_port->pair[1], 103*804Svbart@nginx.com my_port->pid, my_port->id, my_port->pair[0], 104*804Svbart@nginx.com 2); 105*804Svbart@nginx.com 106*804Svbart@nginx.com if (nxt_slow_path(p == end)) { 107*804Svbart@nginx.com nxt_alert(task, "internal error: buffer too small for NXT_UNIT_INIT"); 108*804Svbart@nginx.com 109*804Svbart@nginx.com return NXT_ERROR; 110*804Svbart@nginx.com } 111*804Svbart@nginx.com 112*804Svbart@nginx.com nxt_debug(task, "update "NXT_UNIT_INIT_ENV"=%s", buf); 113*804Svbart@nginx.com 114*804Svbart@nginx.com rc = setenv(NXT_UNIT_INIT_ENV, (char *) buf, 1); 115*804Svbart@nginx.com if (nxt_slow_path(rc == -1)) { 116*804Svbart@nginx.com nxt_alert(task, "setenv("NXT_UNIT_INIT_ENV", %s) failed %E", buf, 117*804Svbart@nginx.com nxt_errno); 118*804Svbart@nginx.com 119*804Svbart@nginx.com return NXT_ERROR; 120*804Svbart@nginx.com } 121*804Svbart@nginx.com 122*804Svbart@nginx.com c = &conf->u.external; 123*804Svbart@nginx.com 124*804Svbart@nginx.com argc = 2; 125*804Svbart@nginx.com size = 0; 126*804Svbart@nginx.com 127*804Svbart@nginx.com if (c->arguments != NULL) { 128*804Svbart@nginx.com 129*804Svbart@nginx.com for (index = 0; /* void */ ; index++) { 130*804Svbart@nginx.com value = nxt_conf_get_array_element(c->arguments, index); 131*804Svbart@nginx.com if (value == NULL) { 132*804Svbart@nginx.com break; 133*804Svbart@nginx.com } 134*804Svbart@nginx.com 135*804Svbart@nginx.com nxt_conf_get_string(value, &str); 136*804Svbart@nginx.com 137*804Svbart@nginx.com size += str.length + 1; 138*804Svbart@nginx.com argc++; 139*804Svbart@nginx.com } 140*804Svbart@nginx.com } 141*804Svbart@nginx.com 142*804Svbart@nginx.com argv = nxt_malloc(argc * sizeof(argv[0]) + size); 143*804Svbart@nginx.com if (nxt_slow_path(argv == NULL)) { 144*804Svbart@nginx.com nxt_alert(task, "failed to allocate arguments"); 145*804Svbart@nginx.com return NXT_ERROR; 146*804Svbart@nginx.com } 147*804Svbart@nginx.com 148*804Svbart@nginx.com argv[0] = c->executable; 149*804Svbart@nginx.com i = 1; 150*804Svbart@nginx.com 151*804Svbart@nginx.com if (c->arguments != NULL) { 152*804Svbart@nginx.com p = (u_char *) &argv[argc]; 153*804Svbart@nginx.com 154*804Svbart@nginx.com for (index = 0; /* void */ ; index++) { 155*804Svbart@nginx.com value = nxt_conf_get_array_element(c->arguments, index); 156*804Svbart@nginx.com if (value == NULL) { 157*804Svbart@nginx.com break; 158*804Svbart@nginx.com } 159*804Svbart@nginx.com 160*804Svbart@nginx.com argv[i++] = (char *) p; 161*804Svbart@nginx.com 162*804Svbart@nginx.com nxt_conf_get_string(value, &str); 163*804Svbart@nginx.com 164*804Svbart@nginx.com p = nxt_cpymem(p, str.start, str.length); 165*804Svbart@nginx.com *p++ = '\0'; 166*804Svbart@nginx.com } 167*804Svbart@nginx.com } 168*804Svbart@nginx.com 169*804Svbart@nginx.com argv[i] = NULL; 170*804Svbart@nginx.com 171*804Svbart@nginx.com (void) execve(c->executable, argv, environ); 172*804Svbart@nginx.com 173*804Svbart@nginx.com nxt_alert(task, "execve(%s) failed %E", c->executable, nxt_errno); 174*804Svbart@nginx.com 175*804Svbart@nginx.com nxt_free(argv); 176*804Svbart@nginx.com 177*804Svbart@nginx.com return NXT_ERROR; 178*804Svbart@nginx.com } 179