1 files changed, 94 insertions, 125 deletions

diff --git a/src/sbservices.c b/src/sbservices.c
index 3596cbd..365e130 100644
--- a/src/sbservices.c
+++ b/src/sbservices.c
@@ -8,17 +8,20 @@
  * 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 
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  */
 
+#ifdef HAVE_CONFIG_H
+#include <config.h>
+#endif
 #include <string.h>
 #include <stdlib.h>
 #include <unistd.h>
@@ -26,28 +29,28 @@
 
 #include "sbservices.h"
 #include "property_list_service.h"
-#include "debug.h"
+#include "common/debug.h"
 
 /**
  * Locks an sbservices client, used for thread safety.
  *
  * @param client sbservices client to lock.
  */
-static void sbs_lock(sbservices_client_t client)
+static void sbservices_lock(sbservices_client_t client)
 {
-        debug_info("SBServices: Locked");
-        g_mutex_lock(client->mutex);
+        debug_info("Locked");
+        mutex_lock(&client->mutex);
 }
 
 /**
  * Unlocks an sbservices client, used for thread safety.
- * 
+ *
  * @param client sbservices client to unlock
  */
-static void sbs_unlock(sbservices_client_t client)
+static void sbservices_unlock(sbservices_client_t client)
 {
-        debug_info("SBServices: Unlocked");
-        g_mutex_unlock(client->mutex);
+        debug_info("Unlocked");
+        mutex_unlock(&client->mutex);
 }
 
 /**
@@ -61,64 +64,44 @@ static void sbs_unlock(sbservices_client_t client)
  */
 static sbservices_error_t sbservices_error(property_list_service_error_t err)
 {
-       switch (err) {
-                case PROPERTY_LIST_SERVICE_E_SUCCESS:
-                        return SBSERVICES_E_SUCCESS;
-                case PROPERTY_LIST_SERVICE_E_INVALID_ARG:
-                        return SBSERVICES_E_INVALID_ARG;
-                case PROPERTY_LIST_SERVICE_E_PLIST_ERROR:
-                        return SBSERVICES_E_PLIST_ERROR;
-                case PROPERTY_LIST_SERVICE_E_MUX_ERROR:
-                        return SBSERVICES_E_CONN_FAILED;
-                default:
-                        break;
-        }
-        return SBSERVICES_E_UNKNOWN_ERROR;
+        switch (err) {
+                case PROPERTY_LIST_SERVICE_E_SUCCESS:
+                        return SBSERVICES_E_SUCCESS;
+                case PROPERTY_LIST_SERVICE_E_INVALID_ARG:
+                        return SBSERVICES_E_INVALID_ARG;
+                case PROPERTY_LIST_SERVICE_E_PLIST_ERROR:
+                        return SBSERVICES_E_PLIST_ERROR;
+                case PROPERTY_LIST_SERVICE_E_MUX_ERROR:
+                        return SBSERVICES_E_CONN_FAILED;
+                default:
+                        break;
+        }
+        return SBSERVICES_E_UNKNOWN_ERROR;
 }
 
-/**
- * Connects to the springboardservices service on the specified device.
- *
- * @param device The device to connect to.
- * @param port Destination port (usually given by lockdownd_start_service).
- * @param client Pointer that will point to a newly allocated
- *     sbservices_client_t upon successful return.
- *
- * @return SBSERVICES_E_SUCCESS on success, SBSERVICES_E_INVALID_ARG when
- *     client is NULL, or an SBSERVICES_E_* error code otherwise.
- */
-sbservices_error_t sbservices_client_new(idevice_t device, uint16_t port, sbservices_client_t *client)
+sbservices_error_t sbservices_client_new(idevice_t device, lockdownd_service_descriptor_t service, sbservices_client_t *client)
 {
-        /* makes sure thread environment is available */
-        if (!g_thread_supported())
-                g_thread_init(NULL);
-
-        if (!device)
-                return SBSERVICES_E_INVALID_ARG;
-
         property_list_service_client_t plistclient = NULL;
-        sbservices_error_t err = sbservices_error(property_list_service_client_new(device, port, &plistclient));
+        sbservices_error_t err = sbservices_error(property_list_service_client_new(device, service, &plistclient));
         if (err != SBSERVICES_E_SUCCESS) {
                 return err;
         }
 
         sbservices_client_t client_loc = (sbservices_client_t) malloc(sizeof(struct sbservices_client_private));
         client_loc->parent = plistclient;
-        client_loc->mutex = g_mutex_new();
+        mutex_init(&client_loc->mutex);
 
         *client = client_loc;
         return SBSERVICES_E_SUCCESS;
 }
 
