-/**
- * ConsoleMessage.c - ConsoleMessage main
- * Wilfredo Sanchez | wsanchez@opensource.apple.com
- * Kevin Van Vechten | kevinvv@uclink4.berkeley.edu
- * $Apple$
- **
- * Copyright (c) 1999-2002 Apple Computer, Inc. All rights reserved.
- *
- * @APPLE_LICENSE_HEADER_START@
- *
- * Portions Copyright (c) 1999 Apple Computer, Inc. All Rights
- * Reserved. This file contains Original Code and/or Modifications of
- * Original Code as defined in and that are subject to the Apple Public
- * Source License Version 1.1 (the "License"). You may not use this file
- * except in compliance with the License. Please obtain a copy of the
- * License at http://www.apple.com/publicsource and read it before using
- * this file.
- *
- * The Original Code and all software distributed under the License are
- * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
- * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
- * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE OR NON- INFRINGEMENT. Please see the
- * License for the specific language governing rights and limitations
- * under the License.
- *
- * @APPLE_LICENSE_HEADER_END@
- **
- * The ConsoleMessage utility sends an IPC message to SystemStarter
- * containing the message specified on the command line. SystemStarter
- * will perform the localization. The message is also printed to
- * the system log.
- **/
-
-#include <unistd.h>
-#include <crt_externs.h>
-#include <syslog.h>
-#include <mach/mach.h>
-#include <mach/message.h>
-#include <mach/mach_error.h>
-#include <servers/bootstrap.h>
-#include <CoreFoundation/CoreFoundation.h>
-#include "SystemStarterIPC.h"
-
-static CFDataRef sendIPCMessage(CFStringRef aPortName, CFDataRef aData, CFStringRef aRunLoopMode);
-
-static void usage() __attribute__((__noreturn__));
-static void
-usage()
-{
- /* char* aProgram = **_NSGetArgv(); */
- fprintf(stderr, "usage:\n"
- "\tConsoleMessage [-v] <message>\n"
- "\tConsoleMessage [-v] -S\n"
- "\tConsoleMessage [-v] -F\n"
- "\tConsoleMessage [-v] -s <service>\n"
- "\tConsoleMessage [-v] -f <service>\n"
- "\tConsoleMessage [-v] -q <setting>\n"
- "\tConsoleMessage [-v] -b <path>\n"
- "\tConsoleMessage [-v] -u\n"
- "\noptions:\n"
- "\t-v: verbose (prints errors to stdout)\n"
- "\t-S: mark all services as successful\n"
- "\t-F: mark all services as failed\n"
- "\t-s: mark a specific service as successful\n"
- "\t-f: mark a specific service as failed\n"
- "\t-q: query a configuration setting\n"
- "\t-b: (ignored)\n"
- "\t-u: (ignored)\n");
- exit(1);
-}
-
-enum {
- kActionConsoleMessage,
- kActionSuccess,
- kActionFailure,
- kActionQuery,
-};
-
-int
-main(int argc, char *argv[])
-{
- int anExitCode = 0;
- int aVerboseFlag = 0;
- int anAction = kActionConsoleMessage;
- char *aProgram = argv[0];
- char *anArgCStr = NULL;
- char c;
- pid_t w4lw_pid = 0;
- FILE *w4lw_f;
-
- /**
- * Handle command line.
- **/
- while ((c = getopt(argc, argv, "?vSFs:f:q:b:u")) != -1) {
- switch (c) {
- case '?':
- usage();
- break;
- case 'v':
- aVerboseFlag = 1;
- break;
- case 'S':
- anAction = kActionSuccess;
- anArgCStr = NULL;
- break;
- case 'F':
- anAction = kActionFailure;
- anArgCStr = NULL;
- break;
- case 's':
- anAction = kActionSuccess;
- anArgCStr = optarg;
- break;
- case 'f':
- anAction = kActionFailure;
- anArgCStr = optarg;
- break;
- case 'q':
- anAction = kActionQuery;
- anArgCStr = optarg;
- break;
- case 'b':
- exit(EXIT_SUCCESS);
- break;
- case 'u':
- w4lw_f = fopen("/var/run/waiting4loginwindow.pid", "r");
- if (w4lw_f) {
- fscanf(w4lw_f, "%d\n", &w4lw_pid);
- if (w4lw_pid)
- kill(w4lw_pid, SIGTERM);
- }
- exit(EXIT_SUCCESS);
- break;
- default:
- fprintf(stderr, "ignoring unknown option '-%c'\n", c);
- break;
- }
- }
- argc -= optind;
- argv += optind;
-
- if ((anAction == kActionConsoleMessage && argc != 1) ||
- (anAction == kActionSuccess && argc != 0) ||
- (anAction == kActionFailure && argc != 0) ||
- (anAction == kActionQuery && argc != 0)) {
- usage();
- }
- if (getuid() != 0) {
- fprintf(stderr, "you must be root to run %s\n", aProgram);
- exit(1);
- } else {
- CFMutableDictionaryRef anIPCMessage = CFDictionaryCreateMutable(NULL, 0, &kCFTypeDictionaryKeyCallBacks,
- &kCFTypeDictionaryValueCallBacks);
-
- if (anIPCMessage) {
- CFStringRef anArg = NULL;
- CFIndex aPID = getppid();
- CFNumberRef aPIDNumber = CFNumberCreate(NULL, kCFNumberCFIndexType, &aPID);
-
- /*
- * Parent process id is the process id of the startup
- * item that called ConsoleMessage.
- */
- CFDictionarySetValue(anIPCMessage, kIPCProcessIDKey, aPIDNumber);
- CFRelease(aPIDNumber);
-
- if (anArgCStr) {
- anArg = CFStringCreateWithCString(NULL, anArgCStr, kCFStringEncodingUTF8);
- }
- if (anAction == kActionSuccess || anAction == kActionFailure) {
- CFBooleanRef aStatus = (anAction == kActionSuccess) ? kCFBooleanTrue : kCFBooleanFalse;
- CFDictionarySetValue(anIPCMessage, kIPCMessageKey, kIPCStatusMessage);
- CFDictionarySetValue(anIPCMessage, kIPCStatusKey, aStatus);
- if (anArg)
- CFDictionarySetValue(anIPCMessage, kIPCServiceNameKey, anArg);
- } else if (anAction == kActionQuery && anArg) {
- CFDictionarySetValue(anIPCMessage, kIPCMessageKey, kIPCQueryMessage);
- CFDictionarySetValue(anIPCMessage, kIPCConfigSettingKey, anArg);
- } else if (anAction == kActionConsoleMessage) {
- char *aConsoleMessageCStr = argv[0];
- CFStringRef aConsoleMessage = CFStringCreateWithCString(NULL, aConsoleMessageCStr, kCFStringEncodingUTF8);
-
- syslog(LOG_INFO, "%s", aConsoleMessageCStr);
-
- CFDictionarySetValue(anIPCMessage, kIPCMessageKey, kIPCConsoleMessage);
- CFDictionarySetValue(anIPCMessage, kIPCConsoleMessageKey, aConsoleMessage);
- CFRelease(aConsoleMessage);
- }
- if (anArg)
- CFRelease(anArg);
-
- {
- CFDataRef aData = CFPropertyListCreateXMLData(NULL, anIPCMessage);
- if (aData) {
- CFDataRef aResultData = sendIPCMessage(CFSTR(kSystemStarterMessagePort), aData, kCFRunLoopDefaultMode);
-
- /* aResultData should be ASCIZ */
- if (aResultData) {
- fprintf(stdout, "%s", CFDataGetBytePtr(aResultData));
- CFRelease(aResultData);
- } else {
- char *aConsoleMessageCStr = argv[0];
- fprintf(stdout, "%s\n", aConsoleMessageCStr);
-
- if (aVerboseFlag)
- fprintf(stderr, "%s could not connect to SystemStarter.\n", aProgram);
- anExitCode = 0;
- }
- CFRelease(aData);
- } else {
- if (aVerboseFlag)
- fprintf(stderr, "%s: not enough memory to create IPC message.\n", aProgram);
- anExitCode = 1;
- }
- }
- } else {
- if (aVerboseFlag)
- fprintf(stderr, "%s: not enough memory to create IPC message.\n", aProgram);
- anExitCode = 1;
- }
- }
- exit(anExitCode);
-}
-
-
-static void
-dummyCallback(void)
-{
-}
-
-static void replyCallback(CFMachPortRef port __attribute__((unused)), void *aPtr, CFIndex aSize __attribute__((unused)), CFDataRef * aReply) {
- SystemStarterIPCMessage *aMessage = (SystemStarterIPCMessage *) aPtr;
-
- if (aReply != NULL &&
- aMessage->aProtocol == kIPCProtocolVersion &&
- aMessage->aByteLength >= 0) {
- *aReply = CFDataCreate(NULL, (UInt8 *) aMessage + aMessage->aByteLength, aMessage->aByteLength);
- } else if (aReply != NULL) {
- *aReply = NULL;
- }
-}
-
-
-static CFDataRef
-sendIPCMessage(CFStringRef aPortName, CFDataRef aData, CFStringRef aRunLoopMode)
-{
- SystemStarterIPCMessage *aMessage = NULL;
- CFRunLoopSourceRef aSource = NULL;
- CFMachPortRef aMachPort = NULL, aReplyPort = NULL;
- CFMachPortContext aContext;
- kern_return_t aKernReturn = KERN_FAILURE;
- mach_port_t aBootstrapPort, aNativePort;
- char *aPortNameUTF8String;
- CFDataRef aReply = NULL;
- SInt32 aStrLen;
-
- aContext.version = 0;
- aContext.info = (void *) NULL;
- aContext.retain = 0;
- aContext.release = 0;
- aContext.copyDescription = 0;
-
- /* Look up the remote port by name */
-
- aStrLen = CFStringGetMaximumSizeForEncoding(CFStringGetLength(aPortName) + 1, kCFStringEncodingUTF8);
- aPortNameUTF8String = malloc(aStrLen);
- if (aPortNameUTF8String) {
- CFStringGetCString(aPortName, aPortNameUTF8String, aStrLen, kCFStringEncodingUTF8);
- task_get_bootstrap_port(mach_task_self(), &aBootstrapPort);
- aKernReturn = bootstrap_look_up(aBootstrapPort, aPortNameUTF8String, &aNativePort);
- aMachPort = (KERN_SUCCESS == aKernReturn) ? CFMachPortCreateWithPort(NULL, aNativePort, (CFMachPortCallBack) dummyCallback, &aContext, NULL) : NULL;
- free(aPortNameUTF8String);
- }
- /* Create a reply port and associated run loop source */
- aContext.info = &aReply;
- aReplyPort = CFMachPortCreate(NULL, (CFMachPortCallBack) replyCallback, &aContext, NULL);
- if (aReplyPort) {
- aSource = CFMachPortCreateRunLoopSource(NULL, aReplyPort, 0);
- if (aSource) {
- CFRunLoopAddSource(CFRunLoopGetCurrent(), aSource, aRunLoopMode);
- }
- }
- /* Allocate a buffer for the message */
- if (aData && aMachPort && aReplyPort) {
- SInt32 aSize = (sizeof(SystemStarterIPCMessage) + (CFDataGetLength(aData) + 3)) & ~0x3;
- aMessage = (SystemStarterIPCMessage *) malloc(aSize);
- if (aMessage) {
- aMessage->aHeader.msgh_id = 1;
- aMessage->aHeader.msgh_size = aSize;
- aMessage->aHeader.msgh_remote_port = CFMachPortGetPort(aMachPort);
- aMessage->aHeader.msgh_local_port = CFMachPortGetPort(aReplyPort);
- aMessage->aHeader.msgh_reserved = 0;
- aMessage->aHeader.msgh_bits = MACH_MSGH_BITS(MACH_MSG_TYPE_COPY_SEND, MACH_MSG_TYPE_MAKE_SEND_ONCE);
- aMessage->aBody.msgh_descriptor_count = 0;
- aMessage->aProtocol = kIPCProtocolVersion;
- aMessage->aByteLength = CFDataGetLength(aData);
- memmove((uint8_t *) aMessage + sizeof(SystemStarterIPCMessage),
- CFDataGetBytePtr(aData),
- CFDataGetLength(aData));
- }
- }
- /* Wait up to 1 second to send the message */
- if (aMessage) {
- aKernReturn = mach_msg((mach_msg_header_t *) aMessage, MACH_SEND_MSG | MACH_SEND_TIMEOUT, aMessage->aHeader.msgh_size, 0, MACH_PORT_NULL, 1000.0, MACH_PORT_NULL);
- free(aMessage);
- }
- /* Wait up to 30 seconds for the reply */
- if (aSource && aKernReturn == MACH_MSG_SUCCESS) {
- CFRetain(aReplyPort);
- CFRunLoopRunInMode(aRunLoopMode, 30.0, true);
- /*
- * aReplyPort's replyCallback will set the local aReply
- * variable
- */
- CFRelease(aReplyPort);
- }
- if (aMachPort)
- CFRelease(aMachPort);
- if (aReplyPort)
- CFRelease(aReplyPort);
- if (aSource) {
- CFRunLoopRemoveSource(CFRunLoopGetCurrent(), aSource, aRunLoopMode);
- CFRelease(aSource);
- }
- return aReply;
-}
projects / apple / launchd.git / blobdiff