src/socket.c

   1 /*-*- Mode: C; c-basic-offset: 8 -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   under the terms of the GNU General Public License as published by
  10   the Free Software Foundation; either version 2 of the License, or
  11   (at your option) any later version.
  12
  13   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <sys/types.h>
  23 #include <sys/stat.h>
  24 #include <unistd.h>
  25 #include <errno.h>
  26 #include <fcntl.h>
  27 #include <sys/epoll.h>
  28 #include <signal.h>
  29 #include <arpa/inet.h>
  30
  31 #include "unit.h"
  32 #include "socket.h"
  33 #include "netinet/tcp.h"
  34 #include "log.h"
  35 #include "load-dropin.h"
  36 #include "load-fragment.h"
  37 #include "strv.h"
  38 #include "unit-name.h"
  39 #include "dbus-socket.h"
  40 #include "missing.h"
  41 #include "special.h"
  42 #include "bus-errors.h"
  43
  44 static const UnitActiveState state_translation_table[_SOCKET_STATE_MAX] = {
  45         [SOCKET_DEAD] = UNIT_INACTIVE,
  46         [SOCKET_START_PRE] = UNIT_ACTIVATING,
  47         [SOCKET_START_POST] = UNIT_ACTIVATING,
  48         [SOCKET_LISTENING] = UNIT_ACTIVE,
  49         [SOCKET_RUNNING] = UNIT_ACTIVE,
  50         [SOCKET_STOP_PRE] = UNIT_DEACTIVATING,
  51         [SOCKET_STOP_PRE_SIGTERM] = UNIT_DEACTIVATING,
  52         [SOCKET_STOP_PRE_SIGKILL] = UNIT_DEACTIVATING,
  53         [SOCKET_STOP_POST] = UNIT_DEACTIVATING,
  54         [SOCKET_FINAL_SIGTERM] = UNIT_DEACTIVATING,
  55         [SOCKET_FINAL_SIGKILL] = UNIT_DEACTIVATING,
  56         [SOCKET_MAINTENANCE] = UNIT_MAINTENANCE
  57 };
  58
  59 static void socket_init(Unit *u) {
  60         Socket *s = SOCKET(u);
  61
  62         assert(u);
  63         assert(u->meta.load_state == UNIT_STUB);
  64
  65         s->backlog = SOMAXCONN;
  66         s->timeout_usec = DEFAULT_TIMEOUT_USEC;
  67         s->directory_mode = 0755;
  68         s->socket_mode = 0666;
  69
  70         s->max_connections = 64;
  71
  72         s->priority = -1;
  73         s->ip_tos = -1;
  74         s->ip_ttl = -1;
  75         s->mark = -1;
  76
  77         exec_context_init(&s->exec_context);
  78
  79         s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
  80 }
  81
  82 static void socket_unwatch_control_pid(Socket *s) {
  83         assert(s);
  84
  85         if (s->control_pid <= 0)
  86                 return;
  87
  88         unit_unwatch_pid(UNIT(s), s->control_pid);
  89         s->control_pid = 0;
  90 }
  91
  92 static void socket_done(Unit *u) {
  93         Socket *s = SOCKET(u);
  94         SocketPort *p;
  95         Meta *i;
  96
  97         assert(s);
  98
  99         while ((p = s->ports)) {
 100                 LIST_REMOVE(SocketPort, port, s->ports, p);
 101
 102                 if (p->fd >= 0) {
 103                         unit_unwatch_fd(UNIT(s), &p->fd_watch);
 104                         close_nointr_nofail(p->fd);
 105                 }
 106
 107                 free(p->path);
 108                 free(p);
 109         }
 110
 111         exec_context_done(&s->exec_context);
 112         exec_command_free_array(s->exec_command, _SOCKET_EXEC_COMMAND_MAX);
 113         s->control_command = NULL;
 114
 115         socket_unwatch_control_pid(s);
 116
 117         s->service = NULL;
 118
 119         free(s->tcp_congestion);
 120         s->tcp_congestion = NULL;
 121
 122         free(s->bind_to_device);
 123         s->bind_to_device = NULL;
 124
 125         unit_unwatch_timer(u, &s->timer_watch);
 126
 127         /* Make sure no service instance refers to us anymore. */
 128         LIST_FOREACH(units_per_type, i, u->meta.manager->units_per_type[UNIT_SERVICE]) {
 129                 Service *service = (Service *) i;
 130
 131                 if (service->socket == s)
 132                         service->socket = NULL;
 133         }
 134 }
 135
 136 static int socket_instantiate_service(Socket *s) {
 137         char *prefix, *name;
 138         int r;
 139         Unit *u;
 140
 141         assert(s);
 142
 143         /* This fills in s->service if it isn't filled in yet. For
 144          * Accept=yes sockets we create the next connection service
 145          * here. For Accept=no this is mostly a NOP since the service
 146          * is figured out at load time anyway. */
 147
 148         if (s->service)
 149                 return 0;
 150
 151         assert(s->accept);
 152
 153         if (!(prefix = unit_name_to_prefix(s->meta.id)))
 154                 return -ENOMEM;
 155
 156         r = asprintf(&name, "%s@%u.service", prefix, s->n_accepted);
 157         free(prefix);
 158
 159         if (r < 0)
 160                 return -ENOMEM;
 161
 162         r = manager_load_unit(s->meta.manager, name, NULL, NULL, &u);
 163         free(name);
 164
 165         if (r < 0)
 166                 return r;
 167
 168         s->service = SERVICE(u);
 169         return 0;
 170 }
 171
 172 static bool have_non_accept_socket(Socket *s) {
 173         SocketPort *p;
 174
 175         assert(s);
 176
 177         if (!s->accept)
 178                 return true;
 179
 180         LIST_FOREACH(port, p, s->ports) {
 181
 182                 if (p->type != SOCKET_SOCKET)
 183                         return true;
 184
 185                 if (!socket_address_can_accept(&p->address))
 186                         return true;
 187         }
 188
 189         return false;
 190 }
 191
 192 static int socket_verify(Socket *s) {
 193         assert(s);
 194
 195         if (s->meta.load_state != UNIT_LOADED)
 196                 return 0;
 197
 198         if (!s->ports) {
 199                 log_error("%s lacks Listen setting. Refusing.", s->meta.id);
 200                 return -EINVAL;
 201         }
 202
 203         if (s->accept && have_non_accept_socket(s)) {
 204                 log_error("%s configured for accepting sockets, but sockets are non-accepting. Refusing.", s->meta.id);
 205                 return -EINVAL;
 206         }
 207
 208         if (s->accept && s->max_connections <= 0) {
 209                 log_error("%s's MaxConnection setting too small. Refusing.", s->meta.id);
 210                 return -EINVAL;
 211         }
 212
 213         if (s->exec_context.pam_name && s->exec_context.kill_mode != KILL_CONTROL_GROUP) {
 214                 log_error("%s has PAM enabled. Kill mode must be set to 'control-group'. Refusing.", s->meta.id);
 215                 return -EINVAL;
 216         }
 217
 218         return 0;
 219 }
 220
 221 static bool socket_needs_mount(Socket *s, const char *prefix) {
 222         SocketPort *p;
 223
 224         assert(s);
 225
 226         LIST_FOREACH(port, p, s->ports) {
 227
 228                 if (p->type == SOCKET_SOCKET) {
 229                         if (socket_address_needs_mount(&p->address, prefix))
 230                                 return true;
 231                 } else {
 232                         assert(p->type == SOCKET_FIFO);
 233                         if (path_startswith(p->path, prefix))
 234                                 return true;
 235                 }
 236         }
 237
 238         return false;
 239 }
 240
 241 int socket_add_one_mount_link(Socket *s, Mount *m) {
 242         int r;
 243
 244         assert(s);
 245         assert(m);
 246
 247         if (s->meta.load_state != UNIT_LOADED ||
 248             m->meta.load_state != UNIT_LOADED)
 249                 return 0;
 250
 251         if (!socket_needs_mount(s, m->where))
 252                 return 0;
 253
 254         if ((r = unit_add_two_dependencies(UNIT(s), UNIT_AFTER, UNIT_REQUIRES, UNIT(m), true)) < 0)
 255                 return r;
 256
 257         return 0;
 258 }
 259
 260 static int socket_add_mount_links(Socket *s) {
 261         Meta *other;
 262         int r;
 263
 264         assert(s);
 265
 266         LIST_FOREACH(units_per_type, other, s->meta.manager->units_per_type[UNIT_MOUNT])
 267                 if ((r = socket_add_one_mount_link(s, (Mount*) other)) < 0)
 268                         return r;
 269
 270         return 0;
 271 }
 272
 273 static int socket_add_device_link(Socket *s) {
 274         char *t;
 275         int r;
 276
 277         assert(s);
 278
 279         if (!s->bind_to_device)
 280                 return 0;
 281
 282         if (asprintf(&t, "/sys/subsystem/net/devices/%s", s->bind_to_device) < 0)
 283                 return -ENOMEM;
 284
 285         r = unit_add_node_link(UNIT(s), t, false);
 286         free(t);
 287
 288         return r;
 289 }
 290
 291 static int socket_add_default_dependencies(Socket *s) {
 292         int r;
 293         assert(s);
 294
 295         if (s->meta.manager->running_as == MANAGER_SYSTEM)
 296                 if ((r = unit_add_two_dependencies_by_name(UNIT(s), UNIT_AFTER, UNIT_REQUIRES, SPECIAL_SYSINIT_TARGET, NULL, true)) < 0)
 297                         return r;
 298
 299         return unit_add_two_dependencies_by_name(UNIT(s), UNIT_BEFORE, UNIT_CONFLICTS, SPECIAL_SHUTDOWN_TARGET, NULL, true);
 300 }
 301
 302 static int socket_load(Unit *u) {
 303         Socket *s = SOCKET(u);
 304         int r;
 305
 306         assert(u);
 307         assert(u->meta.load_state == UNIT_STUB);
 308
 309         if ((r = unit_load_fragment_and_dropin(u)) < 0)
 310                 return r;
 311
 312         /* This is a new unit? Then let's add in some extras */
 313         if (u->meta.load_state == UNIT_LOADED) {
 314
 315                 if (have_non_accept_socket(s)) {
 316                         if ((r = unit_load_related_unit(u, ".service", (Unit**) &s->service)) < 0)
 317                                 return r;
 318
 319                         if ((r = unit_add_dependency(u, UNIT_BEFORE, UNIT(s->service), true)) < 0)
 320                                 return r;
 321                 }
 322
 323                 if ((r = socket_add_mount_links(s)) < 0)
 324                         return r;
 325
 326                 if ((r = socket_add_device_link(s)) < 0)
 327                         return r;
 328
 329                 if ((r = unit_add_exec_dependencies(u, &s->exec_context)) < 0)
 330                         return r;
 331
 332                 if ((r = unit_add_default_cgroup(u)) < 0)
 333                         return r;
 334
 335                 if (s->meta.default_dependencies)
 336                         if ((r = socket_add_default_dependencies(s)) < 0)
 337                                 return r;
 338         }
 339
 340         return socket_verify(s);
 341 }
 342
 343 static const char* listen_lookup(int type) {
 344
 345         if (type == SOCK_STREAM)
 346                 return "ListenStream";
 347         else if (type == SOCK_DGRAM)
 348                 return "ListenDatagram";
 349         else if (type == SOCK_SEQPACKET)
 350                 return "ListenSequentialPacket";
 351
 352         assert_not_reached("Unknown socket type");
 353         return NULL;
 354 }
 355
 356 static void socket_dump(Unit *u, FILE *f, const char *prefix) {
 357
 358         SocketExecCommand c;
 359         Socket *s = SOCKET(u);
 360         SocketPort *p;
 361         const char *prefix2;
 362         char *p2;
 363
 364         assert(s);
 365         assert(f);
 366
 367         p2 = strappend(prefix, "\t");
 368         prefix2 = p2 ? p2 : prefix;
 369
 370         fprintf(f,
 371                 "%sSocket State: %s\n"
 372                 "%sBindIPv6Only: %s\n"
 373                 "%sBacklog: %u\n"
 374                 "%sSocketMode: %04o\n"
 375                 "%sDirectoryMode: %04o\n"
 376                 "%sKeepAlive: %s\n"
 377                 "%sFreeBind: %s\n"
 378                 "%sTCPCongestion: %s\n",
 379                 prefix, socket_state_to_string(s->state),
 380                 prefix, socket_address_bind_ipv6_only_to_string(s->bind_ipv6_only),
 381                 prefix, s->backlog,
 382                 prefix, s->socket_mode,
 383                 prefix, s->directory_mode,
 384                 prefix, yes_no(s->keep_alive),
 385                 prefix, yes_no(s->free_bind),
 386                 prefix, s->tcp_congestion);
 387
 388         if (s->control_pid > 0)
 389                 fprintf(f,
 390                         "%sControl PID: %lu\n",
 391                         prefix, (unsigned long) s->control_pid);
 392
 393         if (s->bind_to_device)
 394                 fprintf(f,
 395                         "%sBindToDevice: %s\n",
 396                         prefix, s->bind_to_device);
 397
 398         if (s->accept)
 399                 fprintf(f,
 400                         "%sAccepted: %u\n"
 401                         "%sNConnections: %u\n"
 402                         "%sMaxConnections: %u\n",
 403                         prefix, s->n_accepted,
 404                         prefix, s->n_connections,
 405                         prefix, s->max_connections);
 406
 407         if (s->priority >= 0)
 408                 fprintf(f,
 409                         "%sPriority: %i\n",
 410                         prefix, s->priority);
 411
 412         if (s->receive_buffer > 0)
 413                 fprintf(f,
 414                         "%sReceiveBuffer: %zu\n",
 415                         prefix, s->receive_buffer);
 416
 417         if (s->send_buffer > 0)
 418                 fprintf(f,
 419                         "%sSendBuffer: %zu\n",
 420                         prefix, s->send_buffer);
 421
 422         if (s->ip_tos >= 0)
 423                 fprintf(f,
 424                         "%sIPTOS: %i\n",
 425                         prefix, s->ip_tos);
 426
 427         if (s->ip_ttl >= 0)
 428                 fprintf(f,
 429                         "%sIPTTL: %i\n",
 430                         prefix, s->ip_ttl);
 431
 432         if (s->pipe_size > 0)
 433                 fprintf(f,
 434                         "%sPipeSize: %zu\n",
 435                         prefix, s->pipe_size);
 436
 437         if (s->mark >= 0)
 438                 fprintf(f,
 439                         "%sMark: %i\n",
 440                         prefix, s->mark);
 441
 442         LIST_FOREACH(port, p, s->ports) {
 443
 444                 if (p->type == SOCKET_SOCKET) {
 445                         const char *t;
 446                         int r;
 447                         char *k;
 448
 449                         if ((r = socket_address_print(&p->address, &k)) < 0)
 450                                 t = strerror(-r);
 451                         else
 452                                 t = k;
 453
 454                         fprintf(f, "%s%s: %s\n", prefix, listen_lookup(p->address.type), k);
 455                         free(k);
 456                 } else
 457                         fprintf(f, "%sListenFIFO: %s\n", prefix, p->path);
 458         }
 459
 460         exec_context_dump(&s->exec_context, f, prefix);
 461
 462         for (c = 0; c < _SOCKET_EXEC_COMMAND_MAX; c++) {
 463                 if (!s->exec_command[c])
 464                         continue;
 465
 466                 fprintf(f, "%s-> %s:\n",
 467                         prefix, socket_exec_command_to_string(c));
 468
 469                 exec_command_dump_list(s->exec_command[c], f, prefix2);
 470         }
 471
 472         free(p2);
 473 }
 474
 475 static int instance_from_socket(int fd, unsigned nr, char **instance) {
 476         socklen_t l;
 477         char *r;
 478         union {
 479                 struct sockaddr sa;
 480                 struct sockaddr_un un;
 481                 struct sockaddr_in in;
 482                 struct sockaddr_in6 in6;
 483                 struct sockaddr_storage storage;
 484         } local, remote;
 485
 486         assert(fd >= 0);
 487         assert(instance);
 488
 489         l = sizeof(local);
 490         if (getsockname(fd, &local.sa, &l) < 0)
 491                 return -errno;
 492
 493         l = sizeof(remote);
 494         if (getpeername(fd, &remote.sa, &l) < 0)
 495                 return -errno;
 496
 497         switch (local.sa.sa_family) {
 498
 499         case AF_INET: {
 500                 uint32_t
 501                         a = ntohl(local.in.sin_addr.s_addr),
 502                         b = ntohl(remote.in.sin_addr.s_addr);
 503
 504                 if (asprintf(&r,
 505                              "%u.%u.%u.%u:%u-%u.%u.%u.%u:%u",
 506                              a >> 24, (a >> 16) & 0xFF, (a >> 8) & 0xFF, a & 0xFF,
 507                              ntohs(local.in.sin_port),
 508                              b >> 24, (b >> 16) & 0xFF, (b >> 8) & 0xFF, b & 0xFF,
 509                              ntohs(remote.in.sin_port)) < 0)
 510                         return -ENOMEM;
 511
 512                 break;
 513         }
 514
 515         case AF_INET6: {
 516                 static const char ipv4_prefix[] = {
 517                         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xFF, 0xFF
 518                 };
 519
 520                 if (memcmp(&local.in6.sin6_addr, ipv4_prefix, sizeof(ipv4_prefix)) == 0 &&
 521                     memcmp(&remote.in6.sin6_addr, ipv4_prefix, sizeof(ipv4_prefix)) == 0) {
 522                         const uint8_t
 523                                 *a = local.in6.sin6_addr.s6_addr+12,
 524                                 *b = remote.in6.sin6_addr.s6_addr+12;
 525
 526                         if (asprintf(&r,
 527                                      "%u.%u.%u.%u:%u-%u.%u.%u.%u:%u",
 528                                      a[0], a[1], a[2], a[3],
 529                                      ntohs(local.in6.sin6_port),
 530                                      b[0], b[1], b[2], b[3],
 531                                      ntohs(remote.in6.sin6_port)) < 0)
 532                                 return -ENOMEM;
 533                 } else {
 534                         char a[INET6_ADDRSTRLEN], b[INET6_ADDRSTRLEN];
 535
 536                         if (asprintf(&r,
 537                                      "%s:%u-%s:%u",
 538                                      inet_ntop(AF_INET6, &local.in6.sin6_addr, a, sizeof(a)),
 539                                      ntohs(local.in6.sin6_port),
 540                                      inet_ntop(AF_INET6, &remote.in6.sin6_addr, b, sizeof(b)),
 541                                      ntohs(remote.in6.sin6_port)) < 0)
 542                                 return -ENOMEM;
 543                 }
 544
 545                 break;
 546         }
 547
 548         case AF_UNIX: {
 549                 struct ucred ucred;
 550
 551                 l = sizeof(ucred);
 552                 if (getsockopt(fd, SOL_SOCKET, SO_PEERCRED, &ucred, &l) < 0)
 553                         return -errno;
 554
 555                 if (asprintf(&r,
 556                              "%u-%lu-%lu",
 557                              nr,
 558                              (unsigned long) ucred.pid,
 559                              (unsigned long) ucred.uid) < 0)
 560                         return -ENOMEM;
 561
 562                 break;
 563         }
 564
 565         default:
 566                 assert_not_reached("Unhandled socket type.");
 567         }
 568
 569         *instance = r;
 570         return 0;
 571 }
 572
 573 static void socket_close_fds(Socket *s) {
 574         SocketPort *p;
 575
 576         assert(s);
 577
 578         LIST_FOREACH(port, p, s->ports) {
 579                 if (p->fd < 0)
 580                         continue;
 581
 582                 unit_unwatch_fd(UNIT(s), &p->fd_watch);
 583                 close_nointr_nofail(p->fd);
 584
 585                 /* One little note: we should never delete any sockets
 586                  * in the file system here! After all some other
 587                  * process we spawned might still have a reference of
 588                  * this fd and wants to continue to use it. Therefore
 589                  * we delete sockets in the file system before we
 590                  * create a new one, not after we stopped using
 591                  * one! */
 592
 593                 p->fd = -1;
 594         }
 595 }
 596
 597 static void socket_apply_socket_options(Socket *s, int fd) {
 598         assert(s);
 599         assert(fd >= 0);
 600
 601         if (s->keep_alive) {
 602                 int b = s->keep_alive;
 603                 if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &b, sizeof(b)) < 0)
 604                         log_warning("SO_KEEPALIVE failed: %m");
 605         }
 606
 607         if (s->priority >= 0)
 608                 if (setsockopt(fd, SOL_SOCKET, SO_PRIORITY, &s->priority, sizeof(s->priority)) < 0)
 609                         log_warning("SO_PRIORITY failed: %m");
 610
 611         if (s->receive_buffer > 0) {
 612                 int value = (int) s->receive_buffer;
 613                 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value)) < 0)
 614                         log_warning("SO_RCVBUF failed: %m");
 615         }
 616
 617         if (s->send_buffer > 0) {
 618                 int value = (int) s->send_buffer;
 619                 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value)) < 0)
 620                         log_warning("SO_SNDBUF failed: %m");
 621         }
 622
 623         if (s->mark >= 0)
 624                 if (setsockopt(fd, SOL_SOCKET, SO_MARK, &s->mark, sizeof(s->mark)) < 0)
 625                         log_warning("SO_MARK failed: %m");
 626
 627         if (s->ip_tos >= 0)
 628                 if (setsockopt(fd, IPPROTO_IP, IP_TOS, &s->ip_tos, sizeof(s->ip_tos)) < 0)
 629                         log_warning("IP_TOS failed: %m");
 630
 631         if (s->ip_ttl >= 0) {
 632                 int r, x;
 633
 634                 r = setsockopt(fd, IPPROTO_IP, IP_TTL, &s->ip_ttl, sizeof(s->ip_ttl));
 635                 x = setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &s->ip_ttl, sizeof(s->ip_ttl));
 636
 637                 if (r < 0 && x < 0)
 638                         log_warning("IP_TTL/IPV6_UNICAST_HOPS failed: %m");
 639         }
 640
 641         if (s->tcp_congestion)
 642                 if (setsockopt(fd, SOL_TCP, TCP_CONGESTION, s->tcp_congestion, strlen(s->tcp_congestion)+1) < 0)
 643                         log_warning("TCP_CONGESTION failed: %m");
 644 }
 645
 646 static void socket_apply_fifo_options(Socket *s, int fd) {
 647         assert(s);
 648         assert(fd >= 0);
 649
 650         if (s->pipe_size > 0)
 651                 if (fcntl(fd, F_SETPIPE_SZ, s->pipe_size) < 0)
 652                         log_warning("F_SETPIPE_SZ: %m");
 653 }
 654
 655
 656 static int fifo_address_create(
 657                 const char *path,
 658                 mode_t directory_mode,
 659                 mode_t socket_mode,
 660                 const char *label,
 661                 int *_fd) {
 662
 663         int fd = -1, r = 0;
 664         struct stat st;
 665         mode_t old_mask;
 666
 667         assert(path);
 668         assert(_fd);
 669
 670         mkdir_parents(path, directory_mode);
 671
 672         if ((r = label_fifofile_set(label, path)) < 0)
 673                 goto fail;
 674
 675         /* Enforce the right access mode for the fifo */
 676         old_mask = umask(~ socket_mode);
 677
 678         /* Include the original umask in our mask */
 679         umask(~socket_mode | old_mask);
 680
 681         r = mkfifo(path, socket_mode);
 682         umask(old_mask);
 683
 684         if (r < 0) {
 685                 r = -errno;
 686                 goto fail;
 687         }
 688
 689         if ((fd = open(path, O_RDWR|O_CLOEXEC|O_NOCTTY|O_NONBLOCK|O_NOFOLLOW)) < 0) {
 690                 r = -errno;
 691                 goto fail;
 692         }
 693
 694         label_file_clear();
 695
 696         if (fstat(fd, &st) < 0) {
 697                 r = -errno;
 698                 goto fail;
 699         }
 700
 701         if (!S_ISFIFO(st.st_mode) ||
 702             (st.st_mode & 0777) != (socket_mode & ~old_mask) ||
 703             st.st_uid != getuid() ||
 704             st.st_gid != getgid()) {
 705
 706                 r = -EEXIST;
 707                 goto fail;
 708         }
 709
 710         *_fd = fd;
 711         return 0;
 712
 713 fail:
 714         label_file_clear();
 715
 716         if (fd >= 0)
 717                 close_nointr_nofail(fd);
 718
 719         return r;
 720 }
 721
 722 static int socket_open_fds(Socket *s) {
 723         SocketPort *p;
 724         int r;
 725         char *label = NULL;
 726
 727         assert(s);
 728
 729         if ((r = socket_instantiate_service(s)) < 0)
 730                 return r;
 731
 732         if ((r = label_get_socket_label_from_exe(s->service->exec_command[SERVICE_EXEC_START]->path, &label)) < 0)
 733                 return r;
 734
 735         LIST_FOREACH(port, p, s->ports) {
 736
 737                 if (p->fd >= 0)
 738                         continue;
 739
 740                 if (p->type == SOCKET_SOCKET) {
 741
 742                         if ((r = socket_address_listen(
 743                                              &p->address,
 744                                              s->backlog,
 745                                              s->bind_ipv6_only,
 746                                              s->bind_to_device,
 747                                              s->free_bind,
 748                                              s->directory_mode,
 749                                              s->socket_mode,
 750                                              label,
 751                                              &p->fd)) < 0)
 752                                 goto rollback;
 753
 754                         socket_apply_socket_options(s, p->fd);
 755
 756                 } else  if (p->type == SOCKET_FIFO) {
 757
 758                         if ((r = fifo_address_create(
 759                                              p->path,
 760                                              s->directory_mode,
 761                                              s->socket_mode,
 762                                              label,
 763                                              &p->fd)) < 0)
 764                                 goto rollback;
 765
 766                         socket_apply_fifo_options(s, p->fd);
 767
 768                 } else
 769                         assert_not_reached("Unknown port type");
 770         }
 771
 772         label_free(label);
 773         return 0;
 774
 775 rollback:
 776         socket_close_fds(s);
 777         label_free(label);
 778         return r;
 779 }
 780
 781 static void socket_unwatch_fds(Socket *s) {
 782         SocketPort *p;
 783
 784         assert(s);
 785
 786         LIST_FOREACH(port, p, s->ports) {
 787                 if (p->fd < 0)
 788                         continue;
 789
 790                 unit_unwatch_fd(UNIT(s), &p->fd_watch);
 791         }
 792 }
 793
 794 static int socket_watch_fds(Socket *s) {
 795         SocketPort *p;
 796         int r;
 797
 798         assert(s);
 799
 800         LIST_FOREACH(port, p, s->ports) {
 801                 if (p->fd < 0)
 802                         continue;
 803
 804                 p->fd_watch.socket_accept =
 805                         s->accept &&
 806                         p->type == SOCKET_SOCKET &&
 807                         socket_address_can_accept(&p->address);
 808
 809                 if ((r = unit_watch_fd(UNIT(s), p->fd, EPOLLIN, &p->fd_watch)) < 0)
 810                         goto fail;
 811         }
 812
 813         return 0;
 814
 815 fail:
 816         socket_unwatch_fds(s);
 817         return r;
 818 }
 819
 820 static void socket_set_state(Socket *s, SocketState state) {
 821         SocketState old_state;
 822         assert(s);
 823
 824         old_state = s->state;
 825         s->state = state;
 826
 827         if (state != SOCKET_START_PRE &&
 828             state != SOCKET_START_POST &&
 829             state != SOCKET_STOP_PRE &&
 830             state != SOCKET_STOP_PRE_SIGTERM &&
 831             state != SOCKET_STOP_PRE_SIGKILL &&
 832             state != SOCKET_STOP_POST &&
 833             state != SOCKET_FINAL_SIGTERM &&
 834             state != SOCKET_FINAL_SIGKILL) {
 835                 unit_unwatch_timer(UNIT(s), &s->timer_watch);
 836                 socket_unwatch_control_pid(s);
 837                 s->control_command = NULL;
 838                 s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
 839         }
 840
 841         if (state != SOCKET_LISTENING)
 842                 socket_unwatch_fds(s);
 843
 844         if (state != SOCKET_START_POST &&
 845             state != SOCKET_LISTENING &&
 846             state != SOCKET_RUNNING &&
 847             state != SOCKET_STOP_PRE &&
 848             state != SOCKET_STOP_PRE_SIGTERM &&
 849             state != SOCKET_STOP_PRE_SIGKILL)
 850                 socket_close_fds(s);
 851
 852         if (state != old_state)
 853                 log_debug("%s changed %s -> %s",
 854                           s->meta.id,
 855                           socket_state_to_string(old_state),
 856                           socket_state_to_string(state));
 857
 858         unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state]);
 859 }
 860
 861 static int socket_coldplug(Unit *u) {
 862         Socket *s = SOCKET(u);
 863         int r;
 864
 865         assert(s);
 866         assert(s->state == SOCKET_DEAD);
 867
 868         if (s->deserialized_state != s->state) {
 869
 870                 if (s->deserialized_state == SOCKET_START_PRE ||
 871                     s->deserialized_state == SOCKET_START_POST ||
 872                     s->deserialized_state == SOCKET_STOP_PRE ||
 873                     s->deserialized_state == SOCKET_STOP_PRE_SIGTERM ||
 874                     s->deserialized_state == SOCKET_STOP_PRE_SIGKILL ||
 875                     s->deserialized_state == SOCKET_STOP_POST ||
 876                     s->deserialized_state == SOCKET_FINAL_SIGTERM ||
 877                     s->deserialized_state == SOCKET_FINAL_SIGKILL) {
 878
 879                         if (s->control_pid <= 0)
 880                                 return -EBADMSG;
 881
 882                         if ((r = unit_watch_pid(UNIT(s), s->control_pid)) < 0)
 883                                 return r;
 884
 885                         if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
 886                                 return r;
 887                 }
 888
 889                 if (s->deserialized_state == SOCKET_START_POST ||
 890                     s->deserialized_state == SOCKET_LISTENING ||
 891                     s->deserialized_state == SOCKET_RUNNING ||
 892                     s->deserialized_state == SOCKET_STOP_PRE ||
 893                     s->deserialized_state == SOCKET_STOP_PRE_SIGTERM ||
 894                     s->deserialized_state == SOCKET_STOP_PRE_SIGKILL)
 895                         if ((r = socket_open_fds(s)) < 0)
 896                                 return r;
 897
 898                 if (s->deserialized_state == SOCKET_LISTENING)
 899                         if ((r = socket_watch_fds(s)) < 0)
 900                                 return r;
 901
 902                 socket_set_state(s, s->deserialized_state);
 903         }
 904
 905         return 0;
 906 }
 907
 908 static int socket_spawn(Socket *s, ExecCommand *c, pid_t *_pid) {
 909         pid_t pid;
 910         int r;
 911         char **argv;
 912
 913         assert(s);
 914         assert(c);
 915         assert(_pid);
 916
 917         if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
 918                 goto fail;
 919
 920         if (!(argv = unit_full_printf_strv(UNIT(s), c->argv))) {
 921                 r = -ENOMEM;
 922                 goto fail;
 923         }
 924
 925         r = exec_spawn(c,
 926                        argv,
 927                        &s->exec_context,
 928                        NULL, 0,
 929                        s->meta.manager->environment,
 930                        true,
 931                        true,
 932                        true,
 933                        s->meta.manager->confirm_spawn,
 934                        s->meta.cgroup_bondings,
 935                        &pid);
 936
 937         strv_free(argv);
 938         if (r < 0)
 939                 goto fail;
 940
 941         if ((r = unit_watch_pid(UNIT(s), pid)) < 0)
 942                 /* FIXME: we need to do something here */
 943                 goto fail;
 944
 945         *_pid = pid;
 946
 947         return 0;
 948
 949 fail:
 950         unit_unwatch_timer(UNIT(s), &s->timer_watch);
 951
 952         return r;
 953 }
 954
 955 static void socket_enter_dead(Socket *s, bool success) {
 956         assert(s);
 957
 958         if (!success)
 959                 s->failure = true;
 960
 961         socket_set_state(s, s->failure ? SOCKET_MAINTENANCE : SOCKET_DEAD);
 962 }
 963
 964 static void socket_enter_signal(Socket *s, SocketState state, bool success);
 965
 966 static void socket_enter_stop_post(Socket *s, bool success) {
 967         int r;
 968         assert(s);
 969
 970         if (!success)
 971                 s->failure = true;
 972
 973         socket_unwatch_control_pid(s);
 974
 975         s->control_command_id = SOCKET_EXEC_STOP_POST;
 976
 977         if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_POST])) {
 978                 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
 979                         goto fail;
 980
 981                 socket_set_state(s, SOCKET_STOP_POST);
 982         } else
 983                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, true);
 984
 985         return;
 986
 987 fail:
 988         log_warning("%s failed to run 'stop-post' task: %s", s->meta.id, strerror(-r));
 989         socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
 990 }
 991
 992 static void socket_enter_signal(Socket *s, SocketState state, bool success) {
 993         int r;
 994         bool sent = false;
 995
 996         assert(s);
 997
 998         if (!success)
 999                 s->failure = true;
1000
1001         if (s->exec_context.kill_mode != KILL_NONE) {
1002                 int sig = (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_FINAL_SIGTERM) ? s->exec_context.kill_signal : SIGKILL;
1003
1004                 if (s->exec_context.kill_mode == KILL_CONTROL_GROUP) {
1005
1006                         if ((r = cgroup_bonding_kill_list(s->meta.cgroup_bondings, sig)) < 0) {
1007                                 if (r != -EAGAIN && r != -ESRCH)
1008                                         goto fail;
1009                         } else
1010                                 sent = true;
1011                 }
1012
1013                 if (!sent && s->control_pid > 0)
1014                         if (kill(s->exec_context.kill_mode == KILL_PROCESS ? s->control_pid : -s->control_pid, sig) < 0 && errno != ESRCH) {
1015                                 r = -errno;
1016                                 goto fail;
1017                         }
1018         }
1019
1020         if (sent && s->control_pid > 0) {
1021                 if ((r = unit_watch_timer(UNIT(s), s->timeout_usec, &s->timer_watch)) < 0)
1022                         goto fail;
1023
1024                 socket_set_state(s, state);
1025         } else if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL)
1026                 socket_enter_stop_post(s, true);
1027         else
1028                 socket_enter_dead(s, true);
1029
1030         return;
1031
1032 fail:
1033         log_warning("%s failed to kill processes: %s", s->meta.id, strerror(-r));
1034
1035         if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL)
1036                 socket_enter_stop_post(s, false);
1037         else
1038                 socket_enter_dead(s, false);
1039 }
1040
1041 static void socket_enter_stop_pre(Socket *s, bool success) {
1042         int r;
1043         assert(s);
1044
1045         if (!success)
1046                 s->failure = true;
1047
1048         socket_unwatch_control_pid(s);
1049
1050         s->control_command_id = SOCKET_EXEC_STOP_PRE;
1051
1052         if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_PRE])) {
1053                 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
1054                         goto fail;
1055
1056                 socket_set_state(s, SOCKET_STOP_PRE);
1057         } else
1058                 socket_enter_stop_post(s, true);
1059
1060         return;
1061
1062 fail:
1063         log_warning("%s failed to run 'stop-pre' task: %s", s->meta.id, strerror(-r));
1064         socket_enter_stop_post(s, false);
1065 }
1066
1067 static void socket_enter_listening(Socket *s) {
1068         int r;
1069         assert(s);
1070
1071         if ((r = socket_watch_fds(s)) < 0) {
1072                 log_warning("%s failed to watch sockets: %s", s->meta.id, strerror(-r));
1073                 goto fail;
1074         }
1075
1076         socket_set_state(s, SOCKET_LISTENING);
1077         return;
1078
1079 fail:
1080         socket_enter_stop_pre(s, false);
1081 }
1082
1083 static void socket_enter_start_post(Socket *s) {
1084         int r;
1085         assert(s);
1086
1087         if ((r = socket_open_fds(s)) < 0) {
1088                 log_warning("%s failed to listen on sockets: %s", s->meta.id, strerror(-r));
1089                 goto fail;
1090         }
1091
1092         socket_unwatch_control_pid(s);
1093
1094         s->control_command_id = SOCKET_EXEC_START_POST;
1095
1096         if ((s->control_command = s->exec_command[SOCKET_EXEC_START_POST])) {
1097                 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0) {
1098                         log_warning("%s failed to run 'start-post' task: %s", s->meta.id, strerror(-r));
1099                         goto fail;
1100                 }
1101
1102                 socket_set_state(s, SOCKET_START_POST);
1103         } else
1104                 socket_enter_listening(s);
1105
1106         return;
1107
1108 fail:
1109         socket_enter_stop_pre(s, false);
1110 }
1111
1112 static void socket_enter_start_pre(Socket *s) {
1113         int r;
1114         assert(s);
1115
1116         socket_unwatch_control_pid(s);
1117
1118         s->control_command_id = SOCKET_EXEC_START_PRE;
1119
1120         if ((s->control_command = s->exec_command[SOCKET_EXEC_START_PRE])) {
1121                 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
1122                         goto fail;
1123
1124                 socket_set_state(s, SOCKET_START_PRE);
1125         } else
1126                 socket_enter_start_post(s);
1127
1128         return;
1129
1130 fail:
1131         log_warning("%s failed to run 'start-pre' task: %s", s->meta.id, strerror(-r));
1132         socket_enter_dead(s, false);
1133 }
1134
1135 static void socket_enter_running(Socket *s, int cfd) {
1136         int r;
1137         DBusError error;
1138
1139         assert(s);
1140         dbus_error_init(&error);
1141
1142         /* We don't take connections anymore if we are supposed to
1143          * shut down anyway */
1144         if (s->meta.job && s->meta.job->type == JOB_STOP) {
1145                 if (cfd >= 0)
1146                         close_nointr_nofail(cfd);
1147                 else  {
1148                         /* Flush all sockets by closing and reopening them */
1149                         socket_close_fds(s);
1150
1151                         if ((r = socket_watch_fds(s)) < 0) {
1152                                 log_warning("%s failed to watch sockets: %s", s->meta.id, strerror(-r));
1153                                 socket_enter_stop_pre(s, false);
1154                         }
1155                 }
1156
1157                 return;
1158         }
1159
1160         if (cfd < 0) {
1161                 if ((r = manager_add_job(s->meta.manager, JOB_START, UNIT(s->service), JOB_REPLACE, true, &error, NULL)) < 0)
1162                         goto fail;
1163
1164                 socket_set_state(s, SOCKET_RUNNING);
1165         } else {
1166                 char *prefix, *instance = NULL, *name;
1167                 Service *service;
1168
1169                 if (s->n_connections >= s->max_connections) {
1170                         log_warning("Too many incoming connections (%u)", s->n_connections);
1171                         close_nointr_nofail(cfd);
1172                         return;
1173                 }
1174
1175                 if ((r = socket_instantiate_service(s)) < 0)
1176                         goto fail;
1177
1178                 if ((r = instance_from_socket(cfd, s->n_accepted, &instance)) < 0)
1179                         goto fail;
1180
1181                 if (!(prefix = unit_name_to_prefix(s->meta.id))) {
1182                         free(instance);
1183                         r = -ENOMEM;
1184                         goto fail;
1185                 }
1186
1187                 name = unit_name_build(prefix, instance, ".service");
1188                 free(prefix);
1189                 free(instance);
1190
1191                 if (!name) {
1192                         r = -ENOMEM;
1193                         goto fail;
1194                 }
1195
1196                 if ((r = unit_add_name(UNIT(s->service), name)) < 0) {
1197                         free(name);
1198                         goto fail;
1199                 }
1200
1201                 service = s->service;
1202                 s->service = NULL;
1203                 s->n_accepted ++;
1204
1205                 unit_choose_id(UNIT(service), name);
1206                 free(name);
1207
1208                 if ((r = service_set_socket_fd(service, cfd, s)) < 0)
1209                         goto fail;
1210
1211                 cfd = -1;
1212                 s->n_connections ++;
1213
1214                 if ((r = manager_add_job(s->meta.manager, JOB_START, UNIT(service), JOB_REPLACE, true, &error, NULL)) < 0)
1215                         goto fail;
1216         }
1217
1218         return;
1219
1220 fail:
1221         log_warning("%s failed to queue socket startup job: %s", s->meta.id, bus_error(&error, r));
1222         socket_enter_stop_pre(s, false);
1223
1224         if (cfd >= 0)
1225                 close_nointr_nofail(cfd);
1226
1227         dbus_error_free(&error);
1228 }
1229
1230 static void socket_run_next(Socket *s, bool success) {
1231         int r;
1232
1233         assert(s);
1234         assert(s->control_command);
1235         assert(s->control_command->command_next);
1236
1237         if (!success)
1238                 s->failure = true;
1239
1240         socket_unwatch_control_pid(s);
1241
1242         s->control_command = s->control_command->command_next;
1243
1244         if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
1245                 goto fail;
1246
1247         return;
1248
1249 fail:
1250         log_warning("%s failed to run next task: %s", s->meta.id, strerror(-r));
1251
1252         if (s->state == SOCKET_START_POST)
1253                 socket_enter_stop_pre(s, false);
1254         else if (s->state == SOCKET_STOP_POST)
1255                 socket_enter_dead(s, false);
1256         else
1257                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
1258 }
1259
1260 static int socket_start(Unit *u) {
1261         Socket *s = SOCKET(u);
1262
1263         assert(s);
1264
1265         /* We cannot fulfill this request right now, try again later
1266          * please! */
1267         if (s->state == SOCKET_STOP_PRE ||
1268             s->state == SOCKET_STOP_PRE_SIGKILL ||
1269             s->state == SOCKET_STOP_PRE_SIGTERM ||
1270             s->state == SOCKET_STOP_POST ||
1271             s->state == SOCKET_FINAL_SIGTERM ||
1272             s->state == SOCKET_FINAL_SIGKILL)
1273                 return -EAGAIN;
1274
1275         if (s->state == SOCKET_START_PRE ||
1276             s->state == SOCKET_START_POST)
1277                 return 0;
1278
1279         /* Cannot run this without the service being around */
1280         if (s->service) {
1281                 if (s->service->meta.load_state != UNIT_LOADED)
1282                         return -ENOENT;
1283
1284                 /* If the service is alredy actvie we cannot start the
1285                  * socket */
1286                 if (s->service->state != SERVICE_DEAD &&
1287                     s->service->state != SERVICE_MAINTENANCE &&
1288                     s->service->state != SERVICE_AUTO_RESTART)
1289                         return -EBUSY;
1290         }
1291
1292         assert(s->state == SOCKET_DEAD || s->state == SOCKET_MAINTENANCE);
1293
1294         s->failure = false;
1295         socket_enter_start_pre(s);
1296         return 0;
1297 }
1298
1299 static int socket_stop(Unit *u) {
1300         Socket *s = SOCKET(u);
1301
1302         assert(s);
1303
1304         /* Already on it */
1305         if (s->state == SOCKET_STOP_PRE ||
1306             s->state == SOCKET_STOP_PRE_SIGTERM ||
1307             s->state == SOCKET_STOP_PRE_SIGKILL ||
1308             s->state == SOCKET_STOP_POST ||
1309             s->state == SOCKET_FINAL_SIGTERM ||
1310             s->state == SOCKET_FINAL_SIGKILL)
1311                 return 0;
1312
1313         /* If there's already something running we go directly into
1314          * kill mode. */
1315         if (s->state == SOCKET_START_PRE ||
1316             s->state == SOCKET_START_POST) {
1317                 socket_enter_signal(s, SOCKET_STOP_PRE_SIGTERM, true);
1318                 return -EAGAIN;
1319         }
1320
1321         assert(s->state == SOCKET_LISTENING || s->state == SOCKET_RUNNING);
1322
1323         socket_enter_stop_pre(s, true);
1324         return 0;
1325 }
1326
1327 static int socket_serialize(Unit *u, FILE *f, FDSet *fds) {
1328         Socket *s = SOCKET(u);
1329         SocketPort *p;
1330         int r;
1331
1332         assert(u);
1333         assert(f);
1334         assert(fds);
1335
1336         unit_serialize_item(u, f, "state", socket_state_to_string(s->state));
1337         unit_serialize_item(u, f, "failure", yes_no(s->failure));
1338         unit_serialize_item_format(u, f, "n-accepted", "%u", s->n_accepted);
1339
1340         if (s->control_pid > 0)
1341                 unit_serialize_item_format(u, f, "control-pid", "%lu", (unsigned long) s->control_pid);
1342
1343         if (s->control_command_id >= 0)
1344                 unit_serialize_item(u, f, "control-command", socket_exec_command_to_string(s->control_command_id));
1345
1346         LIST_FOREACH(port, p, s->ports) {
1347                 int copy;
1348
1349                 if (p->fd < 0)
1350                         continue;
1351
1352                 if ((copy = fdset_put_dup(fds, p->fd)) < 0)
1353                         return copy;
1354
1355                 if (p->type == SOCKET_SOCKET) {
1356                         char *t;
1357
1358                         if ((r = socket_address_print(&p->address, &t)) < 0)
1359                                 return r;
1360
1361                         unit_serialize_item_format(u, f, "socket", "%i %i %s", copy, p->address.type, t);
1362                         free(t);
1363                 } else {
1364                         assert(p->type == SOCKET_FIFO);
1365                         unit_serialize_item_format(u, f, "fifo", "%i %s", copy, p->path);
1366                 }
1367         }
1368
1369         return 0;
1370 }
1371
1372 static int socket_deserialize_item(Unit *u, const char *key, const char *value, FDSet *fds) {
1373         Socket *s = SOCKET(u);
1374         int r;
1375
1376         assert(u);
1377         assert(key);
1378         assert(value);
1379         assert(fds);
1380
1381         if (streq(key, "state")) {
1382                 SocketState state;
1383
1384                 if ((state = socket_state_from_string(value)) < 0)
1385                         log_debug("Failed to parse state value %s", value);
1386                 else
1387                         s->deserialized_state = state;
1388         } else if (streq(key, "failure")) {
1389                 int b;
1390
1391                 if ((b = parse_boolean(value)) < 0)
1392                         log_debug("Failed to parse failure value %s", value);
1393                 else
1394                         s->failure = b || s->failure;
1395
1396         } else if (streq(key, "n-accepted")) {
1397                 unsigned k;
1398
1399                 if ((r = safe_atou(value, &k)) < 0)
1400                         log_debug("Failed to parse n-accepted value %s", value);
1401                 else
1402                         s->n_accepted += k;
1403         } else if (streq(key, "control-pid")) {
1404                 pid_t pid;
1405
1406                 if ((r = parse_pid(value, &pid)) < 0)
1407                         log_debug("Failed to parse control-pid value %s", value);
1408                 else
1409                         s->control_pid = pid;
1410         } else if (streq(key, "control-command")) {
1411                 SocketExecCommand id;
1412
1413                 if ((id = socket_exec_command_from_string(value)) < 0)
1414                         log_debug("Failed to parse exec-command value %s", value);
1415                 else {
1416                         s->control_command_id = id;
1417                         s->control_command = s->exec_command[id];
1418                 }
1419         } else if (streq(key, "fifo")) {
1420                 int fd, skip = 0;
1421                 SocketPort *p;
1422
1423                 if (sscanf(value, "%i %n", &fd, &skip) < 1 || fd < 0 || !fdset_contains(fds, fd))
1424                         log_debug("Failed to parse fifo value %s", value);
1425                 else {
1426
1427                         LIST_FOREACH(port, p, s->ports)
1428                                 if (streq(p->path, value+skip))
1429                                         break;
1430
1431                         if (p) {
1432                                 if (p->fd >= 0)
1433                                         close_nointr_nofail(p->fd);
1434                                 p->fd = fdset_remove(fds, fd);
1435                         }
1436                 }
1437
1438         } else if (streq(key, "socket")) {
1439                 int fd, type, skip = 0;
1440                 SocketPort *p;
1441
1442                 if (sscanf(value, "%i %i %n", &fd, &type, &skip) < 2 || fd < 0 || type < 0 || !fdset_contains(fds, fd))
1443                         log_debug("Failed to parse socket value %s", value);
1444                 else {
1445
1446                         LIST_FOREACH(port, p, s->ports)
1447                                 if (socket_address_is(&p->address, value+skip, type))
1448                                         break;
1449
1450                         if (p) {
1451                                 if (p->fd >= 0)
1452                                         close_nointr_nofail(p->fd);
1453                                 p->fd = fdset_remove(fds, fd);
1454                         }
1455                 }
1456
1457         } else
1458                 log_debug("Unknown serialization key '%s'", key);
1459
1460         return 0;
1461 }
1462
1463 static UnitActiveState socket_active_state(Unit *u) {
1464         assert(u);
1465
1466         return state_translation_table[SOCKET(u)->state];
1467 }
1468
1469 static const char *socket_sub_state_to_string(Unit *u) {
1470         assert(u);
1471
1472         return socket_state_to_string(SOCKET(u)->state);
1473 }
1474
1475 static bool socket_check_gc(Unit *u) {
1476         Socket *s = SOCKET(u);
1477
1478         assert(u);
1479
1480         return s->n_connections > 0;
1481 }
1482
1483 static void socket_fd_event(Unit *u, int fd, uint32_t events, Watch *w) {
1484         Socket *s = SOCKET(u);
1485         int cfd = -1;
1486
1487         assert(s);
1488         assert(fd >= 0);
1489
1490         if (s->state != SOCKET_LISTENING)
1491                 return;
1492
1493         log_debug("Incoming traffic on %s", u->meta.id);
1494
1495         if (events != EPOLLIN) {
1496                 log_error("Got invalid poll event on socket.");
1497                 goto fail;
1498         }
1499
1500         if (w->socket_accept) {
1501                 for (;;) {
1502
1503                         if ((cfd = accept4(fd, NULL, NULL, SOCK_NONBLOCK)) < 0) {
1504
1505                                 if (errno == EINTR)
1506                                         continue;
1507
1508                                 log_error("Failed to accept socket: %m");
1509                                 goto fail;
1510                         }
1511
1512                         break;
1513                 }
1514
1515                 socket_apply_socket_options(s, cfd);
1516         }
1517
1518         socket_enter_running(s, cfd);
1519         return;
1520
1521 fail:
1522         socket_enter_stop_pre(s, false);
1523 }
1524
1525 static void socket_sigchld_event(Unit *u, pid_t pid, int code, int status) {
1526         Socket *s = SOCKET(u);
1527         bool success;
1528
1529         assert(s);
1530         assert(pid >= 0);
1531
1532         if (pid != s->control_pid)
1533                 return;
1534
1535         s->control_pid = 0;
1536
1537         success = is_clean_exit(code, status);
1538
1539         if (s->control_command) {
1540                 exec_status_exit(&s->control_command->exec_status, pid, code, status);
1541
1542                 if (s->control_command->ignore)
1543                         success = true;
1544         }
1545
1546         log_full(success ? LOG_DEBUG : LOG_NOTICE,
1547                  "%s control process exited, code=%s status=%i", u->meta.id, sigchld_code_to_string(code), status);
1548         s->failure = s->failure || !success;
1549
1550         if (s->control_command && s->control_command->command_next && success) {
1551                 log_debug("%s running next command for state %s", u->meta.id, socket_state_to_string(s->state));
1552                 socket_run_next(s, success);
1553         } else {
1554                 s->control_command = NULL;
1555                 s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
1556
1557                 /* No further commands for this step, so let's figure
1558                  * out what to do next */
1559
1560                 log_debug("%s got final SIGCHLD for state %s", u->meta.id, socket_state_to_string(s->state));
1561
1562                 switch (s->state) {
1563
1564                 case SOCKET_START_PRE:
1565                         if (success)
1566                                 socket_enter_start_post(s);
1567                         else
1568                                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
1569                         break;
1570
1571                 case SOCKET_START_POST:
1572                         if (success)
1573                                 socket_enter_listening(s);
1574                         else
1575                                 socket_enter_stop_pre(s, false);
1576                         break;
1577
1578                 case SOCKET_STOP_PRE:
1579                 case SOCKET_STOP_PRE_SIGTERM:
1580                 case SOCKET_STOP_PRE_SIGKILL:
1581                         socket_enter_stop_post(s, success);
1582                         break;
1583
1584                 case SOCKET_STOP_POST:
1585                 case SOCKET_FINAL_SIGTERM:
1586                 case SOCKET_FINAL_SIGKILL:
1587                         socket_enter_dead(s, success);
1588                         break;
1589
1590                 default:
1591                         assert_not_reached("Uh, control process died at wrong time.");
1592                 }
1593         }
1594 }
1595
1596 static void socket_timer_event(Unit *u, uint64_t elapsed, Watch *w) {
1597         Socket *s = SOCKET(u);
1598
1599         assert(s);
1600         assert(elapsed == 1);
1601         assert(w == &s->timer_watch);
1602
1603         switch (s->state) {
1604
1605         case SOCKET_START_PRE:
1606                 log_warning("%s starting timed out. Terminating.", u->meta.id);
1607                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
1608
1609         case SOCKET_START_POST:
1610                 log_warning("%s starting timed out. Stopping.", u->meta.id);
1611                 socket_enter_stop_pre(s, false);
1612                 break;
1613
1614         case SOCKET_STOP_PRE:
1615                 log_warning("%s stopping timed out. Terminating.", u->meta.id);
1616                 socket_enter_signal(s, SOCKET_STOP_PRE_SIGTERM, false);
1617                 break;
1618
1619         case SOCKET_STOP_PRE_SIGTERM:
1620                 log_warning("%s stopping timed out. Killing.", u->meta.id);
1621                 socket_enter_signal(s, SOCKET_STOP_PRE_SIGKILL, false);
1622                 break;
1623
1624         case SOCKET_STOP_PRE_SIGKILL:
1625                 log_warning("%s still around after SIGKILL. Ignoring.", u->meta.id);
1626                 socket_enter_stop_post(s, false);
1627                 break;
1628
1629         case SOCKET_STOP_POST:
1630                 log_warning("%s stopping timed out (2). Terminating.", u->meta.id);
1631                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, false);
1632                 break;
1633
1634         case SOCKET_FINAL_SIGTERM:
1635                 log_warning("%s stopping timed out (2). Killing.", u->meta.id);
1636                 socket_enter_signal(s, SOCKET_FINAL_SIGKILL, false);
1637                 break;
1638
1639         case SOCKET_FINAL_SIGKILL:
1640                 log_warning("%s still around after SIGKILL (2). Entering maintenance mode.", u->meta.id);
1641                 socket_enter_dead(s, false);
1642                 break;
1643
1644         default:
1645                 assert_not_reached("Timeout at wrong time.");
1646         }
1647 }
1648
1649 int socket_collect_fds(Socket *s, int **fds, unsigned *n_fds) {
1650         int *rfds;
1651         unsigned rn_fds, k;
1652         SocketPort *p;
1653
1654         assert(s);
1655         assert(fds);
1656         assert(n_fds);
1657
1658         /* Called from the service code for requesting our fds */
1659
1660         rn_fds = 0;
1661         LIST_FOREACH(port, p, s->ports)
1662                 if (p->fd >= 0)
1663                         rn_fds++;
1664
1665         if (!(rfds = new(int, rn_fds)) < 0)
1666                 return -ENOMEM;
1667
1668         k = 0;
1669         LIST_FOREACH(port, p, s->ports)
1670                 if (p->fd >= 0)
1671                         rfds[k++] = p->fd;
1672
1673         assert(k == rn_fds);
1674
1675         *fds = rfds;
1676         *n_fds = rn_fds;
1677
1678         return 0;
1679 }
1680
1681 void socket_notify_service_dead(Socket *s) {
1682         assert(s);
1683
1684         /* The service is dead. Dang!
1685          *
1686          * This is strictly for one-instance-for-all-connections
1687          * services. */
1688
1689         if (s->state == SOCKET_RUNNING) {
1690                 log_debug("%s got notified about service death.", s->meta.id);
1691                 socket_enter_listening(s);
1692         }
1693 }
1694
1695 void socket_connection_unref(Socket *s) {
1696         assert(s);
1697
1698         /* The service is dead. Yay!
1699          *
1700          * This is strictly for one-onstance-per-connection
1701          * services. */
1702
1703         assert(s->n_connections > 0);
1704         s->n_connections--;
1705
1706         log_debug("%s: One connection closed, %u left.", s->meta.id, s->n_connections);
1707 }
1708
1709 static void socket_reset_maintenance(Unit *u) {
1710         Socket *s = SOCKET(u);
1711
1712         assert(s);
1713
1714         if (s->state == SOCKET_MAINTENANCE)
1715                 socket_set_state(s, SOCKET_DEAD);
1716
1717         s->failure = false;
1718 }
1719
1720 static const char* const socket_state_table[_SOCKET_STATE_MAX] = {
1721         [SOCKET_DEAD] = "dead",
1722         [SOCKET_START_PRE] = "start-pre",
1723         [SOCKET_START_POST] = "start-post",
1724         [SOCKET_LISTENING] = "listening",
1725         [SOCKET_RUNNING] = "running",
1726         [SOCKET_STOP_PRE] = "stop-pre",
1727         [SOCKET_STOP_PRE_SIGTERM] = "stop-pre-sigterm",
1728         [SOCKET_STOP_PRE_SIGKILL] = "stop-pre-sigkill",
1729         [SOCKET_STOP_POST] = "stop-post",
1730         [SOCKET_FINAL_SIGTERM] = "final-sigterm",
1731         [SOCKET_FINAL_SIGKILL] = "final-sigkill",
1732         [SOCKET_MAINTENANCE] = "maintenance"
1733 };
1734
1735 DEFINE_STRING_TABLE_LOOKUP(socket_state, SocketState);
1736
1737 static const char* const socket_exec_command_table[_SOCKET_EXEC_COMMAND_MAX] = {
1738         [SOCKET_EXEC_START_PRE] = "StartPre",
1739         [SOCKET_EXEC_START_POST] = "StartPost",
1740         [SOCKET_EXEC_STOP_PRE] = "StopPre",
1741         [SOCKET_EXEC_STOP_POST] = "StopPost"
1742 };
1743
1744 DEFINE_STRING_TABLE_LOOKUP(socket_exec_command, SocketExecCommand);
1745
1746 const UnitVTable socket_vtable = {
1747         .suffix = ".socket",
1748
1749         .init = socket_init,
1750         .done = socket_done,
1751         .load = socket_load,
1752
1753         .coldplug = socket_coldplug,
1754
1755         .dump = socket_dump,
1756
1757         .start = socket_start,
1758         .stop = socket_stop,
1759
1760         .serialize = socket_serialize,
1761         .deserialize_item = socket_deserialize_item,
1762
1763         .active_state = socket_active_state,
1764         .sub_state_to_string = socket_sub_state_to_string,
1765
1766         .check_gc = socket_check_gc,
1767
1768         .fd_event = socket_fd_event,
1769         .sigchld_event = socket_sigchld_event,
1770         .timer_event = socket_timer_event,
1771
1772         .reset_maintenance = socket_reset_maintenance,
1773
1774         .bus_message_handler = bus_socket_message_handler
1775 };