-/**
- * Disconnects an sbservices client from the device and frees up the
- * sbservices client data.
- *
- * @param client The sbservices client to disconnect and free.
- *
- * @return SBSERVICES_E_SUCCESS on success, SBSERVICES_E_INVALID_ARG when
- *     client is NULL, or an SBSERVICES_E_* error code otherwise.
- */
+sbservices_error_t sbservices_client_start_service(idevice_t device, sbservices_client_t * client, const char* label)
+{
+        sbservices_error_t err = SBSERVICES_E_UNKNOWN_ERROR;
+        service_client_factory_start_service(device, SBSERVICES_SERVICE_NAME, (void**)client, label, SERVICE_CONSTRUCTOR(sbservices_client_new), &err);
+        return err;
+}
+
 sbservices_error_t sbservices_client_free(sbservices_client_t client)
 {
         if (!client)
@@ -126,28 +109,12 @@ sbservices_error_t sbservices_client_free(sbservices_client_t client)
 
         sbservices_error_t err = sbservices_error(property_list_service_client_free(client->parent));
         client->parent = NULL;
-        if (client->mutex) {
-                g_mutex_free(client->mutex);
-        }
+        mutex_destroy(&client->mutex);
         free(client);
 
         return err;
 }
 
-/**
- * Gets the icon state of the connected device.
- *
- * @param client The connected sbservices client to use.
- * @param state Pointer that will point to a newly allocated plist containing
- *     the current icon state. It is up to the caller to free the memory.
- * @param format_version A string to be passed as formatVersion along with
- *     the request, or NULL if no formatVersion should be passed. This is only
- *     supported since iOS 4.0 so for older firmware versions this must be set
- *     to NULL.
- *
- * @return SBSERVICES_E_SUCCESS on success, SBSERVICES_E_INVALID_ARG when
- *     client or state is invalid, or an SBSERVICES_E_* error code otherwise.
- */
 sbservices_error_t sbservices_get_icon_state(sbservices_client_t client, plist_t *state, const char *format_version)
 {
         if (!client || !client->parent || !state)
@@ -156,12 +123,12 @@ sbservices_error_t sbservices_get_icon_state(sbservices_client_t client, plist_t
         sbservices_error_t res = SBSERVICES_E_UNKNOWN_ERROR;
 
         plist_t dict = plist_new_dict();
-        plist_dict_insert_item(dict, "command", plist_new_string("getIconState"));
+        plist_dict_set_item(dict, "command", plist_new_string("getIconState"));
         if (format_version) {
-                plist_dict_insert_item(dict, "formatVersion", plist_new_string(format_version));
+                plist_dict_set_item(dict, "formatVersion", plist_new_string(format_version));
         }
 
-        sbs_lock(client);
+        sbservices_lock(client);
 
         res = sbservices_error(property_list_service_send_binary_plist(client->parent, dict));
         if (res != SBSERVICES_E_SUCCESS) {
@@ -184,19 +151,10 @@ leave_unlock:
         if (dict) {
                 plist_free(dict);
         }
-        sbs_unlock(client);
+        sbservices_unlock(client);
         return res;
 }
 
-/**
- * Sets the icon state of the connected device.
- *
- * @param client The connected sbservices client to use.
- * @param newstate A plist containing the new iconstate.
- *
- * @return SBSERVICES_E_SUCCESS on success, SBSERVICES_E_INVALID_ARG when
- *     client or newstate is NULL, or an SBSERVICES_E_* error code otherwise.
- */
 sbservices_error_t sbservices_set_icon_state(sbservices_client_t client, plist_t newstate)
 {
         if (!client || !client->parent || !newstate)
@@ -205,39 +163,27 @@ sbservices_error_t sbservices_set_icon_state(sbservices_client_t client, plist_t
         sbservices_error_t res = SBSERVICES_E_UNKNOWN_ERROR;
 
         plist_t dict = plist_new_dict();
-        plist_dict_insert_item(dict, "command", plist_new_string("setIconState"));
-        plist_dict_insert_item(dict, "iconState", plist_copy(newstate));
+        plist_dict_set_item(dict, "command", plist_new_string("setIconState"));
+        plist_dict_set_item(dict, "iconState", plist_copy(newstate));
 
-        sbs_lock(client);
+        sbservices_lock(client);
 
         res = sbservices_error(property_list_service_send_binary_plist(client->parent, dict));
         if (res != SBSERVICES_E_SUCCESS) {
                 debug_info("could not send plist, error %d", res);
         }
-        /* NO RESPONSE */
+
+        uint32_t bytes = 0;
+        service_receive_with_timeout(client->parent->parent, malloc(4), 4, &bytes, 2000);
+        debug_info("setIconState response: %u", bytes);
 
         if (dict) {
                 plist_free(dict);
         }
-        sbs_unlock(client);
+        sbservices_unlock(client);
         return res;
 }
 
-/**
- * Get the icon of the specified app as PNG data.
- *
- * @param client The connected sbservices client to use.
- * @param bundleId The bundle identifier of the app to retrieve the icon for.
- * @param pngdata Pointer that will point to a newly allocated buffer
- *     containing the PNG data upon successful return. It is up to the caller
- *     to free the memory.
- * @param pngsize Pointer to a uint64_t that will be set to the size of the
- *     buffer pngdata points to upon successful return.
- *
- * @return SBSERVICES_E_SUCCESS on success, SBSERVICES_E_INVALID_ARG when
- *     client, bundleId, or pngdata are invalid, or an SBSERVICES_E_* error
- *     code otherwise.
- */
 sbservices_error_t sbservices_get_icon_pngdata(sbservices_client_t client, const char *bundleId, char **pngdata, uint64_t *pngsize)
 {
         if (!client || !client->parent || !bundleId || !pngdata)
@@ -246,10 +192,10 @@ sbservices_error_t sbservices_get_icon_pngdata(sbservices_client_t client, const
         sbservices_error_t res = SBSERVICES_E_UNKNOWN_ERROR;
 
         plist_t dict = plist_new_dict();
-        plist_dict_insert_item(dict, "command", plist_new_string("getIconPNGData"));
-        plist_dict_insert_item(dict, "bundleId", plist_new_string(bundleId));
+        plist_dict_set_item(dict, "command", plist_new_string("getIconPNGData"));
+        plist_dict_set_item(dict, "bundleId", plist_new_string(bundleId));
 
-        sbs_lock(client);
+        sbservices_lock(client);
 
         res = sbservices_error(property_list_service_send_binary_plist(client->parent, dict));
         if (res != SBSERVICES_E_SUCCESS) {
@@ -271,25 +217,48 @@ leave_unlock:
         if (dict) {
                 plist_free(dict);
         }
-        sbs_unlock(client);
+        sbservices_unlock(client);
         return res;
+}
+
+sbservices_error_t sbservices_get_interface_orientation(sbservices_client_t client, sbservices_interface_orientation_t* interface_orientation)
+{
+        if (!client || !client->parent || !interface_orientation)
+                return SBSERVICES_E_INVALID_ARG;
+
+        sbservices_error_t res = SBSERVICES_E_UNKNOWN_ERROR;
+
+        plist_t dict = plist_new_dict();
+        plist_dict_set_item(dict, "command", plist_new_string("getInterfaceOrientation"));
+
+        sbservices_lock(client);
+
+        res = sbservices_error(property_list_service_send_binary_plist(client->parent, dict));
+        if (res != SBSERVICES_E_SUCCESS) {
+                debug_info("could not send plist, error %d", res);
+                goto leave_unlock;
+        }
+        plist_free(dict);
+        dict = NULL;
+
+        res = sbservices_error(property_list_service_receive_plist(client->parent, &dict));
+        if (res == SBSERVICES_E_SUCCESS) {
+                plist_t node = plist_dict_get_item(dict, "interfaceOrientation");
+                if (node) {
+                        uint64_t value = SBSERVICES_INTERFACE_ORIENTATION_UNKNOWN;
+                        plist_get_uint_val(node, &value);
+                        *interface_orientation = (sbservices_interface_orientation_t)value;
+                }
+        }
 
+leave_unlock:
+        if (dict) {
+                plist_free(dict);
+        }
+        sbservices_unlock(client);
+        return res;
 }
 
-/**
- * Get the home screen wallpaper as PNG data.
- *
- * @param client The connected sbservices client to use.
- * @param pngdata Pointer that will point to a newly allocated buffer
- *     containing the PNG data upon successful return. It is up to the caller
- *     to free the memory.
- * @param pngsize Pointer to a uint64_t that will be set to the size of the
- *     buffer pngdata points to upon successful return.
- *
- * @return SBSERVICES_E_SUCCESS on success, SBSERVICES_E_INVALID_ARG when
- *     client or pngdata are invalid, or an SBSERVICES_E_* error
- *     code otherwise.
- */
 sbservices_error_t sbservices_get_home_screen_wallpaper_pngdata(sbservices_client_t client, char **pngdata, uint64_t *pngsize)
 {
         if (!client || !client->parent || !pngdata)
@@ -298,9 +267,9 @@ sbservices_error_t sbservices_get_home_screen_wallpaper_pngdata(sbservices_clien
         sbservices_error_t res = SBSERVICES_E_UNKNOWN_ERROR;
 
         plist_t dict = plist_new_dict();
-        plist_dict_insert_item(dict, "command", plist_new_string("getHomeScreenWallpaperPNGData"));
+        plist_dict_set_item(dict, "command", plist_new_string("getHomeScreenWallpaperPNGData"));
 
-        sbs_lock(client);
+        sbservices_lock(client);
 
         res = sbservices_error(property_list_service_send_binary_plist(client->parent, dict));
         if (res != SBSERVICES_E_SUCCESS) {
@@ -322,6 +291,6 @@ leave_unlock:
         if (dict) {
                 plist_free(dict);
         }
-        sbs_unlock(client);
+        sbservices_unlock(client);
         return res;
 }