1/*
2 * Copyright (c) 2000, 2001, 2003-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 * - initial revision
29 */
30
31#include <mach/mach.h>
32#include <mach/mach_error.h>
33
34#include <SystemConfiguration/SystemConfiguration.h>
35#include <SystemConfiguration/SCPrivate.h>
36#include "SCDynamicStoreInternal.h"
37#include "config.h"		/* MiG generated file */
38
39
40Boolean
41SCDynamicStoreSetNotificationKeys(SCDynamicStoreRef	store,
42				  CFArrayRef		keys,
43				  CFArrayRef		patterns)
44{
45	SCDynamicStorePrivateRef	storePrivate = (SCDynamicStorePrivateRef)store;
46	kern_return_t			status;
47	CFDataRef			xmlKeys		= NULL;	/* keys (XML serialized) */
48	xmlData_t			myKeysRef	= NULL;	/* keys (serialized) */
49	CFIndex				myKeysLen	= 0;
50	CFDataRef			xmlPatterns	= NULL;	/* patterns (XML serialized) */
51	xmlData_t			myPatternsRef	= NULL;	/* patterns (serialized) */
52	CFIndex				myPatternsLen	= 0;
53	int				sc_status;
54	CFMutableArrayRef		tmp;
55
56	if (store == NULL) {
57		/* sorry, you must provide a session */
58		_SCErrorSet(kSCStatusNoStoreSession);
59		return FALSE;
60	}
61
62	if (storePrivate->server == MACH_PORT_NULL) {
63		_SCErrorSet(kSCStatusNoStoreServer);
64		return FALSE;	/* you must have an open session to play */
65	}
66
67	/* serialize the keys */
68	if (keys != NULL) {
69		if (!_SCSerialize(keys, &xmlKeys, (void **)&myKeysRef, &myKeysLen)) {
70			_SCErrorSet(kSCStatusFailed);
71			return FALSE;
72		}
73	}
74
75	/* serialize the patterns */
76	if (patterns != NULL) {
77		if (!_SCSerialize(patterns, &xmlPatterns, (void **)&myPatternsRef, &myPatternsLen)) {
78			if (xmlKeys != NULL) CFRelease(xmlKeys);
79			_SCErrorSet(kSCStatusFailed);
80			return FALSE;
81		}
82	}
83
84    retry :
85
86	/* send the keys and patterns, fetch the associated result from the server */
87	status = notifyset(storePrivate->server,
88			   myKeysRef,
89			   (mach_msg_type_number_t)myKeysLen,
90			   myPatternsRef,
91			   (mach_msg_type_number_t)myPatternsLen,
92			   (int *)&sc_status);
93
94	if (__SCDynamicStoreCheckRetryAndHandleError(store,
95						     status,
96						     &sc_status,
97						     "SCDynamicStoreSetNotificationKeys notifyset()")) {
98		goto retry;
99	}
100
101	/* clean up */
102	if (xmlKeys != NULL)		CFRelease(xmlKeys);
103	if (xmlPatterns != NULL)	CFRelease(xmlPatterns);
104
105	if (sc_status != kSCStatusOK) {
106		_SCErrorSet(sc_status);
107		return FALSE;
108	}
109
110	/* in case we need to re-connect, save the keys/patterns */
111	tmp = (keys != NULL) ? CFArrayCreateMutableCopy(NULL, 0, keys) : NULL;
112	if (storePrivate->keys != NULL) CFRelease(storePrivate->keys);
113	storePrivate->keys = tmp;
114
115	tmp = (patterns != NULL) ? CFArrayCreateMutableCopy(NULL, 0, patterns) : NULL;
116	if (storePrivate->patterns != NULL) CFRelease(storePrivate->patterns);
117	storePrivate->patterns = tmp;
118
119	return TRUE;
120}
121