src/core/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 Lesser General Public License as published by
  10   the Free Software Foundation; either version 2.1 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   Lesser General Public License for more details.
  17
  18   You should have received a copy of the GNU Lesser 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 #include <netinet/tcp.h>
  31 #include <mqueue.h>
  32 #ifdef HAVE_XATTR
  33 #include <attr/xattr.h>
  34 #endif
  35
  36 #include "sd-event.h"
  37 #include "log.h"
  38 #include "load-dropin.h"
  39 #include "load-fragment.h"
  40 #include "strv.h"
  41 #include "mkdir.h"
  42 #include "path-util.h"
  43 #include "unit-name.h"
  44 #include "unit-printf.h"
  45 #include "missing.h"
  46 #include "special.h"
  47 #include "label.h"
  48 #include "exit-status.h"
  49 #include "def.h"
  50 #include "smack-util.h"
  51 #include "bus-util.h"
  52 #include "bus-error.h"
  53 #include "dbus-socket.h"
  54 #include "unit.h"
  55 #include "socket.h"
  56
  57 static const UnitActiveState state_translation_table[_SOCKET_STATE_MAX] = {
  58         [SOCKET_DEAD] = UNIT_INACTIVE,
  59         [SOCKET_START_PRE] = UNIT_ACTIVATING,
  60         [SOCKET_START_POST] = UNIT_ACTIVATING,
  61         [SOCKET_LISTENING] = UNIT_ACTIVE,
  62         [SOCKET_RUNNING] = UNIT_ACTIVE,
  63         [SOCKET_STOP_PRE] = UNIT_DEACTIVATING,
  64         [SOCKET_STOP_PRE_SIGTERM] = UNIT_DEACTIVATING,
  65         [SOCKET_STOP_PRE_SIGKILL] = UNIT_DEACTIVATING,
  66         [SOCKET_STOP_POST] = UNIT_DEACTIVATING,
  67         [SOCKET_FINAL_SIGTERM] = UNIT_DEACTIVATING,
  68         [SOCKET_FINAL_SIGKILL] = UNIT_DEACTIVATING,
  69         [SOCKET_FAILED] = UNIT_FAILED
  70 };
  71
  72 static int socket_dispatch_io(sd_event_source *source, int fd, uint32_t revents, void *userdata);
  73 static int socket_dispatch_timer(sd_event_source *source, usec_t usec, void *userdata);
  74
  75 static void socket_init(Unit *u) {
  76         Socket *s = SOCKET(u);
  77
  78         assert(u);
  79         assert(u->load_state == UNIT_STUB);
  80
  81         s->backlog = SOMAXCONN;
  82         s->timeout_usec = u->manager->default_timeout_start_usec;
  83         s->directory_mode = 0755;
  84         s->socket_mode = 0666;
  85
  86         s->max_connections = 64;
  87
  88         s->priority = -1;
  89         s->ip_tos = -1;
  90         s->ip_ttl = -1;
  91         s->mark = -1;
  92
  93         exec_context_init(&s->exec_context);
  94         s->exec_context.std_output = u->manager->default_std_output;
  95         s->exec_context.std_error = u->manager->default_std_error;
  96         kill_context_init(&s->kill_context);
  97         cgroup_context_init(&s->cgroup_context);
  98
  99         s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
 100 }
 101
 102 static void socket_unwatch_control_pid(Socket *s) {
 103         assert(s);
 104
 105         if (s->control_pid <= 0)
 106                 return;
 107
 108         unit_unwatch_pid(UNIT(s), s->control_pid);
 109         s->control_pid = 0;
 110 }
 111
 112 void socket_free_ports(Socket *s) {
 113         SocketPort *p;
 114
 115         assert(s);
 116
 117         while ((p = s->ports)) {
 118                 LIST_REMOVE(port, s->ports, p);
 119
 120                 sd_event_source_unref(p->event_source);
 121
 122                 if (p->fd >= 0)
 123                         close_nointr_nofail(p->fd);
 124
 125                 free(p->path);
 126                 free(p);
 127         }
 128 }
 129
 130 static void socket_done(Unit *u) {
 131         Socket *s = SOCKET(u);
 132
 133         assert(s);
 134
 135         socket_free_ports(s);
 136
 137         cgroup_context_done(&s->cgroup_context);
 138         exec_context_done(&s->exec_context);
 139         s->exec_runtime = exec_runtime_unref(s->exec_runtime);
 140         exec_command_free_array(s->exec_command, _SOCKET_EXEC_COMMAND_MAX);
 141         s->control_command = NULL;
 142
 143         socket_unwatch_control_pid(s);
 144
 145         unit_ref_unset(&s->service);
 146
 147         free(s->tcp_congestion);
 148         s->tcp_congestion = NULL;
 149
 150         free(s->bind_to_device);
 151         s->bind_to_device = NULL;
 152
 153         free(s->smack);
 154         free(s->smack_ip_in);
 155         free(s->smack_ip_out);
 156
 157         s->timer_event_source = sd_event_source_unref(s->timer_event_source);
 158 }
 159
 160 static int socket_arm_timer(Socket *s) {
 161         int r;
 162
 163         assert(s);
 164
 165         if (s->timeout_usec <= 0) {
 166                 s->timer_event_source = sd_event_source_unref(s->timer_event_source);
 167                 return 0;
 168         }
 169
 170         if (s->timer_event_source) {
 171                 r = sd_event_source_set_time(s->timer_event_source, now(CLOCK_MONOTONIC) + s->timeout_usec);
 172                 if (r < 0)
 173                         return r;
 174
 175                 return sd_event_source_set_enabled(s->timer_event_source, SD_EVENT_ONESHOT);
 176         }
 177
 178         return sd_event_add_monotonic(UNIT(s)->manager->event, now(CLOCK_MONOTONIC) + s->timeout_usec, 0, socket_dispatch_timer, s, &s->timer_event_source);
 179 }
 180
 181 static int socket_instantiate_service(Socket *s) {
 182         _cleanup_free_ char *prefix = NULL;
 183         _cleanup_free_ char *name = NULL;
 184         int r;
 185         Unit *u;
 186
 187         assert(s);
 188
 189         /* This fills in s->service if it isn't filled in yet. For
 190          * Accept=yes sockets we create the next connection service
 191          * here. For Accept=no this is mostly a NOP since the service
 192          * is figured out at load time anyway. */
 193
 194         if (UNIT_DEREF(s->service))
 195                 return 0;
 196
 197         assert(s->accept);
 198
 199         if (!(prefix = unit_name_to_prefix(UNIT(s)->id)))
 200                 return -ENOMEM;
 201
 202         r = asprintf(&name, "%s@%u.service", prefix, s->n_accepted);
 203
 204         if (r < 0)
 205                 return -ENOMEM;
 206
 207         r = manager_load_unit(UNIT(s)->manager, name, NULL, NULL, &u);
 208
 209         if (r < 0)
 210                 return r;
 211
 212 #ifdef HAVE_SYSV_COMPAT
 213         if (SERVICE(u)->is_sysv) {
 214                 log_error("Using SysV services for socket activation is not supported. Refusing.");
 215                 return -ENOENT;
 216         }
 217 #endif
 218
 219         u->no_gc = true;
 220         unit_ref_set(&s->service, u);
 221
 222         return unit_add_two_dependencies(UNIT(s), UNIT_BEFORE, UNIT_TRIGGERS, u, false);
 223 }
 224
 225 static bool have_non_accept_socket(Socket *s) {
 226         SocketPort *p;
 227
 228         assert(s);
 229
 230         if (!s->accept)
 231                 return true;
 232
 233         LIST_FOREACH(port, p, s->ports) {
 234
 235                 if (p->type != SOCKET_SOCKET)
 236                         return true;
 237
 238                 if (!socket_address_can_accept(&p->address))
 239                         return true;
 240         }
 241
 242         return false;
 243 }
 244
 245 static int socket_add_mount_links(Socket *s) {
 246         SocketPort *p;
 247         int r;
 248
 249         assert(s);
 250
 251         LIST_FOREACH(port, p, s->ports) {
 252                 const char *path = NULL;
 253
 254                 if (p->type == SOCKET_SOCKET)
 255                         path = socket_address_get_path(&p->address);
 256                 else if (p->type == SOCKET_FIFO || p->type == SOCKET_SPECIAL)
 257                         path = p->path;
 258
 259                 if (!path)
 260                         continue;
 261
 262                 r = unit_require_mounts_for(UNIT(s), path);
 263                 if (r < 0)
 264                         return r;
 265         }
 266
 267         return 0;
 268 }
 269
 270 static int socket_add_device_link(Socket *s) {
 271         char *t;
 272
 273         assert(s);
 274
 275         if (!s->bind_to_device || streq(s->bind_to_device, "lo"))
 276                 return 0;
 277
 278         t = strappenda("/sys/subsystem/net/devices/", s->bind_to_device);
 279         return unit_add_node_link(UNIT(s), t, false);
 280 }
 281
 282 static int socket_add_default_dependencies(Socket *s) {
 283         int r;
 284         assert(s);
 285
 286         r = unit_add_dependency_by_name(UNIT(s), UNIT_BEFORE, SPECIAL_SOCKETS_TARGET, NULL, true);
 287         if (r < 0)
 288                 return r;
 289
 290         if (UNIT(s)->manager->running_as == SYSTEMD_SYSTEM) {
 291                 r = unit_add_two_dependencies_by_name(UNIT(s), UNIT_AFTER, UNIT_REQUIRES, SPECIAL_SYSINIT_TARGET, NULL, true);
 292                 if (r < 0)
 293                         return r;
 294         }
 295
 296         return unit_add_two_dependencies_by_name(UNIT(s), UNIT_BEFORE, UNIT_CONFLICTS, SPECIAL_SHUTDOWN_TARGET, NULL, true);
 297 }
 298
 299 _pure_ static bool socket_has_exec(Socket *s) {
 300         unsigned i;
 301         assert(s);
 302
 303         for (i = 0; i < _SOCKET_EXEC_COMMAND_MAX; i++)
 304                 if (s->exec_command[i])
 305                         return true;
 306
 307         return false;
 308 }
 309
 310 static int socket_add_extras(Socket *s) {
 311         Unit *u = UNIT(s);
 312         int r;
 313
 314         assert(s);
 315
 316         if (have_non_accept_socket(s)) {
 317
 318                 if (!UNIT_DEREF(s->service)) {
 319                         Unit *x;
 320
 321                         r = unit_load_related_unit(u, ".service", &x);
 322                         if (r < 0)
 323                                 return r;
 324
 325                         unit_ref_set(&s->service, x);
 326                 }
 327
 328                 r = unit_add_two_dependencies(u, UNIT_BEFORE, UNIT_TRIGGERS, UNIT_DEREF(s->service), true);
 329                 if (r < 0)
 330                         return r;
 331         }
 332
 333         r = socket_add_mount_links(s);
 334         if (r < 0)
 335                 return r;
 336
 337         r = socket_add_device_link(s);
 338         if (r < 0)
 339                 return r;
 340
 341         r = unit_exec_context_defaults(u, &s->exec_context);
 342         if (r < 0)
 343                 return r;
 344
 345         if (socket_has_exec(s)) {
 346                 r = unit_add_exec_dependencies(u, &s->exec_context);
 347                 if (r < 0)
 348                         return r;
 349
 350                 r = unit_add_default_slice(u);
 351                 if (r < 0)
 352                         return r;
 353         }
 354
 355         if (u->default_dependencies) {
 356                 r = socket_add_default_dependencies(s);
 357                 if (r < 0)
 358                         return r;
 359         }
 360
 361         return 0;
 362 }
 363
 364 static int socket_verify(Socket *s) {
 365         assert(s);
 366
 367         if (UNIT(s)->load_state != UNIT_LOADED)
 368                 return 0;
 369
 370         if (!s->ports) {
 371                 log_error_unit(UNIT(s)->id, "%s lacks Listen setting. Refusing.", UNIT(s)->id);
 372                 return -EINVAL;
 373         }
 374
 375         if (s->accept && have_non_accept_socket(s)) {
 376                 log_error_unit(UNIT(s)->id, "%s configured for accepting sockets, but sockets are non-accepting. Refusing.",
 377                                UNIT(s)->id);
 378                 return -EINVAL;
 379         }
 380
 381         if (s->accept && s->max_connections <= 0) {
 382                 log_error_unit(UNIT(s)->id, "%s's MaxConnection setting too small. Refusing.", UNIT(s)->id);
 383                 return -EINVAL;
 384         }
 385
 386         if (s->accept && UNIT_DEREF(s->service)) {
 387                 log_error_unit(UNIT(s)->id, "Explicit service configuration for accepting sockets not supported on %s. Refusing.", UNIT(s)->id);
 388                 return -EINVAL;
 389         }
 390
 391         if (s->exec_context.pam_name && s->kill_context.kill_mode != KILL_CONTROL_GROUP) {
 392                 log_error_unit(UNIT(s)->id, "%s has PAM enabled. Kill mode must be set to 'control-group'. Refusing.", UNIT(s)->id);
 393                 return -EINVAL;
 394         }
 395
 396         return 0;
 397 }
 398
 399 static int socket_load(Unit *u) {
 400         Socket *s = SOCKET(u);
 401         int r;
 402
 403         assert(u);
 404         assert(u->load_state == UNIT_STUB);
 405
 406         r = unit_load_fragment_and_dropin(u);
 407         if (r < 0)
 408                 return r;
 409
 410         if (u->load_state == UNIT_LOADED) {
 411                 /* This is a new unit? Then let's add in some extras */
 412                 r = socket_add_extras(s);
 413                 if (r < 0)
 414                         return r;
 415         }
 416
 417         return socket_verify(s);
 418 }
 419
 420 _const_ static const char* listen_lookup(int family, int type) {
 421
 422         if (family == AF_NETLINK)
 423                 return "ListenNetlink";
 424
 425         if (type == SOCK_STREAM)
 426                 return "ListenStream";
 427         else if (type == SOCK_DGRAM)
 428                 return "ListenDatagram";
 429         else if (type == SOCK_SEQPACKET)
 430                 return "ListenSequentialPacket";
 431
 432         assert_not_reached("Unknown socket type");
 433         return NULL;
 434 }
 435
 436 static void socket_dump(Unit *u, FILE *f, const char *prefix) {
 437         SocketExecCommand c;
 438         Socket *s = SOCKET(u);
 439         SocketPort *p;
 440         const char *prefix2;
 441
 442         assert(s);
 443         assert(f);
 444
 445         prefix2 = strappenda(prefix, "\t");
 446
 447         fprintf(f,
 448                 "%sSocket State: %s\n"
 449                 "%sResult: %s\n"
 450                 "%sBindIPv6Only: %s\n"
 451                 "%sBacklog: %u\n"
 452                 "%sSocketMode: %04o\n"
 453                 "%sDirectoryMode: %04o\n"
 454                 "%sKeepAlive: %s\n"
 455                 "%sFreeBind: %s\n"
 456                 "%sTransparent: %s\n"
 457                 "%sBroadcast: %s\n"
 458                 "%sPassCredentials: %s\n"
 459                 "%sPassSecurity: %s\n"
 460                 "%sTCPCongestion: %s\n",
 461                 prefix, socket_state_to_string(s->state),
 462                 prefix, socket_result_to_string(s->result),
 463                 prefix, socket_address_bind_ipv6_only_to_string(s->bind_ipv6_only),
 464                 prefix, s->backlog,
 465                 prefix, s->socket_mode,
 466                 prefix, s->directory_mode,
 467                 prefix, yes_no(s->keep_alive),
 468                 prefix, yes_no(s->free_bind),
 469                 prefix, yes_no(s->transparent),
 470                 prefix, yes_no(s->broadcast),
 471                 prefix, yes_no(s->pass_cred),
 472                 prefix, yes_no(s->pass_sec),
 473                 prefix, strna(s->tcp_congestion));
 474
 475         if (s->control_pid > 0)
 476                 fprintf(f,
 477                         "%sControl PID: %lu\n",
 478                         prefix, (unsigned long) s->control_pid);
 479
 480         if (s->bind_to_device)
 481                 fprintf(f,
 482                         "%sBindToDevice: %s\n",
 483                         prefix, s->bind_to_device);
 484
 485         if (s->accept)
 486                 fprintf(f,
 487                         "%sAccepted: %u\n"
 488                         "%sNConnections: %u\n"
 489                         "%sMaxConnections: %u\n",
 490                         prefix, s->n_accepted,
 491                         prefix, s->n_connections,
 492                         prefix, s->max_connections);
 493
 494         if (s->priority >= 0)
 495                 fprintf(f,
 496                         "%sPriority: %i\n",
 497                         prefix, s->priority);
 498
 499         if (s->receive_buffer > 0)
 500                 fprintf(f,
 501                         "%sReceiveBuffer: %zu\n",
 502                         prefix, s->receive_buffer);
 503
 504         if (s->send_buffer > 0)
 505                 fprintf(f,
 506                         "%sSendBuffer: %zu\n",
 507                         prefix, s->send_buffer);
 508
 509         if (s->ip_tos >= 0)
 510                 fprintf(f,
 511                         "%sIPTOS: %i\n",
 512                         prefix, s->ip_tos);
 513
 514         if (s->ip_ttl >= 0)
 515                 fprintf(f,
 516                         "%sIPTTL: %i\n",
 517                         prefix, s->ip_ttl);
 518
 519         if (s->pipe_size > 0)
 520                 fprintf(f,
 521                         "%sPipeSize: %zu\n",
 522                         prefix, s->pipe_size);
 523
 524         if (s->mark >= 0)
 525                 fprintf(f,
 526                         "%sMark: %i\n",
 527                         prefix, s->mark);
 528
 529         if (s->mq_maxmsg > 0)
 530                 fprintf(f,
 531                         "%sMessageQueueMaxMessages: %li\n",
 532                         prefix, s->mq_maxmsg);
 533
 534         if (s->mq_msgsize > 0)
 535                 fprintf(f,
 536                         "%sMessageQueueMessageSize: %li\n",
 537                         prefix, s->mq_msgsize);
 538
 539         if (s->reuse_port)
 540                 fprintf(f,
 541                         "%sReusePort: %s\n",
 542                          prefix, yes_no(s->reuse_port));
 543
 544         if (s->smack)
 545                 fprintf(f,
 546                         "%sSmackLabel: %s\n",
 547                         prefix, s->smack);
 548
 549         if (s->smack_ip_in)
 550                 fprintf(f,
 551                         "%sSmackLabelIPIn: %s\n",
 552                         prefix, s->smack_ip_in);
 553
 554         if (s->smack_ip_out)
 555                 fprintf(f,
 556                         "%sSmackLabelIPOut: %s\n",
 557                         prefix, s->smack_ip_out);
 558
 559         LIST_FOREACH(port, p, s->ports) {
 560
 561                 if (p->type == SOCKET_SOCKET) {
 562                         const char *t;
 563                         int r;
 564                         char *k = NULL;
 565
 566                         if ((r = socket_address_print(&p->address, &k)) < 0)
 567                                 t = strerror(-r);
 568                         else
 569                                 t = k;
 570
 571                         fprintf(f, "%s%s: %s\n", prefix, listen_lookup(socket_address_family(&p->address), p->address.type), t);
 572                         free(k);
 573                 } else if (p->type == SOCKET_SPECIAL)
 574                         fprintf(f, "%sListenSpecial: %s\n", prefix, p->path);
 575                 else if (p->type == SOCKET_MQUEUE)
 576                         fprintf(f, "%sListenMessageQueue: %s\n", prefix, p->path);
 577                 else
 578                         fprintf(f, "%sListenFIFO: %s\n", prefix, p->path);
 579         }
 580
 581         exec_context_dump(&s->exec_context, f, prefix);
 582         kill_context_dump(&s->kill_context, f, prefix);
 583
 584         for (c = 0; c < _SOCKET_EXEC_COMMAND_MAX; c++) {
 585                 if (!s->exec_command[c])
 586                         continue;
 587
 588                 fprintf(f, "%s-> %s:\n",
 589                         prefix, socket_exec_command_to_string(c));
 590
 591                 exec_command_dump_list(s->exec_command[c], f, prefix2);
 592         }
 593 }
 594
 595 static int instance_from_socket(int fd, unsigned nr, char **instance) {
 596         socklen_t l;
 597         char *r;
 598         union {
 599                 struct sockaddr sa;
 600                 struct sockaddr_un un;
 601                 struct sockaddr_in in;
 602                 struct sockaddr_in6 in6;
 603                 struct sockaddr_storage storage;
 604         } local, remote;
 605
 606         assert(fd >= 0);
 607         assert(instance);
 608
 609         l = sizeof(local);
 610         if (getsockname(fd, &local.sa, &l) < 0)
 611                 return -errno;
 612
 613         l = sizeof(remote);
 614         if (getpeername(fd, &remote.sa, &l) < 0)
 615                 return -errno;
 616
 617         switch (local.sa.sa_family) {
 618
 619         case AF_INET: {
 620                 uint32_t
 621                         a = ntohl(local.in.sin_addr.s_addr),
 622                         b = ntohl(remote.in.sin_addr.s_addr);
 623
 624                 if (asprintf(&r,
 625                              "%u-%u.%u.%u.%u:%u-%u.%u.%u.%u:%u",
 626                              nr,
 627                              a >> 24, (a >> 16) & 0xFF, (a >> 8) & 0xFF, a & 0xFF,
 628                              ntohs(local.in.sin_port),
 629                              b >> 24, (b >> 16) & 0xFF, (b >> 8) & 0xFF, b & 0xFF,
 630                              ntohs(remote.in.sin_port)) < 0)
 631                         return -ENOMEM;
 632
 633                 break;
 634         }
 635
 636         case AF_INET6: {
 637                 static const unsigned char ipv4_prefix[] = {
 638                         0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xFF, 0xFF
 639                 };
 640
 641                 if (memcmp(&local.in6.sin6_addr, ipv4_prefix, sizeof(ipv4_prefix)) == 0 &&
 642                     memcmp(&remote.in6.sin6_addr, ipv4_prefix, sizeof(ipv4_prefix)) == 0) {
 643                         const uint8_t
 644                                 *a = local.in6.sin6_addr.s6_addr+12,
 645                                 *b = remote.in6.sin6_addr.s6_addr+12;
 646
 647                         if (asprintf(&r,
 648                                      "%u-%u.%u.%u.%u:%u-%u.%u.%u.%u:%u",
 649                                      nr,
 650                                      a[0], a[1], a[2], a[3],
 651                                      ntohs(local.in6.sin6_port),
 652                                      b[0], b[1], b[2], b[3],
 653                                      ntohs(remote.in6.sin6_port)) < 0)
 654                                 return -ENOMEM;
 655                 } else {
 656                         char a[INET6_ADDRSTRLEN], b[INET6_ADDRSTRLEN];
 657
 658                         if (asprintf(&r,
 659                                      "%u-%s:%u-%s:%u",
 660                                      nr,
 661                                      inet_ntop(AF_INET6, &local.in6.sin6_addr, a, sizeof(a)),
 662                                      ntohs(local.in6.sin6_port),
 663                                      inet_ntop(AF_INET6, &remote.in6.sin6_addr, b, sizeof(b)),
 664                                      ntohs(remote.in6.sin6_port)) < 0)
 665                                 return -ENOMEM;
 666                 }
 667
 668                 break;
 669         }
 670
 671         case AF_UNIX: {
 672                 struct ucred ucred;
 673                 int k;
 674
 675                 k = getpeercred(fd, &ucred);
 676                 if (k < 0)
 677                         return k;
 678
 679                 if (asprintf(&r,
 680                              "%u-%lu-%lu",
 681                              nr,
 682                              (unsigned long) ucred.pid,
 683                              (unsigned long) ucred.uid) < 0)
 684                         return -ENOMEM;
 685
 686                 break;
 687         }
 688
 689         default:
 690                 assert_not_reached("Unhandled socket type.");
 691         }
 692
 693         *instance = r;
 694         return 0;
 695 }
 696
 697 static void socket_close_fds(Socket *s) {
 698         SocketPort *p;
 699
 700         assert(s);
 701
 702         LIST_FOREACH(port, p, s->ports) {
 703
 704                 p->event_source = sd_event_source_unref(p->event_source);
 705
 706                 if (p->fd < 0)
 707                         continue;
 708
 709                 close_nointr_nofail(p->fd);
 710
 711                 /* One little note: we should never delete any sockets
 712                  * in the file system here! After all some other
 713                  * process we spawned might still have a reference of
 714                  * this fd and wants to continue to use it. Therefore
 715                  * we delete sockets in the file system before we
 716                  * create a new one, not after we stopped using
 717                  * one! */
 718
 719                 p->fd = -1;
 720         }
 721 }
 722
 723 static void socket_apply_socket_options(Socket *s, int fd) {
 724         assert(s);
 725         assert(fd >= 0);
 726
 727         if (s->keep_alive) {
 728                 int b = s->keep_alive;
 729                 if (setsockopt(fd, SOL_SOCKET, SO_KEEPALIVE, &b, sizeof(b)) < 0)
 730                         log_warning_unit(UNIT(s)->id, "SO_KEEPALIVE failed: %m");
 731         }
 732
 733         if (s->broadcast) {
 734                 int one = 1;
 735                 if (setsockopt(fd, SOL_SOCKET, SO_BROADCAST, &one, sizeof(one)) < 0)
 736                         log_warning_unit(UNIT(s)->id, "SO_BROADCAST failed: %m");
 737         }
 738
 739         if (s->pass_cred) {
 740                 int one = 1;
 741                 if (setsockopt(fd, SOL_SOCKET, SO_PASSCRED, &one, sizeof(one)) < 0)
 742                         log_warning_unit(UNIT(s)->id, "SO_PASSCRED failed: %m");
 743         }
 744
 745         if (s->pass_sec) {
 746                 int one = 1;
 747                 if (setsockopt(fd, SOL_SOCKET, SO_PASSSEC, &one, sizeof(one)) < 0)
 748                         log_warning_unit(UNIT(s)->id, "SO_PASSSEC failed: %m");
 749         }
 750
 751         if (s->priority >= 0)
 752                 if (setsockopt(fd, SOL_SOCKET, SO_PRIORITY, &s->priority, sizeof(s->priority)) < 0)
 753                         log_warning_unit(UNIT(s)->id, "SO_PRIORITY failed: %m");
 754
 755         if (s->receive_buffer > 0) {
 756                 int value = (int) s->receive_buffer;
 757
 758                 /* We first try with SO_RCVBUFFORCE, in case we have the perms for that */
 759
 760                 if (setsockopt(fd, SOL_SOCKET, SO_RCVBUFFORCE, &value, sizeof(value)) < 0)
 761                         if (setsockopt(fd, SOL_SOCKET, SO_RCVBUF, &value, sizeof(value)) < 0)
 762                                 log_warning_unit(UNIT(s)->id, "SO_RCVBUF failed: %m");
 763         }
 764
 765         if (s->send_buffer > 0) {
 766                 int value = (int) s->send_buffer;
 767                 if (setsockopt(fd, SOL_SOCKET, SO_SNDBUFFORCE, &value, sizeof(value)) < 0)
 768                         if (setsockopt(fd, SOL_SOCKET, SO_SNDBUF, &value, sizeof(value)) < 0)
 769                                 log_warning_unit(UNIT(s)->id, "SO_SNDBUF failed: %m");
 770         }
 771
 772         if (s->mark >= 0)
 773                 if (setsockopt(fd, SOL_SOCKET, SO_MARK, &s->mark, sizeof(s->mark)) < 0)
 774                         log_warning_unit(UNIT(s)->id, "SO_MARK failed: %m");
 775
 776         if (s->ip_tos >= 0)
 777                 if (setsockopt(fd, IPPROTO_IP, IP_TOS, &s->ip_tos, sizeof(s->ip_tos)) < 0)
 778                         log_warning_unit(UNIT(s)->id, "IP_TOS failed: %m");
 779
 780         if (s->ip_ttl >= 0) {
 781                 int r, x;
 782
 783                 r = setsockopt(fd, IPPROTO_IP, IP_TTL, &s->ip_ttl, sizeof(s->ip_ttl));
 784
 785                 if (socket_ipv6_is_supported())
 786                         x = setsockopt(fd, IPPROTO_IPV6, IPV6_UNICAST_HOPS, &s->ip_ttl, sizeof(s->ip_ttl));
 787                 else {
 788                         x = -1;
 789                         errno = EAFNOSUPPORT;
 790                 }
 791
 792                 if (r < 0 && x < 0)
 793                         log_warning_unit(UNIT(s)->id,
 794                                          "IP_TTL/IPV6_UNICAST_HOPS failed: %m");
 795         }
 796
 797         if (s->tcp_congestion)
 798                 if (setsockopt(fd, SOL_TCP, TCP_CONGESTION, s->tcp_congestion, strlen(s->tcp_congestion)+1) < 0)
 799                         log_warning_unit(UNIT(s)->id, "TCP_CONGESTION failed: %m");
 800
 801         if (s->reuse_port) {
 802                 int b = s->reuse_port;
 803                 if (setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &b, sizeof(b)) < 0)
 804                         log_warning_unit(UNIT(s)->id, "SO_REUSEPORT failed: %m");
 805         }
 806
 807         if (s->smack_ip_in)
 808                 if (smack_label_ip_in_fd(fd, s->smack_ip_in) < 0)
 809                         log_error_unit(UNIT(s)->id, "smack_label_ip_in_fd: %m");
 810
 811         if (s->smack_ip_out)
 812                 if (smack_label_ip_out_fd(fd, s->smack_ip_out) < 0)
 813                         log_error_unit(UNIT(s)->id, "smack_label_ip_out_fd: %m");
 814 }
 815
 816 static void socket_apply_fifo_options(Socket *s, int fd) {
 817         assert(s);
 818         assert(fd >= 0);
 819
 820         if (s->pipe_size > 0)
 821                 if (fcntl(fd, F_SETPIPE_SZ, s->pipe_size) < 0)
 822                         log_warning_unit(UNIT(s)->id,
 823                                          "F_SETPIPE_SZ: %m");
 824
 825         if (s->smack)
 826                 if (smack_label_fd(fd, s->smack) < 0)
 827                         log_error_unit(UNIT(s)->id, "smack_label_fd: %m");
 828 }
 829
 830 static int fifo_address_create(
 831                 const char *path,
 832                 mode_t directory_mode,
 833                 mode_t socket_mode,
 834                 int *_fd) {
 835
 836         int fd = -1, r = 0;
 837         struct stat st;
 838         mode_t old_mask;
 839
 840         assert(path);
 841         assert(_fd);
 842
 843         mkdir_parents_label(path, directory_mode);
 844
 845         r = label_context_set(path, S_IFIFO);
 846         if (r < 0)
 847                 goto fail;
 848
 849         /* Enforce the right access mode for the fifo */
 850         old_mask = umask(~ socket_mode);
 851
 852         /* Include the original umask in our mask */
 853         umask(~socket_mode | old_mask);
 854
 855         r = mkfifo(path, socket_mode);
 856         umask(old_mask);
 857
 858         if (r < 0 && errno != EEXIST) {
 859                 r = -errno;
 860                 goto fail;
 861         }
 862
 863         if ((fd = open(path, O_RDWR|O_CLOEXEC|O_NOCTTY|O_NONBLOCK|O_NOFOLLOW)) < 0) {
 864                 r = -errno;
 865                 goto fail;
 866         }
 867
 868         label_context_clear();
 869
 870         if (fstat(fd, &st) < 0) {
 871                 r = -errno;
 872                 goto fail;
 873         }
 874
 875         if (!S_ISFIFO(st.st_mode) ||
 876             (st.st_mode & 0777) != (socket_mode & ~old_mask) ||
 877             st.st_uid != getuid() ||
 878             st.st_gid != getgid()) {
 879
 880                 r = -EEXIST;
 881                 goto fail;
 882         }
 883
 884         *_fd = fd;
 885         return 0;
 886
 887 fail:
 888         label_context_clear();
 889
 890         if (fd >= 0)
 891                 close_nointr_nofail(fd);
 892
 893         return r;
 894 }
 895
 896 static int special_address_create(
 897                 const char *path,
 898                 int *_fd) {
 899
 900         int fd = -1, r = 0;
 901         struct stat st;
 902
 903         assert(path);
 904         assert(_fd);
 905
 906         if ((fd = open(path, O_RDONLY|O_CLOEXEC|O_NOCTTY|O_NONBLOCK|O_NOFOLLOW)) < 0) {
 907                 r = -errno;
 908                 goto fail;
 909         }
 910
 911         if (fstat(fd, &st) < 0) {
 912                 r = -errno;
 913                 goto fail;
 914         }
 915
 916         /* Check whether this is a /proc, /sys or /dev file or char device */
 917         if (!S_ISREG(st.st_mode) && !S_ISCHR(st.st_mode)) {
 918                 r = -EEXIST;
 919                 goto fail;
 920         }
 921
 922         *_fd = fd;
 923         return 0;
 924
 925 fail:
 926         if (fd >= 0)
 927                 close_nointr_nofail(fd);
 928
 929         return r;
 930 }
 931
 932 static int mq_address_create(
 933                 const char *path,
 934                 mode_t mq_mode,
 935                 long maxmsg,
 936                 long msgsize,
 937                 int *_fd) {
 938
 939         int fd = -1, r = 0;
 940         struct stat st;
 941         mode_t old_mask;
 942         struct mq_attr _attr, *attr = NULL;
 943
 944         assert(path);
 945         assert(_fd);
 946
 947         if (maxmsg > 0 && msgsize > 0) {
 948                 zero(_attr);
 949                 _attr.mq_flags = O_NONBLOCK;
 950                 _attr.mq_maxmsg = maxmsg;
 951                 _attr.mq_msgsize = msgsize;
 952                 attr = &_attr;
 953         }
 954
 955         /* Enforce the right access mode for the mq */
 956         old_mask = umask(~ mq_mode);
 957
 958         /* Include the original umask in our mask */
 959         umask(~mq_mode | old_mask);
 960
 961         fd = mq_open(path, O_RDONLY|O_CLOEXEC|O_NONBLOCK|O_CREAT, mq_mode, attr);
 962         umask(old_mask);
 963
 964         if (fd < 0) {
 965                 r = -errno;
 966                 goto fail;
 967         }
 968
 969         if (fstat(fd, &st) < 0) {
 970                 r = -errno;
 971                 goto fail;
 972         }
 973
 974         if ((st.st_mode & 0777) != (mq_mode & ~old_mask) ||
 975             st.st_uid != getuid() ||
 976             st.st_gid != getgid()) {
 977
 978                 r = -EEXIST;
 979                 goto fail;
 980         }
 981
 982         *_fd = fd;
 983         return 0;
 984
 985 fail:
 986         if (fd >= 0)
 987                 close_nointr_nofail(fd);
 988
 989         return r;
 990 }
 991
 992 static int socket_open_fds(Socket *s) {
 993         SocketPort *p;
 994         int r;
 995         char *label = NULL;
 996         bool know_label = false;
 997
 998         assert(s);
 999
1000         LIST_FOREACH(port, p, s->ports) {
1001
1002                 if (p->fd >= 0)
1003                         continue;
1004
1005                 if (p->type == SOCKET_SOCKET) {
1006
1007                         if (!know_label) {
1008
1009                                 if ((r = socket_instantiate_service(s)) < 0)
1010                                         return r;
1011
1012                                 if (UNIT_ISSET(s->service) &&
1013                                     SERVICE(UNIT_DEREF(s->service))->exec_command[SERVICE_EXEC_START]) {
1014                                         r = label_get_create_label_from_exe(SERVICE(UNIT_DEREF(s->service))->exec_command[SERVICE_EXEC_START]->path, &label);
1015
1016                                         if (r < 0) {
1017                                                 if (r != -EPERM)
1018                                                         return r;
1019                                         }
1020                                 }
1021
1022                                 know_label = true;
1023                         }
1024
1025                         r = socket_address_listen(
1026                                         &p->address,
1027                                         SOCK_CLOEXEC|SOCK_NONBLOCK,
1028                                         s->backlog,
1029                                         s->bind_ipv6_only,
1030                                         s->bind_to_device,
1031                                         s->free_bind,
1032                                         s->transparent,
1033                                         s->directory_mode,
1034                                         s->socket_mode,
1035                                         label);
1036                         if (r < 0)
1037                                 goto rollback;
1038
1039                         p->fd = r;
1040                         socket_apply_socket_options(s, p->fd);
1041
1042                 } else  if (p->type == SOCKET_SPECIAL) {
1043
1044                         r = special_address_create(
1045                                         p->path,
1046                                         &p->fd);
1047                         if (r < 0)
1048                                 goto rollback;
1049
1050                 } else  if (p->type == SOCKET_FIFO) {
1051
1052                         r = fifo_address_create(
1053                                         p->path,
1054                                         s->directory_mode,
1055                                         s->socket_mode,
1056                                         &p->fd);
1057                         if (r < 0)
1058                                 goto rollback;
1059
1060                         socket_apply_fifo_options(s, p->fd);
1061                 } else if (p->type == SOCKET_MQUEUE) {
1062
1063                         r = mq_address_create(
1064                                         p->path,
1065                                         s->socket_mode,
1066                                         s->mq_maxmsg,
1067                                         s->mq_msgsize,
1068                                         &p->fd);
1069                         if (r < 0)
1070                                 goto rollback;
1071                 } else
1072                         assert_not_reached("Unknown port type");
1073         }
1074
1075         label_free(label);
1076         return 0;
1077
1078 rollback:
1079         socket_close_fds(s);
1080         label_free(label);
1081         return r;
1082 }
1083
1084 static void socket_unwatch_fds(Socket *s) {
1085         SocketPort *p;
1086         int r;
1087
1088         assert(s);
1089
1090         LIST_FOREACH(port, p, s->ports) {
1091                 if (p->fd < 0)
1092                         continue;
1093
1094                 if (p->event_source) {
1095                         r = sd_event_source_set_enabled(p->event_source, SD_EVENT_OFF);
1096                         if (r < 0)
1097                                 log_debug_unit(UNIT(s)->id, "Failed to disable event source.");
1098                 }
1099         }
1100 }
1101
1102 static int socket_watch_fds(Socket *s) {
1103         SocketPort *p;
1104         int r;
1105
1106         assert(s);
1107
1108         LIST_FOREACH(port, p, s->ports) {
1109                 if (p->fd < 0)
1110                         continue;
1111
1112                 if (p->event_source)
1113                         r = sd_event_source_set_enabled(p->event_source, SD_EVENT_ON);
1114                 else
1115                         r = sd_event_add_io(UNIT(s)->manager->event, p->fd, EPOLLIN, socket_dispatch_io, p, &p->event_source);
1116
1117                 if (r < 0) {
1118                         log_warning_unit(UNIT(s)->id, "Failed to watch listening fds: %s", strerror(-r));
1119                         goto fail;
1120                 }
1121         }
1122
1123         return 0;
1124
1125 fail:
1126         socket_unwatch_fds(s);
1127         return r;
1128 }
1129
1130 static void socket_set_state(Socket *s, SocketState state) {
1131         SocketState old_state;
1132         assert(s);
1133
1134         old_state = s->state;
1135         s->state = state;
1136
1137         if (state != SOCKET_START_PRE &&
1138             state != SOCKET_START_POST &&
1139             state != SOCKET_STOP_PRE &&
1140             state != SOCKET_STOP_PRE_SIGTERM &&
1141             state != SOCKET_STOP_PRE_SIGKILL &&
1142             state != SOCKET_STOP_POST &&
1143             state != SOCKET_FINAL_SIGTERM &&
1144             state != SOCKET_FINAL_SIGKILL) {
1145
1146                 s->timer_event_source = sd_event_source_unref(s->timer_event_source);
1147                 socket_unwatch_control_pid(s);
1148                 s->control_command = NULL;
1149                 s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
1150         }
1151
1152         if (state != SOCKET_LISTENING)
1153                 socket_unwatch_fds(s);
1154
1155         if (state != SOCKET_START_POST &&
1156             state != SOCKET_LISTENING &&
1157             state != SOCKET_RUNNING &&
1158             state != SOCKET_STOP_PRE &&
1159             state != SOCKET_STOP_PRE_SIGTERM &&
1160             state != SOCKET_STOP_PRE_SIGKILL)
1161                 socket_close_fds(s);
1162
1163         if (state != old_state)
1164                 log_debug_unit(UNIT(s)->id, "%s changed %s -> %s",
1165                                UNIT(s)->id, socket_state_to_string(old_state), socket_state_to_string(state));
1166
1167         unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state], true);
1168 }
1169
1170 static int socket_coldplug(Unit *u) {
1171         Socket *s = SOCKET(u);
1172         int r;
1173
1174         assert(s);
1175         assert(s->state == SOCKET_DEAD);
1176
1177         if (s->deserialized_state == s->state)
1178                 return 0;
1179
1180         if (s->deserialized_state == SOCKET_START_PRE ||
1181             s->deserialized_state == SOCKET_START_POST ||
1182             s->deserialized_state == SOCKET_STOP_PRE ||
1183             s->deserialized_state == SOCKET_STOP_PRE_SIGTERM ||
1184             s->deserialized_state == SOCKET_STOP_PRE_SIGKILL ||
1185             s->deserialized_state == SOCKET_STOP_POST ||
1186             s->deserialized_state == SOCKET_FINAL_SIGTERM ||
1187             s->deserialized_state == SOCKET_FINAL_SIGKILL) {
1188
1189                 if (s->control_pid <= 0)
1190                         return -EBADMSG;
1191
1192                 r = unit_watch_pid(UNIT(s), s->control_pid);
1193                 if (r < 0)
1194                         return r;
1195
1196                 r = socket_arm_timer(s);
1197                 if (r < 0)
1198                         return r;
1199         }
1200
1201         if (s->deserialized_state == SOCKET_START_POST ||
1202             s->deserialized_state == SOCKET_LISTENING ||
1203             s->deserialized_state == SOCKET_RUNNING ||
1204             s->deserialized_state == SOCKET_STOP_PRE ||
1205             s->deserialized_state == SOCKET_STOP_PRE_SIGTERM ||
1206             s->deserialized_state == SOCKET_STOP_PRE_SIGKILL) {
1207                 r = socket_open_fds(s);
1208                 if (r < 0)
1209                         return r;
1210         }
1211
1212         if (s->deserialized_state == SOCKET_LISTENING) {
1213                 r = socket_watch_fds(s);
1214                 if (r < 0)
1215                         return r;
1216         }
1217
1218         socket_set_state(s, s->deserialized_state);
1219         return 0;
1220 }
1221
1222 static int socket_spawn(Socket *s, ExecCommand *c, pid_t *_pid) {
1223         pid_t pid;
1224         int r;
1225         char **argv;
1226
1227         assert(s);
1228         assert(c);
1229         assert(_pid);
1230
1231         unit_realize_cgroup(UNIT(s));
1232
1233         r = unit_setup_exec_runtime(UNIT(s));
1234         if (r < 0)
1235                 goto fail;
1236
1237         r = socket_arm_timer(s);
1238         if (r < 0)
1239                 goto fail;
1240
1241         r = unit_full_printf_strv(UNIT(s), c->argv, &argv);
1242         if (r < 0)
1243                 goto fail;
1244
1245         r = exec_spawn(c,
1246                        argv,
1247                        &s->exec_context,
1248                        NULL, 0,
1249                        UNIT(s)->manager->environment,
1250                        true,
1251                        true,
1252                        true,
1253                        UNIT(s)->manager->confirm_spawn,
1254                        UNIT(s)->manager->cgroup_supported,
1255                        UNIT(s)->cgroup_path,
1256                        UNIT(s)->id,
1257                        0,
1258                        NULL,
1259                        s->exec_runtime,
1260                        &pid);
1261
1262         strv_free(argv);
1263         if (r < 0)
1264                 goto fail;
1265
1266         r = unit_watch_pid(UNIT(s), pid);
1267         if (r < 0)
1268                 /* FIXME: we need to do something here */
1269                 goto fail;
1270
1271         *_pid = pid;
1272
1273         return 0;
1274
1275 fail:
1276         s->timer_event_source = sd_event_source_unref(s->timer_event_source);
1277
1278         return r;
1279 }
1280
1281 static void socket_enter_dead(Socket *s, SocketResult f) {
1282         assert(s);
1283
1284         if (f != SOCKET_SUCCESS)
1285                 s->result = f;
1286
1287         exec_runtime_destroy(s->exec_runtime);
1288         s->exec_runtime = exec_runtime_unref(s->exec_runtime);
1289
1290         socket_set_state(s, s->result != SOCKET_SUCCESS ? SOCKET_FAILED : SOCKET_DEAD);
1291 }
1292
1293 static void socket_enter_signal(Socket *s, SocketState state, SocketResult f);
1294
1295 static void socket_enter_stop_post(Socket *s, SocketResult f) {
1296         int r;
1297         assert(s);
1298
1299         if (f != SOCKET_SUCCESS)
1300                 s->result = f;
1301
1302         socket_unwatch_control_pid(s);
1303
1304         s->control_command_id = SOCKET_EXEC_STOP_POST;
1305
1306         if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_POST])) {
1307                 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
1308                         goto fail;
1309
1310                 socket_set_state(s, SOCKET_STOP_POST);
1311         } else
1312                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, SOCKET_SUCCESS);
1313
1314         return;
1315
1316 fail:
1317         log_warning_unit(UNIT(s)->id,
1318                          "%s failed to run 'stop-post' task: %s",
1319                          UNIT(s)->id, strerror(-r));
1320         socket_enter_signal(s, SOCKET_FINAL_SIGTERM, SOCKET_FAILURE_RESOURCES);
1321 }
1322
1323 static void socket_enter_signal(Socket *s, SocketState state, SocketResult f) {
1324         int r;
1325
1326         assert(s);
1327
1328         if (f != SOCKET_SUCCESS)
1329                 s->result = f;
1330
1331         r = unit_kill_context(
1332                         UNIT(s),
1333                         &s->kill_context,
1334                         state != SOCKET_STOP_PRE_SIGTERM && state != SOCKET_FINAL_SIGTERM,
1335                         -1,
1336                         s->control_pid,
1337                         false);
1338         if (r < 0)
1339                 goto fail;
1340
1341         if (r > 0) {
1342                 r = socket_arm_timer(s);
1343                 if (r < 0)
1344                         goto fail;
1345
1346                 socket_set_state(s, state);
1347         } else if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL)
1348                 socket_enter_stop_post(s, SOCKET_SUCCESS);
1349         else
1350                 socket_enter_dead(s, SOCKET_SUCCESS);
1351
1352         return;
1353
1354 fail:
1355         log_warning_unit(UNIT(s)->id, "%s failed to kill processes: %s", UNIT(s)->id, strerror(-r));
1356
1357         if (state == SOCKET_STOP_PRE_SIGTERM || state == SOCKET_STOP_PRE_SIGKILL)
1358                 socket_enter_stop_post(s, SOCKET_FAILURE_RESOURCES);
1359         else
1360                 socket_enter_dead(s, SOCKET_FAILURE_RESOURCES);
1361 }
1362
1363 static void socket_enter_stop_pre(Socket *s, SocketResult f) {
1364         int r;
1365         assert(s);
1366
1367         if (f != SOCKET_SUCCESS)
1368                 s->result = f;
1369
1370         socket_unwatch_control_pid(s);
1371
1372         s->control_command_id = SOCKET_EXEC_STOP_PRE;
1373
1374         if ((s->control_command = s->exec_command[SOCKET_EXEC_STOP_PRE])) {
1375                 if ((r = socket_spawn(s, s->control_command, &s->control_pid)) < 0)
1376                         goto fail;
1377
1378                 socket_set_state(s, SOCKET_STOP_PRE);
1379         } else
1380                 socket_enter_stop_post(s, SOCKET_SUCCESS);
1381
1382         return;
1383
1384 fail:
1385         log_warning_unit(UNIT(s)->id, "%s failed to run 'stop-pre' task: %s", UNIT(s)->id, strerror(-r));
1386         socket_enter_stop_post(s, SOCKET_FAILURE_RESOURCES);
1387 }
1388
1389 static void socket_enter_listening(Socket *s) {
1390         int r;
1391         assert(s);
1392
1393         r = socket_watch_fds(s);
1394         if (r < 0) {
1395                 log_warning_unit(UNIT(s)->id, "%s failed to watch sockets: %s", UNIT(s)->id, strerror(-r));
1396                 goto fail;
1397         }
1398
1399         socket_set_state(s, SOCKET_LISTENING);
1400         return;
1401
1402 fail:
1403         socket_enter_stop_pre(s, SOCKET_FAILURE_RESOURCES);
1404 }
1405
1406 static void socket_enter_start_post(Socket *s) {
1407         int r;
1408         assert(s);
1409
1410         r = socket_open_fds(s);
1411         if (r < 0) {
1412                 log_warning_unit(UNIT(s)->id, "%s failed to listen on sockets: %s", UNIT(s)->id, strerror(-r));
1413                 goto fail;
1414         }
1415
1416         socket_unwatch_control_pid(s);
1417
1418         s->control_command_id = SOCKET_EXEC_START_POST;
1419
1420         if ((s->control_command = s->exec_command[SOCKET_EXEC_START_POST])) {
1421                 r = socket_spawn(s, s->control_command, &s->control_pid);
1422                 if (r < 0) {
1423                         log_warning_unit(UNIT(s)->id, "%s failed to run 'start-post' task: %s", UNIT(s)->id, strerror(-r));
1424                         goto fail;
1425                 }
1426
1427                 socket_set_state(s, SOCKET_START_POST);
1428         } else
1429                 socket_enter_listening(s);
1430
1431         return;
1432
1433 fail:
1434         socket_enter_stop_pre(s, SOCKET_FAILURE_RESOURCES);
1435 }
1436
1437 static void socket_enter_start_pre(Socket *s) {
1438         int r;
1439         assert(s);
1440
1441         socket_unwatch_control_pid(s);
1442
1443         s->control_command_id = SOCKET_EXEC_START_PRE;
1444
1445         if ((s->control_command = s->exec_command[SOCKET_EXEC_START_PRE])) {
1446                 r = socket_spawn(s, s->control_command, &s->control_pid);
1447                 if (r < 0)
1448                         goto fail;
1449
1450                 socket_set_state(s, SOCKET_START_PRE);
1451         } else
1452                 socket_enter_start_post(s);
1453
1454         return;
1455
1456 fail:
1457         log_warning_unit(UNIT(s)->id, "%s failed to run 'start-pre' task: %s", UNIT(s)->id, strerror(-r));
1458         socket_enter_dead(s, SOCKET_FAILURE_RESOURCES);
1459 }
1460
1461 static void socket_enter_running(Socket *s, int cfd) {
1462         _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
1463         int r;
1464
1465         assert(s);
1466
1467         /* We don't take connections anymore if we are supposed to
1468          * shut down anyway */
1469         if (unit_stop_pending(UNIT(s))) {
1470
1471                 log_debug_unit(UNIT(s)->id, "Suppressing connection request on %s since unit stop is scheduled.", UNIT(s)->id);
1472
1473                 if (cfd >= 0)
1474                         close_nointr_nofail(cfd);
1475                 else  {
1476                         /* Flush all sockets by closing and reopening them */
1477                         socket_close_fds(s);
1478
1479                         r = socket_open_fds(s);
1480                         if (r < 0) {
1481                                 log_warning_unit(UNIT(s)->id, "%s failed to listen on sockets: %s", UNIT(s)->id, strerror(-r));
1482                                 socket_enter_stop_pre(s, SOCKET_FAILURE_RESOURCES);
1483                                 return;
1484                         }
1485
1486                         r = socket_watch_fds(s);
1487                         if (r < 0) {
1488                                 log_warning_unit(UNIT(s)->id, "%s failed to watch sockets: %s", UNIT(s)->id, strerror(-r));
1489                                 socket_enter_stop_pre(s, SOCKET_FAILURE_RESOURCES);
1490                         }
1491                 }
1492
1493                 return;
1494         }
1495
1496         if (cfd < 0) {
1497                 Iterator i;
1498                 Unit *other;
1499                 bool pending = false;
1500
1501                 /* If there's already a start pending don't bother to
1502                  * do anything */
1503                 SET_FOREACH(other, UNIT(s)->dependencies[UNIT_TRIGGERS], i)
1504                         if (unit_active_or_pending(other)) {
1505                                 pending = true;
1506                                 break;
1507                         }
1508
1509                 if (!pending) {
1510                         r = manager_add_job(UNIT(s)->manager, JOB_START, UNIT_DEREF(s->service), JOB_REPLACE, true, &error, NULL);
1511                         if (r < 0)
1512                                 goto fail;
1513                 }
1514
1515                 socket_set_state(s, SOCKET_RUNNING);
1516         } else {
1517                 _cleanup_free_ char *prefix = NULL, *instance = NULL, *name = NULL;
1518                 Service *service;
1519
1520                 if (s->n_connections >= s->max_connections) {
1521                         log_warning_unit(UNIT(s)->id, "%s: Too many incoming connections (%u)", UNIT(s)->id, s->n_connections);
1522                         close_nointr_nofail(cfd);
1523                         return;
1524                 }
1525
1526                 r = socket_instantiate_service(s);
1527                 if (r < 0)
1528                         goto fail;
1529
1530                 r = instance_from_socket(cfd, s->n_accepted, &instance);
1531                 if (r < 0) {
1532                         if (r != -ENOTCONN)
1533                                 goto fail;
1534
1535                         /* ENOTCONN is legitimate if TCP RST was received.
1536                          * This connection is over, but the socket unit lives on. */
1537                         close_nointr_nofail(cfd);
1538                         return;
1539                 }
1540
1541                 prefix = unit_name_to_prefix(UNIT(s)->id);
1542                 if (!prefix) {
1543                         r = -ENOMEM;
1544                         goto fail;
1545                 }
1546
1547                 name = unit_name_build(prefix, instance, ".service");
1548                 if (!name) {
1549                         r = -ENOMEM;
1550                         goto fail;
1551                 }
1552
1553                 r = unit_add_name(UNIT_DEREF(s->service), name);
1554                 if (r < 0)
1555                         goto fail;
1556
1557                 service = SERVICE(UNIT_DEREF(s->service));
1558                 unit_ref_unset(&s->service);
1559                 s->n_accepted ++;
1560
1561                 UNIT(service)->no_gc = false;
1562
1563                 unit_choose_id(UNIT(service), name);
1564
1565                 r = service_set_socket_fd(service, cfd, s);
1566                 if (r < 0)
1567                         goto fail;
1568
1569                 cfd = -1;
1570                 s->n_connections ++;
1571
1572                 r = manager_add_job(UNIT(s)->manager, JOB_START, UNIT(service), JOB_REPLACE, true, &error, NULL);
1573                 if (r < 0)
1574                         goto fail;
1575
1576                 /* Notify clients about changed counters */
1577                 unit_add_to_dbus_queue(UNIT(s));
1578         }
1579
1580         return;
1581
1582 fail:
1583         log_warning_unit(UNIT(s)->id, "%s failed to queue service startup job (Maybe the service file is missing or not a %s unit?): %s",
1584                          UNIT(s)->id, cfd >= 0 ? "template" : "non-template",
1585                          bus_error_message(&error, r));
1586
1587         socket_enter_stop_pre(s, SOCKET_FAILURE_RESOURCES);
1588
1589         if (cfd >= 0)
1590                 close_nointr_nofail(cfd);
1591 }
1592
1593 static void socket_run_next(Socket *s) {
1594         int r;
1595
1596         assert(s);
1597         assert(s->control_command);
1598         assert(s->control_command->command_next);
1599
1600         socket_unwatch_control_pid(s);
1601
1602         s->control_command = s->control_command->command_next;
1603
1604         r = socket_spawn(s, s->control_command, &s->control_pid);
1605         if (r < 0)
1606                 goto fail;
1607
1608         return;
1609
1610 fail:
1611         log_warning_unit(UNIT(s)->id, "%s failed to run next task: %s", UNIT(s)->id, strerror(-r));
1612
1613         if (s->state == SOCKET_START_POST)
1614                 socket_enter_stop_pre(s, SOCKET_FAILURE_RESOURCES);
1615         else if (s->state == SOCKET_STOP_POST)
1616                 socket_enter_dead(s, SOCKET_FAILURE_RESOURCES);
1617         else
1618                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, SOCKET_FAILURE_RESOURCES);
1619 }
1620
1621 static int socket_start(Unit *u) {
1622         Socket *s = SOCKET(u);
1623
1624         assert(s);
1625
1626         /* We cannot fulfill this request right now, try again later
1627          * please! */
1628         if (s->state == SOCKET_STOP_PRE ||
1629             s->state == SOCKET_STOP_PRE_SIGKILL ||
1630             s->state == SOCKET_STOP_PRE_SIGTERM ||
1631             s->state == SOCKET_STOP_POST ||
1632             s->state == SOCKET_FINAL_SIGTERM ||
1633             s->state == SOCKET_FINAL_SIGKILL)
1634                 return -EAGAIN;
1635
1636         if (s->state == SOCKET_START_PRE ||
1637             s->state == SOCKET_START_POST)
1638                 return 0;
1639
1640         /* Cannot run this without the service being around */
1641         if (UNIT_ISSET(s->service)) {
1642                 Service *service;
1643
1644                 service = SERVICE(UNIT_DEREF(s->service));
1645
1646                 if (UNIT(service)->load_state != UNIT_LOADED) {
1647                         log_error_unit(u->id, "Socket service %s not loaded, refusing.", UNIT(service)->id);
1648                         return -ENOENT;
1649                 }
1650
1651                 /* If the service is already active we cannot start the
1652                  * socket */
1653                 if (service->state != SERVICE_DEAD &&
1654                     service->state != SERVICE_FAILED &&
1655                     service->state != SERVICE_AUTO_RESTART) {
1656                         log_error_unit(u->id, "Socket service %s already active, refusing.", UNIT(service)->id);
1657                         return -EBUSY;
1658                 }
1659
1660 #ifdef HAVE_SYSV_COMPAT
1661                 if (service->is_sysv) {
1662                         log_error_unit(u->id,
1663                                        "Using SysV services for socket activation is not supported. Refusing.");
1664                         return -ENOENT;
1665                 }
1666 #endif
1667         }
1668
1669         assert(s->state == SOCKET_DEAD || s->state == SOCKET_FAILED);
1670
1671         s->result = SOCKET_SUCCESS;
1672         socket_enter_start_pre(s);
1673
1674         return 0;
1675 }
1676
1677 static int socket_stop(Unit *u) {
1678         Socket *s = SOCKET(u);
1679
1680         assert(s);
1681
1682         /* Already on it */
1683         if (s->state == SOCKET_STOP_PRE ||
1684             s->state == SOCKET_STOP_PRE_SIGTERM ||
1685             s->state == SOCKET_STOP_PRE_SIGKILL ||
1686             s->state == SOCKET_STOP_POST ||
1687             s->state == SOCKET_FINAL_SIGTERM ||
1688             s->state == SOCKET_FINAL_SIGKILL)
1689                 return 0;
1690
1691         /* If there's already something running we go directly into
1692          * kill mode. */
1693         if (s->state == SOCKET_START_PRE ||
1694             s->state == SOCKET_START_POST) {
1695                 socket_enter_signal(s, SOCKET_STOP_PRE_SIGTERM, SOCKET_SUCCESS);
1696                 return -EAGAIN;
1697         }
1698
1699         assert(s->state == SOCKET_LISTENING || s->state == SOCKET_RUNNING);
1700
1701         socket_enter_stop_pre(s, SOCKET_SUCCESS);
1702         return 0;
1703 }
1704
1705 static int socket_serialize(Unit *u, FILE *f, FDSet *fds) {
1706         Socket *s = SOCKET(u);
1707         SocketPort *p;
1708         int r;
1709
1710         assert(u);
1711         assert(f);
1712         assert(fds);
1713
1714         unit_serialize_item(u, f, "state", socket_state_to_string(s->state));
1715         unit_serialize_item(u, f, "result", socket_result_to_string(s->result));
1716         unit_serialize_item_format(u, f, "n-accepted", "%u", s->n_accepted);
1717
1718         if (s->control_pid > 0)
1719                 unit_serialize_item_format(u, f, "control-pid", "%lu", (unsigned long) s->control_pid);
1720
1721         if (s->control_command_id >= 0)
1722                 unit_serialize_item(u, f, "control-command", socket_exec_command_to_string(s->control_command_id));
1723
1724         LIST_FOREACH(port, p, s->ports) {
1725                 int copy;
1726
1727                 if (p->fd < 0)
1728                         continue;
1729
1730                 copy = fdset_put_dup(fds, p->fd);
1731                 if (copy < 0)
1732                         return copy;
1733
1734                 if (p->type == SOCKET_SOCKET) {
1735                         _cleanup_free_ char *t = NULL;
1736
1737                         r = socket_address_print(&p->address, &t);
1738                         if (r < 0)
1739                                 return r;
1740
1741                         if (socket_address_family(&p->address) == AF_NETLINK)
1742                                 unit_serialize_item_format(u, f, "netlink", "%i %s", copy, t);
1743                         else
1744                                 unit_serialize_item_format(u, f, "socket", "%i %i %s", copy, p->address.type, t);
1745
1746                 } else if (p->type == SOCKET_SPECIAL)
1747                         unit_serialize_item_format(u, f, "special", "%i %s", copy, p->path);
1748                 else if (p->type == SOCKET_MQUEUE)
1749                         unit_serialize_item_format(u, f, "mqueue", "%i %s", copy, p->path);
1750                 else {
1751                         assert(p->type == SOCKET_FIFO);
1752                         unit_serialize_item_format(u, f, "fifo", "%i %s", copy, p->path);
1753                 }
1754         }
1755
1756         return 0;
1757 }
1758
1759 static int socket_deserialize_item(Unit *u, const char *key, const char *value, FDSet *fds) {
1760         Socket *s = SOCKET(u);
1761
1762         assert(u);
1763         assert(key);
1764         assert(value);
1765
1766         if (streq(key, "state")) {
1767                 SocketState state;
1768
1769                 state = socket_state_from_string(value);
1770                 if (state < 0)
1771                         log_debug_unit(u->id, "Failed to parse state value %s", value);
1772                 else
1773                         s->deserialized_state = state;
1774         } else if (streq(key, "result")) {
1775                 SocketResult f;
1776
1777                 f = socket_result_from_string(value);
1778                 if (f < 0)
1779                         log_debug_unit(u->id, "Failed to parse result value %s", value);
1780                 else if (f != SOCKET_SUCCESS)
1781                         s->result = f;
1782
1783         } else if (streq(key, "n-accepted")) {
1784                 unsigned k;
1785
1786                 if (safe_atou(value, &k) < 0)
1787                         log_debug_unit(u->id, "Failed to parse n-accepted value %s", value);
1788                 else
1789                         s->n_accepted += k;
1790         } else if (streq(key, "control-pid")) {
1791                 pid_t pid;
1792
1793                 if (parse_pid(value, &pid) < 0)
1794                         log_debug_unit(u->id, "Failed to parse control-pid value %s", value);
1795                 else
1796                         s->control_pid = pid;
1797         } else if (streq(key, "control-command")) {
1798                 SocketExecCommand id;
1799
1800                 id = socket_exec_command_from_string(value);
1801                 if (id < 0)
1802                         log_debug_unit(u->id, "Failed to parse exec-command value %s", value);
1803                 else {
1804                         s->control_command_id = id;
1805                         s->control_command = s->exec_command[id];
1806                 }
1807         } else if (streq(key, "fifo")) {
1808                 int fd, skip = 0;
1809                 SocketPort *p;
1810
1811                 if (sscanf(value, "%i %n", &fd, &skip) < 1 || fd < 0 || !fdset_contains(fds, fd))
1812                         log_debug_unit(u->id, "Failed to parse fifo value %s", value);
1813                 else {
1814
1815                         LIST_FOREACH(port, p, s->ports)
1816                                 if (p->type == SOCKET_FIFO &&
1817                                     streq_ptr(p->path, value+skip))
1818                                         break;
1819
1820                         if (p) {
1821                                 if (p->fd >= 0)
1822                                         close_nointr_nofail(p->fd);
1823                                 p->fd = fdset_remove(fds, fd);
1824                         }
1825                 }
1826
1827         } else if (streq(key, "special")) {
1828                 int fd, skip = 0;
1829                 SocketPort *p;
1830
1831                 if (sscanf(value, "%i %n", &fd, &skip) < 1 || fd < 0 || !fdset_contains(fds, fd))
1832                         log_debug_unit(u->id, "Failed to parse special value %s", value);
1833                 else {
1834
1835                         LIST_FOREACH(port, p, s->ports)
1836                                 if (p->type == SOCKET_SPECIAL &&
1837                                     streq_ptr(p->path, value+skip))
1838                                         break;
1839
1840                         if (p) {
1841                                 if (p->fd >= 0)
1842                                         close_nointr_nofail(p->fd);
1843                                 p->fd = fdset_remove(fds, fd);
1844                         }
1845                 }
1846
1847         } else if (streq(key, "mqueue")) {
1848                 int fd, skip = 0;
1849                 SocketPort *p;
1850
1851                 if (sscanf(value, "%i %n", &fd, &skip) < 1 || fd < 0 || !fdset_contains(fds, fd))
1852                         log_debug_unit(u->id, "Failed to parse mqueue value %s", value);
1853                 else {
1854
1855                         LIST_FOREACH(port, p, s->ports)
1856                                 if (p->type == SOCKET_MQUEUE &&
1857                                     streq_ptr(p->path, value+skip))
1858                                         break;
1859
1860                         if (p) {
1861                                 if (p->fd >= 0)
1862                                         close_nointr_nofail(p->fd);
1863                                 p->fd = fdset_remove(fds, fd);
1864                         }
1865                 }
1866
1867         } else if (streq(key, "socket")) {
1868                 int fd, type, skip = 0;
1869                 SocketPort *p;
1870
1871                 if (sscanf(value, "%i %i %n", &fd, &type, &skip) < 2 || fd < 0 || type < 0 || !fdset_contains(fds, fd))
1872                         log_debug_unit(u->id, "Failed to parse socket value %s", value);
1873                 else {
1874
1875                         LIST_FOREACH(port, p, s->ports)
1876                                 if (socket_address_is(&p->address, value+skip, type))
1877                                         break;
1878
1879                         if (p) {
1880                                 if (p->fd >= 0)
1881                                         close_nointr_nofail(p->fd);
1882                                 p->fd = fdset_remove(fds, fd);
1883                         }
1884                 }
1885
1886         } else if (streq(key, "netlink")) {
1887                 int fd, skip = 0;
1888                 SocketPort *p;
1889
1890                 if (sscanf(value, "%i %n", &fd, &skip) < 1 || fd < 0 || !fdset_contains(fds, fd))
1891                         log_debug_unit(u->id, "Failed to parse socket value %s", value);
1892                 else {
1893
1894                         LIST_FOREACH(port, p, s->ports)
1895                                 if (socket_address_is_netlink(&p->address, value+skip))
1896                                         break;
1897
1898                         if (p) {
1899                                 if (p->fd >= 0)
1900                                         close_nointr_nofail(p->fd);
1901                                 p->fd = fdset_remove(fds, fd);
1902                         }
1903                 }
1904         } else
1905                 log_debug_unit(UNIT(s)->id, "Unknown serialization key '%s'", key);
1906
1907         return 0;
1908 }
1909
1910 static int socket_distribute_fds(Unit *u, FDSet *fds) {
1911         Socket *s = SOCKET(u);
1912         SocketPort *p;
1913
1914         assert(u);
1915
1916         LIST_FOREACH(port, p, s->ports) {
1917                 Iterator i;
1918                 int fd;
1919
1920                 if (p->type != SOCKET_SOCKET)
1921                         continue;
1922
1923                 if (p->fd >= 0)
1924                         continue;
1925
1926                 FDSET_FOREACH(fd, fds, i) {
1927                         if (socket_address_matches_fd(&p->address, fd)) {
1928                                 p->fd = fdset_remove(fds, fd);
1929                                 s->deserialized_state = SOCKET_LISTENING;
1930                                 break;
1931                         }
1932                 }
1933         }
1934
1935         return 0;
1936 }
1937
1938 _pure_ static UnitActiveState socket_active_state(Unit *u) {
1939         assert(u);
1940
1941         return state_translation_table[SOCKET(u)->state];
1942 }
1943
1944 _pure_ static const char *socket_sub_state_to_string(Unit *u) {
1945         assert(u);
1946
1947         return socket_state_to_string(SOCKET(u)->state);
1948 }
1949
1950 const char* socket_port_type_to_string(SocketPort *p) {
1951
1952         assert(p);
1953
1954         switch (p->type) {
1955
1956         case SOCKET_SOCKET:
1957
1958                 switch (p->address.type) {
1959
1960                 case SOCK_STREAM:
1961                         return "Stream";
1962
1963                 case SOCK_DGRAM:
1964                         return "Datagram";
1965
1966                 case SOCK_SEQPACKET:
1967                         return "SequentialPacket";
1968
1969                 case SOCK_RAW:
1970                         if (socket_address_family(&p->address) == AF_NETLINK)
1971                                 return "Netlink";
1972
1973                 default:
1974                         return NULL;
1975                 }
1976
1977         case SOCKET_SPECIAL:
1978                 return "Special";
1979
1980         case SOCKET_MQUEUE:
1981                 return "MessageQueue";
1982
1983         case SOCKET_FIFO:
1984                 return "FIFO";
1985
1986         default:
1987                 return NULL;
1988         }
1989 }
1990
1991 _pure_ static bool socket_check_gc(Unit *u) {
1992         Socket *s = SOCKET(u);
1993
1994         assert(u);
1995
1996         return s->n_connections > 0;
1997 }
1998
1999 static int socket_dispatch_io(sd_event_source *source, int fd, uint32_t revents, void *userdata) {
2000         SocketPort *p = userdata;
2001         int cfd = -1;
2002
2003         assert(p);
2004         assert(fd >= 0);
2005
2006         if (p->socket->state != SOCKET_LISTENING)
2007                 return 0;
2008
2009         log_debug_unit(UNIT(p->socket)->id, "Incoming traffic on %s", UNIT(p->socket)->id);
2010
2011         if (revents != EPOLLIN) {
2012
2013                 if (revents & EPOLLHUP)
2014                         log_error_unit(UNIT(p->socket)->id, "%s: Got POLLHUP on a listening socket. The service probably invoked shutdown() on it, and should better not do that.",
2015                                        UNIT(p->socket)->id);
2016                 else
2017                         log_error_unit(UNIT(p->socket)->id, "%s: Got unexpected poll event (0x%x) on socket.",
2018                                        UNIT(p->socket)->id, revents);
2019
2020                 goto fail;
2021         }
2022
2023         if (p->socket->accept &&
2024             p->type == SOCKET_SOCKET &&
2025             socket_address_can_accept(&p->address)) {
2026
2027                 for (;;) {
2028
2029                         cfd = accept4(fd, NULL, NULL, SOCK_NONBLOCK);
2030                         if (cfd < 0) {
2031
2032                                 if (errno == EINTR)
2033                                         continue;
2034
2035                                 log_error_unit(UNIT(p->socket)->id,
2036                                                "Failed to accept socket: %m");
2037                                 goto fail;
2038                         }
2039
2040                         break;
2041                 }
2042
2043                 socket_apply_socket_options(p->socket, cfd);
2044         }
2045
2046         socket_enter_running(p->socket, cfd);
2047         return 0;
2048
2049 fail:
2050         socket_enter_stop_pre(p->socket, SOCKET_FAILURE_RESOURCES);
2051         return 0;
2052 }
2053
2054 static void socket_sigchld_event(Unit *u, pid_t pid, int code, int status) {
2055         Socket *s = SOCKET(u);
2056         SocketResult f;
2057
2058         assert(s);
2059         assert(pid >= 0);
2060
2061         if (pid != s->control_pid)
2062                 return;
2063
2064         s->control_pid = 0;
2065
2066         if (is_clean_exit(code, status, NULL))
2067                 f = SOCKET_SUCCESS;
2068         else if (code == CLD_EXITED)
2069                 f = SOCKET_FAILURE_EXIT_CODE;
2070         else if (code == CLD_KILLED)
2071                 f = SOCKET_FAILURE_SIGNAL;
2072         else if (code == CLD_DUMPED)
2073                 f = SOCKET_FAILURE_CORE_DUMP;
2074         else
2075                 assert_not_reached("Unknown code");
2076
2077         if (s->control_command) {
2078                 exec_status_exit(&s->control_command->exec_status, &s->exec_context, pid, code, status);
2079
2080                 if (s->control_command->ignore)
2081                         f = SOCKET_SUCCESS;
2082         }
2083
2084         log_full_unit(f == SOCKET_SUCCESS ? LOG_DEBUG : LOG_NOTICE,
2085                       u->id,
2086                       "%s control process exited, code=%s status=%i",
2087                       u->id, sigchld_code_to_string(code), status);
2088
2089         if (f != SOCKET_SUCCESS)
2090                 s->result = f;
2091
2092         if (s->control_command &&
2093             s->control_command->command_next &&
2094             f == SOCKET_SUCCESS) {
2095
2096                 log_debug_unit(u->id,
2097                                "%s running next command for state %s",
2098                                u->id, socket_state_to_string(s->state));
2099                 socket_run_next(s);
2100         } else {
2101                 s->control_command = NULL;
2102                 s->control_command_id = _SOCKET_EXEC_COMMAND_INVALID;
2103
2104                 /* No further commands for this step, so let's figure
2105                  * out what to do next */
2106
2107                 log_debug_unit(u->id,
2108                                "%s got final SIGCHLD for state %s",
2109                                u->id, socket_state_to_string(s->state));
2110
2111                 switch (s->state) {
2112
2113                 case SOCKET_START_PRE:
2114                         if (f == SOCKET_SUCCESS)
2115                                 socket_enter_start_post(s);
2116                         else
2117                                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, f);
2118                         break;
2119
2120                 case SOCKET_START_POST:
2121                         if (f == SOCKET_SUCCESS)
2122                                 socket_enter_listening(s);
2123                         else
2124                                 socket_enter_stop_pre(s, f);
2125                         break;
2126
2127                 case SOCKET_STOP_PRE:
2128                 case SOCKET_STOP_PRE_SIGTERM:
2129                 case SOCKET_STOP_PRE_SIGKILL:
2130                         socket_enter_stop_post(s, f);
2131                         break;
2132
2133                 case SOCKET_STOP_POST:
2134                 case SOCKET_FINAL_SIGTERM:
2135                 case SOCKET_FINAL_SIGKILL:
2136                         socket_enter_dead(s, f);
2137                         break;
2138
2139                 default:
2140                         assert_not_reached("Uh, control process died at wrong time.");
2141                 }
2142         }
2143
2144         /* Notify clients about changed exit status */
2145         unit_add_to_dbus_queue(u);
2146 }
2147
2148 static int socket_dispatch_timer(sd_event_source *source, usec_t usec, void *userdata) {
2149         Socket *s = SOCKET(userdata);
2150
2151         assert(s);
2152         assert(s->timer_event_source == source);
2153
2154         switch (s->state) {
2155
2156         case SOCKET_START_PRE:
2157                 log_warning_unit(UNIT(s)->id,
2158                                  "%s starting timed out. Terminating.", UNIT(s)->id);
2159                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, SOCKET_FAILURE_TIMEOUT);
2160                 break;
2161
2162         case SOCKET_START_POST:
2163                 log_warning_unit(UNIT(s)->id,
2164                                  "%s starting timed out. Stopping.", UNIT(s)->id);
2165                 socket_enter_stop_pre(s, SOCKET_FAILURE_TIMEOUT);
2166                 break;
2167
2168         case SOCKET_STOP_PRE:
2169                 log_warning_unit(UNIT(s)->id,
2170                                  "%s stopping timed out. Terminating.", UNIT(s)->id);
2171                 socket_enter_signal(s, SOCKET_STOP_PRE_SIGTERM, SOCKET_FAILURE_TIMEOUT);
2172                 break;
2173
2174         case SOCKET_STOP_PRE_SIGTERM:
2175                 if (s->kill_context.send_sigkill) {
2176                         log_warning_unit(UNIT(s)->id,
2177                                          "%s stopping timed out. Killing.", UNIT(s)->id);
2178                         socket_enter_signal(s, SOCKET_STOP_PRE_SIGKILL, SOCKET_FAILURE_TIMEOUT);
2179                 } else {
2180                         log_warning_unit(UNIT(s)->id,
2181                                          "%s stopping timed out. Skipping SIGKILL. Ignoring.",
2182                                          UNIT(s)->id);
2183                         socket_enter_stop_post(s, SOCKET_FAILURE_TIMEOUT);
2184                 }
2185                 break;
2186
2187         case SOCKET_STOP_PRE_SIGKILL:
2188                 log_warning_unit(UNIT(s)->id,
2189                                  "%s still around after SIGKILL. Ignoring.", UNIT(s)->id);
2190                 socket_enter_stop_post(s, SOCKET_FAILURE_TIMEOUT);
2191                 break;
2192
2193         case SOCKET_STOP_POST:
2194                 log_warning_unit(UNIT(s)->id,
2195                                  "%s stopping timed out (2). Terminating.", UNIT(s)->id);
2196                 socket_enter_signal(s, SOCKET_FINAL_SIGTERM, SOCKET_FAILURE_TIMEOUT);
2197                 break;
2198
2199         case SOCKET_FINAL_SIGTERM:
2200                 if (s->kill_context.send_sigkill) {
2201                         log_warning_unit(UNIT(s)->id,
2202                                          "%s stopping timed out (2). Killing.", UNIT(s)->id);
2203                         socket_enter_signal(s, SOCKET_FINAL_SIGKILL, SOCKET_FAILURE_TIMEOUT);
2204                 } else {
2205                         log_warning_unit(UNIT(s)->id,
2206                                          "%s stopping timed out (2). Skipping SIGKILL. Ignoring.",
2207                                          UNIT(s)->id);
2208                         socket_enter_dead(s, SOCKET_FAILURE_TIMEOUT);
2209                 }
2210                 break;
2211
2212         case SOCKET_FINAL_SIGKILL:
2213                 log_warning_unit(UNIT(s)->id,
2214                                  "%s still around after SIGKILL (2). Entering failed mode.",
2215                                  UNIT(s)->id);
2216                 socket_enter_dead(s, SOCKET_FAILURE_TIMEOUT);
2217                 break;
2218
2219         default:
2220                 assert_not_reached("Timeout at wrong time.");
2221         }
2222
2223         return 0;
2224 }
2225
2226 int socket_collect_fds(Socket *s, int **fds, unsigned *n_fds) {
2227         int *rfds;
2228         unsigned rn_fds, k;
2229         SocketPort *p;
2230
2231         assert(s);
2232         assert(fds);
2233         assert(n_fds);
2234
2235         /* Called from the service code for requesting our fds */
2236
2237         rn_fds = 0;
2238         LIST_FOREACH(port, p, s->ports)
2239                 if (p->fd >= 0)
2240                         rn_fds++;
2241
2242         if (rn_fds <= 0) {
2243                 *fds = NULL;
2244                 *n_fds = 0;
2245                 return 0;
2246         }
2247
2248         if (!(rfds = new(int, rn_fds)))
2249                 return -ENOMEM;
2250
2251         k = 0;
2252         LIST_FOREACH(port, p, s->ports)
2253                 if (p->fd >= 0)
2254                         rfds[k++] = p->fd;
2255
2256         assert(k == rn_fds);
2257
2258         *fds = rfds;
2259         *n_fds = rn_fds;
2260
2261         return 0;
2262 }
2263
2264 static void socket_reset_failed(Unit *u) {
2265         Socket *s = SOCKET(u);
2266
2267         assert(s);
2268
2269         if (s->state == SOCKET_FAILED)
2270                 socket_set_state(s, SOCKET_DEAD);
2271
2272         s->result = SOCKET_SUCCESS;
2273 }
2274
2275 static void socket_notify_service_dead(Socket *s, bool failed_permanent) {
2276         assert(s);
2277
2278         /* The service is dead. Dang!
2279          *
2280          * This is strictly for one-instance-for-all-connections
2281          * services. */
2282
2283         if (s->state == SOCKET_RUNNING) {
2284                 log_debug_unit(UNIT(s)->id, "%s got notified about service death (failed permanently: %s)", UNIT(s)->id, yes_no(failed_permanent));
2285                 if (failed_permanent)
2286                         socket_enter_stop_pre(s, SOCKET_FAILURE_SERVICE_FAILED_PERMANENT);
2287                 else
2288                         socket_enter_listening(s);
2289         }
2290 }
2291
2292 void socket_connection_unref(Socket *s) {
2293         assert(s);
2294
2295         /* The service is dead. Yay!
2296          *
2297          * This is strictly for one-instance-per-connection
2298          * services. */
2299
2300         assert(s->n_connections > 0);
2301         s->n_connections--;
2302
2303         log_debug_unit(UNIT(s)->id, "%s: One connection closed, %u left.", UNIT(s)->id, s->n_connections);
2304 }
2305
2306 static void socket_trigger_notify(Unit *u, Unit *other) {
2307         Socket *s = SOCKET(u);
2308         Service *se;
2309
2310         assert(u);
2311         assert(other);
2312
2313         /* Don't propagate state changes from the service if we are
2314            already down or accepting connections */
2315         if ((s->state !=  SOCKET_RUNNING &&
2316             s->state != SOCKET_LISTENING) ||
2317             s->accept)
2318                 return;
2319
2320         if (other->load_state != UNIT_LOADED ||
2321             other->type != UNIT_SERVICE)
2322                 return;
2323
2324         se = SERVICE(other);
2325
2326         if (se->state == SERVICE_FAILED)
2327                 socket_notify_service_dead(s, se->result == SERVICE_FAILURE_START_LIMIT);
2328
2329         if (se->state == SERVICE_DEAD ||
2330             se->state == SERVICE_STOP ||
2331             se->state == SERVICE_STOP_SIGTERM ||
2332             se->state == SERVICE_STOP_SIGKILL ||
2333             se->state == SERVICE_STOP_POST ||
2334             se->state == SERVICE_FINAL_SIGTERM ||
2335             se->state == SERVICE_FINAL_SIGKILL ||
2336             se->state == SERVICE_AUTO_RESTART)
2337                 socket_notify_service_dead(s, false);
2338
2339         if (se->state == SERVICE_RUNNING)
2340                 socket_set_state(s, SOCKET_RUNNING);
2341 }
2342
2343 static int socket_kill(Unit *u, KillWho who, int signo, sd_bus_error *error) {
2344         return unit_kill_common(u, who, signo, -1, SOCKET(u)->control_pid, error);
2345 }
2346
2347 static int socket_get_timeout(Unit *u, uint64_t *timeout) {
2348         Socket *s = SOCKET(u);
2349         int r;
2350
2351         if (!s->timer_event_source)
2352                 return 0;
2353
2354         r = sd_event_source_get_time(s->timer_event_source, timeout);
2355         if (r < 0)
2356                 return r;
2357
2358         return 1;
2359 }
2360
2361 static const char* const socket_state_table[_SOCKET_STATE_MAX] = {
2362         [SOCKET_DEAD] = "dead",
2363         [SOCKET_START_PRE] = "start-pre",
2364         [SOCKET_START_POST] = "start-post",
2365         [SOCKET_LISTENING] = "listening",
2366         [SOCKET_RUNNING] = "running",
2367         [SOCKET_STOP_PRE] = "stop-pre",
2368         [SOCKET_STOP_PRE_SIGTERM] = "stop-pre-sigterm",
2369         [SOCKET_STOP_PRE_SIGKILL] = "stop-pre-sigkill",
2370         [SOCKET_STOP_POST] = "stop-post",
2371         [SOCKET_FINAL_SIGTERM] = "final-sigterm",
2372         [SOCKET_FINAL_SIGKILL] = "final-sigkill",
2373         [SOCKET_FAILED] = "failed"
2374 };
2375
2376 DEFINE_STRING_TABLE_LOOKUP(socket_state, SocketState);
2377
2378 static const char* const socket_exec_command_table[_SOCKET_EXEC_COMMAND_MAX] = {
2379         [SOCKET_EXEC_START_PRE] = "StartPre",
2380         [SOCKET_EXEC_START_POST] = "StartPost",
2381         [SOCKET_EXEC_STOP_PRE] = "StopPre",
2382         [SOCKET_EXEC_STOP_POST] = "StopPost"
2383 };
2384
2385 DEFINE_STRING_TABLE_LOOKUP(socket_exec_command, SocketExecCommand);
2386
2387 static const char* const socket_result_table[_SOCKET_RESULT_MAX] = {
2388         [SOCKET_SUCCESS] = "success",
2389         [SOCKET_FAILURE_RESOURCES] = "resources",
2390         [SOCKET_FAILURE_TIMEOUT] = "timeout",
2391         [SOCKET_FAILURE_EXIT_CODE] = "exit-code",
2392         [SOCKET_FAILURE_SIGNAL] = "signal",
2393         [SOCKET_FAILURE_CORE_DUMP] = "core-dump",
2394         [SOCKET_FAILURE_SERVICE_FAILED_PERMANENT] = "service-failed-permanent"
2395 };
2396
2397 DEFINE_STRING_TABLE_LOOKUP(socket_result, SocketResult);
2398
2399 const UnitVTable socket_vtable = {
2400         .object_size = sizeof(Socket),
2401         .exec_context_offset = offsetof(Socket, exec_context),
2402         .cgroup_context_offset = offsetof(Socket, cgroup_context),
2403         .kill_context_offset = offsetof(Socket, kill_context),
2404         .exec_runtime_offset = offsetof(Socket, exec_runtime),
2405
2406         .sections =
2407                 "Unit\0"
2408                 "Socket\0"
2409                 "Install\0",
2410         .private_section = "Socket",
2411
2412         .init = socket_init,
2413         .done = socket_done,
2414         .load = socket_load,
2415
2416         .coldplug = socket_coldplug,
2417
2418         .dump = socket_dump,
2419
2420         .start = socket_start,
2421         .stop = socket_stop,
2422
2423         .kill = socket_kill,
2424
2425         .get_timeout = socket_get_timeout,
2426
2427         .serialize = socket_serialize,
2428         .deserialize_item = socket_deserialize_item,
2429         .distribute_fds = socket_distribute_fds,
2430
2431         .active_state = socket_active_state,
2432         .sub_state_to_string = socket_sub_state_to_string,
2433
2434         .check_gc = socket_check_gc,
2435
2436         .sigchld_event = socket_sigchld_event,
2437
2438         .trigger_notify = socket_trigger_notify,
2439
2440         .reset_failed = socket_reset_failed,
2441
2442         .bus_interface = "org.freedesktop.systemd1.Socket",
2443         .bus_vtable = bus_socket_vtable,
2444         .bus_set_property = bus_socket_set_property,
2445         .bus_commit_properties = bus_socket_commit_properties,
2446
2447         .status_message_formats = {
2448                 /*.starting_stopping = {
2449                         [0] = "Starting socket %s...",
2450                         [1] = "Stopping socket %s...",
2451                 },*/
2452                 .finished_start_job = {
2453                         [JOB_DONE]       = "Listening on %s.",
2454                         [JOB_FAILED]     = "Failed to listen on %s.",
2455                         [JOB_DEPENDENCY] = "Dependency failed for %s.",
2456                         [JOB_TIMEOUT]    = "Timed out starting %s.",
2457                 },
2458                 .finished_stop_job = {
2459                         [JOB_DONE]       = "Closed %s.",
2460                         [JOB_FAILED]     = "Failed stopping %s.",
2461                         [JOB_TIMEOUT]    = "Timed out stopping %s.",
2462                 },
2463         },
2464 };