1/*
2 * Copyright (c) 2000-2005, 2009-2011, 2013 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 * March 24, 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
43static void
44addKey(CFMutableArrayRef *keysP, CFStringRef key)
45{
46	if (*keysP == NULL) {
47		*keysP = CFArrayCreateMutable(NULL, 0, &kCFTypeArrayCallBacks);
48	}
49
50	CFArrayAppendValue(*keysP, key);
51	return;
52}
53
54
55Boolean
56SCDynamicStoreAddWatchedKey(SCDynamicStoreRef store, CFStringRef key, Boolean isRegex)
57{
58	SCDynamicStorePrivateRef	storePrivate = (SCDynamicStorePrivateRef)store;
59	kern_return_t			status;
60	CFDataRef			utfKey;		/* serialized key */
61	xmlData_t			myKeyRef;
62	CFIndex				myKeyLen;
63	int				sc_status;
64
65	if (store == NULL) {
66		/* sorry, you must provide a session */
67		_SCErrorSet(kSCStatusNoStoreSession);
68		return FALSE;
69	}
70
71	if (storePrivate->server == MACH_PORT_NULL) {
72		/* sorry, you must have an open session to play */
73		_SCErrorSet(kSCStatusNoStoreServer);
74		return FALSE;
75	}
76
77	/* serialize the key */
78	if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
79		_SCErrorSet(kSCStatusFailed);
80		return FALSE;
81	}
82
83    retry :
84
85	/* send the key to the server */
86	status = notifyadd(storePrivate->server,
87			   myKeyRef,
88			   (mach_msg_type_number_t)myKeyLen,
89			   isRegex,
90			   (int *)&sc_status);
91
92	if (__SCDynamicStoreCheckRetryAndHandleError(store,
93						     status,
94						     &sc_status,
95						     "SCDynamicStoreAddWatchedKey notifyadd()")) {
96		goto retry;
97	}
98
99	/* clean up */
100	CFRelease(utfKey);
101
102	if (sc_status != kSCStatusOK) {
103		_SCErrorSet(sc_status);
104		return FALSE;
105	}
106
107	if (isRegex) {
108		addKey(&storePrivate->patterns, key);
109	} else {
110		addKey(&storePrivate->keys, key);
111	}
112	return TRUE;
113}
114