初窥内存管理(三)伙伴算法

        假设系统的可利用内存空间容量为2m个字(地址从0到2m-1),则在开始运行时,整个内存区是一个大小为2m的空闲块,在运行了一段时间之后,被分隔成若干占用块和空闲块。为了在分配时查找方便起见,我们将所有大小相同的空闲块建于一张子表中。每个子表是一个双重链表,这样的链表可能有m+1个,将这m+1个表头指针用向量结构组织成一个表,这就是伙伴系统中的可利用空间表,如图所示:



分配算法: 

       当用户提出大小为n的内存请求时,首先在可利用表上寻找结点大小与n相匹配的子表,若此子表非空,则将子表中任意一个结点分配之即可;若此子表为空,则需从结点更大的非空子表中去查找,直至找到一个空闲块,则将其中一部分分配给用户,而将剩余部分插入相应的子表中。
       若2k-1 < n ≤ 2k-1,又第k+1个子表不空,则只要删除此链表中第一个结点并分配给用户即可;若 2k-2 < n ≤ 2k-1-1,此时由于结点大小为2k-1 的子表为空,则需从结点大小为2k 的子表中取出一块,将其中一半分配给用户,剩余的一半作为一个新结点插入在结点大小为2k-1的子表中,若2k-i-1 < n ≤ 2k-i-1(i为小于是的整数),并且所有结点小于2k的子表均为空,则同样需从结点大小为2k的子表中取出一块,将其中2k-i的一小部分分配给用户,剩余部分分割成若干个结点分别插入在结点大小为2k-1 、 2k-2 、…、 2k-i的子表中。

回收算法:

        在用户释放不再使用的占用块时,系统需将这新的空闲块插入到可利用空间表中去。这里,同样有一个地址相邻的空闲块归并成大块的问题。但是在伙伴系统中仅考虑互为“伙伴”的两个空闲块的归并。
        何谓“伙伴”?如前所述,在分配时经常需要将一个大的空闲块分裂成两个大小相等的存储区,这两个由同一大块分裂出来的小块就称之“互为伙伴”。例如:假设p为大小为pow(2,k)的空闲块的初始地址,且p MOD pow(2,k+1)=0,则初始地址为p和p+pow(2,k)的两个空闲块互为伙伴。在伙伴系统中回收空闲块时,只当其伙伴为空闲块时才归并成大块。也就是说,若有两个空闲块,即使大小相同且地址相邻,但不是由同一大块分裂出来的,也不归并在一起。
        由此,在回收空闲块时,应首先判别其伙伴是否为空闲块,若否,则只要将释放的空闲块简单插入在相应子表中即可;若是,则需在相应子表中找到其伙伴并删除之,然后再判别合并后的空闲块的伙伴是否是空闲块。依此重复,直到归并所得空闲块的伙伴不是空闲块时,再插入到相应的子表中去。


代码如下(只用了单链表):


#include <stdio.h>
#include <stdlib.h>
#include <time.h>

#define MIN_MOMORY_SIZE 536870912   //随机产生的最小内存空间:512M (最大为1G) 
#define INDEX_SIZE 30               //哈希索引表大小 
#define WORKTIME 1500               //系统工作时间 
#define MAX_REQ_SIZE 268435456      //申请空闲内存分配的最大容量:256M 
#define MIN_DUE 30                  //使用内存块的最短时间
#define MAX_DUE 90                  //使用内存块的最长时间
#define OCCUPY_INTERVAL 60          //每次分配的最大间隔 
#define USED 1                      //内存块被使用 
#define UNUSED 0                    //内存块未被使用

//内存块链表结点结构 
typedef struct buddy_node {
	int flag;                       //标记空间是否被使用
	int base;                       //本块儿内存的基地址 
	int occupy;                     //实际使用空间大小 
	int fragment;                   //碎片大小 
	int duetime;                    //使用时间
	struct buddy_node *nextPtr;     //指向下一个结点 
} Buddy, *BuddyPtr;

//哈希索引表结构 
typedef struct hash_table {
	int nodesize;
	BuddyPtr headPtr;
} IndexTable;

IndexTable table[INDEX_SIZE];//使用哈希表管理伙伴系统 
int ready = 0;               //需要分配内存的时刻 
int availSpace;              //可分配空间大小 
int totalFragment = 0;       //总碎片大小 

//函数:根据k的值计算哈希表项链接的内存块的大小
int get_size (int k)
{
    int i, nodesize = 1;
    
    for (i = 0; i < k; i ++)
        nodesize *= 2;

    return nodesize;
}

//函数:初始化哈希索引表 
void ini_index (void)
{
    int i;
    
    for (i = 0; i < INDEX_SIZE; i ++) {
        table[i].nodesize = get_size(i);
        table[i].headPtr = NULL;
    } 
}

