Added timeSetCurrentTime() and attempt to initialize time-serv with '…

…time:s' first.
MrKcyre · Feb 21, 2018 · d61154f · d61154f
1 parent 0dcaeec
commit d61154f
Show file tree

Hide file tree

Showing 2 changed files with 63 additions and 8 deletions.
diff --git a/nx/include/switch/services/time.h b/nx/include/switch/services/time.h
@@ -9,6 +9,7 @@
 #include "../types.h"
 #include "../services/sm.h"
 
+/// Time clock type.
 typedef enum {
     TimeType_UserSystemClock,
     TimeType_NetworkSystemClock,
@@ -22,3 +23,11 @@ void timeExit(void);
 Service* timeGetSessionService(void);
 
 Result timeGetCurrentTime(TimeType type, u64 *timestamp);
+
+/**
+ * @brief Sets the time for the specified clock.
+ * @param[in] type Clock to use.
+ * @param[in] timestamp POSIX UTC timestamp.
+ * @return Result code.
+ */
+Result timeSetCurrentTime(TimeType type, u64 timestamp);
diff --git a/nx/source/services/time.c b/nx/source/services/time.c
@@ -20,7 +20,10 @@ Result timeInitialize(void)
 
     Result rc;
 
-    rc = smGetService(&g_timeSrv, "time:u");
+    rc = smGetService(&g_timeSrv, "time:s");
+    if (R_FAILED(rc))
+        rc = smGetService(&g_timeSrv, "time:u");
+
     if (R_FAILED(rc))
         return rc;
 
@@ -92,21 +95,26 @@ static Result _timeGetSession(Service* srv_out, u64 cmd_id) {
     return rc;
 }
 
-Result timeGetCurrentTime(TimeType type, u64 *timestamp) {
-    Service *srv = NULL;
-
+static Service* _timeGetClockSession(TimeType type) {
     if (type==TimeType_UserSystemClock) {
-        srv = &g_timeUserSystemClock;
+        return &g_timeUserSystemClock;
     }
     else if (type==TimeType_NetworkSystemClock) {
-        srv = &g_timeNetworkSystemClock;
+        return &g_timeNetworkSystemClock;
     }
     else if (type==TimeType_LocalSystemClock) {
-        srv = &g_timeLocalSystemClock;
+        return &g_timeLocalSystemClock;
     }
     else {
-        return MAKERESULT(Module_Libnx, LibnxError_BadInput);
+        return NULL;
     }
+}
+
+Result timeGetCurrentTime(TimeType type, u64 *timestamp) {
+    Service *srv = _timeGetClockSession(type);
+
+    if (srv==NULL)
+        return MAKERESULT(Module_Libnx, LibnxError_BadInput);
 
     IpcCommand c;
     ipcInitialize(&c);
@@ -141,3 +149,41 @@ Result timeGetCurrentTime(TimeType type, u64 *timestamp) {
     return rc;
 }
 
+Result timeSetCurrentTime(TimeType type, u64 timestamp) {
+    Service *srv = _timeGetClockSession(type);
+
+    if (srv==NULL)
+        return MAKERESULT(Module_Libnx, LibnxError_BadInput);
+
+    IpcCommand c;
+    ipcInitialize(&c);
+
+    struct {
+        u64 magic;
+        u64 cmd_id;
+        u64 timestamp;
+    } *raw;
+
+    raw = ipcPrepareHeader(&c, sizeof(*raw));
+
+    raw->magic = SFCI_MAGIC;
+    raw->cmd_id = 1;
+    raw->timestamp = timestamp;
+
+    Result rc = serviceIpcDispatch(srv);
+
+    if (R_SUCCEEDED(rc)) {
+        IpcParsedCommand r;
+        ipcParse(&r);
+
+        struct {
+            u64 magic;
+            u64 result;
+        } *resp = r.Raw;
+
+        rc = resp->result;
+    }
+
+    return rc;
+}
+