source3/printing/spoolssd.c

   1 /*
   2    Unix SMB/Netbios implementation.
   3    SPOOLSS Daemon
   4    Copyright (C) Simo Sorce <idra@samba.org> 2010-2011
   5
   6    This program is free software; you can redistribute it and/or modify
   7    it under the terms of the GNU General Public License as published by
   8    the Free Software Foundation; either version 3 of the License, or
   9    (at your option) any later version.
  10
  11    This program is distributed in the hope that it will be useful,
  12    but WITHOUT ANY WARRANTY; without even the implied warranty of
  13    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14    GNU General Public License for more details.
  15
  16    You should have received a copy of the GNU General Public License
  17    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  18 */
  19 #include "includes.h"
  20 #include "serverid.h"
  21 #include "smbd/smbd.h"
  22
  23 #include "messages.h"
  24 #include "include/printing.h"
  25 #include "printing/nt_printing_migrate_internal.h"
  26 #include "printing/queue_process.h"
  27 #include "printing/pcap.h"
  28 #include "printing/load.h"
  29 #include "ntdomain.h"
  30 #include "librpc/gen_ndr/srv_winreg.h"
  31 #include "librpc/gen_ndr/srv_spoolss.h"
  32 #include "rpc_server/rpc_server.h"
  33 #include "rpc_server/rpc_ep_register.h"
  34 #include "rpc_server/spoolss/srv_spoolss_nt.h"
  35 #include "librpc/rpc/dcerpc_ep.h"
  36 #include "lib/server_prefork.h"
  37 #include "lib/server_prefork_util.h"
  38
  39 #define SPOOLSS_PIPE_NAME "spoolss"
  40 #define DAEMON_NAME "spoolssd"
  41
  42 static struct server_id parent_id;
  43 static struct prefork_pool *spoolss_pool = NULL;
  44 static int spoolss_child_id = 0;
  45
  46 static struct pf_daemon_config default_pf_spoolss_cfg = {
  47         .prefork_status = PFH_INIT,
  48         .min_children = 5,
  49         .max_children = 25,
  50         .spawn_rate = 5,
  51         .max_allowed_clients = 100,
  52         .child_min_life = 60 /* 1 minute minimum life time */
  53 };
  54 static struct pf_daemon_config pf_spoolss_cfg = { 0 };
  55
  56 pid_t start_spoolssd(struct tevent_context *ev_ctx,
  57                      struct messaging_context *msg_ctx);
  58
  59 static void spoolss_reopen_logs(int child_id)
  60 {
  61         char *lfile = lp_logfile();
  62         char *ext;
  63         int rc;
  64
  65         if (child_id) {
  66                 rc = asprintf(&ext, "%s.%d", DAEMON_NAME, child_id);
  67         } else {
  68                 rc = asprintf(&ext, "%s", DAEMON_NAME);
  69         }
  70
  71         if (rc == -1) {
  72                 return;
  73         }
  74
  75         rc = 0;
  76         if (lfile == NULL || lfile[0] == '\0') {
  77                 rc = asprintf(&lfile, "%s/log.%s",
  78                               get_dyn_LOGFILEBASE(), ext);
  79         } else {
  80                 if (strstr(lfile, ext) == NULL) {
  81                         if (child_id) {
  82                                 rc = asprintf(&lfile, "%s.%d",
  83                                               lp_logfile(), child_id);
  84                         } else {
  85                                 rc = asprintf(&lfile, "%s.%s",
  86                                               lp_logfile(), ext);
  87                         }
  88                 }
  89         }
  90
  91         if (rc > 0) {
  92                 lp_set_logfile(lfile);
  93                 SAFE_FREE(lfile);
  94         }
  95
  96         SAFE_FREE(ext);
  97
  98         reopen_logs();
  99 }
 100
 101 static void update_conf(struct tevent_context *ev,
 102                         struct messaging_context *msg)
 103 {
 104         change_to_root_user();
 105         lp_load(get_dyn_CONFIGFILE(), true, false, false, true);
 106         reload_printers(ev, msg);
 107
 108         spoolss_reopen_logs(spoolss_child_id);
 109         if (spoolss_child_id == 0) {
 110                 pfh_daemon_config(DAEMON_NAME,
 111                                   &pf_spoolss_cfg,
 112                                   &default_pf_spoolss_cfg);
 113                 pfh_manage_pool(ev, msg, &pf_spoolss_cfg, spoolss_pool);
 114         }
 115 }
 116
 117 static void smb_conf_updated(struct messaging_context *msg,
 118                              void *private_data,
 119                              uint32_t msg_type,
 120                              struct server_id server_id,
 121                              DATA_BLOB *data)
 122 {
 123         struct tevent_context *ev_ctx = talloc_get_type_abort(private_data,
 124                                                              struct tevent_context);
 125
 126         DEBUG(10, ("Got message saying smb.conf was updated. Reloading.\n"));
 127         update_conf(ev_ctx, msg);
 128 }
 129
 130 static void update_pcap(struct tevent_context *ev_ctx,
 131                         struct messaging_context *msg_ctx)
 132 {
 133         change_to_root_user();
 134         reload_printers(ev_ctx, msg_ctx);
 135 }
 136
 137 static void pcap_updated(struct messaging_context *msg,
 138                          void *private_data,
 139                          uint32_t msg_type,
 140                          struct server_id server_id,
 141                          DATA_BLOB *data)
 142 {
 143         struct tevent_context *ev_ctx;
 144
 145         ev_ctx = talloc_get_type_abort(private_data, struct tevent_context);
 146
 147         DEBUG(10, ("Got message that pcap updated. Reloading.\n"));
 148         update_pcap(ev_ctx, msg);
 149 }
 150
 151 static void spoolss_sig_term_handler(struct tevent_context *ev,
 152                                      struct tevent_signal *se,
 153                                      int signum,
 154                                      int count,
 155                                      void *siginfo,
 156                                      void *private_data)
 157 {
 158         exit_server_cleanly("termination signal");
 159 }
 160
 161 static void spoolss_setup_sig_term_handler(struct tevent_context *ev_ctx)
 162 {
 163         struct tevent_signal *se;
 164
 165         se = tevent_add_signal(ev_ctx,
 166                                ev_ctx,
 167                                SIGTERM, 0,
 168                                spoolss_sig_term_handler,
 169                                NULL);
 170         if (!se) {
 171                 exit_server("failed to setup SIGTERM handler");
 172         }
 173 }
 174
 175 static void spoolss_sig_hup_handler(struct tevent_context *ev,
 176                                     struct tevent_signal *se,
 177                                     int signum,
 178                                     int count,
 179                                     void *siginfo,
 180                                     void *pvt)
 181 {
 182         struct messaging_context *msg_ctx;
 183
 184         msg_ctx = talloc_get_type_abort(pvt, struct messaging_context);
 185
 186         DEBUG(1,("Reloading printers after SIGHUP\n"));
 187         update_conf(ev, msg_ctx);
 188
 189         /* relay to all children */
 190         if (spoolss_pool) {
 191                 prefork_send_signal_to_all(spoolss_pool, SIGHUP);
 192         }
 193 }
 194
 195 static void spoolss_setup_sig_hup_handler(struct tevent_context *ev_ctx,
 196                                           struct messaging_context *msg_ctx)
 197 {
 198         struct tevent_signal *se;
 199
 200         se = tevent_add_signal(ev_ctx,
 201                                ev_ctx,
 202                                SIGHUP, 0,
 203                                spoolss_sig_hup_handler,
 204                                msg_ctx);
 205         if (!se) {
 206                 exit_server("failed to setup SIGHUP handler");
 207         }
 208 }
 209
 210 static bool spoolss_init_cb(void *ptr)
 211 {
 212         struct messaging_context *msg_ctx = talloc_get_type_abort(
 213                 ptr, struct messaging_context);
 214
 215         return nt_printing_tdb_migrate(msg_ctx);
 216 }
 217
 218 static bool spoolss_shutdown_cb(void *ptr)
 219 {
 220         srv_spoolss_cleanup();
 221
 222         return true;
 223 }
 224
 225 /* Children */
 226
 227 struct spoolss_chld_sig_hup_ctx {
 228         struct messaging_context *msg_ctx;
 229         struct pf_worker_data *pf;
 230 };
 231
 232 static void spoolss_chld_sig_hup_handler(struct tevent_context *ev,
 233                                          struct tevent_signal *se,
 234                                          int signum,
 235                                          int count,
 236                                          void *siginfo,
 237                                          void *pvt)
 238 {
 239         struct spoolss_chld_sig_hup_ctx *shc;
 240
 241         shc = talloc_get_type_abort(pvt, struct spoolss_chld_sig_hup_ctx);
 242
 243         /* avoid wasting CPU cycles if we are going to exit soon anyways */
 244         if (shc->pf != NULL &&
 245             shc->pf->cmds == PF_SRV_MSG_EXIT) {
 246                 return;
 247         }
 248
 249         change_to_root_user();
 250         DEBUG(1,("Reloading printers after SIGHUP\n"));
 251         reload_printers(ev, shc->msg_ctx);
 252         spoolss_reopen_logs(spoolss_child_id);
 253 }
 254
 255 static bool spoolss_setup_chld_hup_handler(struct tevent_context *ev_ctx,
 256                                            struct messaging_context *msg_ctx,
 257                                            struct pf_worker_data *pf)
 258 {
 259         struct spoolss_chld_sig_hup_ctx *shc;
 260         struct tevent_signal *se;
 261
 262         shc = talloc(ev_ctx, struct spoolss_chld_sig_hup_ctx);
 263         if (!shc) {
 264                 DEBUG(1, ("failed to setup SIGHUP handler"));
 265                 return false;
 266         }
 267         shc->pf = pf;
 268         shc->msg_ctx = msg_ctx;
 269
 270         se = tevent_add_signal(ev_ctx,
 271                                ev_ctx,
 272                                SIGHUP, 0,
 273                                spoolss_chld_sig_hup_handler,
 274                                shc);
 275         if (!se) {
 276                 DEBUG(1, ("failed to setup SIGHUP handler"));
 277                 return false;
 278         }
 279
 280         return true;
 281 }
 282
 283 static void parent_ping(struct messaging_context *msg_ctx,
 284                         void *private_data,
 285                         uint32_t msg_type,
 286                         struct server_id server_id,
 287                         DATA_BLOB *data)
 288 {
 289
 290         /* The fact we received this message is enough to let make the event
 291          * loop if it was idle. spoolss_children_main will cycle through
 292          * spoolss_next_client at least once. That function will take whatever
 293          * action is necessary */
 294
 295         DEBUG(10, ("Got message that the parent changed status.\n"));
 296         return;
 297 }
 298
 299 static bool spoolss_child_init(struct tevent_context *ev_ctx,
 300                                int child_id, struct pf_worker_data *pf)
 301 {
 302         NTSTATUS status;
 303         struct rpc_srv_callbacks spoolss_cb;
 304         struct messaging_context *msg_ctx = server_messaging_context();
 305         bool ok;
 306
 307         status = reinit_after_fork(msg_ctx, ev_ctx,
 308                                    procid_self(), true);
 309         if (!NT_STATUS_IS_OK(status)) {
 310                 DEBUG(0,("reinit_after_fork() failed\n"));
 311                 smb_panic("reinit_after_fork() failed");
 312         }
 313
 314         spoolss_child_id = child_id;
 315         spoolss_reopen_logs(child_id);
 316
 317         ok = spoolss_setup_chld_hup_handler(ev_ctx, msg_ctx, pf);
 318         if (!ok) {
 319                 return false;
 320         }
 321
 322         if (!serverid_register(procid_self(),
 323                                 FLAG_MSG_GENERAL |
 324                                 FLAG_MSG_PRINT_GENERAL)) {
 325                 return false;
 326         }
 327
 328         if (!locking_init()) {
 329                 return false;
 330         }
 331
 332         messaging_register(msg_ctx, ev_ctx,
 333                            MSG_SMB_CONF_UPDATED, smb_conf_updated);
 334         messaging_register(msg_ctx, ev_ctx, MSG_PRINTER_PCAP,
 335                            pcap_updated);
 336         messaging_register(msg_ctx, ev_ctx,
 337                            MSG_PREFORK_PARENT_EVENT, parent_ping);
 338
 339         /* As soon as messaging is up check if pcap has been loaded already.
 340          * If so then we probably missed a message and should load_printers()
 341          * ourselves. If pcap has not been loaded yet, then ignore, we will get
 342          * a message as soon as the bq process completes the reload. */
 343         if (pcap_cache_loaded()) {
 344                 load_printers(ev_ctx, msg_ctx);
 345         }
 346
 347         /* try to reinit rpc queues */
 348         spoolss_cb.init = spoolss_init_cb;
 349         spoolss_cb.shutdown = spoolss_shutdown_cb;
 350         spoolss_cb.private_data = msg_ctx;
 351
 352         status = rpc_winreg_init(NULL);
 353         if (!NT_STATUS_IS_OK(status)) {
 354                 DEBUG(0, ("Failed to register winreg rpc inteface! (%s)\n",
 355                           nt_errstr(status)));
 356                 return false;
 357         }
 358
 359         status = rpc_spoolss_init(&spoolss_cb);
 360         if (!NT_STATUS_IS_OK(status)) {
 361                 DEBUG(0, ("Failed to register spoolss rpc inteface! (%s)\n",
 362                           nt_errstr(status)));
 363                 return false;
 364         }
 365
 366         return true;
 367 }
 368
 369 struct spoolss_children_data {
 370         struct tevent_context *ev_ctx;
 371         struct messaging_context *msg_ctx;
 372         struct pf_worker_data *pf;
 373         int listen_fd_size;
 374         int *listen_fds;
 375 };
 376
 377 static void spoolss_next_client(void *pvt);
 378
 379 static int spoolss_children_main(struct tevent_context *ev_ctx,
 380                                  struct messaging_context *msg_ctx,
 381                                  struct pf_worker_data *pf,
 382                                  int child_id,
 383                                  int listen_fd_size,
 384                                  int *listen_fds,
 385                                  void *private_data)
 386 {
 387         struct spoolss_children_data *data;
 388         bool ok;
 389         int ret;
 390
 391         ok = spoolss_child_init(ev_ctx, child_id, pf);
 392         if (!ok) {
 393                 return 1;
 394         }
 395
 396         data = talloc(ev_ctx, struct spoolss_children_data);
 397         if (!data) {
 398                 return 1;
 399         }
 400         data->pf = pf;
 401         data->ev_ctx = ev_ctx;
 402         data->msg_ctx = msg_ctx;
 403         data->listen_fd_size = listen_fd_size;
 404         data->listen_fds = listen_fds;
 405
 406         /* loop until it is time to exit */
 407         while (pf->status != PF_WORKER_EXITING) {
 408                 /* try to see if it is time to schedule the next client */
 409                 spoolss_next_client(data);
 410
 411                 ret = tevent_loop_once(ev_ctx);
 412                 if (ret != 0) {
 413                         DEBUG(0, ("tevent_loop_once() exited with %d: %s\n",
 414                                   ret, strerror(errno)));
 415                         pf->status = PF_WORKER_EXITING;
 416                 }
 417         }
 418
 419         return ret;
 420 }
 421
 422 static void spoolss_client_terminated(void *pvt)
 423 {
 424         struct spoolss_children_data *data;
 425
 426         data = talloc_get_type_abort(pvt, struct spoolss_children_data);
 427
 428         pfh_client_terminated(data->pf);
 429
 430         spoolss_next_client(pvt);
 431 }
 432
 433 struct spoolss_new_client {
 434         struct spoolss_children_data *data;
 435         struct tsocket_address *srv_addr;
 436         struct tsocket_address *cli_addr;
 437 };
 438
 439 static void spoolss_handle_client(struct tevent_req *req);
 440
 441 static void spoolss_next_client(void *pvt)
 442 {
 443         struct tevent_req *req;
 444         struct spoolss_children_data *data;
 445         struct spoolss_new_client *next;
 446
 447         data = talloc_get_type_abort(pvt, struct spoolss_children_data);
 448
 449         if (!pfh_child_allowed_to_accept(data->pf)) {
 450                 /* nothing to do for now we are already listening
 451                  * or we are not allowed to listen further */
 452                 return;
 453         }
 454
 455         next = talloc_zero(data, struct spoolss_new_client);
 456         if (!next) {
 457                 DEBUG(1, ("Out of memory!?\n"));
 458                 return;
 459         }
 460         next->data = data;
 461
 462         req = prefork_listen_send(next, data->ev_ctx, data->pf,
 463                                   data->listen_fd_size,
 464                                   data->listen_fds);
 465         if (!req) {
 466                 DEBUG(1, ("Failed to make listening request!?\n"));
 467                 talloc_free(next);
 468                 return;
 469         }
 470         tevent_req_set_callback(req, spoolss_handle_client, next);
 471 }
 472
 473 static void spoolss_handle_client(struct tevent_req *req)
 474 {
 475         struct spoolss_children_data *data;
 476         struct spoolss_new_client *client;
 477         const DATA_BLOB ping = data_blob_null;
 478         int ret;
 479         int sd;
 480
 481         client = tevent_req_callback_data(req, struct spoolss_new_client);
 482         data = client->data;
 483
 484         ret = prefork_listen_recv(req, client, &sd,
 485                                   &client->srv_addr, &client->cli_addr);
 486
 487         /* this will free the request too */
 488         talloc_free(client);
 489
 490         if (ret != 0) {
 491                 DEBUG(6, ("No client connection was available after all!\n"));
 492                 return;
 493         }
 494
 495         /* Warn parent that our status changed */
 496         messaging_send(data->msg_ctx, parent_id,
 497                         MSG_PREFORK_CHILD_EVENT, &ping);
 498
 499         DEBUG(2, ("Spoolss preforked child %d got client connection!\n",
 500                   (int)(data->pf->pid)));
 501
 502         named_pipe_accept_function(data->ev_ctx, data->msg_ctx,
 503                                    SPOOLSS_PIPE_NAME, sd,
 504                                    spoolss_client_terminated, data);
 505 }
 506
 507 /* ==== Main Process Functions ==== */
 508
 509 extern pid_t background_lpq_updater_pid;
 510 static char *bq_logfile;
 511
 512 static void check_updater_child(void)
 513 {
 514         int status;
 515         pid_t pid;
 516
 517         if (background_lpq_updater_pid == -1) {
 518                 return;
 519         }
 520
 521         pid = sys_waitpid(background_lpq_updater_pid, &status, WNOHANG);
 522         if (pid > 0) {
 523                 DEBUG(2, ("The background queue child died... Restarting!\n"));
 524                 pid = start_background_queue(server_event_context(),
 525                                              server_messaging_context(),
 526                                              bq_logfile);
 527                 background_lpq_updater_pid = pid;
 528         }
 529 }
 530
 531 static void child_ping(struct messaging_context *msg_ctx,
 532                         void *private_data,
 533                         uint32_t msg_type,
 534                         struct server_id server_id,
 535                         DATA_BLOB *data)
 536 {
 537         struct tevent_context *ev_ctx;
 538
 539         ev_ctx = talloc_get_type_abort(private_data, struct tevent_context);
 540
 541         DEBUG(10, ("Got message that a child changed status.\n"));
 542         pfh_manage_pool(ev_ctx, msg_ctx, &pf_spoolss_cfg, spoolss_pool);
 543 }
 544
 545 static bool spoolssd_schedule_check(struct tevent_context *ev_ctx,
 546                                     struct messaging_context *msg_ctx,
 547                                     struct timeval current_time);
 548 static void spoolssd_check_children(struct tevent_context *ev_ctx,
 549                                     struct tevent_timer *te,
 550                                     struct timeval current_time,
 551                                     void *pvt);
 552
 553 static void spoolssd_sigchld_handler(struct tevent_context *ev_ctx,
 554                                      struct prefork_pool *pfp,
 555                                      void *pvt)
 556 {
 557         struct messaging_context *msg_ctx;
 558
 559         msg_ctx = talloc_get_type_abort(pvt, struct messaging_context);
 560
 561         /* run pool management so we can fork/retire or increase
 562          * the allowed connections per child based on load */
 563         pfh_manage_pool(ev_ctx, msg_ctx, &pf_spoolss_cfg, spoolss_pool);
 564
 565         /* also check if the updater child is alive and well */
 566         check_updater_child();
 567 }
 568
 569 static bool spoolssd_setup_children_monitor(struct tevent_context *ev_ctx,
 570                                             struct messaging_context *msg_ctx)
 571 {
 572         bool ok;
 573
 574         /* add our oun sigchld callback */
 575         prefork_set_sigchld_callback(spoolss_pool,
 576                                      spoolssd_sigchld_handler, msg_ctx);
 577
 578         ok = spoolssd_schedule_check(ev_ctx, msg_ctx,
 579                                      tevent_timeval_current());
 580         return ok;
 581 }
 582
 583 static bool spoolssd_schedule_check(struct tevent_context *ev_ctx,
 584                                     struct messaging_context *msg_ctx,
 585                                     struct timeval current_time)
 586 {
 587         struct tevent_timer *te;
 588         struct timeval next_event;
 589
 590         /* check situation again in 10 seconds */
 591         next_event = tevent_timeval_current_ofs(10, 0);
 592
 593         /* TODO: check when the socket becomes readable, so that children
 594          * are checked only when there is some activity ? */
 595         te = tevent_add_timer(ev_ctx, spoolss_pool, next_event,
 596                                 spoolssd_check_children, msg_ctx);
 597         if (!te) {
 598                 DEBUG(2, ("Failed to set up children monitoring!\n"));
 599                 return false;
 600         }
 601
 602         return true;
 603 }
 604
 605 static void spoolssd_check_children(struct tevent_context *ev_ctx,
 606                                     struct tevent_timer *te,
 607                                     struct timeval current_time,
 608                                     void *pvt)
 609 {
 610         struct messaging_context *msg_ctx;
 611
 612         msg_ctx = talloc_get_type_abort(pvt, struct messaging_context);
 613
 614         pfh_manage_pool(ev_ctx, msg_ctx, &pf_spoolss_cfg, spoolss_pool);
 615
 616         spoolssd_schedule_check(ev_ctx, msg_ctx, current_time);
 617 }
 618
 619 static void print_queue_forward(struct messaging_context *msg,
 620                                 void *private_data,
 621                                 uint32_t msg_type,
 622                                 struct server_id server_id,
 623                                 DATA_BLOB *data)
 624 {
 625         messaging_send_buf(msg, pid_to_procid(background_lpq_updater_pid),
 626                            MSG_PRINTER_UPDATE, data->data, data->length);
 627 }
 628
 629 static char *get_bq_logfile(void)
 630 {
 631         char *lfile = lp_logfile();
 632         int rc;
 633
 634         if (lfile == NULL || lfile[0] == '\0') {
 635                 rc = asprintf(&lfile, "%s/log.%s.bq",
 636                                         get_dyn_LOGFILEBASE(), DAEMON_NAME);
 637         } else {
 638                 rc = asprintf(&lfile, "%s.bq", lp_logfile());
 639         }
 640         if (rc == -1) {
 641                 lfile = NULL;
 642         }
 643         return lfile;
 644 }
 645
 646 pid_t start_spoolssd(struct tevent_context *ev_ctx,
 647                     struct messaging_context *msg_ctx)
 648 {
 649         struct rpc_srv_callbacks spoolss_cb;
 650         struct dcerpc_binding_vector *v;
 651         TALLOC_CTX *mem_ctx;
 652         pid_t pid;
 653         NTSTATUS status;
 654         int listen_fd;
 655         int ret;
 656         bool ok;
 657
 658         DEBUG(1, ("Forking SPOOLSS Daemon\n"));
 659
 660         /*
 661          * Block signals before forking child as it will have to
 662          * set its own handlers. Child will re-enable SIGHUP as
 663          * soon as the handlers are set up.
 664          */
 665         BlockSignals(true, SIGTERM);
 666         BlockSignals(true, SIGHUP);
 667
 668         pid = sys_fork();
 669
 670         if (pid == -1) {
 671                 DEBUG(0, ("Failed to fork SPOOLSS [%s]\n",
 672                            strerror(errno)));
 673         }
 674
 675         /* parent or error */
 676         if (pid != 0) {
 677
 678                 /* Re-enable SIGHUP before returnig */
 679                 BlockSignals(false, SIGTERM);
 680                 BlockSignals(false, SIGHUP);
 681                 return pid;
 682         }
 683
 684         /* child */
 685         close_low_fds(false);
 686
 687         status = reinit_after_fork(msg_ctx,
 688                                    ev_ctx,
 689                                    procid_self(), true);
 690         if (!NT_STATUS_IS_OK(status)) {
 691                 DEBUG(0,("reinit_after_fork() failed\n"));
 692                 smb_panic("reinit_after_fork() failed");
 693         }
 694
 695         /* save the parent process id so the children can use it later */
 696         parent_id = procid_self();
 697
 698         spoolss_reopen_logs(0);
 699         pfh_daemon_config(DAEMON_NAME,
 700                           &pf_spoolss_cfg,
 701                           &default_pf_spoolss_cfg);
 702
 703         spoolss_setup_sig_term_handler(ev_ctx);
 704         spoolss_setup_sig_hup_handler(ev_ctx, msg_ctx);
 705
 706         BlockSignals(false, SIGTERM);
 707         BlockSignals(false, SIGHUP);
 708
 709         /* always start the backgroundqueue listner in spoolssd */
 710         bq_logfile = get_bq_logfile();
 711         pid = start_background_queue(ev_ctx, msg_ctx, bq_logfile);
 712         if (pid > 0) {
 713                 background_lpq_updater_pid = pid;
 714         }
 715
 716         /* the listening fd must be created before the children are actually
 717          * forked out. */
 718         listen_fd = create_named_pipe_socket(SPOOLSS_PIPE_NAME);
 719         if (listen_fd == -1) {
 720                 exit(1);
 721         }
 722
 723         ret = listen(listen_fd, pf_spoolss_cfg.max_allowed_clients);
 724         if (ret == -1) {
 725                 DEBUG(0, ("Failed to listen on spoolss pipe - %s\n",
 726                           strerror(errno)));
 727                 exit(1);
 728         }
 729
 730         /* start children before any more initialization is done */
 731         ok = prefork_create_pool(ev_ctx, /* mem_ctx */
 732                                  ev_ctx, msg_ctx,
 733                                  1, &listen_fd,
 734                                  pf_spoolss_cfg.min_children,
 735                                  pf_spoolss_cfg.max_children,
 736                                  &spoolss_children_main, NULL,
 737                                  &spoolss_pool);
 738         if (!ok) {
 739                 exit(1);
 740         }
 741
 742         if (!serverid_register(procid_self(),
 743                                 FLAG_MSG_GENERAL |
 744                                 FLAG_MSG_PRINT_GENERAL)) {
 745                 exit(1);
 746         }
 747
 748         if (!locking_init()) {
 749                 exit(1);
 750         }
 751
 752         messaging_register(msg_ctx, ev_ctx,
 753                            MSG_SMB_CONF_UPDATED, smb_conf_updated);
 754         messaging_register(msg_ctx, NULL, MSG_PRINTER_UPDATE,
 755                            print_queue_forward);
 756         messaging_register(msg_ctx, ev_ctx, MSG_PRINTER_PCAP,
 757                            pcap_updated);
 758         messaging_register(msg_ctx, ev_ctx,
 759                            MSG_PREFORK_CHILD_EVENT, child_ping);
 760
 761         /* As soon as messaging is up check if pcap has been loaded already.
 762          * If so then we probably missed a message and should load_printers()
 763          * ourselves. If pcap has not been loaded yet, then ignore, we will get
 764          * a message as soon as the bq process completes the reload. */
 765         if (pcap_cache_loaded()) {
 766                 load_printers(ev_ctx, msg_ctx);
 767         }
 768
 769         mem_ctx = talloc_new(NULL);
 770         if (mem_ctx == NULL) {
 771                 exit(1);
 772         }
 773
 774         /*
 775          * Initialize spoolss with an init function to convert printers first.
 776          * static_init_rpc will try to initialize the spoolss server too but you
 777          * can't register it twice.
 778          */
 779         spoolss_cb.init = spoolss_init_cb;
 780         spoolss_cb.shutdown = spoolss_shutdown_cb;
 781         spoolss_cb.private_data = msg_ctx;
 782
 783         status = rpc_winreg_init(NULL);
 784         if (!NT_STATUS_IS_OK(status)) {
 785                 DEBUG(0, ("Failed to register winreg rpc inteface! (%s)\n",
 786                           nt_errstr(status)));
 787                 exit(1);
 788         }
 789
 790         status = rpc_spoolss_init(&spoolss_cb);
 791         if (!NT_STATUS_IS_OK(status)) {
 792                 DEBUG(0, ("Failed to register spoolss rpc inteface! (%s)\n",
 793                           nt_errstr(status)));
 794                 exit(1);
 795         }
 796
 797         status = dcerpc_binding_vector_new(mem_ctx, &v);
 798         if (!NT_STATUS_IS_OK(status)) {
 799                 DEBUG(0, ("Failed to create binding vector (%s)\n",
 800                           nt_errstr(status)));
 801                 exit(1);
 802         }
 803
 804         status = dcerpc_binding_vector_add_np_default(&ndr_table_spoolss, v);
 805         if (!NT_STATUS_IS_OK(status)) {
 806                 DEBUG(0, ("Failed to add np to binding vector (%s)\n",
 807                           nt_errstr(status)));
 808                 exit(1);
 809         }
 810
 811         status = rpc_ep_register(ev_ctx, msg_ctx, &ndr_table_spoolss, v);
 812         if (!NT_STATUS_IS_OK(status)) {
 813                 DEBUG(0, ("Failed to register spoolss endpoint! (%s)\n",
 814                           nt_errstr(status)));
 815                 exit(1);
 816         }
 817
 818         talloc_free(mem_ctx);
 819
 820         ok = spoolssd_setup_children_monitor(ev_ctx, msg_ctx);
 821         if (!ok) {
 822                 DEBUG(0, ("Failed to setup children monitoring!\n"));
 823                 exit(1);
 824         }
 825
 826         DEBUG(1, ("SPOOLSS Daemon Started (%d)\n", getpid()));
 827
 828         pfh_manage_pool(ev_ctx, msg_ctx, &pf_spoolss_cfg, spoolss_pool);
 829
 830         /* loop forever */
 831         ret = tevent_loop_wait(ev_ctx);
 832
 833         /* should not be reached */
 834         DEBUG(0,("spoolssd tevent_loop_wait() exited with %d - %s\n",
 835                  ret, (ret == 0) ? "out of events" : strerror(errno)));
 836         exit(1);
 837 }