第3章 文件I/O（4）_dup、dup2、fcntl和ioctl函数

#ifndef __IO_H__
#define __IO_H__

extern void copy(int fdin, int fdout);//文件复制
extern void set_fl(int fd, int flag); //设置文件状态标志
extern void clr_fl(int fd, int flag); //取消文件状态标志


#endif

#include "io.h"
#include <unistd.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>

//编译命令：gcc -o obj/io.o -Iinclude -c src/io.c

#define BUFFER_LEN 1024 //与分区文件块大小一致。可以通过
                        //tune2fs -l /dev/sda1命令查看
                        

void copy(int fdin, int fdout)
{
    char buffer[BUFFER_LEN];
    
    ssize_t size;

    //保证从文件开始处复制
    lseek(fdin, 0L, SEEK_SET); 
    lseek(fdout, 0L, SEEK_SET); 

    while((size = read(fdin, buffer, BUFFER_LEN)) > 0){
        if(write(fdout, buffer, size) != size)
        {
           fprintf(stderr, "write error: %s \n", strerror(errno));
           exit(1);
        }
    }

    if (size < 0 )
    {
        fprintf(stderr, "read error: %s\n",strerror(errno));
        exit(1);  //return 1;
    }   
}

void set_fl(int fd, int flag) //设置文件状态标志
{
    //获取原来的文件状态标志
    int val = fcntl(fd, F_GETFL);

    //增加新的文件状态标志
    val |= flag;

    //重新设置文件状态标志
    if(fcntl(fd, F_SETFL, val) < 0)
    {
        perror("fcntl error");
    }
}

void clr_fl(int fd, int flag) //取消文件状态标志
{

    //获取原来的文件状态标志
    int val = fcntl(fd, F_GETFL, val);

    //清除指定的文件状态标志（置0）
    val &= ~flag;

    //重新设置文件状态标志
    if(fcntl(fd, F_SETFL, val) < 0 )
    {
        perror("fcntl error");
    }
}

#include "io.h"
#include <unistd.h>
#include <fcntl.h>
#include <stdio.h>

//模拟cat命令，设编译后的程序名也叫cat
//实验1：bin/cat file1.txt file2.txt
//实验2：bin/cat < file1.txt
//实验3：bin/cat > file1.txt

int main(int argc, char* argv[])
{
    int fd_in = STDIN_FILENO;    //0
    int fd_out = STDOUT_FILENO;  //1

    int i = 0;
    for(i=1; i<argc; i++){
        //1.输入cat file1.txt file2.txt时会将file1和file2
        //  分别拷贝到标准输出，即输出到屏幕
        fd_in = open(argv[1], O_RDONLY);
        if(fd_in < 0)
        {
            perror("open error");
            continue;
        }

        copy(fd_in, fd_out);
        close(fd_in);
    }
    //2.如果cat命令后面不带参数，则直接接受键盘输入。
    //3.如果输入cat < file1.txt，由于<表示输入重定向，Linux
    //会将file1.txt这个参数作为输入，复制给STDIN_FILENO。而不会将其
    //传给argv参数。即这种情况下，argc==1，表示程序本身，argv中不包含file1.txt。
    if(argc == 1) copy(fd_in, fd_out);
    
    return 0;
}

#include "io.h"
#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <stdlib.h>
/*
 * bin/mcat + file1.txt (+为输入重定向)
 * bin/mcat - file1.txt (-为输出重定向)
 */

int main(int argc, char* argv[])
{
    int fd_in;
    int fd_out;

    int i=0;

    for(i=1; i<argc; i++){
        if(!strcmp("+", argv[i])){
            fd_in = open(argv[++i], O_RDONLY);
            if(fd_in < 0){
                perror("open error");
                exit(1);
            }

            //将标准输入重定向到文件(即，将fd_in的文件表项指针复制给STDIN_FILENO
            //因此，STDIN_FILENO文件描述符与fd_in就同时指向同一个文件表项，也就
            //同时指向了file1.txt这个文件。
            if(dup2(fd_in, STDIN_FILENO) != STDIN_FILENO){
                perror("dup2 error");
                exit(1);
            }

            close(fd_in);
        }else if(!strcmp("-", argv[i])){

            fd_out = open(argv[++i], O_WRONLY | O_CREAT | O_TRUNC, 0777);
            if(fd_out < 0){
                perror("open error");
                exit(1);
            }

            //将标准输出重定向到文件(即，将fd_out的文件表项指针复制给STDOUT_FILENO
            //因此，STDOUT_FILENO文件描述符与fd_out就同时指向同一个文件表项，也就
            //同时指向了file1.txt这个文件。
            if(dup2(fd_out, STDOUT_FILENO) != STDOUT_FILENO){
                perror("dup2 error");
                exit(1);
            }

            close(fd_out);
        }else{
              fd_in = open(argv[i], O_RDONLY);
              if(fd_in < 0){
                  perror("open error");
                  exit(1);
              }

              if(dup2(fd_in, STDIN_FILENO) != STDIN_FILENO)
              {
                  perror("dup2 error");
                  exit(1);
              }

              close(fd_in);
        }

        copy(STDIN_FILENO, STDOUT_FILENO);
    }

   //命令后面不带参数的情况
   if(argc ==1)
      copy(STDIN_FILENO, STDOUT_FILENO);
      
    return 0;
}

#include "io.h"
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h> //exit
#include <string.h> //strlen
#include <fcntl.h>  //O_WRONLY

int main(int argc, char* argv[]){
    if( argc < 3){
        printf("usage: %s content destfile\n", argv[0]);
        exit(1);
    }

    int fd = open(argv[2], O_WRONLY); //注意，这里没设置为追加
    
    if(fd < 0){
        perror("open error");
        exit(1);
    }

    //设置追加文件标志
    set_fl(fd, O_APPEND);
    
    sleep(10); //为了把定位与写入过程隔开，以便演示多进程同时写入同一文件
               //时会出现后启动进程格覆盖之前进程写过的内容。
               
    //往文件尾部追加内容
    size_t size = strlen(argv[1])*sizeof(char);
    if(write(fd, argv[1], size)!=size){
        perror("write error");
        exit(1);
    }

    close(fd);

    return 0;
}

#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/ioctl.h>  //for ioctl  
#include <linux/input.h> //keyboard

//说明本程序须在虚拟机，而不能在远程终端上运行！

int main(int argc, char * argv[])
{
    int fd = -1;
    char name[256]="unknow";
    struct input_event event;

    int ret = 0;

    if(( fd = open("/dev/input/event2", O_RDONLY)) < 0){
        perror("open error");
        exit(1);
    }

    //通用EVIOCGNAME命令获取设备名字
    if(ioctl(fd, EVIOCGNAME(sizeof(name)), name) < 0){
        perror("evdev ioctl error\n");
        exit(1);
    }

    printf("The device‘s name is %s\n", name);
    
    while(1)
    {   
        ret = read(fd, &event, sizeof(event));
        if(ret < 0){
            printf("read event error\n");
        }

        if(EV_KEY == event.type) //EV_KEY表示按键类事件，类似的有EV_PWR(电源)、EV_SND（声音）事件
        {
            //if the event is a key code
            printf(" key code is %d \n", event.code);

            //quit，press ‘q‘ to quit this application.
            if(event.code == 16){
                break;
            }
        }
    }

    return 0;
}