mittorn
5 years ago
4 changed files with 632 additions and 5 deletions
@ -0,0 +1,488 @@ |
|||||||
|
/* $NetBSD: malloc.c,v 1.8 1997/04/07 03:12:14 christos Exp $ */ |
||||||
|
|
||||||
|
/*
|
||||||
|
* Copyright (c) 1983 Regents of the University of California. |
||||||
|
* All rights reserved. |
||||||
|
* |
||||||
|
* Redistribution and use in source and binary forms, with or without |
||||||
|
* modification, are permitted provided that the following conditions |
||||||
|
* are met: |
||||||
|
* 1. Redistributions of source code must retain the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer. |
||||||
|
* 2. Redistributions in binary form must reproduce the above copyright |
||||||
|
* notice, this list of conditions and the following disclaimer in the |
||||||
|
* documentation and/or other materials provided with the distribution. |
||||||
|
* 3. All advertising materials mentioning features or use of this software |
||||||
|
* must display the following acknowledgement: |
||||||
|
* This product includes software developed by the University of |
||||||
|
* California, Berkeley and its contributors. |
||||||
|
* 4. Neither the name of the University nor the names of its contributors |
||||||
|
* may be used to endorse or promote products derived from this software |
||||||
|
* without specific prior written permission. |
||||||
|
* |
||||||
|
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND |
||||||
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
||||||
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
||||||
|
* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
||||||
|
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
||||||
|
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
||||||
|
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
||||||
|
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
||||||
|
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
||||||
|
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
||||||
|
* SUCH DAMAGE. |
||||||
|
*/ |
||||||
|
|
||||||
|
#if defined(LIBC_SCCS) && !defined(lint) |
||||||
|
#if 0 |
||||||
|
static char *sccsid = "from: @(#)malloc.c 5.11 (Berkeley) 2/23/91"; |
||||||
|
#else |
||||||
|
static char *rcsid = "$NetBSD: malloc.c,v 1.8 1997/04/07 03:12:14 christos Exp $"; |
||||||
|
#endif |
||||||
|
#endif /* LIBC_SCCS and not lint */ |
||||||
|
|
||||||
|
/*
|
||||||
|
* malloc.c (Caltech) 2/21/82 |
||||||
|
* Chris Kingsley, kingsley@cit-20. |
||||||
|
* |
||||||
|
* This is a very fast storage allocator. It allocates blocks of a small |
||||||
|
* number of different sizes, and keeps free lists of each size. Blocks that |
||||||
|
* don't exactly fit are passed up to the next larger size. In this |
||||||
|
* implementation, the available sizes are 2^n-4 (or 2^n-10) bytes long. |
||||||
|
* This is designed for use in a virtual memory environment. |
||||||
|
*/ |
||||||
|
|
||||||
|
#include <sys/types.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <string.h> |
||||||
|
#include "platform/swap/swap.h" |
||||||
|
/* #define MSTATS 1 */ |
||||||
|
|
||||||
|
#ifndef _WIN32 |
||||||
|
#include <unistd.h> |
||||||
|
#define NULL 0 |
||||||
|
#else |
||||||
|
|
||||||
|
#define u_char unsigned char |
||||||
|
#define u_long unsigned long |
||||||
|
#define getpagesize() 4096 |
||||||
|
#define caddr_t size_t |
||||||
|
#define bcopy(a,b,c) memcpy(b,a,c) |
||||||
|
|
||||||
|
#endif |
||||||
|
|
||||||
|
|
||||||
|
static void morecore(int); |
||||||
|
/*
|
||||||
|
* The overhead on a block is at least 4 bytes. When free, this space |
||||||
|
* contains a pointer to the next free block, and the bottom two bits must |
||||||
|
* be zero. When in use, the first byte is set to MAGIC, and the second |
||||||
|
* byte is the size index. The remaining bytes are for alignment. |
||||||
|
* If range checking is enabled then a second word holds the size of the |
||||||
|
* requested block, less 1, rounded up to a multiple of sizeof(RMAGIC). |
||||||
|
* The order of elements is critical: ov_magic must overlay the low order |
||||||
|
* bits of ov_next, and ov_magic can not be a valid ov_next bit pattern. |
||||||
|
*/ |
||||||
|
union overhead { |
||||||
|
union overhead *ov_next; /* when free */ |
||||||
|
struct { |
||||||
|
u_char ovu_magic; /* magic number */ |
||||||
|
u_char ovu_index; /* bucket # */ |
||||||
|
#ifdef RCHECK |
||||||
|
u_short ovu_rmagic; /* range magic number */ |
||||||
|
u_long ovu_size; /* actual block size */ |
||||||
|
#endif |
||||||
|
} ovu; |
||||||
|
#define ov_magic ovu.ovu_magic |
||||||
|
#define ov_index ovu.ovu_index |
||||||
|
#define ov_rmagic ovu.ovu_rmagic |
||||||
|
#define ov_size ovu.ovu_size |
||||||
|
}; |
||||||
|
|
||||||
|
static int findbucket( union overhead *freep, |
||||||
|
int srchlen); |
||||||
|
#define MAGIC 0xef /* magic # on accounting info */ |
||||||
|
#define RMAGIC 0x5555 /* magic # on range info */ |
||||||
|
|
||||||
|
#ifdef RCHECK |
||||||
|
#define RSLOP sizeof (u_short) |
||||||
|
#else |
||||||
|
#define RSLOP 0 |
||||||
|
#endif |
||||||
|
|
||||||
|
/*
|
||||||
|
* nextf[i] is the pointer to the next free block of size 2^(i+3). The |
||||||
|
* smallest allocatable block is 8 bytes. The overhead information |
||||||
|
* precedes the data area returned to the user. |
||||||
|
*/ |
||||||
|
#define NBUCKETS 30 |
||||||
|
static union overhead *nextf[NBUCKETS]; |
||||||
|
/* extern char *sbrk(); */ |
||||||
|
|
||||||
|
static int pagesz; /* page size */ |
||||||
|
static int pagebucket; /* page size bucket */ |
||||||
|
|
||||||
|
#ifdef MSTATS |
||||||
|
/*
|
||||||
|
* nmalloc[i] is the difference between the number of mallocs and frees |
||||||
|
* for a given block size. |
||||||
|
*/ |
||||||
|
static unsigned int nmalloc[NBUCKETS]; |
||||||
|
#include <stdio.h> |
||||||
|
#endif |
||||||
|
|
||||||
|
#if defined(DEBUG) || defined(RCHECK) |
||||||
|
#define ASSERT(p) if (!(p)) botch(__STRING(p)) |
||||||
|
#include /* <stdio.h> */ |
||||||
|
static |
||||||
|
botch(s) |
||||||
|
char *s; |
||||||
|
{ |
||||||
|
fprintf(stderr, "\r\nassertion botched: %s\r\n", s); |
||||||
|
(void) fflush(stderr); /* just in case user buffered it */ |
||||||
|
abort(); |
||||||
|
} |
||||||
|
#else |
||||||
|
#define ASSERT(p) |
||||||
|
#endif |
||||||
|
|
||||||
|
#include <stdio.h> |
||||||
|
|
||||||
|
/* */ |
||||||
|
/* malloc */ |
||||||
|
/* malloc */ |
||||||
|
void * SWAP_Malloc( size_t nbytes ) |
||||||
|
{ |
||||||
|
register union overhead *op; |
||||||
|
register int bucket; |
||||||
|
register long n; |
||||||
|
register unsigned amt; |
||||||
|
|
||||||
|
/*
|
||||||
|
* First time malloc is called, setup page size and |
||||||
|
* align break pointer so all data will be page aligned. |
||||||
|
*/ |
||||||
|
if (pagesz == 0) { |
||||||
|
pagesz = n = getpagesize(); |
||||||
|
op = (union overhead *)SWAP_Sbrk( 0 ); |
||||||
|
n = n - sizeof (*op) - ((long)op & (n - 1)); |
||||||
|
if (n < 0) |
||||||
|
n += pagesz; |
||||||
|
if (n) { |
||||||
|
if (SWAP_Sbrk(n) == (char *)-1) { |
||||||
|
return (NULL); |
||||||
|
} |
||||||
|
} |
||||||
|
bucket = 0; |
||||||
|
amt = 8; |
||||||
|
while (pagesz > amt) { |
||||||
|
amt <<= 1; |
||||||
|
bucket++; |
||||||
|
} |
||||||
|
pagebucket = bucket; |
||||||
|
} |
||||||
|
/*
|
||||||
|
* Convert amount of memory requested into closest block size |
||||||
|
* stored in hash buckets which satisfies request. |
||||||
|
* Account for space used per block for accounting. |
||||||
|
*/ |
||||||
|
if (nbytes <= (n = pagesz - sizeof (*op) - RSLOP)) { |
||||||
|
#ifndef RCHECK |
||||||
|
amt = 8; /* size of first bucket */ |
||||||
|
bucket = 0; |
||||||
|
#else |
||||||
|
amt = 16; /* size of first bucket */ |
||||||
|
bucket = 1; |
||||||
|
#endif |
||||||
|
n = -((long)sizeof (*op) + RSLOP); |
||||||
|
} else { |
||||||
|
amt = pagesz; |
||||||
|
bucket = pagebucket; |
||||||
|
} |
||||||
|
while (nbytes > amt + n) { |
||||||
|
amt <<= 1; |
||||||
|
if (amt == 0) { |
||||||
|
return (NULL); |
||||||
|
} |
||||||
|
bucket++; |
||||||
|
} |
||||||
|
/*
|
||||||
|
* If nothing in hash bucket right now, |
||||||
|
* request more memory from the system. |
||||||
|
*/ |
||||||
|
if ((op = nextf[bucket]) == NULL) { |
||||||
|
morecore(bucket); |
||||||
|
if ((op = nextf[bucket]) == NULL) { |
||||||
|
return (NULL); |
||||||
|
} |
||||||
|
} |
||||||
|
/* remove from linked list */ |
||||||
|
nextf[bucket] = op->ov_next; |
||||||
|
op->ov_magic = MAGIC; |
||||||
|
op->ov_index = bucket; |
||||||
|
#ifdef MSTATS |
||||||
|
nmalloc[bucket]++; |
||||||
|
#endif |
||||||
|
#ifdef RCHECK |
||||||
|
/*
|
||||||
|
* Record allocated size of block and |
||||||
|
* bound space with magic numbers. |
||||||
|
*/ |
||||||
|
op->ov_size = (nbytes + RSLOP - 1) & ~(RSLOP - 1); |
||||||
|
op->ov_rmagic = RMAGIC; |
||||||
|
*(u_short *)((caddr_t)(op + 1) + op->ov_size) = RMAGIC; |
||||||
|
#endif |
||||||
|
return ((char *)(op + 1)); |
||||||
|
} |
||||||
|
|
||||||
|
void * SWAP_Calloc(size_t nelem, size_t elsize) { |
||||||
|
void * ptr = SWAP_Malloc (nelem * elsize); |
||||||
|
// Zero out the malloc'd block.
|
||||||
|
memset (ptr, 0, nelem * elsize); |
||||||
|
return ptr; |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Allocate more memory to the indicated bucket. |
||||||
|
*/ |
||||||
|
static void |
||||||
|
morecore(int bucket) |
||||||
|
{ |
||||||
|
register union overhead *op; |
||||||
|
register long sz; /* size of desired block */ |
||||||
|
long amt; /* amount to allocate */ |
||||||
|
int nblks; /* how many blocks we get */ |
||||||
|
|
||||||
|
/*
|
||||||
|
* sbrk_size <= 0 only for big, FLUFFY, requests (about |
||||||
|
* 2^30 bytes on a VAX, I think) or for a negative arg. |
||||||
|
*/ |
||||||
|
sz = 1 << (bucket + 3); |
||||||
|
#ifdef DEBUG |
||||||
|
ASSERT(sz > 0); |
||||||
|
#else |
||||||
|
if (sz <= 0) |
||||||
|
return; |
||||||
|
#endif |
||||||
|
if (sz < pagesz) { |
||||||
|
amt = pagesz; |
||||||
|
nblks = amt / sz; |
||||||
|
} else { |
||||||
|
amt = sz + pagesz; |
||||||
|
nblks = 1; |
||||||
|
} |
||||||
|
op = (union overhead *)SWAP_Sbrk(amt); |
||||||
|
/* no more room! */ |
||||||
|
if ((long)op == -1) |
||||||
|
return; |
||||||
|
/*
|
||||||
|
* Add new memory allocated to that on |
||||||
|
* free list for this hash bucket. |
||||||
|
*/ |
||||||
|
nextf[bucket] = op; |
||||||
|
while (--nblks > 0) { |
||||||
|
op->ov_next = (union overhead *)((caddr_t)op + sz); |
||||||
|
op = (union overhead *)((caddr_t)op + sz); |
||||||
|
} |
||||||
|
} |
||||||
|
/* */ |
||||||
|
/* free */ |
||||||
|
/* free */ |
||||||
|
void SWAP_Free( void *cp) |
||||||
|
|
||||||
|
{ |
||||||
|
register long size; |
||||||
|
register union overhead *op; |
||||||
|
|
||||||
|
if (cp == NULL) |
||||||
|
return; |
||||||
|
op = (union overhead *)((caddr_t)cp - sizeof (union overhead)); |
||||||
|
#ifdef DEBUG |
||||||
|
ASSERT(op->ov_magic == MAGIC); /* make sure it was in use */ |
||||||
|
#else |
||||||
|
if (op->ov_magic != MAGIC) |
||||||
|
return; /* sanity */ |
||||||
|
#endif |
||||||
|
#ifdef RCHECK |
||||||
|
ASSERT(op->ov_rmagic == RMAGIC); |
||||||
|
ASSERT(*(u_short *)((caddr_t)(op + 1) + op->ov_size) == RMAGIC); |
||||||
|
#endif |
||||||
|
size = op->ov_index; |
||||||
|
ASSERT(size < NBUCKETS); |
||||||
|
op->ov_next = nextf[size]; /* also clobbers ov_magic */ |
||||||
|
nextf[size] = op; |
||||||
|
#ifdef MSTATS |
||||||
|
nmalloc[size]--; |
||||||
|
#endif |
||||||
|
} |
||||||
|
|
||||||
|
|
||||||
|
/* EDB: added size lookup */ |
||||||
|
|
||||||
|
size_t SWAP_MallocUsableSize(void * cp) |
||||||
|
{ |
||||||
|
register long size; |
||||||
|
register union overhead *op; |
||||||
|
|
||||||
|
if (cp == NULL) |
||||||
|
return 0; |
||||||
|
op = (union overhead *)((caddr_t)cp - sizeof (union overhead)); |
||||||
|
#ifdef DEBUG |
||||||
|
ASSERT(op->ov_magic == MAGIC); /* make sure it was in use */ |
||||||
|
#else |
||||||
|
if (op->ov_magic != MAGIC) |
||||||
|
return 0; /* sanity */ |
||||||
|
#endif |
||||||
|
#ifdef RCHECK |
||||||
|
ASSERT(op->ov_rmagic == RMAGIC); |
||||||
|
ASSERT(*(u_short *)((caddr_t)(op + 1) + op->ov_size) == RMAGIC); |
||||||
|
#endif |
||||||
|
return op->ov_index; |
||||||
|
} |
||||||
|
/* End EDB */ |
||||||
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
* When a program attempts "storage compaction" as mentioned in the |
||||||
|
* old malloc man page, it realloc's an already freed block. Usually |
||||||
|
* this is the last block it freed; occasionally it might be farther |
||||||
|
* back. We have to search all the free lists for the block in order |
||||||
|
* to determine its bucket: 1st we make one pass thru the lists |
||||||
|
* checking only the first block in each; if that fails we search |
||||||
|
* ``realloc_srchlen'' blocks in each list for a match (the variable |
||||||
|
* is extern so the caller can modify it). If that fails we just copy |
||||||
|
* however many bytes was given to realloc() and hope it's not huge. |
||||||
|
*/ |
||||||
|
int realloc_srchlen = 4; /* 4 should be plenty, -1 =>'s whole list */ |
||||||
|
|
||||||
|
/* */ |
||||||
|
/* realloc */ |
||||||
|
/* realloc */ |
||||||
|
void * |
||||||
|
SWAP_Realloc( void *cp, |
||||||
|
size_t nbytes) |
||||||
|
|
||||||
|
{ |
||||||
|
register u_long onb; |
||||||
|
register long i; |
||||||
|
union overhead *op; |
||||||
|
char *res; |
||||||
|
int was_alloced = 0; |
||||||
|
|
||||||
|
if (cp == NULL) |
||||||
|
return (SWAP_Malloc(nbytes)); |
||||||
|
if (nbytes == 0) { |
||||||
|
SWAP_Free (cp); |
||||||
|
return NULL; |
||||||
|
} |
||||||
|
op = (union overhead *)((caddr_t)cp - sizeof (union overhead)); |
||||||
|
if (op->ov_magic == MAGIC) { |
||||||
|
was_alloced++; |
||||||
|
i = op->ov_index; |
||||||
|
} else { |
||||||
|
/*
|
||||||
|
* Already free, doing "compaction". |
||||||
|
* |
||||||
|
* Search for the old block of memory on the |
||||||
|
* free list. First, check the most common |
||||||
|
* case (last element free'd), then (this failing) |
||||||
|
* the last ``realloc_srchlen'' items free'd. |
||||||
|
* If all lookups fail, then assume the size of |
||||||
|
* the memory block being realloc'd is the |
||||||
|
* largest possible (so that all "nbytes" of new |
||||||
|
* memory are copied into). Note that this could cause |
||||||
|
* a memory fault if the old area was tiny, and the moon |
||||||
|
* is gibbous. However, that is very unlikely. |
||||||
|
*/ |
||||||
|
if ((i = findbucket(op, 1)) < 0 && |
||||||
|
(i = findbucket(op, realloc_srchlen)) < 0) |
||||||
|
i = NBUCKETS; |
||||||
|
} |
||||||
|
onb = 1 << (i + 3); |
||||||
|
if (onb < pagesz) |
||||||
|
onb -= sizeof (*op) + RSLOP; |
||||||
|
else |
||||||
|
onb += pagesz - sizeof (*op) - RSLOP; |
||||||
|
/* avoid the copy if same size block */ |
||||||
|
if (was_alloced) { |
||||||
|
if (i) { |
||||||
|
i = 1 << (i + 2); |
||||||
|
if (i < pagesz) |
||||||
|
i -= sizeof (*op) + RSLOP; |
||||||
|
else |
||||||
|
i += pagesz - sizeof (*op) - RSLOP; |
||||||
|
} |
||||||
|
if (nbytes <= onb && nbytes > i) { |
||||||
|
#ifdef RCHECK |
||||||
|
op->ov_size = (nbytes + RSLOP - 1) & ~(RSLOP - 1); |
||||||
|
*(u_short *)((caddr_t)(op + 1) + op->ov_size) = RMAGIC; |
||||||
|
#endif |
||||||
|
return(cp); |
||||||
|
} else |
||||||
|
SWAP_Free(cp); |
||||||
|
} |
||||||
|
if ((res = SWAP_Malloc(nbytes)) == NULL) |
||||||
|
return (NULL); |
||||||
|
if (cp != res) /* common optimization if "compacting" */ |
||||||
|
bcopy(cp, res, (nbytes < onb) ? nbytes : onb); |
||||||
|
return (res); |
||||||
|
} |
||||||
|
|
||||||
|
/*
|
||||||
|
* Search ``srchlen'' elements of each free list for a block whose |
||||||
|
* header starts at ``freep''. If srchlen is -1 search the whole list. |
||||||
|
* Return bucket number, or -1 if not found. |
||||||
|
*/ |
||||||
|
static int |
||||||
|
findbucket( union overhead *freep, |
||||||
|
int srchlen) |
||||||
|
|
||||||
|
{ |
||||||
|
register union overhead *p; |
||||||
|
register int i, j; |
||||||
|
|
||||||
|
for (i = 0; i < NBUCKETS; i++) { |
||||||
|
j = 0; |
||||||
|
for (p = nextf[i]; p && j != srchlen; p = p->ov_next) { |
||||||
|
if (p == freep) |
||||||
|
return (i); |
||||||
|
j++; |
||||||
|
} |
||||||
|
} |
||||||
|
return (-1); |
||||||
|
} |
||||||
|
|
||||||
|
#ifdef MSTATS |
||||||
|
/*
|
||||||
|
* mstats - print out statistics about malloc |
||||||
|
* |
||||||
|
* Prints two lines of numbers, one showing the length of the free list |
||||||
|
* for each size category, the second showing the number of mallocs - |
||||||
|
* frees for each size category. |
||||||
|
*/ |
||||||
|
/* */ |
||||||
|
mstats(void) |
||||||
|
{ |
||||||
|
register int i, j; |
||||||
|
register union overhead *p; |
||||||
|
int totfree = 0, |
||||||
|
totused = 0; |
||||||
|
|
||||||
|
fprintf(stderr, "Memory allocation statistics\nfree:\t"); |
||||||
|
for (i = 0; i < NBUCKETS; i++) { |
||||||
|
for (j = 0, p = nextf[i]; p; p = p->ov_next, j++) |
||||||
|
; |
||||||
|
fprintf(stderr, " %d", j); |
||||||
|
totfree += j * (1 << (i + 3)); |
||||||
|
} |
||||||
|
fprintf(stderr, "\nused:\t"); |
||||||
|
for (i = 0; i < NBUCKETS; i++) { |
||||||
|
fprintf(stderr, " %d", nmalloc[i]); |
||||||
|
totused += nmalloc[i] * (1 << (i + 3)); |
||||||
|
} |
||||||
|
fprintf(stderr, "\n\tTotal in use: %d, total free: %d, total allocated: %d\n", |
||||||
|
totused, totfree, totused + totfree); |
||||||
|
} |
||||||
|
#endif |
||||||
|
|
@ -0,0 +1,108 @@ |
|||||||
|
/*
|
||||||
|
sbrk.c - swap memory allocation |
||||||
|
Copyright (C) 2019 mittorn |
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify |
||||||
|
it under the terms of the GNU General Public License as published by |
||||||
|
the Free Software Foundation, either version 3 of the License, or |
||||||
|
(at your option) any later version. |
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful, |
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
GNU General Public License for more details. |
||||||
|
*/ |
||||||
|
#define _GNU_SOURCE |
||||||
|
#include <dlfcn.h> |
||||||
|
#include <sys/mman.h> |
||||||
|
#include <unistd.h> |
||||||
|
#include <sys/types.h> |
||||||
|
#include <fcntl.h> |
||||||
|
#include <stdio.h> |
||||||
|
#include <stdlib.h> |
||||||
|
#include <errno.h> |
||||||
|
#include <signal.h> |
||||||
|
#define _GNU_SOURCE |
||||||
|
#include <dlfcn.h> |
||||||
|
#include "platform/swap/swap.h" |
||||||
|
#include "string.h" |
||||||
|
|
||||||
|
#ifndef XASH_DEFAULT_SWAP_PATH |
||||||
|
#define XASH_DEFAULT_SWAP_PATH "/tmp/xash3d-swap" |
||||||
|
#endif |
||||||
|
#define PAGE_SIZE 4096 |
||||||
|
static struct sbrk_state_s |
||||||
|
{ |
||||||
|
void *top; |
||||||
|
int fd; |
||||||
|
size_t size; |
||||||
|
size_t prealloc; |
||||||
|
pid_t pid; |
||||||
|
} s; |
||||||
|
|
||||||
|
static void SWAP_Initialize(void) |
||||||
|
{ |
||||||
|
char *path; |
||||||
|
char *prealloc = getenv("SWAP_SIZE"); |
||||||
|
int fd; |
||||||
|
|
||||||
|
if( s.top ) |
||||||
|
return; |
||||||
|
|
||||||
|
path = getenv("SWAP_PATH"); |
||||||
|
if( !path ) |
||||||
|
path = XASH_DEFAULT_SWAP_PATH; |
||||||
|
fd = open( path, O_CREAT|O_RDWR, 0600 ); |
||||||
|
|
||||||
|
if( prealloc ) s.prealloc = atoi(prealloc); |
||||||
|
else s.prealloc = 128*1024*1024; |
||||||
|
s.prealloc &= ~(PAGE_SIZE - 1); |
||||||
|
|
||||||
|
s.fd = fd; |
||||||
|
ftruncate( fd, s.prealloc ); |
||||||
|
s.top = mmap( 0, s.prealloc, PROT_READ|PROT_WRITE, MAP_SHARED, fd, 0 ); |
||||||
|
|
||||||
|
// space will be freed on exit
|
||||||
|
//unlink(path);
|
||||||
|
} |
||||||
|
|
||||||
|
void *SWAP_Sbrk(size_t size) |
||||||
|
{ |
||||||
|
char buf[64]; |
||||||
|
SWAP_Initialize(); |
||||||
|
|
||||||
|
if( size == 0 ) |
||||||
|
return s.top; |
||||||
|
else if( size > 0 ) |
||||||
|
{ |
||||||
|
void *res; |
||||||
|
|
||||||
|
//write(1, buf, snprintf(buf, 32, "allocating %d\n", size) );
|
||||||
|
res = s.top; |
||||||
|
s.size += size; |
||||||
|
s.top = res + size; |
||||||
|
if( s.size + size > s.prealloc ) |
||||||
|
return (void*)-1; |
||||||
|
|
||||||
|
memset( res, 0, size ); |
||||||
|
return res; |
||||||
|
|
||||||
|
} |
||||||
|
else |
||||||
|
{ |
||||||
|
void *res = s.top; |
||||||
|
|
||||||
|
if( -size > s.size ) |
||||||
|
res = (void*)-1; |
||||||
|
else |
||||||
|
{ |
||||||
|
s.top += size; |
||||||
|
s.size += size; |
||||||
|
//write(1, buf, snprintf(buf, 32, "freed %d\n", -size) );
|
||||||
|
} |
||||||
|
|
||||||
|
return res; |
||||||
|
} |
||||||
|
} |
||||||
|
|
||||||
|
|
@ -0,0 +1,21 @@ |
|||||||
|
/*
|
||||||
|
sbrk_swap.h - swap memory allocation |
||||||
|
Copyright (C) 2019 mittorn |
||||||
|
|
||||||
|
This program is free software: you can redistribute it and/or modify |
||||||
|
it under the terms of the GNU General Public License as published by |
||||||
|
the Free Software Foundation, either version 3 of the License, or |
||||||
|
(at your option) any later version. |
||||||
|
|
||||||
|
This program is distributed in the hope that it will be useful, |
||||||
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||||
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||||||
|
GNU General Public License for more details. |
||||||
|
*/ |
||||||
|
//#include <stdint.h>
|
||||||
|
void *SWAP_Sbrk( size_t size ); |
||||||
|
void *SWAP_Malloc( size_t size ); |
||||||
|
void *SWAP_Calloc( size_t nelem, size_t size ); |
||||||
|
void SWAP_Free( void *cp ); |
||||||
|
void *SWAP_Realloc( void *cp, size_t size ); |
||||||
|
size_t SWAP_MallocUsableSize( void * cp ); |
Loading…
Reference in new issue