sources/splicecp.c

   1 /* splicecp.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 splicecp.c:
  22  * A sample program that copy a file using splice
  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
  41 #include "macros.h"
  42 /*
  43  * Function and globals definitions
  44  */
  45 void usage(void);  /* Help printing routine */
  46 /*
  47  * Main program
  48  */
  49 int main(int argc, char *argv[])
  50 {
  51    /*
  52     * Variables definition
  53     */
  54     int i;
  55     int size = 4096;
  56     int pipefd[2];
  57     int in_fd, out_fd;
  58     int nread, nwrite;
  59     /*
  60      * Input section: decode command line parameters
  61      * Use getopt function
  62      */
  63     opterr = 0;  /* don't want writing to stderr */
  64     while ( (i = getopt(argc, argv, "hs:")) != -1) {
  65         switch (i) {
  66         /*
  67          * Handling options
  68          */
  69         case 'h':      /* help option */
  70             printf("Wrong -h option use\n");
  71             usage();
  72             return -1;
  73             break;
  74         case 's':      /* take wait time for childen */
  75             size = strtol(optarg, NULL, 10);    /* convert input */
  76             break;
  77         case '?':      /* unrecognized options */
  78             printf("Unrecognized options -%c\n", optopt);
  79             usage();
  80         default:       /* should not reached */
  81             usage();
  82         }
  83     }
  84    /*
  85     * Main body
  86     */
  87     if ((argc - optind) != 2) { /* There must two argument */
  88         printf("Wrong number of arguments %d\n", argc - optind);
  89         usage();
  90     }
  91     /* open pipe, input and output file */
  92     in_fd = open(argv[optind], O_RDONLY);
  93     if (in_fd < 0) {
  94         printf("Input error %s on %s\n", strerror(errno), argv[optind]);
  95         exit(EXIT_FAILURE);
  96     }
  97     out_fd = open(argv[optind+1], O_CREAT|O_RDWR|O_TRUNC, 0644);
  98     if (out_fd < 0) {
  99         printf("Cannot open %s, error %s\n", argv[optind+1], strerror(errno));
 100         exit(EXIT_FAILURE);
 101     }
 102     if (pipe(pipefd) == -1) {
 103         perror("Cannot create buffer pipe");
 104         exit(EXIT_FAILURE);
 105     }
 106     /* copy loop */
 107     debug("Size %d\n", size);
 108     while (1) {
 109         nread = splice(in_fd, NULL, pipefd[1], NULL, size,
 110                        SPLICE_F_MOVE|SPLICE_F_MORE);
 111         debug("read %d bytes\n", nread);
 112         if (nread == 0) break;
 113         if (nread < 0) {
 114             if (errno == EINTR) {
 115                 continue;
 116             } else {
 117                 perror("read error");
 118                 exit(EXIT_FAILURE);
 119             }
 120         }
 121         while (nread > 0) {
 122             nwrite = splice(pipefd[0], NULL, out_fd, NULL, nread,
 123                             SPLICE_F_MOVE|SPLICE_F_MORE);
 124             debug("write %d bytes\n", nwrite);
 125             if (nwrite < 0) {
 126                 if (errno == EINTR)
 127                     continue;
 128                 else {
 129                     perror("write error");
 130                     exit(EXIT_FAILURE);
 131                 }
 132             }
 133             nread -= nwrite;
 134             debug("left %d bytes\n", nread);
 135         }
 136     }
 137     return EXIT_SUCCESS;
 138 }
 139 /*
 140  * routine to print usage info and exit
 141  */
 142 void usage(void) {
 143     printf("Program splicecp: copy two file using splice syscall\n");
 144     printf("Usage:\n");
 145     printf("  splicecp [-h] [-s N] filesrc filedst \n");
 146     printf("  -h           print this help\n");
 147     printf("  -s N         set a buffer size of N bytes \n");
 148     exit(1);
 149 }