模仿C库，自己封装一个最简单的文件接口 FILE
1. 创建makefile
创建makefile
testfile: main.c mystdio.c gcc -o $@ $^ .PHONY:clean clean: rm -f testfile
2. mystdio.h ——接口的声明
创建MY_FILE结构体 内部包含文件描述符fd，输出缓冲区ou’tputbuffer 、flags刷新方法
分别通过C库中fopen 、fwrite、fclose 接口的实现，设计属于自己的接口
3. mystdio.c —— 接口的实现
1. MY_fopen的实现
1.识别标志位
分别实现了读、写追加方式
2. 尝试打开文件
若想打开文件，需要调用open函数
若需要创建文件，则需调用第二个open函数
由于open中的mode参数受umask影响，所以设置一个默认的mode
若不需要创建文件，则调用第一个open函数
3. 给用户返回MY_FILE对象，需要先创建对象
判断对象是否创建成功，若失败需要将文件关闭
4.初始化MY_FILE对象
将自己设置的结构体MY_FILE内部的fd赋值为 open函数打开的返回值fd
刷新方法设置成行缓冲
outputbuffer缓冲区中全部初始化为0
current代表缓冲区中没有数据
5.返回打开的文件
当关闭文件的时候，fclose（FILE*) 将C语言当中的文件指针传进来
当关闭文件的时候，C要自己帮助我们进行冲刷缓冲区
为了方便表述，在MY_FILE结构体添加current变量
current代表下次写入时应该写入什么位置
如 outputbuffer中有5个字符 ，对应下标0 1 2 3 4 ，所以cuurrent代表下标5
2.MY_close 的实现
冲刷缓冲区自己实现一个fflush(刷新缓冲区)，叫做MY_fflush
判断缓冲区是否有数据，若有数据就刷新出去
缓冲区为ptr，单个单元的大小为size，nmemb代表想要写入几个单元，写入对应的流中
实际上是往缓冲区里写的
1.缓冲区如果已经满了，就直接写入流中
刷新流的缓冲区
2.根据缓冲区剩余情况，进行拷贝
共分为两种情况，若剩余空间足够，则调用if语句，将用户从ptr拷贝的数据全部拷贝给缓冲区
同时由于缓冲区加入user_size个字节，要更新current的位置
若剩余空间不足够，则调用else语句，将从ptr拷贝的数据填满剩余空间即可
同时由于缓冲区加入MY_size个字节，要更新current的位置
通过调用sriten 代表实际写了多少字节，为了充当最后的的返回值
3. 开始计划刷新
主要分为全刷新和行刷新两种情况，其他不考虑
全刷新判断缓冲区是否满了，若满了则直接刷新缓冲区
行刷新判断是否遇见\n，若遇见\n则直接刷新缓冲区
对之前内容清空
为了防止出现每次打印都会有之前的内容情况，所以刷新之后要清空
在这种情况下，之前的内容会被打印出来
将current置为0后，下次写入就可以覆盖上次缓冲区内容
4. 整体代码
1. main.c
#include"mystdio.h" #include<string.h> #include<unistd.h> #define MYFILE "log.txt" int main() { MY_FILE*fp=MY_fopen(MYFILE,"w"); if(fp==NULL) return 1; const char*str="hello world"; int cnt=5; //操作文件 while(1) { char buffer[1024]; snprintf(buffer,sizeof(buffer),"%s:%d\n",str,cnt--); size_t size=MY_fwrite(buffer,strlen(buffer),1,fp); sleep(1); printf("当前成功写入:%lu个字节\n",size); } MY_fclose(fp); return 0; }
2. mystdio.h
#include<stdio.h> #define NUM 1024 #define BUFF_NONE 0x1 //表示无缓冲 #define BUFF_LINE 0x2 //行缓冲 #define BUFF_ALL 0x4 //全缓冲 typedef struct MY_FILE { int fd;//文件描述符 int flags;//刷新方法 char outputbuffer[1024];//输出缓冲区 int current; }MY_FILE; MY_FILE *MY_fopen(const char *path, const char *mode);//自己写fopen size_t MY_fwrite(const void *ptr, size_t size, size_t nmemb, MY_FILE *stream);//自己写的fwrite int MY_fclose(MY_FILE *fp);//自己写的fwrite int MY_fflush (MY_FILE*fp);//自己实现的缓冲区
3. mystdio.c
#include"mystdio.h" #include<string.h> #include<sys/types.h> #include<sys/stat.h> #include<fcntl.h> #include<stdlib.h> #include<unistd.h> #include<assert.h> MY_FILE*MY_fopen(const char *path, const char *mode)//自己写fopen { int flag=0; if(strcmp(mode,"r")==0)//说明当前使用读方式打开文件 flag |= O_RDONLY;//读取 else if(strcmp(mode,"w")==0) flag |=(O_CREAT | O_WRONLY | O_TRUNC);//创建文件 以写的方式打开文件 清空文件 else if(strcmp(mode,"a")==0) flag |=(O_CREAT | O_WRONLY | O_APPEND); //创建文件 以写的方式打开文件 追加 else { //其他不考虑 } //2. 尝试打开文件 mode_t m=0666; int fd=0; //flag代表模式 r w a if(flag & O_CREAT) fd=open(path,flag,m); else //说明不需要打开 fd=open(path,flag); if(fd<0)//当前打开文件失败 return NULL; //3.给用户返回MY_FILE对象，需要先进行构建 MY_FILE*mf=(MY_FILE*)malloc(sizeof(MY_FILE)); if(mf==NULL)//申请空间失败 { close(fd);//关闭文件 return NULL; } // 4. 初始化 MY_FILE对象 mf->fd=fd;//将上述的fd传入结构体的fd中 mf->flags=0; mf->flags=BUFF_LINE;//设置成行缓冲 memset(mf->outputbuffer,'\0',sizeof(mf->outputbuffer));//将outputbufeer中的内容全部初始化为0 mf->current=0;//代表缓冲区中没有数据 W>} size_t MY_fwrite(const void *ptr, size_t size, size_t nmemb,MY_FILE *stream) { // 1. 缓冲区如果已经满了，就直接写入 if(stream->current == NUM) MY_fflush(stream); // 2. 根据缓冲区剩余情况，进行数据拷贝即可 size_t user_size = size * nmemb; size_t my_size = NUM - stream->current; // 100 - 10 = 90 size_t writen = 0; if(my_size >= user_size) { memcpy(stream->outputbuffer+stream->current, ptr, user_size); //3. 更新计数器字段 stream->current += user_size; writen = user_size; } else { memcpy(stream->outputbuffer+stream->current, ptr, my_size); //3. 更新计数器字段 stream->current += my_size; writen = my_size; } // 4. 开始计划刷新, 他们高效体现在哪里 -- TODO // 不发生刷新的本质，不进行写入，就是不进行IO,不进行调用系统调用，所以MY_fwrite函数调用会非常快,数据会暂时保存在缓冲区中 // 可以在缓冲区中积压多份数据，统一进行刷新写入，本质：就是一次IO可以IO更多的数据，提高IO效率 if(stream->flags & BUFF_ALL) { if(stream->current == NUM) MY_fflush(stream); } else if(stream->flags & BUFF_LINE) { if(stream->outputbuffer[stream->current-1] == '\n') MY_fflush(stream); } else { //TODO } return writen; } int MY_fflush(MY_FILE *fp) { assert(fp); W> int n= write(fp->fd,fp->outputbuffer,fp->current);//将缓冲区中的current个数传入fd中 fp->current=0; return 0; } int MY_fclose(MY_FILE *fp)//自己写的fwrite { assert(fp);//首先要保证fp不为空 //1. 冲刷缓冲区 if(fp->current>0)//说明缓冲区有数据 MY_fflush(fp); //2. 关闭文件 close(fp->fd); //3.释放堆空间 free(fp); //4.指针置为NULL fp=NULL; return 0; }
