src/socket.c

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