基于Linux系统的流量控制程序的C语言代码

基于Linux系统的流量控制程序的C语言代码

mytbf.h头文件

ifndef MYTBF_H_
#define MYTBF_H_

typedef void mytbf_t;

mytbf_t *tbf_init(int bits,int max);

int gettoken(mytbf_t *tbf,int tknum);

int returntoken(mytbf_t *tbf,int rtnum);

int tbffree(mytbf_t *tbf);

#endif

mytbf.c文件

#include <stdio.h>
#include <stdlib.h>
#include <mytbf.h>
#include <signal.h>

struct tbf_t
{
    int bits;
    int token;
    int maxtoken;
};

struct tbf_t *data=NULL;
int initflag=1;

void sigdo(int s)
{
    data->token=data->token+data->bits;
    if(data->token>=data->maxtoken)
        data->token=data->token;
    alarm(1);
}

mytbf_t *tbf_init(int bits,int max)
{
    data=malloc(sizeof(*data));
    if(bits < 0)
        return NULL;
    if(bits > max)
        return NULL;

    data->bits=bits;
    data->token=bits;
    data->maxtoken=max;

    if(initflag)
    {
        signal(SIGALRM,sigdo);
        alarm(1);
        initflag=0;
    }

    return data;
}

int gettoken(mytbf_t *tbf,int tknum)
{
    int min;
    data=(struct tbf_t *)tbf;
    if(tknum < 0)
        return -1;
    if(tknum > data->maxtoken)
        return -2;

    while(data->token == 0)
        pause();

    min=data->token > tknum ? tknum:data->token;
    data->token=data->token-min;

    return min;
}

int returntoken(mytbf_t *tbf,int rtnum)
{
    data=(struct tbf_t *)tbf;
    if(rtnum < 0)
        return -3;
    if(data->token > data->maxtoken-rtnum)
        data->token=data->maxtoken;
    else
        data->token=data->token+rtnum;
    return 0;
}

int tbffree(mytbf_t *tbf)
{
    free(tbf);
    return 0;
}

mycat.c文件：实现Linux终端命令cat的功能，显示文件内容，不过为这个程序加上了流量控制。

#include <stdio.h>
#include <sys/stat.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <errno.h>
#include <signal.h>

#include <mytbf.h>

#define SIZE 20
#define MAXSIZE 500
static int fd;

void func_closefile()
{
    close(fd);
}

int main(int argc,char **argv)
{
    char buff[SIZE];
    int res;
    int res1;
    int retnum;
    mytbf_t *tbf;
    if(argc<2)
    {
        fprintf(stdout,"出错，命令后没有文件名！\n");
        exit(-1);
    }

    do{
        fd=open(argv[1],O_RDONLY);
        if(fd < 0)
        {
            if(errno == EINTR)
                continue;
            perror("open()");
            exit(-2);
        }
    }while(fd < 0);

    atexit(func_closefile);

    tbf = tbf_init(SIZE,MAXSIZE);

    while(1)
    {
        retnum=gettoken(tbf,MAXSIZE);
        res = read(fd,buff,retnum);
        if(retnum-res > 0)
        {
            returntoken(tbf,retnum-res);
        }

        if(res==0)
            break;
        if(res<0)
        {
            if(errno == EINTR)
                continue;
            perror("read()");
            exit(-3);
        }

        do{
            res1 = write(1,buff,res);
            if(res1 < 0)
            {
                if(errno == EINTR)
                    continue;
                perror("write()");
                exit(-4);
            }
        }while(res1<0);
    }
    tbffree(tbf);
    exit(0);
}

`