SystemConfiguration.fproj/SCDNotifierInformViaFD.c

   1 /*
   2  * Copyright (c) 2000, 2001, 2004, 2005, 2008-2010 Apple Inc. All rights reserved.
   3  *
   4  * @APPLE_LICENSE_HEADER_START@
   5  *
   6  * This file contains Original Code and/or Modifications of Original Code
   7  * as defined in and that are subject to the Apple Public Source License
   8  * Version 2.0 (the 'License'). You may not use this file except in
   9  * compliance with the License. Please obtain a copy of the License at
  10  * http://www.opensource.apple.com/apsl/ and read it before using this
  11  * file.
  12  *
  13  * The Original Code and all software distributed under the License are
  14  * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
  15  * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
  16  * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
  18  * Please see the License for the specific language governing rights and
  19  * limitations under the License.
  20  *
  21  * @APPLE_LICENSE_HEADER_END@
  22  */
  23
  24 /*
  25  * Modification History
  26  *
  27  * June 1, 2001                 Allan Nathanson <ajn@apple.com>
  28  * - public API conversion
  29  *
  30  * April 5, 2000                Allan Nathanson <ajn@apple.com>
  31  * - initial revision
  32  */
  33
  34 #include <mach/mach.h>
  35 #include <mach/mach_error.h>
  36
  37 #include <SystemConfiguration/SystemConfiguration.h>
  38 #include <SystemConfiguration/SCPrivate.h>
  39 #include "SCDynamicStoreInternal.h"
  40 #include "config.h"             /* MiG generated file */
  41
  42 #include <paths.h>
  43 #include <unistd.h>
  44 #include <sys/types.h>
  45 #include <sys/socket.h>
  46 #include <sys/un.h>
  47
  48 Boolean
  49 SCDynamicStoreNotifyFileDescriptor(SCDynamicStoreRef    store,
  50                                    int32_t              identifier,
  51                                    int                  *fd)
  52 {
  53         size_t                          n;
  54         int                             sc_status;
  55         int                             sock;
  56         SCDynamicStorePrivateRef        storePrivate = (SCDynamicStorePrivateRef)store;
  57         kern_return_t                   status;
  58         char                            tmpdir[PATH_MAX];
  59         struct sockaddr_un              un;
  60
  61         if (store == NULL) {
  62                 /* sorry, you must provide a session */
  63                 _SCErrorSet(kSCStatusNoStoreSession);
  64                 return FALSE;
  65         }
  66
  67         if (storePrivate->server == MACH_PORT_NULL) {
  68                 /* sorry, you must have an open session to play */
  69                 _SCErrorSet(kSCStatusNoStoreServer);
  70                 return FALSE;
  71         }
  72
  73         if (storePrivate->notifyStatus != NotifierNotRegistered) {
  74                 /* sorry, you can only have one notification registered at once */
  75                 _SCErrorSet(kSCStatusNotifierActive);
  76                 return FALSE;
  77         }
  78
  79         if ((sock = socket(AF_UNIX, SOCK_STREAM, 0)) == -1) {
  80                 _SCErrorSet(errno);
  81                 SCLog(TRUE, LOG_NOTICE, CFSTR("SCDynamicStoreNotifyFileDescriptor socket(): %s"), strerror(errno));
  82                 return FALSE;
  83         }
  84
  85         /* establish a UNIX domain socket for server->client notification */
  86
  87         n = confstr(_CS_DARWIN_USER_TEMP_DIR, tmpdir, sizeof(tmpdir));
  88         if ((n <= 0) || (n >= sizeof(tmpdir))) {
  89                 (void) strlcpy(tmpdir, _PATH_TMP, sizeof(tmpdir));
  90         }
  91
  92         bzero(&un, sizeof(un));
  93         un.sun_family = AF_UNIX;
  94         snprintf(un.sun_path,
  95                  sizeof(un.sun_path)-1,
  96                  "%s%s-%d-%d",
  97                  tmpdir,
  98                  "SCDynamicStoreNotifyFileDescriptor",
  99                  getpid(),
 100                  storePrivate->server);
 101
 102         /* ensure that the path does not already exist */
 103         (void) unlink(un.sun_path);
 104
 105         if (bind(sock, (struct sockaddr *)&un, sizeof(un)) == -1) {
 106                 _SCErrorSet(errno);
 107                 SCLog(TRUE, LOG_NOTICE, CFSTR("SCDynamicStoreNotifyFileDescriptor bind(): %s"), strerror(errno));
 108                 (void) unlink(un.sun_path);
 109                 (void) close(sock);
 110                 return FALSE;
 111         }
 112
 113         if (listen(sock, 0) == -1) {
 114                 _SCErrorSet(errno);
 115                 SCLog(TRUE, LOG_NOTICE, CFSTR("SCDynamicStoreNotifyFileDescriptor listen(): %s"), strerror(errno));
 116                 (void) unlink(un.sun_path);
 117                 (void) close(sock);
 118                 return FALSE;
 119         }
 120
 121     retry :
 122
 123         status = notifyviafd(storePrivate->server,
 124                              un.sun_path,
 125                              strlen(un.sun_path),
 126                              identifier,
 127                              (int *)&sc_status);
 128
 129         if (status != KERN_SUCCESS) {
 130                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 131                         /* the server's gone and our session port's dead, remove the dead name right */
 132                         (void) mach_port_deallocate(mach_task_self(), storePrivate->server);
 133                 } else {
 134                         /* we got an unexpected error, leave the [session] port alone */
 135                         SCLog(TRUE, LOG_ERR, CFSTR("SCDynamicStoreNotifyFileDescriptor notifyviafd(): %s"), mach_error_string(status));
 136                 }
 137                 storePrivate->server = MACH_PORT_NULL;
 138                 if ((status == MACH_SEND_INVALID_DEST) || (status == MIG_SERVER_DIED)) {
 139                         if (__SCDynamicStoreReconnect(store)) {
 140                                 goto retry;
 141                         }
 142                 }
 143                 _SCErrorSet(status);
 144                 return FALSE;
 145         }
 146
 147         (void) unlink(un.sun_path);
 148
 149         if (sc_status != kSCStatusOK) {
 150                 _SCErrorSet(sc_status);
 151                 SCLog(TRUE, LOG_NOTICE,
 152                       CFSTR("SCDynamicStoreNotifyFileDescriptor server error: %s"),
 153                       SCErrorString(sc_status));
 154                 (void) close(sock);
 155                 return FALSE;
 156         }
 157
 158         *fd = accept(sock, 0, 0);
 159         if (*fd == -1) {
 160                 _SCErrorSet(errno);
 161                 SCLog(TRUE, LOG_NOTICE, CFSTR("SCDynamicStoreNotifyFileDescriptor accept(): %s"), strerror(errno));
 162                 (void) close(sock);
 163                 return FALSE;
 164         }
 165         (void) close(sock);
 166
 167         /* set notifier active */
 168         storePrivate->notifyStatus = Using_NotifierInformViaFD;
 169
 170         return TRUE;
 171 }