sem.c~

来自「操作系统实验一到四 哈尔滨工业大学计算机学院的」· C~ 代码 · 共 120 行

#include <errno.h>
#include <fcntl.h>
#include <sys/types.h>
#include <utime.h>
#include <sys/stat.h>

#include <linux/sched.h>
#include <linux/tty.h>
#include <linux/kernel.h>
#include <asm/segment.h>


#define NR_MAX 21 //信号量名称长度
#define NR_SEMS 10 //系统支持信号量资源个数
typedef struct semaphore
{
    char name[NR_MAX]; //信号量的名字
    int used; //该信号量是否空闲
    int value; //信号量的值
    struct task_struct * wait; //进程在这个指针上等待该信号量
} sem_t;
sem_t array[NR_SEMS];
char myname[NR_MAX];
int in=0,out=0;
int sys_sem_open(const char *name, unsigned int value)
{
    int size,i,pos=-1;
    for(i=0;i<NR_MAX;i++)
		myname[i]='\0';
    for(size=0;(c=get_fs_byte(name+size))!='\0';size++)
    {
		if(size>=NR_MAX-1)
		{
		    sti();
			return NULL;
		}
		else
                	myname[size]=c;
    }
    for(i=0;i<NR_SEMS;i++)
    {
        if(pos==-1&&!array[i].used)
            pos=i;
        if(!strcmp(myname,array[i].name))
        {
            sti();
            return array+i;
        }
    }
    if(pos==-1)
    {
        sti();
        return NULL;
    }
    array[pos].value=value;
    array[pos].wait=NULL;
    for(i=0;i<size;i++)
        array[pos].name[i]=myname[i];
    sti();
    return array+pos;
}

int sys_sem_wait(sem_t *sem)
{
	cli();
	sem.value--;
	if(sem.value<0)
    {
		sleep_on(&sem->wait[in]);
		in=(in+1)%10;
    }
    sti();
	return 0;
}

int sys_sem_post(sem_t *sem)
{
    cli();
    sem.value++;
    if(sem.value<=0)
    {
        wake_up(&sem->wait[out]);
        out=(out+1)%10;
    }
    sti();
    return 0;
}

int sys_sem_unlink(const char *name)
{
    int i,j,flag=0,size;
    cli();
    for(i=0;i<NR_MAX;i++)
		myname[i]='\0';
    for(size=0;(c=get_fs_byte(name+size))!='\0';size++)
    {
		if(size>=NR_MAX-1)
		{
		    sti();
			return -1;
		}
		else
                	myname[size]=c;
    }
    for(i=0;i<NR_SEMS;i++)
    {
        if(!strcmp(myname,array[i].name))
        {
            flag=1;
            array[i].used=0;
            array[i].value=0;
            array[i].wait=NULL;
            sti();
            return 0;
        }
    }
    sti();
    return -1;
}