clang-format
This commit is contained in:
@@ -54,95 +54,85 @@
|
||||
* threads at once.
|
||||
*/
|
||||
|
||||
#define NTRIES 1200
|
||||
#define ITEMSIZE 997
|
||||
#define NTHREADS 8
|
||||
#define NTRIES 1200
|
||||
#define ITEMSIZE 997
|
||||
#define NTHREADS 8
|
||||
|
||||
static
|
||||
void
|
||||
kmallocthread(void *sm, unsigned long num)
|
||||
{
|
||||
struct semaphore *sem = sm;
|
||||
void *ptr;
|
||||
void *oldptr=NULL;
|
||||
void *oldptr2=NULL;
|
||||
int i;
|
||||
static void kmallocthread(void *sm, unsigned long num) {
|
||||
struct semaphore *sem = sm;
|
||||
void *ptr;
|
||||
void *oldptr = NULL;
|
||||
void *oldptr2 = NULL;
|
||||
int i;
|
||||
|
||||
for (i=0; i<NTRIES; i++) {
|
||||
ptr = kmalloc(ITEMSIZE);
|
||||
if (ptr==NULL) {
|
||||
if (sem) {
|
||||
kprintf("thread %lu: kmalloc returned NULL\n",
|
||||
num);
|
||||
goto done;
|
||||
}
|
||||
kprintf("kmalloc returned null; test failed.\n");
|
||||
goto done;
|
||||
}
|
||||
if (oldptr2) {
|
||||
kfree(oldptr2);
|
||||
}
|
||||
oldptr2 = oldptr;
|
||||
oldptr = ptr;
|
||||
}
|
||||
for (i = 0; i < NTRIES; i++) {
|
||||
ptr = kmalloc(ITEMSIZE);
|
||||
if (ptr == NULL) {
|
||||
if (sem) {
|
||||
kprintf("thread %lu: kmalloc returned NULL\n", num);
|
||||
goto done;
|
||||
}
|
||||
kprintf("kmalloc returned null; test failed.\n");
|
||||
goto done;
|
||||
}
|
||||
if (oldptr2) {
|
||||
kfree(oldptr2);
|
||||
}
|
||||
oldptr2 = oldptr;
|
||||
oldptr = ptr;
|
||||
}
|
||||
done:
|
||||
if (oldptr2) {
|
||||
kfree(oldptr2);
|
||||
}
|
||||
if (oldptr) {
|
||||
kfree(oldptr);
|
||||
}
|
||||
if (sem) {
|
||||
V(sem);
|
||||
}
|
||||
if (oldptr2) {
|
||||
kfree(oldptr2);
|
||||
}
|
||||
if (oldptr) {
|
||||
kfree(oldptr);
|
||||
}
|
||||
if (sem) {
|
||||
V(sem);
|
||||
}
|
||||
}
|
||||
|
||||
int
|
||||
kmalloctest(int nargs, char **args)
|
||||
{
|
||||
(void)nargs;
|
||||
(void)args;
|
||||
int kmalloctest(int nargs, char **args) {
|
||||
(void)nargs;
|
||||
(void)args;
|
||||
|
||||
kprintf("Starting kmalloc test...\n");
|
||||
kmallocthread(NULL, 0);
|
||||
kprintf("kmalloc test done\n");
|
||||
kprintf("Starting kmalloc test...\n");
|
||||
kmallocthread(NULL, 0);
|
||||
kprintf("kmalloc test done\n");
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
int
|
||||
kmallocstress(int nargs, char **args)
|
||||
{
|
||||
struct semaphore *sem;
|
||||
int i, result;
|
||||
int kmallocstress(int nargs, char **args) {
|
||||
struct semaphore *sem;
|
||||
int i, result;
|
||||
|
||||
(void)nargs;
|
||||
(void)args;
|
||||
(void)nargs;
|
||||
(void)args;
|
||||
|
||||
sem = sem_create("kmallocstress", 0);
|
||||
if (sem == NULL) {
|
||||
panic("kmallocstress: sem_create failed\n");
|
||||
}
|
||||
sem = sem_create("kmallocstress", 0);
|
||||
if (sem == NULL) {
|
||||
panic("kmallocstress: sem_create failed\n");
|
||||
}
|
||||
|
||||
kprintf("Starting kmalloc stress test...\n");
|
||||
kprintf("Starting kmalloc stress test...\n");
|
||||
|
||||
for (i=0; i<NTHREADS; i++) {
|
||||
result = thread_fork("kmallocstress", NULL,
|
||||
kmallocthread, sem, i);
|
||||
if (result) {
|
||||
panic("kmallocstress: thread_fork failed: %s\n",
|
||||
strerror(result));
|
||||
}
|
||||
}
|
||||
for (i = 0; i < NTHREADS; i++) {
|
||||
result = thread_fork("kmallocstress", NULL, kmallocthread, sem, i);
|
||||
if (result) {
|
||||
panic("kmallocstress: thread_fork failed: %s\n", strerror(result));
|
||||
}
|
||||
}
|
||||
|
||||
for (i=0; i<NTHREADS; i++) {
|
||||
P(sem);
|
||||
}
|
||||
for (i = 0; i < NTHREADS; i++) {
|
||||
P(sem);
|
||||
}
|
||||
|
||||
sem_destroy(sem);
|
||||
kprintf("kmalloc stress test done\n");
|
||||
sem_destroy(sem);
|
||||
kprintf("kmalloc stress test done\n");
|
||||
|
||||
return 0;
|
||||
return 0;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////
|
||||
@@ -169,212 +159,199 @@ kmallocstress(int nargs, char **args)
|
||||
* Having set this up, the test just allocates and then frees all the
|
||||
* pointers in order, setting and checking the contents.
|
||||
*/
|
||||
int
|
||||
kmalloctest3(int nargs, char **args)
|
||||
{
|
||||
int kmalloctest3(int nargs, char **args) {
|
||||
#define NUM_KM3_SIZES 5
|
||||
static const unsigned sizes[NUM_KM3_SIZES] = { 32, 41, 109, 86, 9 };
|
||||
unsigned numptrs;
|
||||
size_t ptrspace;
|
||||
size_t blocksize;
|
||||
unsigned numptrblocks;
|
||||
void ***ptrblocks;
|
||||
unsigned curblock, curpos, cursizeindex, cursize;
|
||||
size_t totalsize;
|
||||
unsigned i, j;
|
||||
unsigned char *ptr;
|
||||
static const unsigned sizes[NUM_KM3_SIZES] = {32, 41, 109, 86, 9};
|
||||
unsigned numptrs;
|
||||
size_t ptrspace;
|
||||
size_t blocksize;
|
||||
unsigned numptrblocks;
|
||||
void ***ptrblocks;
|
||||
unsigned curblock, curpos, cursizeindex, cursize;
|
||||
size_t totalsize;
|
||||
unsigned i, j;
|
||||
unsigned char *ptr;
|
||||
|
||||
if (nargs != 2) {
|
||||
kprintf("kmalloctest3: usage: km3 numobjects\n");
|
||||
return EINVAL;
|
||||
}
|
||||
if (nargs != 2) {
|
||||
kprintf("kmalloctest3: usage: km3 numobjects\n");
|
||||
return EINVAL;
|
||||
}
|
||||
|
||||
/* Figure out how many pointers we'll get and the space they need. */
|
||||
numptrs = atoi(args[1]);
|
||||
ptrspace = numptrs * sizeof(void *);
|
||||
/* Figure out how many pointers we'll get and the space they need. */
|
||||
numptrs = atoi(args[1]);
|
||||
ptrspace = numptrs * sizeof(void *);
|
||||
|
||||
/* Figure out how many blocks in the lower tier. */
|
||||
blocksize = PAGE_SIZE / 4;
|
||||
numptrblocks = DIVROUNDUP(ptrspace, blocksize);
|
||||
/* Figure out how many blocks in the lower tier. */
|
||||
blocksize = PAGE_SIZE / 4;
|
||||
numptrblocks = DIVROUNDUP(ptrspace, blocksize);
|
||||
|
||||
kprintf("kmalloctest3: %u objects, %u pointer blocks\n",
|
||||
numptrs, numptrblocks);
|
||||
kprintf("kmalloctest3: %u objects, %u pointer blocks\n", numptrs,
|
||||
numptrblocks);
|
||||
|
||||
/* Allocate the upper tier. */
|
||||
ptrblocks = kmalloc(numptrblocks * sizeof(ptrblocks[0]));
|
||||
if (ptrblocks == NULL) {
|
||||
panic("kmalloctest3: failed on pointer block array\n");
|
||||
}
|
||||
/* Allocate the lower tier. */
|
||||
for (i=0; i<numptrblocks; i++) {
|
||||
ptrblocks[i] = kmalloc(blocksize);
|
||||
if (ptrblocks[i] == NULL) {
|
||||
panic("kmalloctest3: failed on pointer block %u\n", i);
|
||||
}
|
||||
}
|
||||
/* Allocate the upper tier. */
|
||||
ptrblocks = kmalloc(numptrblocks * sizeof(ptrblocks[0]));
|
||||
if (ptrblocks == NULL) {
|
||||
panic("kmalloctest3: failed on pointer block array\n");
|
||||
}
|
||||
/* Allocate the lower tier. */
|
||||
for (i = 0; i < numptrblocks; i++) {
|
||||
ptrblocks[i] = kmalloc(blocksize);
|
||||
if (ptrblocks[i] == NULL) {
|
||||
panic("kmalloctest3: failed on pointer block %u\n", i);
|
||||
}
|
||||
}
|
||||
|
||||
/* Allocate the objects. */
|
||||
curblock = 0;
|
||||
curpos = 0;
|
||||
cursizeindex = 0;
|
||||
totalsize = 0;
|
||||
for (i=0; i<numptrs; i++) {
|
||||
cursize = sizes[cursizeindex];
|
||||
ptr = kmalloc(cursize);
|
||||
if (ptr == NULL) {
|
||||
kprintf("kmalloctest3: failed on object %u size %u\n",
|
||||
i, cursize);
|
||||
kprintf("kmalloctest3: pos %u in pointer block %u\n",
|
||||
curpos, curblock);
|
||||
kprintf("kmalloctest3: total so far %zu\n", totalsize);
|
||||
panic("kmalloctest3: failed.\n");
|
||||
}
|
||||
/* Fill the object with its number. */
|
||||
for (j=0; j<cursize; j++) {
|
||||
ptr[j] = (unsigned char) i;
|
||||
}
|
||||
/* Move to the next slot in the tree. */
|
||||
ptrblocks[curblock][curpos] = ptr;
|
||||
curpos++;
|
||||
if (curpos >= blocksize / sizeof(void *)) {
|
||||
curblock++;
|
||||
curpos = 0;
|
||||
}
|
||||
/* Update the running total, and rotate the size. */
|
||||
totalsize += cursize;
|
||||
cursizeindex = (cursizeindex + 1) % NUM_KM3_SIZES;
|
||||
}
|
||||
/* Allocate the objects. */
|
||||
curblock = 0;
|
||||
curpos = 0;
|
||||
cursizeindex = 0;
|
||||
totalsize = 0;
|
||||
for (i = 0; i < numptrs; i++) {
|
||||
cursize = sizes[cursizeindex];
|
||||
ptr = kmalloc(cursize);
|
||||
if (ptr == NULL) {
|
||||
kprintf("kmalloctest3: failed on object %u size %u\n", i, cursize);
|
||||
kprintf("kmalloctest3: pos %u in pointer block %u\n", curpos, curblock);
|
||||
kprintf("kmalloctest3: total so far %zu\n", totalsize);
|
||||
panic("kmalloctest3: failed.\n");
|
||||
}
|
||||
/* Fill the object with its number. */
|
||||
for (j = 0; j < cursize; j++) {
|
||||
ptr[j] = (unsigned char)i;
|
||||
}
|
||||
/* Move to the next slot in the tree. */
|
||||
ptrblocks[curblock][curpos] = ptr;
|
||||
curpos++;
|
||||
if (curpos >= blocksize / sizeof(void *)) {
|
||||
curblock++;
|
||||
curpos = 0;
|
||||
}
|
||||
/* Update the running total, and rotate the size. */
|
||||
totalsize += cursize;
|
||||
cursizeindex = (cursizeindex + 1) % NUM_KM3_SIZES;
|
||||
}
|
||||
|
||||
kprintf("kmalloctest3: %zu bytes allocated\n", totalsize);
|
||||
kprintf("kmalloctest3: %zu bytes allocated\n", totalsize);
|
||||
|
||||
/* Free the objects. */
|
||||
curblock = 0;
|
||||
curpos = 0;
|
||||
cursizeindex = 0;
|
||||
for (i=0; i<numptrs; i++) {
|
||||
cursize = sizes[cursizeindex];
|
||||
ptr = ptrblocks[curblock][curpos];
|
||||
KASSERT(ptr != NULL);
|
||||
for (j=0; j<cursize; j++) {
|
||||
if (ptr[j] == (unsigned char) i) {
|
||||
continue;
|
||||
}
|
||||
kprintf("kmalloctest3: failed on object %u size %u\n",
|
||||
i, cursize);
|
||||
kprintf("kmalloctest3: pos %u in pointer block %u\n",
|
||||
curpos, curblock);
|
||||
kprintf("kmalloctest3: at object offset %u\n", j);
|
||||
kprintf("kmalloctest3: expected 0x%x, found 0x%x\n",
|
||||
ptr[j], (unsigned char) i);
|
||||
panic("kmalloctest3: failed.\n");
|
||||
}
|
||||
kfree(ptr);
|
||||
curpos++;
|
||||
if (curpos >= blocksize / sizeof(void *)) {
|
||||
curblock++;
|
||||
curpos = 0;
|
||||
}
|
||||
KASSERT(totalsize > 0);
|
||||
totalsize -= cursize;
|
||||
cursizeindex = (cursizeindex + 1) % NUM_KM3_SIZES;
|
||||
}
|
||||
KASSERT(totalsize == 0);
|
||||
/* Free the objects. */
|
||||
curblock = 0;
|
||||
curpos = 0;
|
||||
cursizeindex = 0;
|
||||
for (i = 0; i < numptrs; i++) {
|
||||
cursize = sizes[cursizeindex];
|
||||
ptr = ptrblocks[curblock][curpos];
|
||||
KASSERT(ptr != NULL);
|
||||
for (j = 0; j < cursize; j++) {
|
||||
if (ptr[j] == (unsigned char)i) {
|
||||
continue;
|
||||
}
|
||||
kprintf("kmalloctest3: failed on object %u size %u\n", i, cursize);
|
||||
kprintf("kmalloctest3: pos %u in pointer block %u\n", curpos, curblock);
|
||||
kprintf("kmalloctest3: at object offset %u\n", j);
|
||||
kprintf("kmalloctest3: expected 0x%x, found 0x%x\n", ptr[j],
|
||||
(unsigned char)i);
|
||||
panic("kmalloctest3: failed.\n");
|
||||
}
|
||||
kfree(ptr);
|
||||
curpos++;
|
||||
if (curpos >= blocksize / sizeof(void *)) {
|
||||
curblock++;
|
||||
curpos = 0;
|
||||
}
|
||||
KASSERT(totalsize > 0);
|
||||
totalsize -= cursize;
|
||||
cursizeindex = (cursizeindex + 1) % NUM_KM3_SIZES;
|
||||
}
|
||||
KASSERT(totalsize == 0);
|
||||
|
||||
/* Free the lower tier. */
|
||||
for (i=0; i<numptrblocks; i++) {
|
||||
KASSERT(ptrblocks[i] != NULL);
|
||||
kfree(ptrblocks[i]);
|
||||
}
|
||||
/* Free the upper tier. */
|
||||
kfree(ptrblocks);
|
||||
/* Free the lower tier. */
|
||||
for (i = 0; i < numptrblocks; i++) {
|
||||
KASSERT(ptrblocks[i] != NULL);
|
||||
kfree(ptrblocks[i]);
|
||||
}
|
||||
/* Free the upper tier. */
|
||||
kfree(ptrblocks);
|
||||
|
||||
kprintf("kmalloctest3: passed\n");
|
||||
return 0;
|
||||
kprintf("kmalloctest3: passed\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
////////////////////////////////////////////////////////////
|
||||
// km4
|
||||
|
||||
static
|
||||
void
|
||||
kmalloctest4thread(void *sm, unsigned long num)
|
||||
{
|
||||
static void kmalloctest4thread(void *sm, unsigned long num) {
|
||||
#define NUM_KM4_SIZES 5
|
||||
static const unsigned sizes[NUM_KM4_SIZES] = { 1, 3, 5, 2, 4 };
|
||||
static const unsigned sizes[NUM_KM4_SIZES] = {1, 3, 5, 2, 4};
|
||||
|
||||
struct semaphore *sem = sm;
|
||||
void *ptrs[NUM_KM4_SIZES];
|
||||
unsigned p, q;
|
||||
unsigned i;
|
||||
struct semaphore *sem = sm;
|
||||
void *ptrs[NUM_KM4_SIZES];
|
||||
unsigned p, q;
|
||||
unsigned i;
|
||||
|
||||
for (i=0; i<NUM_KM4_SIZES; i++) {
|
||||
ptrs[i] = NULL;
|
||||
}
|
||||
p = 0;
|
||||
q = NUM_KM4_SIZES / 2;
|
||||
for (i = 0; i < NUM_KM4_SIZES; i++) {
|
||||
ptrs[i] = NULL;
|
||||
}
|
||||
p = 0;
|
||||
q = NUM_KM4_SIZES / 2;
|
||||
|
||||
for (i=0; i<NTRIES; i++) {
|
||||
if (ptrs[q] != NULL) {
|
||||
kfree(ptrs[q]);
|
||||
ptrs[q] = NULL;
|
||||
}
|
||||
ptrs[p] = kmalloc(sizes[p] * PAGE_SIZE);
|
||||
if (ptrs[p] == NULL) {
|
||||
panic("kmalloctest4: thread %lu: "
|
||||
"allocating %u pages failed\n",
|
||||
num, sizes[p]);
|
||||
}
|
||||
p = (p + 1) % NUM_KM4_SIZES;
|
||||
q = (q + 1) % NUM_KM4_SIZES;
|
||||
}
|
||||
for (i = 0; i < NTRIES; i++) {
|
||||
if (ptrs[q] != NULL) {
|
||||
kfree(ptrs[q]);
|
||||
ptrs[q] = NULL;
|
||||
}
|
||||
ptrs[p] = kmalloc(sizes[p] * PAGE_SIZE);
|
||||
if (ptrs[p] == NULL) {
|
||||
panic("kmalloctest4: thread %lu: "
|
||||
"allocating %u pages failed\n",
|
||||
num, sizes[p]);
|
||||
}
|
||||
p = (p + 1) % NUM_KM4_SIZES;
|
||||
q = (q + 1) % NUM_KM4_SIZES;
|
||||
}
|
||||
|
||||
for (i=0; i<NUM_KM4_SIZES; i++) {
|
||||
if (ptrs[i] != NULL) {
|
||||
kfree(ptrs[i]);
|
||||
}
|
||||
}
|
||||
for (i = 0; i < NUM_KM4_SIZES; i++) {
|
||||
if (ptrs[i] != NULL) {
|
||||
kfree(ptrs[i]);
|
||||
}
|
||||
}
|
||||
|
||||
V(sem);
|
||||
V(sem);
|
||||
}
|
||||
|
||||
int
|
||||
kmalloctest4(int nargs, char **args)
|
||||
{
|
||||
struct semaphore *sem;
|
||||
unsigned nthreads;
|
||||
unsigned i;
|
||||
int result;
|
||||
int kmalloctest4(int nargs, char **args) {
|
||||
struct semaphore *sem;
|
||||
unsigned nthreads;
|
||||
unsigned i;
|
||||
int result;
|
||||
|
||||
(void)nargs;
|
||||
(void)args;
|
||||
(void)nargs;
|
||||
(void)args;
|
||||
|
||||
kprintf("Starting multipage kmalloc test...\n");
|
||||
kprintf("Starting multipage kmalloc test...\n");
|
||||
#if OPT_DUMBVM
|
||||
kprintf("(This test will not work with dumbvm)\n");
|
||||
kprintf("(This test will not work with dumbvm)\n");
|
||||
#endif
|
||||
|
||||
sem = sem_create("kmalloctest4", 0);
|
||||
if (sem == NULL) {
|
||||
panic("kmalloctest4: sem_create failed\n");
|
||||
}
|
||||
sem = sem_create("kmalloctest4", 0);
|
||||
if (sem == NULL) {
|
||||
panic("kmalloctest4: sem_create failed\n");
|
||||
}
|
||||
|
||||
/* use 6 instead of 8 threads */
|
||||
nthreads = (3*NTHREADS)/4;
|
||||
/* use 6 instead of 8 threads */
|
||||
nthreads = (3 * NTHREADS) / 4;
|
||||
|
||||
for (i=0; i<nthreads; i++) {
|
||||
result = thread_fork("kmalloctest4", NULL,
|
||||
kmalloctest4thread, sem, i);
|
||||
if (result) {
|
||||
panic("kmallocstress: thread_fork failed: %s\n",
|
||||
strerror(result));
|
||||
}
|
||||
}
|
||||
for (i = 0; i < nthreads; i++) {
|
||||
result = thread_fork("kmalloctest4", NULL, kmalloctest4thread, sem, i);
|
||||
if (result) {
|
||||
panic("kmallocstress: thread_fork failed: %s\n", strerror(result));
|
||||
}
|
||||
}
|
||||
|
||||
for (i=0; i<nthreads; i++) {
|
||||
P(sem);
|
||||
}
|
||||
for (i = 0; i < nthreads; i++) {
|
||||
P(sem);
|
||||
}
|
||||
|
||||
sem_destroy(sem);
|
||||
kprintf("Multipage kmalloc test done\n");
|
||||
return 0;
|
||||
sem_destroy(sem);
|
||||
kprintf("Multipage kmalloc test done\n");
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user