Merge branch 'master' of https://github.com/switchbrew/libnx

MrKcyre · Feb 21, 2018 · f8a86bf · f8a86bf
2 parents e25aae5 + d61154f
commit f8a86bf
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 10 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/runtime/newlib.c b/nx/source/runtime/newlib.c
@@ -28,6 +28,8 @@ static struct _reent* __libnx_get_reent(void) {
     return tv->reent;
 }
 
+//TODO: timeGetCurrentTime() returns UTC time. How to handle timezones?
+
 int __libnx_gtod(struct _reent *ptr, struct timeval *tp, struct timezone *tz) {
     if (tp != NULL) {
         u64 now=0;
@@ -40,10 +42,10 @@ int __libnx_gtod(struct _reent *ptr, struct timeval *tp, struct timezone *tz) {
         }
 
         tp->tv_sec =  now;
-        tp->tv_usec = 0;//timeGetCurrentTime() only returns seconds.
+        tp->tv_usec = now*1000000;//timeGetCurrentTime() only returns seconds.
     }
 
-    if (tz != NULL) {//TODO: This needs handled properly, timeGetCurrentTime() returns UTC time.
+    if (tz != NULL) {
         tz->tz_minuteswest = 0;
         tz->tz_dsttime = 0;
     }

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;
+}
+