Add forgotten file.

1 files changed, 256 insertions, 0 deletions

diff --git a/src/NotificationProxy.c b/src/NotificationProxy.c
new file mode 100644
index 0000000..b0c10ea
--- /dev/null
+++ b/src/NotificationProxy.c
@@ -0,0 +1,256 @@
+/*
+ * NotificationProxy.c
+ * Notification Proxy implementation.
+ *
+ * Copyright (c) 2009 Nikias Bassen, All Rights Reserved.
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ * 
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ * 
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA 
+ */
+
+#include <stdio.h>
+#include "NotificationProxy.h"
+#include "plist.h"
+#include "utils.h"
+
+/** Locks an NP client, done for thread safety stuff.
+ *
+ * @param client The NP
+ */
+static void np_lock(iphone_np_client_t client)
+{
+	log_debug_msg("NP: Locked\n");
+	g_mutex_lock(client->mutex);
+}
+
+/** Unlocks an NP client, done for thread safety stuff.
+ * 
+ * @param client The NP
+ */
+static void np_unlock(iphone_np_client_t client)
+{
+	log_debug_msg("NP: Unlocked\n");
+	g_mutex_unlock(client->mutex);
+}
+
+/** Makes a connection to the NP service on the phone. 
+ * 
+ * @param phone The iPhone to connect on.
+ * @param s_port The source port. 
+ * @param d_port The destination port. 
+ * 
+ * @return A handle to the newly-connected client or NULL upon error.
+ */
+iphone_error_t iphone_np_new_client ( iphone_device_t device, int src_port, int dst_port, iphone_np_client_t *client )
+{
+	int ret = IPHONE_E_SUCCESS;
+
+	//makes sure thread environment is available
+	if (!g_thread_supported())
+		g_thread_init(NULL);
+	iphone_np_client_t client_loc = (iphone_np_client_t) malloc(sizeof(struct iphone_np_client_int));
+
+	if (!device)
+		return IPHONE_E_INVALID_ARG;
+
+	// Attempt connection
+	client_loc->connection = NULL;
+	ret = iphone_mux_new_client(device, src_port, dst_port, &client_loc->connection);
+	if (IPHONE_E_SUCCESS != ret || !client_loc->connection) {
+		free(client_loc);
+		return ret;
+	}
+
+	client_loc->mutex = g_mutex_new();
+
+	*client = client_loc;
+	return IPHONE_E_SUCCESS;	
+}
+
+/** Disconnects an NP client from the phone.
+ * 
+ * @param client The client to disconnect.
+ */
+iphone_error_t iphone_np_free_client ( iphone_np_client_t client )
+{
+	if (!client || !client->connection )
+		return IPHONE_E_INVALID_ARG;
+
+	iphone_mux_free_client(client->connection);
+	free(client);
+	return IPHONE_E_SUCCESS;
+}
+
+/** Sends a notification to the NP client.
+ *
+ * notification messages seen so far:
+ *   com.apple.itunes-mobdev.syncWillStart
+ *   com.apple.itunes-mobdev.syncDidStart
+ *
+ * @param client The client to send to
+ * @param notification The notification Message
+ */
+iphone_error_t iphone_np_post_notification( iphone_np_client_t client, const char *notification )
+{
+	xmlDocPtr plist;
+	xmlNode *dict, *key;
+    	char *XML_content;
+	uint32_t length;
+	int bytes;
+	iphone_error_t ret;
+	unsigned char sndbuf[4096];
+	int sndlen = 0;
+	int nlen;
+
+	if (!client || !notification) {
+		return IPHONE_E_INVALID_ARG;
+	}
+	np_lock(client);
+
+	plist = new_plist();
+	dict = add_child_to_plist(plist, "dict", "\n", NULL, 0);
+	key = add_key_str_dict_element(plist, dict, "Command", "PostNotification", 1);
+	key = add_key_str_dict_element(plist, dict, "Name", notification, 1);
+	xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length);
+
+	nlen = htonl(length);
+
+	memcpy(sndbuf+sndlen, &nlen, 4);
+	sndlen += 4;
+	memcpy(sndbuf+sndlen, XML_content, length);
+	sndlen += length;
+ 
+	xmlFree(XML_content);
+	xmlFreeDoc(plist);
+
+	plist = new_plist();
+	dict = add_child_to_plist(plist, "dict", "\n", NULL, 0);
+	key = add_key_str_dict_element(plist, dict, "Command", "Shutdown", 1);
+	xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length);
+
+	nlen = htonl(length);
+
+	memcpy(sndbuf+sndlen, &nlen, 4);
+	sndlen+=4;
+
+	memcpy(sndbuf+sndlen, XML_content, length);
+	sndlen+=length;
+
+	xmlFree(XML_content);
+	xmlFreeDoc(plist);
+	plist = NULL;
+
+	log_debug_buffer(sndbuf, sndlen);
+
+	iphone_mux_send(client->connection, sndbuf, sndlen, &bytes);
+        if (bytes <= 0) {
+		np_unlock(client);
+		return bytes;
+	}
+
+	np_unlock(client);
+	return bytes;
+}
+
+/** Notifies the iphone to send a notification on certain events.
+ *
+ * observation messages seen so far:
+ *   com.apple.itunes-client.syncCancelRequest
+ *   com.apple.itunes-client.syncSuspendRequest
+ *   com.apple.itunes-client.syncResumeRequest
+ *   com.apple.mobile.lockdown.phone_number_changed
+ *   com.apple.mobile.lockdown.device_name_changed
+ *   com.apple.springboard.attemptactivation
+ *   com.apple.mobile.data_sync.domain_changed
+ *   com.apple.mobile.application_installed
+ *   com.apple.mobile.application_uninstalled
+ *
+ * @param client The client to send to
+ */
+iphone_error_t iphone_np_observe_notification( iphone_np_client_t client )
+{
+	xmlDocPtr plist;
+	xmlNode *dict, *key;
+    	char *XML_content;
+	uint32_t length;
+	int bytes;
+	iphone_error_t ret;
+	unsigned char sndbuf[4096];
+	int sndlen = 0;
+	int nlen;
+	int i=0;
+	char *notifications[10] = {
+	    "com.apple.itunes-client.syncCancelRequest",
+	    "com.apple.itunes-client.syncSuspendRequest",
+	    "com.apple.itunes-client.syncResumeRequest",
+	    "com.apple.mobile.lockdown.phone_number_changed",
+	    "com.apple.mobile.lockdown.device_name_changed",
+	    "com.apple.springboard.attemptactivation",
+	    "com.apple.mobile.data_sync.domain_changed",
+	    "com.apple.mobile.application_installed",
+	    "com.apple.mobile.application_uninstalled",
+	    NULL};
+
+	sndlen = 0;
+
+	if (!client) {
+		return IPHONE_E_INVALID_ARG;
+	}
+	np_lock(client);
+
+	while (notifications[i]) {
+		plist = new_plist();
+		dict = add_child_to_plist(plist, "dict", "\n", NULL, 0);
+		key = add_key_str_dict_element(plist, dict, "Command", "ObserveNotification", 1);
+		key = add_key_str_dict_element(plist, dict, "Name", notifications[i++], 1);
+		xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length);
+
+		nlen = htonl(length);
+		memcpy(sndbuf+sndlen, &nlen, 4);
+		sndlen += 4;
+		memcpy(sndbuf+sndlen, XML_content, length);
+		sndlen += length;
+ 
+		xmlFree(XML_content);
+		xmlFreeDoc(plist);
+	}
+
+	plist = new_plist();
+	dict = add_child_to_plist(plist, "dict", "\n", NULL, 0);
+	key = add_key_str_dict_element(plist, dict, "Command", "Shutdown", 1);
+	xmlDocDumpMemory(plist, (xmlChar **) & XML_content, &length);
+
+	nlen = htonl(length);
+
+	memcpy(sndbuf+sndlen, &nlen, 4);
+	sndlen+=4;
+
+	memcpy(sndbuf+sndlen, XML_content, length);
+	sndlen+=length;
+
+	xmlFree(XML_content);
+	xmlFreeDoc(plist);
+	plist = NULL;
+
+	log_debug_buffer(sndbuf, sndlen);
+
+	iphone_mux_send(client->connection, sndbuf, sndlen, &bytes);
+        if (bytes <= 0) {
+		np_unlock(client);
+		return bytes;
+	}
+
+	np_unlock(client);
+	return bytes;
+}