]> git.meshlink.io Git - meshlink/blob - src/control.c
Fix sending bulk data starting with a newline.
[meshlink] / src / control.c
1 /*
2     control.c -- Control socket handling.
3     Copyright (C) 2013 Guus Sliepen <guus@tinc-vpn.org>
4
5     This program is free software; you can redistribute it and/or modify
6     it under the terms of the GNU General Public License as published by
7     the Free Software Foundation; either version 2 of the License, or
8     (at your option) any later version.
9
10     This program is distributed in the hope that it will be useful,
11     but WITHOUT ANY WARRANTY; without even the implied warranty of
12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13     GNU General Public License for more details.
14
15     You should have received a copy of the GNU General Public License along
16     with this program; if not, write to the Free Software Foundation, Inc.,
17     51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
20 #include "system.h"
21 #include "crypto.h"
22 #include "conf.h"
23 #include "control.h"
24 #include "control_common.h"
25 #include "graph.h"
26 #include "logger.h"
27 #include "meta.h"
28 #include "names.h"
29 #include "net.h"
30 #include "netutl.h"
31 #include "protocol.h"
32 #include "route.h"
33 #include "utils.h"
34 #include "xalloc.h"
35
36 char controlcookie[65];
37
38 static bool control_return(connection_t *c, int type, int error) {
39         return send_request(c, "%d %d %d", CONTROL, type, error);
40 }
41
42 static bool control_ok(connection_t *c, int type) {
43         return control_return(c, type, 0);
44 }
45
46 bool control_h(connection_t *c, const char *request) {
47         int type;
48
49         if(!c->status.control || c->allow_request != CONTROL) {
50                 logger(DEBUG_ALWAYS, LOG_ERR, "Unauthorized control request from %s (%s)", c->name, c->hostname);
51                 return false;
52         }
53
54         if(sscanf(request, "%*d %d", &type) != 1) {
55                 logger(DEBUG_ALWAYS, LOG_ERR, "Got bad %s from %s (%s)", "CONTROL", c->name, c->hostname);
56                 return false;
57         }
58
59         switch (type) {
60                 case REQ_STOP:
61                         event_exit();
62                         return control_ok(c, REQ_STOP);
63
64                 case REQ_DUMP_NODES:
65                         return dump_nodes(c);
66
67                 case REQ_DUMP_EDGES:
68                         return dump_edges(c);
69
70                 case REQ_DUMP_SUBNETS:
71                         return dump_subnets(c);
72
73                 case REQ_DUMP_CONNECTIONS:
74                         return dump_connections(c);
75
76                 case REQ_PURGE:
77                         purge();
78                         return control_ok(c, REQ_PURGE);
79
80                 case REQ_SET_DEBUG: {
81                         int new_level;
82                         if(sscanf(request, "%*d %*d %d", &new_level) != 1)
83                                 return false;
84                         send_request(c, "%d %d %d", CONTROL, REQ_SET_DEBUG, debug_level);
85                         if(new_level >= 0)
86                                 debug_level = new_level;
87                         return true;
88                 }
89
90                 case REQ_RETRY:
91                         retry();
92                         return control_ok(c, REQ_RETRY);
93
94                 case REQ_RELOAD:
95                         logger(DEBUG_ALWAYS, LOG_NOTICE, "Got '%s' command", "reload");
96                         int result = reload_configuration();
97                         return control_return(c, REQ_RELOAD, result);
98
99                 case REQ_DISCONNECT: {
100                         char name[MAX_STRING_SIZE];
101                         bool found = false;
102
103                         if(sscanf(request, "%*d %*d " MAX_STRING, name) != 1)
104                                 return control_return(c, REQ_DISCONNECT, -1);
105
106                         for list_each(connection_t, other, connection_list) {
107                                 if(strcmp(other->name, name))
108                                         continue;
109                                 terminate_connection(other, other->status.active);
110                                 found = true;
111                         }
112
113                         return control_return(c, REQ_DISCONNECT, found ? 0 : -2);
114                 }
115
116                 case REQ_DUMP_TRAFFIC:
117                         return dump_traffic(c);
118
119                 case REQ_PCAP:
120                         sscanf(request, "%*d %*d %d", &c->outmaclength);
121                         c->status.pcap = true;
122                         pcap = true;
123                         return true;
124
125                 case REQ_LOG:
126                         sscanf(request, "%*d %*d %d", &c->outcompression);
127                         c->status.log = true;
128                         logcontrol = true;
129                         return true;
130
131                 default:
132                         return send_request(c, "%d %d", CONTROL, REQ_INVALID);
133         }
134 }
135
136 bool init_control(void) {
137         randomize(controlcookie, sizeof controlcookie / 2);
138         bin2hex(controlcookie, controlcookie, sizeof controlcookie / 2);
139
140         mode_t mask = umask(0);
141         umask(mask | 077);
142         FILE *f = fopen(pidfilename, "w");
143         umask(mask);
144
145         if(!f) {
146                 logger(DEBUG_ALWAYS, LOG_ERR, "Cannot write control socket cookie file %s: %s", pidfilename, strerror(errno));
147                 return false;
148         }
149
150         // Get the address and port of the first listening socket
151
152         char *localhost = NULL;
153         sockaddr_t sa;
154         socklen_t len = sizeof sa;
155
156         // Make sure we have a valid address, and map 0.0.0.0 and :: to 127.0.0.1 and ::1.
157
158         if(getsockname(listen_socket[0].tcp.fd, (struct sockaddr *)&sa, &len)) {
159                 xasprintf(&localhost, "127.0.0.1 port %s", myport);
160         } else {
161                 if(sa.sa.sa_family == AF_INET) {
162                         if(sa.in.sin_addr.s_addr == 0)
163                                 sa.in.sin_addr.s_addr = htonl(0x7f000001);
164                 } else if(sa.sa.sa_family == AF_INET6) {
165                         static const uint8_t zero[16] = {0};
166                         if(!memcmp(sa.in6.sin6_addr.s6_addr, zero, sizeof zero))
167                                 sa.in6.sin6_addr.s6_addr[15] = 1;
168                 }
169
170                 localhost = sockaddr2hostname(&sa);
171         }
172
173         fprintf(f, "%d %s %s\n", (int)getpid(), controlcookie, localhost);
174
175         free(localhost);
176         fclose(f);
177
178 #ifndef HAVE_MINGW
179         int unix_fd = socket(AF_UNIX, SOCK_STREAM, 0);
180         if(unix_fd < 0) {
181                 logger(DEBUG_ALWAYS, LOG_ERR, "Could not create UNIX socket: %s", sockstrerror(errno));
182                 return false;
183         }
184
185         struct sockaddr_un sun;
186         sun.sun_family = AF_UNIX;
187         strncpy(sun.sun_path, unixsocketname, sizeof sun.sun_path);
188
189         if(connect(unix_fd, (struct sockaddr *)&sun, sizeof sun) >= 0) {
190                 logger(DEBUG_ALWAYS, LOG_ERR, "UNIX socket %s is still in use!", unixsocketname);
191                 return false;
192         }
193
194         unlink(unixsocketname);
195
196         umask(mask | 077);
197         int result = bind(unix_fd, (struct sockaddr *)&sun, sizeof sun);
198         umask(mask);
199
200         if(result < 0) {
201                 logger(DEBUG_ALWAYS, LOG_ERR, "Could not bind UNIX socket to %s: %s", unixsocketname, sockstrerror(errno));
202                 return false;
203         }
204
205         if(listen(unix_fd, 3) < 0) {
206                 logger(DEBUG_ALWAYS, LOG_ERR, "Could not listen on UNIX socket %s: %s", unixsocketname, sockstrerror(errno));
207                 return false;
208         }
209
210         io_add(&unix_socket, handle_new_unix_connection, &unix_socket, unix_fd, IO_READ);
211 #endif
212
213         return true;
214 }
215
216 void exit_control(void) {
217 #ifndef HAVE_MINGW
218         unlink(unixsocketname);
219         io_del(&unix_socket);
220         close(unix_socket.fd);
221 #endif
222
223         unlink(pidfilename);
224 }