Boolean
SCDynamicStoreAddWatchedKey(SCDynamicStoreRef store, CFStringRef key, Boolean isRegex)
{
SCDynamicStorePrivateRef storePrivate = (SCDynamicStorePrivateRef)store;
kern_return_t status;
CFDataRef utfKey; /* serialized key */
xmlData_t myKeyRef;
CFIndex myKeyLen;
int sc_status;
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;
}
/* serialize the key */
if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
_SCErrorSet(kSCStatusFailed);
return FALSE;
}
retry :
/* send the key to the server */
status = notifyadd(storePrivate->server,
myKeyRef,
myKeyLen,
isRegex,
(int *)&sc_status);
if (__SCDynamicStoreCheckRetryAndHandleError(store,
status,
&sc_status,
"SCDynamicStoreAddWatchedKey notifyadd()")) {
goto retry;
}
/* clean up */
CFRelease(utfKey);
if (sc_status != kSCStatusOK) {
_SCErrorSet(sc_status);
return FALSE;
}
if (isRegex) {
addKey(&storePrivate->patterns, key);
} else {
addKey(&storePrivate->keys, key);
}
return TRUE;
}
Boolean
SCDynamicStoreRemoveValue(SCDynamicStoreRef store, CFStringRef key)
{
SCDynamicStorePrivateRef storePrivate;
kern_return_t status;
CFDataRef utfKey; /* serialized key */
xmlData_t myKeyRef;
CFIndex myKeyLen;
int sc_status;
if (store == NULL) {
store = __SCDynamicStoreNullSession();
if (store == NULL) {
/* sorry, you must provide a session */
_SCErrorSet(kSCStatusNoStoreSession);
return FALSE;
}
}
storePrivate = (SCDynamicStorePrivateRef)store;
if (storePrivate->server == MACH_PORT_NULL) {
/* sorry, you must have an open session to play */
_SCErrorSet(kSCStatusNoStoreServer);
return FALSE;
}
/* serialize the key */
if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
_SCErrorSet(kSCStatusFailed);
return FALSE;
}
retry :
/* send the key to the server */
status = configremove(storePrivate->server,
myKeyRef,
myKeyLen,
(int *)&sc_status);
if (__SCDynamicStoreCheckRetryAndHandleError(store,
status,
&sc_status,
"SCDynamicStoreRemoveValue configremove()")) {
goto retry;
}
/* clean up */
CFRelease(utfKey);
if (sc_status != kSCStatusOK) {
_SCErrorSet(sc_status);
return FALSE;
}
return TRUE;
}
static Boolean
__SCDynamicStoreAddSession(SCDynamicStorePrivateRef storePrivate)
{
kern_return_t kr = KERN_SUCCESS;
CFDataRef myName; /* serialized name */
xmlData_t myNameRef;
CFIndex myNameLen;
CFDataRef myOptions = NULL; /* serialized options */
xmlData_t myOptionsRef = NULL;
CFIndex myOptionsLen = 0;
int sc_status = kSCStatusFailed;
mach_port_t server;
if (!_SCSerializeString(storePrivate->name, &myName, (void **)&myNameRef, &myNameLen)) {
goto done;
}
/* serialize the options */
if (storePrivate->options != NULL) {
if (!_SCSerialize(storePrivate->options, &myOptions, (void **)&myOptionsRef, &myOptionsLen)) {
CFRelease(myName);
goto done;
}
}
/* open a new session with the server */
server = MACH_PORT_NULL;
updateServerPort(storePrivate, &server, &sc_status);
while (server != MACH_PORT_NULL) {
// if SCDynamicStore server available
if (!storePrivate->serverNullSession) {
// if SCDynamicStore session
kr = configopen(server,
myNameRef,
(mach_msg_type_number_t)myNameLen,
myOptionsRef,
(mach_msg_type_number_t)myOptionsLen,
&storePrivate->server,
(int *)&sc_status);
} else {
// if NULL session
if (storePrivate->server == MACH_PORT_NULL) {
// use the [main] SCDynamicStore server port
kr = mach_port_mod_refs(mach_task_self(), server, MACH_PORT_RIGHT_SEND, +1);
if (kr == KERN_SUCCESS) {
storePrivate->server = server;
sc_status = kSCStatusOK;
} else {
storePrivate->server = MACH_PORT_NULL;
}
} else {
// if the server port we used returned an error
storePrivate->server = MACH_PORT_NULL;
kr = MACH_SEND_INVALID_DEST;
}
}
if (kr == KERN_SUCCESS) {
break;
}
// our [cached] server port is not valid
if ((kr != MACH_SEND_INVALID_DEST) && (kr != MIG_SERVER_DIED)) {
// if we got an unexpected error, don't retry
sc_status = kr;
break;
}
updateServerPort(storePrivate, &server, &sc_status);
}
__MACH_PORT_DEBUG(TRUE, "*** SCDynamicStoreAddSession", storePrivate->server);
// clean up
CFRelease(myName);
if (myOptions != NULL) CFRelease(myOptions);
done :
switch (sc_status) {
case kSCStatusOK :
return TRUE;
case BOOTSTRAP_UNKNOWN_SERVICE :
SCLog(TRUE,
(kr == KERN_SUCCESS) ? LOG_DEBUG : LOG_ERR,
CFSTR("SCDynamicStore server not available"));
sc_status = kSCStatusNoStoreServer;
break;
default :
SCLog(TRUE,
(kr == KERN_SUCCESS) ? LOG_DEBUG : LOG_ERR,
CFSTR("SCDynamicStoreAddSession configopen(): %s"),
SCErrorString(sc_status));
break;
}
_SCErrorSet(sc_status);
return FALSE;
}
Boolean
SCDynamicStoreSetValue(SCDynamicStoreRef store, CFStringRef key, CFPropertyListRef value)
{
SCDynamicStorePrivateRef storePrivate;
kern_return_t status;
CFDataRef utfKey; /* serialized key */
xmlData_t myKeyRef;
CFIndex myKeyLen;
CFDataRef xmlData; /* serialized data */
xmlData_t myDataRef;
CFIndex myDataLen;
int sc_status;
int newInstance;
if (store == NULL) {
store = __SCDynamicStoreNullSession();
if (store == NULL) {
/* sorry, you must provide a session */
_SCErrorSet(kSCStatusNoStoreSession);
return FALSE;
}
}
storePrivate = (SCDynamicStorePrivateRef)store;
if (storePrivate->server == MACH_PORT_NULL) {
/* sorry, you must have an open session to play */
_SCErrorSet(kSCStatusNoStoreServer);
return FALSE;
}
/* serialize the key */
if (!_SCSerializeString(key, &utfKey, (void **)&myKeyRef, &myKeyLen)) {
_SCErrorSet(kSCStatusInvalidArgument);
return FALSE;
}
/* serialize the data */
if (!_SCSerialize(value, &xmlData, (void **)&myDataRef, &myDataLen)) {
CFRelease(utfKey);
_SCErrorSet(kSCStatusInvalidArgument);
return FALSE;
}
retry :
/* send the key & data to the server, get new instance id */
status = configset(storePrivate->server,
myKeyRef,
myKeyLen,
myDataRef,
myDataLen,
0,
&newInstance,
(int *)&sc_status);
if (__SCDynamicStoreCheckRetryAndHandleError(store,
status,
&sc_status,
"SCDynamicStoreSetValue configset()")) {
goto retry;
}
/* clean up */
CFRelease(utfKey);
CFRelease(xmlData);
if (sc_status != kSCStatusOK) {
_SCErrorSet(sc_status);
return FALSE;
}
return TRUE;
}
