Note per il kernel 3.5
[gapil.git] / sources / tee.c
1 /* tee.c
2  * 
3  * Copyright (C) 2007 Simone Piccardi
4  * 
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License as published by
7  * the Free Software Foundation; either version 2 of the License, or (at
8  * your option) any later version.
9  * 
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License for more details.
14  * 
15  * You should have received a copy of the GNU General Public License
16  * along with this program; if not, write to the Free Software
17  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
18  */
19 /****************************************************************
20  *
21  * Program tee.c: 
22  * A sample program that duolicete stdin to stdout and a file
23  *
24  * Author: Simone Piccardi
25  * Aug. 2007
26  *
27  ****************************************************************/
28 /* 
29  * Include needed headers
30  */
31 #define _GNU_SOURCE
32 #include <fcntl.h>       /* file control functions */
33 #include <unistd.h>      /* unix standard library */
34 #include <stdlib.h>      /* C standard library */
35 #include <errno.h>       /* error definitions and routines */
36 #include <stdio.h>       /* standard I/O library */
37 #include <string.h>      /* C strings library */
38 #include <sys/stat.h>    /* file characteristics constants and functions */
39 #include <sys/types.h>   /* primitive system data types */
40 #include <limits.h>      /* system limits constants, types and functions */
41
42 #include "macros.h"
43 /* 
44  * Function and globals definitions
45  */
46 void usage(void);  /* Help printing routine */
47 /*
48  * Main program
49  */
50 int main(int argc, char *argv[])
51 {
52    /*
53     * Variables definition
54     */
55     int i;
56     size_t size = 4096;
57     int fd;
58     int len, nwrite;
59     struct stat fdata;
60     /*
61      * Input section: decode command line parameters 
62      * Use getopt function
63      */
64     opterr = 0;  /* don't want writing to stderr */
65     while ( (i = getopt(argc, argv, "hs:")) != -1) {
66         switch (i) {
67         /* 
68          * Handling options 
69          */ 
70         case 'h':      /* help option */
71             printf("Wrong -h option use\n");
72             usage();
73             return -1;
74             break;
75         case 's':      /* buffer size */
76             size = strtol(optarg, NULL, 10);    /* convert input */
77             break;
78         case '?':      /* unrecognized options */
79             printf("Unrecognized options -%c\n", optopt);
80             usage();
81         default:       /* should not reached */
82             usage();
83         }
84     }
85    /*
86     * Main body
87     */
88     if ((argc - optind) != 1) { /* There must be one argument */
89         printf("Wrong number of arguments %d\n", argc - optind);
90         usage();
91     }
92     /* open destination file and check stdin and stdout */
93     fd = open(argv[1], O_WRONLY|O_CREAT|O_TRUNC, 0644);
94     if (fd == -1) {
95         printf("cannot open destination file %s, %s", argv[1], 
96                strerror(errno));
97         exit(EXIT_FAILURE);
98     }
99     if (fstat(STDIN_FILENO, &fdata) < 0) {
100         perror("stat");
101         exit(EXIT_FAILURE);
102     }
103     if (!S_ISFIFO(fdata.st_mode)) {
104         fprintf(stderr, "stdin must be a pipe\n");
105         exit(EXIT_FAILURE);
106     }
107     if (fstat(STDOUT_FILENO, &fdata) < 0) {
108         perror("stat");
109         exit(EXIT_FAILURE);
110     }
111     if (!S_ISFIFO(fdata.st_mode)) {
112         fprintf(stderr, "stdout must be a pipe\n");
113         exit(EXIT_FAILURE);
114     }
115     /* tee loop */
116     while (1) {
117         /* copy stdin to stdout */
118         len = tee(STDIN_FILENO, STDOUT_FILENO, size, 0);
119         fprintf(stderr, "Copied %d byte\n", len); /* debug (use stderr!) */
120         if (len == 0) break;
121         if (len < 0) {
122             if (errno == EAGAIN) {
123                 continue;
124             } else {
125                 perror("error on tee stdin to stdout");
126                 exit(EXIT_FAILURE);
127             }
128         }
129         /* write data to the file using splice */
130         while (len > 0) {
131             nwrite = splice(STDIN_FILENO, NULL, fd, NULL, len, SPLICE_F_MOVE);
132             if (nwrite < 0) {
133                 perror("error on splice stdin to file");
134                 break;
135             }
136             len -= nwrite;
137         }
138     }
139     close(fd);
140     exit(EXIT_SUCCESS);
141 }
142
143 /*
144  * routine to print usage info and exit 
145  */
146 void usage(void) {
147     printf("Program tee: duplicate stdin to stdout and a file\n");
148     printf("Usage:\n");
149     printf("  tee [-h] [-s N] filename\n");
150     printf("  -h           print this help\n");
151     printf("  -s N         set a buffer size of N bytes \n");
152     exit(1);
153 }
154