git://scm.lunaixsky.com
/
lunaix-os.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Implement APIC, RTC, basic ACPI parser and timer support
[lunaix-os.git]
/
lunaix-os
/
kernel
/
mm
/
kalloc.c
diff --git
a/lunaix-os/kernel/mm/kalloc.c
b/lunaix-os/kernel/mm/kalloc.c
index 675a51f4033fe0191a5fba16c46e0f558e22f40e..6f04e2edebcabe6a2497e00b5288e4e1509af0bb 100644
(file)
--- a/
lunaix-os/kernel/mm/kalloc.c
+++ b/
lunaix-os/kernel/mm/kalloc.c
@@
-17,7
+17,7
@@
#include <lunaix/constants.h>
#include <lunaix/spike.h>
#include <lunaix/constants.h>
#include <lunaix/spike.h>
-#include <libc/string.h>
+#include <
k
libc/string.h>
#include <stdint.h>
#include <stdint.h>
@@
-82,13
+82,20
@@
lxmalloc(size_t size) {
}
void*
}
void*
-lxcalloc(size_t size) {
- void* ptr = lxmalloc(size);
+lxcalloc(size_t n, size_t elem) {
+ size_t pd = n * elem;
+
+ // overflow detection
+ if (pd < elem || pd < n) {
+ return NULL;
+ }
+
+ void* ptr = lxmalloc(pd);
if (!ptr) {
return NULL;
}
if (!ptr) {
return NULL;
}
- return memset(ptr, 0,
size
);
+ return memset(ptr, 0,
pd
);
}
void
}
void