标签：readcnt 流控 -- int while exit fd 漏桶 include
漏桶示例：slowcat.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#define READLENTH 10 //设置一次读取长度
static volatile int loop =1; //标志
static void alrm_handler(int s) //闹钟事件处理
{
alarm(1); //循环设置闹钟
loop = 0;
}
int main(int argc, char**argv)
{
int fd;
int readcnt;
int writecnt;
int pos;
char readbuff[READLENTH];
if(argc<2) //参数判断
{
fprintf(stderr,"Usage:%s <filename>\n",argv[0]);
exit(1);
}
signal(SIGALRM,alrm_handler); //注册信号处理函数
alarm(1); //创建闹钟信号
while(1)
{
fd = open(argv[1],O_RDONLY); //打开文件
if(fd <0 && errno !=EINTR) //当不是中断导致打开出错时
{
perror("open()");
exit(1);
}
else if(fd >=0) //正常打开
break;
}
while(1)
{
while(loop) //判断是否有闹钟事件到
pause();
loop = 1;
while((readcnt = read(fd, readbuff, READLENTH))<0) //读取数据
{
if(errno!=EINTR) //不为中断导致读取出错时
{
perror("read()");
exit(1);
}
else
continue;
}
if(readcnt ==0)
break;
pos = 0;
while(readcnt>0)
{
writecnt = write(1,readbuff+pos,readcnt); //向标准输出写入数据
if(writecnt<0) //写入错误时
{
if(errno == EINTR)
continue;
perror("write()");
exit(1);
}
pos +=writecnt;
readcnt -=writecnt;
}
}
close(fd);
exit(0);
}
令牌桶示例：slowcat2.c
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <signal.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <errno.h>
#define READLENTH 10
#define MAXTOKEN 100 //最大令牌数
static volatile int token =0; //令牌数
static void alrm_handler(int s) //闹钟信号处理函数
{
alarm(1); //循环产生闹钟事件
token++; //令牌数自加
if(token>=MAXTOKEN) //限制令牌最大数
token == MAXTOKEN;
}
int main(int argc, char**argv)
{
int fd;
int readcnt;
int writecnt;
int pos;
char readbuff[READLENTH];
if(argc<2) //判断输入参数
{
fprintf(stderr,"Usage:%s <filename>\n",argv[0]);
exit(1);
}
signal(SIGALRM,alrm_handler); //注册闹钟处理函数
alarm(1); //创建闹钟
while(1)
{
fd = open(argv[1],O_RDONLY); //打开文件
if(fd <0 && errno !=EINTR)
{
perror("open()");
exit(1);
}
else if(fd >=0)
break;
}
while(1)
{
while(token<=0) //没有令牌数时
pause();
token--; //令牌数自减
while((readcnt = read(fd, readbuff, READLENTH))<0) //读取数据
{
if(errno!=EINTR)
{
perror("read()");
exit(1);
}
else
continue;
}
if(readcnt ==0)
break;
pos = 0;
while(readcnt>0) //循环写入数据
{
writecnt = write(1,readbuff+pos,readcnt);
if(writecnt<0)
{
if(errno == EINTR)
continue;
perror("write()");
exit(1);
}
pos +=writecnt;
readcnt -=writecnt;
}
}
close(fd);
exit(0);
}
标签：readcnt,流控,--,int,while,exit,fd,漏桶,include 来源： https://www.cnblogs.com/linux-learn/p/16485308.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。