62
62
#include <inttypes.h> /* PRIu16, PRIdMAX, intmax_t,
64
64
#include <assert.h> /* assert() */
65
#include <errno.h> /* perror(), errno */
65
#include <errno.h> /* perror(), errno, program_invocation_short_name */
66
66
#include <time.h> /* nanosleep(), time(), sleep() */
67
67
#include <net/if.h> /* ioctl, ifreq, SIOCGIFFLAGS, IFF_UP,
68
68
SIOCSIFFLAGS, if_indextoname(),
130
130
static const char sys_class_net[] = "/sys/class/net";
131
131
char *connect_to = NULL;
133
/* Doubly linked list that need to be circular linked when ever used */
134
typedef struct server{
137
AvahiIfIndex if_index;
139
struct timespec last_seen;
133
144
/* Used for passing in values through the Avahi callback functions */
135
146
AvahiSimplePoll *simple_poll;
139
150
gnutls_dh_params_t dh_params;
140
151
const char *priority;
153
server *current_server;
142
154
} mandos_context;
144
156
/* global context so signal handler can reach it*/
145
157
mandos_context mc = { .simple_poll = NULL, .server = NULL,
146
158
.dh_bits = 1024, .priority = "SECURE256"
147
":!CTYPE-X.509:+CTYPE-OPENPGP" };
159
":!CTYPE-X.509:+CTYPE-OPENPGP", .current_server = NULL };
149
161
sig_atomic_t quit_now = 0;
150
162
int signal_received = 0;
164
/* Function to use when printing errors */
165
void perror_plus(const char *print_text){
166
fprintf(stderr, "Mandos plugin %s: ", program_invocation_short_name);
153
171
* Make additional room in "buffer" for at least BUFFER_SIZE more
154
172
* bytes. "buffer_capacity" is how much is currently allocated,
166
184
return buffer_capacity;
187
int add_server(const char *ip, uint16_t port,
188
AvahiIfIndex if_index,
191
server *new_server = malloc(sizeof(server));
192
if(new_server == NULL){
193
perror_plus("malloc");
196
*new_server = (server){ .ip = strdup(ip),
198
.if_index = if_index,
200
if(new_server->ip == NULL){
201
perror_plus("strdup");
204
/* uniqe case of first server */
205
if (mc.current_server == NULL){
206
new_server->next = new_server;
207
new_server->prev = new_server;
208
mc.current_server = new_server;
209
/* Placing the new server last in the list */
211
new_server->next = mc.current_server;
212
new_server->prev = mc.current_server->prev;
213
new_server->prev->next = new_server;
214
mc.current_server->prev = new_server;
216
ret = clock_gettime(CLOCK_MONOTONIC, &mc.current_server->last_seen);
218
perror_plus("clock_gettime");
170
225
* Initialize GPGME.
337
392
/* Seek back to the beginning of the GPGME plaintext data buffer */
338
393
if(gpgme_data_seek(dh_plain, (off_t)0, SEEK_SET) == -1){
339
perror("gpgme_data_seek");
394
perror_plus("gpgme_data_seek");
340
395
plaintext_length = -1;
341
396
goto decrypt_end;
651
706
if(af == AF_INET6 and if_index != AVAHI_IF_UNSPEC){
652
707
char interface[IF_NAMESIZE];
653
708
if(if_indextoname((unsigned int)if_index, interface) == NULL){
654
perror("if_indextoname");
709
perror_plus("if_indextoname");
656
711
fprintf(stderr, "Connection to: %s%%%s, port %" PRIu16 "\n",
657
712
ip, interface, port);
671
726
sizeof(addrstr));
673
728
if(pcret == NULL){
729
perror_plus("inet_ntop");
676
731
if(strcmp(addrstr, ip) != 0){
677
732
fprintf(stderr, "Canonical address form: %s\n", addrstr);
952
1007
avahi_proto_to_af(proto));
954
1009
avahi_simple_poll_quit(mc.simple_poll);
1011
ret = add_server(ip, port, interface,
1012
avahi_proto_to_af(proto));
1203
int avahi_loop_with_timeout(AvahiSimplePoll *s, int retry_interval){
1205
struct timespec now;
1206
struct timespec waited_time;
1207
intmax_t block_time;
1210
if(mc.current_server == NULL){
1212
fprintf(stderr, "Wait until first server is found. No timeout!\n");
1214
ret = avahi_simple_poll_iterate(s, -1);
1217
fprintf(stderr, "Check current_server if we should run it, or wait\n");
1219
/* the current time */
1220
ret = clock_gettime(CLOCK_MONOTONIC, &now);
1222
perror_plus("clock_gettime");
1225
/* Calculating in ms how long time between now and server
1226
who we visted longest time ago. Now - last seen. */
1227
waited_time.tv_sec = now.tv_sec - mc.current_server->last_seen.tv_sec;
1228
waited_time.tv_nsec = now.tv_nsec - mc.current_server->last_seen.tv_nsec;
1229
/* total time is 10s/10000ms. Converting to s to ms by 1000/s, and ns to ms by divind by 1000000. */
1230
block_time = (retry_interval - ((intmax_t)waited_time.tv_sec * 1000)) - ((intmax_t)waited_time.tv_nsec / 1000000);
1233
fprintf(stderr, "Blocking for %ld ms\n", block_time);
1236
if(block_time <= 0){
1237
ret = start_mandos_communication(mc.current_server->ip,
1238
mc.current_server->port,
1239
mc.current_server->if_index,
1240
mc.current_server->af);
1242
avahi_simple_poll_quit(mc.simple_poll);
1245
ret = clock_gettime(CLOCK_MONOTONIC, &mc.current_server->last_seen);
1247
perror_plus("clock_gettime");
1250
mc.current_server = mc.current_server->next;
1251
block_time = 0; /* call avahi to find new mandos servers, but dont block */
1254
ret = avahi_simple_poll_iterate(s, (int)block_time);
1257
if (ret > 0 or errno != EINTR) {
1258
return (ret != 1) ? ret : 0;
1144
1264
int main(int argc, char *argv[]){
1145
1265
AvahiSServiceBrowser *sb = NULL;
1163
1283
bool gnutls_initialized = false;
1164
1284
bool gpgme_initialized = false;
1165
1285
float delay = 2.5f;
1286
double retry_interval = 10; /* 10s between retrying a server and checking again*/
1167
1288
struct sigaction old_sigterm_action = { .sa_handler = SIG_DFL };
1168
1289
struct sigaction sigterm_action = { .sa_handler = handle_sigterm };
1222
1343
.arg = "SECONDS",
1223
1344
.doc = "Maximum delay to wait for interface startup",
1346
{ .name = "retry", .key = 132,
1348
.doc = "Retry interval used when denied by the mandos server",
1226
1351
* These reproduce what we would get without ARGP_NO_HELP
1271
1396
if(errno != 0 or tmp == arg or *tmp != '\0'){
1272
1397
argp_error(state, "Bad delay");
1399
case 132: /* --retry */
1401
retry_interval = strtod(arg, &tmp);
1402
if(errno != 0 or tmp == arg or *tmp != '\0'
1403
or (retry_interval * 1000) > INT_MAX){
1404
argp_error(state, "Bad retry interval");
1276
1408
* These reproduce what we would get without ARGP_NO_HELP
1356
1488
sigemptyset(&sigterm_action.sa_mask);
1357
1489
ret = sigaddset(&sigterm_action.sa_mask, SIGINT);
1359
perror("sigaddset");
1491
perror_plus("sigaddset");
1360
1492
exitcode = EX_OSERR;
1363
1495
ret = sigaddset(&sigterm_action.sa_mask, SIGHUP);
1365
perror("sigaddset");
1497
perror_plus("sigaddset");
1366
1498
exitcode = EX_OSERR;
1369
1501
ret = sigaddset(&sigterm_action.sa_mask, SIGTERM);
1371
perror("sigaddset");
1503
perror_plus("sigaddset");
1372
1504
exitcode = EX_OSERR;
1379
1511
ret = sigaction(SIGINT, NULL, &old_sigterm_action);
1381
perror("sigaction");
1513
perror_plus("sigaction");
1382
1514
return EX_OSERR;
1384
1516
if(old_sigterm_action.sa_handler != SIG_IGN){
1385
1517
ret = sigaction(SIGINT, &sigterm_action, NULL);
1387
perror("sigaction");
1519
perror_plus("sigaction");
1388
1520
exitcode = EX_OSERR;
1392
1524
ret = sigaction(SIGHUP, NULL, &old_sigterm_action);
1394
perror("sigaction");
1526
perror_plus("sigaction");
1395
1527
return EX_OSERR;
1397
1529
if(old_sigterm_action.sa_handler != SIG_IGN){
1398
1530
ret = sigaction(SIGHUP, &sigterm_action, NULL);
1400
perror("sigaction");
1532
perror_plus("sigaction");
1401
1533
exitcode = EX_OSERR;
1405
1537
ret = sigaction(SIGTERM, NULL, &old_sigterm_action);
1407
perror("sigaction");
1539
perror_plus("sigaction");
1408
1540
return EX_OSERR;
1410
1542
if(old_sigterm_action.sa_handler != SIG_IGN){
1411
1543
ret = sigaction(SIGTERM, &sigterm_action, NULL);
1413
perror("sigaction");
1545
perror_plus("sigaction");
1414
1546
exitcode = EX_OSERR;
1443
1575
bool restore_loglevel = true;
1445
1577
restore_loglevel = false;
1578
perror_plus("klogctl");
1448
1580
#endif /* __linux__ */
1450
1582
sd = socket(PF_INET6, SOCK_DGRAM, IPPROTO_IP);
1584
perror_plus("socket");
1453
1585
exitcode = EX_OSERR;
1454
1586
#ifdef __linux__
1455
1587
if(restore_loglevel){
1456
1588
ret = klogctl(7, NULL, 0);
1590
perror_plus("klogctl");
1461
1593
#endif /* __linux__ */
1464
1596
ret = seteuid(uid);
1598
perror_plus("seteuid");
1470
1602
strcpy(network.ifr_name, interface);
1471
1603
ret = ioctl(sd, SIOCGIFFLAGS, &network);
1473
perror("ioctl SIOCGIFFLAGS");
1605
perror_plus("ioctl SIOCGIFFLAGS");
1474
1606
#ifdef __linux__
1475
1607
if(restore_loglevel){
1476
1608
ret = klogctl(7, NULL, 0);
1610
perror_plus("klogctl");
1481
1613
#endif /* __linux__ */
1494
1626
ret = ioctl(sd, SIOCSIFFLAGS, &network);
1496
1628
take_down_interface = false;
1497
perror("ioctl SIOCSIFFLAGS +IFF_UP");
1629
perror_plus("ioctl SIOCSIFFLAGS +IFF_UP");
1498
1630
exitcode = EX_OSERR;
1499
1631
#ifdef __linux__
1500
1632
if(restore_loglevel){
1501
1633
ret = klogctl(7, NULL, 0);
1635
perror_plus("klogctl");
1506
1638
#endif /* __linux__ */
1509
1641
ret = seteuid(uid);
1643
perror_plus("seteuid");
1516
/* sleep checking until interface is running */
1648
/* sleep checking until interface is running. Check every 0.25s, up to total time of delay */
1517
1649
for(int i=0; i < delay * 4; i++){
1518
1650
ret = ioctl(sd, SIOCGIFFLAGS, &network);
1520
perror("ioctl SIOCGIFFLAGS");
1652
perror_plus("ioctl SIOCGIFFLAGS");
1521
1653
} else if(network.ifr_flags & IFF_RUNNING){
1524
1656
struct timespec sleeptime = { .tv_nsec = 250000000 };
1525
1657
ret = nanosleep(&sleeptime, NULL);
1526
1658
if(ret == -1 and errno != EINTR){
1527
perror("nanosleep");
1659
perror_plus("nanosleep");
1530
1662
if(not take_down_interface){
1531
1663
/* We won't need the socket anymore */
1532
1664
ret = (int)TEMP_FAILURE_RETRY(close(sd));
1666
perror_plus("close");
1537
1669
#ifdef __linux__
1712
1844
fprintf(stderr, "Starting Avahi loop search\n");
1715
avahi_simple_poll_loop(mc.simple_poll);
1847
ret = avahi_loop_with_timeout(mc.simple_poll, (int)(retry_interval * 1000));
1849
fprintf(stderr, "avahi_loop_with_timeout exited %s\n",
1850
(ret == 0) ? "successfully" : "with error");
1739
1875
if(gpgme_initialized){
1740
1876
gpgme_release(mc.ctx);
1879
/* cleans up the circular linked list of mandos servers the client has seen */
1880
if(mc.current_server != NULL){
1881
mc.current_server->prev->next = NULL;
1882
while(mc.current_server != NULL){
1883
server *next = mc.current_server->next;
1884
free(mc.current_server);
1885
mc.current_server = next;
1743
1889
/* Take down the network interface */
1744
1890
if(take_down_interface){
1747
1893
ret = seteuid(0);
1895
perror_plus("seteuid");
1751
1897
if(geteuid() == 0){
1752
1898
ret = ioctl(sd, SIOCGIFFLAGS, &network);
1754
perror("ioctl SIOCGIFFLAGS");
1900
perror_plus("ioctl SIOCGIFFLAGS");
1755
1901
} else if(network.ifr_flags & IFF_UP) {
1756
1902
network.ifr_flags &= ~(short)IFF_UP; /* clear flag */
1757
1903
ret = ioctl(sd, SIOCSIFFLAGS, &network);
1759
perror("ioctl SIOCSIFFLAGS -IFF_UP");
1905
perror_plus("ioctl SIOCSIFFLAGS -IFF_UP");
1762
1908
ret = (int)TEMP_FAILURE_RETRY(close(sd));
1910
perror_plus("close");
1766
1912
/* Lower privileges permanently */
1768
1914
ret = setuid(uid);
1916
perror_plus("setuid");