Fix tests

Reviewed-on: #9

core/thread.c

@@ -232,16 +232,17 @@ int threadOnJieffiesTick()
     disable_IRQs(flags);
     list_foreach(currentThread, nextThread, idx)
     {
-        if (nextThread->state == SLEEPING && nextThread->jiffiesSleeping) {
+        if (nextThread->state == SLEEPING) {
+            if (nextThread->jiffiesSleeping)
                 nextThread->jiffiesSleeping--;
-            if (!nextThread->jiffiesSleeping) {
+            if (!nextThread->jiffiesSleeping)
                 nextThread->state = READY;
         }
     }
-    }
     list_foreach_named(threadWithTimeout, nextThread, idx, timePrev, timeNext)
     {
-        if (nextThread->state == WAITING && nextThread->jiffiesSleeping) {
+        if (nextThread->state == WAITING) {
+            if (nextThread->jiffiesSleeping)
                 nextThread->jiffiesSleeping--;
             if (!nextThread->jiffiesSleeping) {
                 nextThread->sleepHaveTimeouted = 1;
@@ -332,8 +333,6 @@ int threadUsleep(unsigned long usec)
     current->state              = SLEEPING;
     current->sleepHaveTimeouted = 0;
     current->jiffiesSleeping    = usecs_to_jiffies(usec);
-    if (!current->jiffiesSleeping) // sleep at least 1 jiffies
-        current->jiffiesSleeping = 1;
     next                        = threadSelectNext();
 
     assert(next != current);

drivers/zero.c

@@ -1,14 +1,41 @@
 #include "zero.h"
 
 #include "alloc.h"
+#include "errno.h"
 #include "kernel.h"
 #include "klibc.h"
 #include "mem.h"
+#include "list.h"
+#include "paging.h"
+#include "types.h"
+
+struct zeroMappedPage {
+    uaddr_t mappedAddr;
+    paddr_t phyAddr;
+
+    struct zeroMappedPage *prev, *next;
+};
 
 struct zeroMappedEntry {
     int refCnt;
+
+    struct zeroMappedPage *listMapped;
 };
 
+static int insertMappedPage(struct zeroMappedEntry *entry, uaddr_t vAddr, paddr_t pAddr){
+    struct zeroMappedPage *info = (struct zeroMappedPage *)zalloc(sizeof(struct zeroMappedPage));
+
+    if(!info)
+        return -ENOMEM;
+
+    info->mappedAddr = vAddr;
+    info->phyAddr = pAddr;
+
+    list_add_tail(entry->listMapped, info);
+    return 0;
+}
+
+
 static int zeroOpen(struct uAddrVirtualReg *vreg)
 {
     struct zeroMappedEntry *ent = (struct zeroMappedEntry *)vreg->res->customData;
@@ -21,6 +48,11 @@ static int zeroClose(struct uAddrVirtualReg *vreg)
     struct zeroMappedEntry *ent = (struct zeroMappedEntry *)vreg->res->customData;
     ent->refCnt--;
     if (ent->refCnt == 0) {
+        struct zeroMappedPage *pageInfo;
+        list_collapse(ent->listMapped, pageInfo){
+            pageUnmap(pageInfo->mappedAddr);
+            free(pageInfo);
+        }
         free(vreg->res->customData);
         free(vreg->res);
     }
@@ -29,17 +61,26 @@ static int zeroClose(struct uAddrVirtualReg *vreg)
 
 static int zeroNoPage(struct uAddrVirtualReg *vreg, uaddr_t addr, int right)
 {
-    (void)vreg;
-
     int ret            = 0;
     paddr_t ppage      = allocPhyPage(1);
-    ret           = pageMap(ALIGN_DOWN(addr, PAGE_SIZE), ppage, right | PAGING_MEM_USER) ;
+    uaddr_t mappedAddr = ALIGN_DOWN(addr, PAGE_SIZE);
+    ret                = pageMap(mappedAddr, ppage, right | PAGING_MEM_USER);
+
+    struct zeroMappedEntry *ent = (struct zeroMappedEntry *)vreg->res->customData;
 
     unrefPhyPage(ppage);
     if (ret) {
         return ret;
     }
-    memset((void *)ALIGN_DOWN(addr, PAGE_SIZE), 0, PAGE_SIZE);
+
+    ret = insertMappedPage(ent, mappedAddr, ppage);
+    if (ret) {
+        pageUnmap(mappedAddr);
+        return ret;
+    }
+
+    memset((void *)mappedAddr, 0, PAGE_SIZE);
+
     return ret;
 }
 
@@ -65,6 +106,8 @@ int zeroMmap(struct uAddrSpace *as, uaddr_t *uaddr, size_t size, uint32_t rights
     struct zeroMappedEntry *cust =
         (struct zeroMappedEntry *)zalloc(sizeof(struct zeroMappedEntry));
 
+    list_init(cust->listMapped);
+
     res->allowedRight = PAGING_MEM_READ | PAGING_MEM_WRITE | PAGING_MEM_EXEC | PAGING_MEM_USER;
     res->ops          = &zeroOps;
     res->customData   = cust;

tests/test.c

@@ -306,9 +306,8 @@ void sleepThread(void *arg)
     }
     unsigned long ellapsedTime = jiffies_to_msecs(jiffies - initialJiffies);
     assertmsg(ellapsedTime >= 500 && ellapsedTime < 510, "ellapsedTime %lu\n", ellapsedTime);
-    threadMsleep(0);
-    printf("I should never be showed\n");
-    assert(1);
+    threadMsleep(ULONG_MAX);
+    assert(0);
 }
 
 struct mutex mutexTest;