//函数:初始化伙伴系统
void int_system (int memory_size)
{
    int i, addr = 0;
    int left = memory_size;
    BuddyPtr newnodePtr = NULL;
     
    //初始化可分配空间大小和碎片大小 
    availSpace = memory_size; 
    totalFragment = 0;
    
	//将内存分割成尽量大的2倍数的空闲块并插入到相应的索引表项中
	for (i = INDEX_SIZE-1; left > 0; i --) {
		if (left / table[i].nodesize == 1) {
            newnodePtr = (BuddyPtr) malloc (sizeof (Buddy));//分配结点 
            newnodePtr->flag = UNUSED;
            newnodePtr->base = addr;
            newnodePtr->occupy = 0; 
            newnodePtr->fragment = 0;
            newnodePtr->duetime = 0;
            newnodePtr->nextPtr = NULL;
            table[i].headPtr = newnodePtr;
            addr += table[i].nodesize;
            left -= table[i].nodesize;
        }
    } 
}

//函数:程序运行结束后释放所有结点
void free_system (void)
{
    int i;
    BuddyPtr tempPtr = NULL, tofreePtr = NULL;
     
    for (i = 0; i < INDEX_SIZE; i ++) {
        if (table[i].headPtr) {
            tempPtr = table[i].headPtr;
            table[i].headPtr = NULL;
            //依次释放所有结点
            while (tempPtr) {
                tofreePtr = tempPtr;
                tempPtr = tempPtr->nextPtr;
                free (tofreePtr);
            }
        }
    }
}

//函数:添加结点(形参为内存块结点的信息) 
void insert_node (int i, int inbase, int f, int occ, int frag, int d)
{
    BuddyPtr newnodePtr = NULL, prePtr = NULL, curPtr = NULL;
    
    newnodePtr = (BuddyPtr) malloc (sizeof (Buddy));//分配结点
    newnodePtr->base = inbase;
    newnodePtr->flag = f;
    newnodePtr->occupy = occ; 
    newnodePtr->fragment = frag;
    newnodePtr->duetime = d;
    newnodePtr->nextPtr = NULL; 
    if (table[i].headPtr == NULL)
        table[i].headPtr = newnodePtr;
    else {
        curPtr = table[i].headPtr;
        prePtr = NULL;
        //按地址顺序插入内存块 
        while (curPtr && curPtr->base < inbase) {
            prePtr = curPtr;
            curPtr = curPtr->nextPtr;  
        }
        if (prePtr == NULL) {          //插在最前 
            newnodePtr->nextPtr = curPtr;
            table[i].headPtr = newnodePtr;
        }
        else if (curPtr == NULL) {     //插在最后 
            prePtr->nextPtr = newnodePtr;
        }
        else {                         //插在中间 
            prePtr->nextPtr = newnodePtr;
            newnodePtr->nextPtr = curPtr;
        } 
    }
}

//函数:删除结点
int delete_node (int i, BuddyPtr delPtr)
{
    BuddyPtr prePtr = NULL, curPtr = NULL;
    int basehold = delPtr->base;
    
    curPtr = table[i].headPtr;
    while (curPtr != delPtr) {  //寻找要删除的结点的位置 
        prePtr = curPtr;
        curPtr = curPtr->nextPtr;  
    }
    if (prePtr == NULL)         //要删除的结点在最前 
        table[i].headPtr = curPtr->nextPtr;
    else                        //要删除的结点不在链表的最前 
        prePtr->nextPtr = curPtr->nextPtr;

    free (curPtr);              //释放结点 
    
    return basehold;            //返回删除的内存块结点的基地址 
} 

