/*
- * Copyright (c) 2000 Apple Computer, Inc. All rights reserved.
+ * Copyright (c) 2000, 2001, 2003-2005, 2009-2011, 2013, 2016, 2017, 2019 Apple Inc. All rights reserved.
*
* @APPLE_LICENSE_HEADER_START@
*
- * The contents of this file constitute Original Code as defined in and
- * 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.
+ * 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 2.0 (the 'License'). You may not use this file except in
+ * compliance with the License. Please obtain a copy of the License at
+ * http://www.opensource.apple.com/apsl/ and read it before using this
+ * file.
*
- * This Original Code and all software distributed under the License are
- * distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY KIND, EITHER
+ * 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.
+ * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
+ * Please see the License for the specific language governing rights and
+ * limitations under the License.
*
* @APPLE_LICENSE_HEADER_END@
*/
* - initial revision
*/
-#include <mach/mach.h>
-#include <mach/mach_error.h>
-
-#include <SystemConfiguration/SystemConfiguration.h>
-#include <SystemConfiguration/SCPrivate.h>
#include "SCDynamicStoreInternal.h"
#include "config.h" /* MiG generated file */
-
-static __inline__ void
-my_CFSetApplyFunction(CFSetRef theSet,
- CFSetApplierFunction applier,
- void *context)
-{
- CFAllocatorRef myAllocator;
- CFSetRef mySet;
-
- myAllocator = CFGetAllocator(theSet);
- mySet = CFSetCreateCopy(myAllocator, theSet);
- CFSetApplyFunction(mySet, applier, context);
- CFRelease(mySet);
- return;
-}
-
-
-/*
- * "context" argument for removeOldKey() and addNewKey()
- */
-typedef struct {
- SCDynamicStoreRef store;
- CFArrayRef newKeys; /* for removeOldKey */
- Boolean isRegex;
- Boolean ok;
-} updateKeysContext, *updateKeysContextRef;
-
-
-static void
-removeOldKey(const void *value, void *context)
-{
- CFStringRef oldKey = (CFStringRef)value;
- updateKeysContextRef myContextRef = (updateKeysContextRef)context;
-
- if (!myContextRef->ok) {
- return;
- }
-
- if (!myContextRef->newKeys ||
- !CFArrayContainsValue(myContextRef->newKeys,
- CFRangeMake(0, CFArrayGetCount(myContextRef->newKeys)),
- oldKey)) {
- /* the old notification key is not being retained, remove it */
- myContextRef->ok = SCDynamicStoreRemoveWatchedKey(myContextRef->store,
- oldKey,
- myContextRef->isRegex);
- }
-
- return;
-}
-
-
-static void
-addNewKey(const void *value, void *context)
-{
- CFStringRef newKey = (CFStringRef)value;
- updateKeysContextRef myContextRef = (updateKeysContextRef)context;
- SCDynamicStorePrivateRef storePrivate = (SCDynamicStorePrivateRef)myContextRef->store;
-
- if (!myContextRef->ok) {
- return;
- }
-
- if (myContextRef->isRegex) {
- if (!CFSetContainsValue(storePrivate->reKeys, newKey)) {
- /* add pattern to this sessions notifier list */
- myContextRef->ok = SCDynamicStoreAddWatchedKey(myContextRef->store, newKey, TRUE);
- }
- } else {
- if (!CFSetContainsValue(storePrivate->keys, newKey)) {
- /* add key to this sessions notifier list */
- myContextRef->ok = SCDynamicStoreAddWatchedKey(myContextRef->store, newKey, FALSE);
- }
- }
-
- return;
-}
-
Boolean
SCDynamicStoreSetNotificationKeys(SCDynamicStoreRef store,
CFArrayRef keys,
CFArrayRef patterns)
{
- updateKeysContext myContext;
SCDynamicStorePrivateRef storePrivate = (SCDynamicStorePrivateRef)store;
-
- SCLog(_sc_verbose, LOG_DEBUG, CFSTR("SCDynamicStoreSetNotificationKeys:"));
- SCLog(_sc_verbose, LOG_DEBUG, CFSTR(" keys = %@"), keys);
- SCLog(_sc_verbose, LOG_DEBUG, CFSTR(" patterns = %@"), patterns);
-
- if (!store) {
+ kern_return_t status;
+ CFDataRef xmlKeys = NULL; /* keys (XML serialized) */
+ xmlData_t myKeysRef = NULL; /* keys (serialized) */
+ CFIndex myKeysLen = 0;
+ CFDataRef xmlPatterns = NULL; /* patterns (XML serialized) */
+ xmlData_t myPatternsRef = NULL; /* patterns (serialized) */
+ CFIndex myPatternsLen = 0;
+ int sc_status;
+ CFMutableArrayRef tmp;
+
+ if (store == NULL) {
/* sorry, you must provide a session */
_SCErrorSet(kSCStatusNoStoreSession);
return FALSE;
}
if (storePrivate->server == MACH_PORT_NULL) {
- /* sorry, you must have an open session to play */
_SCErrorSet(kSCStatusNoStoreServer);
- return FALSE;
+ return FALSE; /* you must have an open session to play */
+ }
+
+ /* serialize the keys */
+ if (keys != NULL) {
+ if (!_SCSerialize(keys, &xmlKeys, (void **)&myKeysRef, &myKeysLen)) {
+ _SCErrorSet(kSCStatusFailed);
+ return FALSE;
+ }
+ }
+
+ /* serialize the patterns */
+ if (patterns != NULL) {
+ if (!_SCSerialize(patterns, &xmlPatterns, (void **)&myPatternsRef, &myPatternsLen)) {
+ if (xmlKeys != NULL) CFRelease(xmlKeys);
+ _SCErrorSet(kSCStatusFailed);
+ return FALSE;
+ }
}
- myContext.ok = TRUE;
- myContext.store = store;
- /* remove any previously registered keys */
- myContext.newKeys = keys;
- myContext.isRegex = FALSE;
- my_CFSetApplyFunction(storePrivate->keys, removeOldKey, &myContext);
+ retry :
- /* register any new keys */
- if (keys) {
- CFArrayApplyFunction(keys,
- CFRangeMake(0, CFArrayGetCount(keys)),
- addNewKey,
- &myContext);
+ /* send the keys and patterns, fetch the associated result from the server */
+ status = notifyset(storePrivate->server,
+ myKeysRef,
+ (mach_msg_type_number_t)myKeysLen,
+ myPatternsRef,
+ (mach_msg_type_number_t)myPatternsLen,
+ (int *)&sc_status);
+
+ if (__SCDynamicStoreCheckRetryAndHandleError(store,
+ status,
+ &sc_status,
+ "SCDynamicStoreSetNotificationKeys notifyset()")) {
+ goto retry;
}
- /* remove any previously registered patterns */
- myContext.newKeys = patterns;
- myContext.isRegex = TRUE;
- my_CFSetApplyFunction(storePrivate->reKeys, removeOldKey, &myContext);
-
- /* register any new patterns */
- if (patterns) {
- CFArrayApplyFunction(patterns,
- CFRangeMake(0, CFArrayGetCount(patterns)),
- addNewKey,
- &myContext);
+ /* clean up */
+ if (xmlKeys != NULL) CFRelease(xmlKeys);
+ if (xmlPatterns != NULL) CFRelease(xmlPatterns);
+
+ if (sc_status != kSCStatusOK) {
+ _SCErrorSet(sc_status);
+ return FALSE;
}
- return myContext.ok;
+ /* in case we need to re-connect, save the keys/patterns */
+ tmp = (keys != NULL) ? CFArrayCreateMutableCopy(NULL, 0, keys) : NULL;
+ if (storePrivate->keys != NULL) CFRelease(storePrivate->keys);
+ storePrivate->keys = tmp;
+
+ tmp = (patterns != NULL) ? CFArrayCreateMutableCopy(NULL, 0, patterns) : NULL;
+ if (storePrivate->patterns != NULL) CFRelease(storePrivate->patterns);
+ storePrivate->patterns = tmp;
+
+ return TRUE;
}
projects / apple / configd.git / blobdiff