xref: /unit/src/nxt_external.c (revision 804)
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