//函数:伙伴系统的分配算法 
void buddy_allocate (int time_slice)
{
    int i, j, size, due;
    int state = 0;              //分配状态:0为未分配,1为已分配 
    int inbase, basehold;
    BuddyPtr curPtr = NULL;
    
    if (ready == time_slice) {  //到达分配内存的时刻 
        printf ("Time %d:", time_slice);
        size = 1 + rand () % MAX_REQ_SIZE;            //申请使用内存的大小 
        due = MIN_DUE + rand ()%(MAX_DUE - MIN_DUE);  //申请使用内存的时间
        
        if (availSpace > size) {//在可分配空间大于申请空间时分配
            //依次寻找可分配的内存块 
            for (i = 0; (i < INDEX_SIZE) && (state == 0); i ++) {
                //找到一个不小于申请大小的块索引
                if (table[i].nodesize >= size && table[i].headPtr) {
                    curPtr = table[i].headPtr;
                    //遍历相应的循环链表中 
                    while (curPtr && (state == 0)) {
                        //找到空闲块
                        if (curPtr->flag == UNUSED) {
                            //空闲块的大小小于申请大小的2倍,分配 
                            if (table[i].nodesize / size == 1) {
                                //在分配的内存块上设置信息 
                                curPtr->flag = USED;
                                curPtr->occupy = size;
                                curPtr->fragment = table[i].nodesize - size;
                                curPtr->duetime = due + ready;
                                //修改可系统分配空间和碎片大小 
                                availSpace -= table[i].nodesize;
                                totalFragment += curPtr->fragment;
                                state = 1;//标记已分配 
                                break;
                            }
                            //空闲块的大小刚大于申请大小的2倍
                            else {
                                basehold = delete_node (i, curPtr);//删除较大的空闲块并保留其基地址 
                                inbase = basehold + table[i].nodesize;
                                j = i;
                                //分割空闲块 
                                do {
                                    j --;
                                    inbase -= table[j].nodesize;   //设置要添加内存块结点的基地址 
                                    insert_node (j, inbase, UNUSED, 0, 0, 0);//添加较小的空闲块 
                                    printf ("A block cut takes place\n");
                                } while (table[j].nodesize / size > 1);
                                //分配 
                                insert_node (j, basehold, USED, size, table[j].nodesize - size, due + ready);
                                //修改可系统分配空间和碎片大小 
                                availSpace -= table[j].nodesize;
                                totalFragment += table[j].nodesize - size;
                                state = 1;//标记已分配 
                            }
                        }
                        //块被占用,查看下一结点 
                        else
                            curPtr = curPtr->nextPtr;
                    }
                }            
            }
            printf ("Allocated %d,Fragment %d,Due %d\n", size, totalFragment, ready+due);
        }
            
        else if ((availSpace < size) && ((availSpace + totalFragment) >= size))
            printf ("Allocation failed because of fragment!\n");
        else
            printf ("Allocation failed because of no enough unused space!\n");
            
        ready += (1 + rand() % OCCUPY_INTERVAL);  //下次需要分配内存的时刻
    }
}

//函数:伙伴系统的回收算法
void buddy_retrieve (int time_slice)
{
    int i, basehold, dif;
    int f = 0;
	int Modnext=0;
    BuddyPtr curPtr = NULL, todelPtr = NULL; 
    
    //依次查找,并回收需要回收的块 
    for (i = 0; i < INDEX_SIZE; i ++) {
        if (table[i].headPtr) {
            curPtr = table[i].headPtr;
            while (curPtr) {
                if ((curPtr->flag == USED) && (curPtr->duetime == time_slice)) {//需要回收 
                    //修改可系统分配空间和碎片大小 
                    availSpace += table[i].nodesize;
                    totalFragment -= curPtr->fragment;
                    //回收为空闲块 
                    curPtr->flag = UNUSED;
                    curPtr->occupy = 0; 
                    curPtr->fragment = 0;
                    curPtr->duetime = 0;
                    printf ("Time %d:Retrieve %d,Fragment %d\n", time_slice, table[i].nodesize, totalFragment);
                }
                curPtr = curPtr->nextPtr;
            }
        }
    }
    
    //合并空闲块
    for (i = 0; i < INDEX_SIZE; i ++) {
        if (table[i].headPtr) {            
            curPtr = table[i].headPtr;
            while (curPtr && curPtr->nextPtr) {
                //将地址连续且都为空闲的块合并后加入下一级的链表中 
                if (curPtr->flag == UNUSED && (curPtr->nextPtr)->flag == UNUSED) {
                    dif = (curPtr->nextPtr)->base - curPtr->base;
					Modnext = ((int)(curPtr->nextPtr->base))%(2*table[i].nodesize);
                    if ((dif == table[i].nodesize)&&(Modnext==0)) {
                        //删除两个结点 
                        todelPtr = curPtr;
                        curPtr = curPtr->nextPtr;
                        basehold = delete_node (i, todelPtr);
                        todelPtr = curPtr;
                        curPtr = curPtr->nextPtr;
                        delete_node (i, todelPtr);
                        insert_node (i+1, basehold, UNUSED, 0, 0, 0);//添加合并后的结点 
                        printf ("Two blocks merge\n");
                    }
                    else
                        curPtr = curPtr->nextPtr;
                }
                else
                    curPtr = curPtr->nextPtr;
            }
        }
    }
}

//函数:伙伴系统的处理过程 
void buddy_system (void)
{
    int time_slice = 0;
    
    //在每个时间片内使用分配算法和回收算法 
    for (; time_slice < WORKTIME; time_slice ++) {
        buddy_allocate (time_slice);         //分配算法 
        buddy_retrieve (time_slice);         //回收算法 
    }
}


int main(int argc, char *argv[])
{
    int memory_size;
    
    ini_index ();               //初始化哈希索引表 
    srand (time (NULL));        //设置随机数种子
    //随机产生需要管理的内存大小:512M ~ 1G 
    memory_size = MIN_MOMORY_SIZE + rand() % MIN_MOMORY_SIZE; 
    printf ("The size of memory is:%d\n", memory_size);
    
    int_system (memory_size);   //初始化伙伴系统 
    
    buddy_system ();            //伙伴系统的处理过程
    printf ("Time %d:System execution stops and the spaces are all freed.\n", WORKTIME);
    
    free_system ();             //释放所有结点 
    
    return 0;
}



评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值