libsysio/tests/test_copy.c

   1 /*
   2  *    This Cplant(TM) source code is the property of Sandia National
   3  *    Laboratories.
   4  *
   5  *    This Cplant(TM) source code is copyrighted by Sandia National
   6  *    Laboratories.
   7  *
   8  *    The redistribution of this Cplant(TM) source code is subject to the
   9  *    terms of the GNU Lesser General Public License
  10  *    (see cit/LGPL or http://www.gnu.org/licenses/lgpl.html)
  11  *
  12  *    Cplant(TM) Copyright 1998-2003 Sandia Corporation.
  13  *    Under the terms of Contract DE-AC04-94AL85000, there is a non-exclusive
  14  *    license for use of this work by or on behalf of the US Government.
  15  *    Export of this program may require a license from the United States
  16  *    Government.
  17  */
  18
  19 /*
  20  * This library is free software; you can redistribute it and/or
  21  * modify it under the terms of the GNU Lesser General Public
  22  * License as published by the Free Software Foundation; either
  23  * version 2.1 of the License, or (at your option) any later version.
  24  *
  25  * This library is distributed in the hope that it will be useful,
  26  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  27  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  28  * Lesser General Public License for more details.
  29  *
  30  * You should have received a copy of the GNU Lesser General Public
  31  * License along with this library; if not, write to the Free Software
  32  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  33  *
  34  * Questions or comments about this library should be sent to:
  35  *
  36  * Lee Ward
  37  * Sandia National Laboratories, New Mexico
  38  * P.O. Box 5800
  39  * Albuquerque, NM 87185-1110
  40  *
  41  * lee@sandia.gov
  42  */
  43
  44 #define _BSD_SOURCE
  45
  46 #include <stdio.h>
  47 #include <stdlib.h>
  48 #include <unistd.h>
  49 #include <errno.h>
  50 #include <sys/types.h>
  51 #include <sys/stat.h>
  52 #include <fcntl.h>
  53 #include <sys/uio.h>
  54 #include <sys/queue.h>
  55
  56 #if defined(SYSIO_LABEL_NAMES)
  57 #include "sysio.h"
  58 #endif
  59 #include "xtio.h"
  60 #include "test.h"
  61
  62 /*
  63  * Copy one file to another.
  64  *
  65  * Usage: test_copy [-o] <src> <dest>
  66  *
  67  * Destination will not be overwritten if it already exist.
  68  */
  69
  70 static int overwrite = 0;                               /* over-write? */
  71
  72 void    usage(void);
  73 int     copy_file(const char *spath, const char *dpath);
  74
  75 int
  76 main(int argc, char * const argv[])
  77 {
  78         int     i;
  79         int     err;
  80         const char *spath, *dpath;
  81
  82         /*
  83          * Parse command-line args.
  84          */
  85         while ((i = getopt(argc,
  86                            argv,
  87                            "o"
  88                            )) != -1)
  89                 switch (i) {
  90
  91                 case 'o':
  92                         overwrite = 1;
  93                         break;
  94                 default:
  95                         usage();
  96                 }
  97
  98         if (!(argc - optind))
  99                 usage();
 100         err = _test_sysio_startup();
 101         if (err) {
 102                 errno = -err;
 103                 perror("sysio startup");
 104                 exit(1);
 105         }
 106
 107         /*
 108          * Source
 109          */
 110         spath = argv[optind++];
 111         if (!(argc - optind))
 112                 usage();
 113         /*
 114          * Destination
 115          */
 116         dpath = argv[optind++];
 117         if (argc - optind)
 118                 usage();
 119
 120         err = copy_file(spath, dpath);
 121
 122         _test_sysio_shutdown();
 123
 124         return err;
 125 }
 126
 127 void
 128 usage()
 129 {
 130
 131         (void )fprintf(stderr,
 132                        "Usage: test_copy "
 133                        " source destination\n");
 134         exit(1);
 135 }
 136
 137 int
 138 open_file(const char *path, int flags, mode_t mode)
 139 {
 140         int     fd;
 141
 142         fd = SYSIO_INTERFACE_NAME(open)(path, flags, mode);
 143         if (fd < 0)
 144                 perror(path);
 145
 146         return fd;
 147 }
 148
 149 int
 150 copy_file(const char *spath, const char *dpath)
 151 {
 152         int     sfd, dfd;
 153         int     flags;
 154         int     rtn;
 155         struct stat stat;
 156         char    *buf;
 157         size_t  bufsiz;
 158         ssize_t cc, wcc;
 159
 160         sfd = dfd = -1;
 161         rtn = -1;
 162         buf = NULL;
 163
 164         sfd = open_file(spath, O_RDONLY, 0);
 165         if (sfd < 0)
 166                 goto out;
 167         flags = O_CREAT|O_WRONLY;
 168         if (!overwrite)
 169                 flags |= O_EXCL;
 170         dfd = open_file(dpath, flags, 0666);
 171         if (dfd < 0)
 172                 goto out;
 173
 174         rtn = SYSIO_INTERFACE_NAME(fstat)(dfd, &stat);
 175         if (rtn != 0) {
 176                 perror(dpath);
 177                 goto out;
 178         }
 179         bufsiz = stat.st_blksize;
 180         if (bufsiz < (64 * 1024))
 181                 bufsiz =
 182                     (((64 * 1024) / stat.st_blksize - 1) + 1) * (64 * 1024);
 183         buf = malloc(bufsiz);
 184         if (!buf) {
 185                 perror(dpath);
 186                 goto out;
 187         }
 188
 189         while ((cc = SYSIO_INTERFACE_NAME(read)(sfd, buf, bufsiz)) > 0)
 190                 if ((wcc = SYSIO_INTERFACE_NAME(write)(dfd, buf, cc)) != cc) {
 191                         if (wcc < 0) {
 192                                 perror(dpath);
 193                                 break;
 194                         }
 195                         (void )fprintf(stderr,
 196                                        "%s: short write (%u/%u)\n",
 197                                        dpath,
 198                                        (unsigned )wcc,
 199                                        (unsigned )cc);
 200                         break;
 201                 }
 202         if (cc < 0) {
 203                 perror(spath);
 204                 rtn = -1;
 205         }
 206
 207 out:
 208         if (buf)
 209                 free(buf);
 210         if (sfd >= 0 && SYSIO_INTERFACE_NAME(close)(sfd) != 0)
 211                 perror(spath);
 212         if (dfd >= 0 &&
 213             (SYSIO_INTERFACE_NAME(fsync)(dfd) != 0 ||
 214              SYSIO_INTERFACE_NAME(close)(dfd) != 0))
 215                 perror(dpath);
 216
 217         return rtn;
 218 }