LU-6142 tests: Fix style issues for statmany.c

[fs/lustre-release.git] / lustre / tests / sendfile.c

diff --git a/lustre/tests/sendfile.c b/lustre/tests/sendfile.c
index 21ae58a..028f02c 100644 (file)
--- a/lustre/tests/sendfile.c
+++ b/lustre/tests/sendfile.c
@@ -1,20 +1,45 @@
+/*
+ * GPL HEADER START
+ *
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 only,
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * General Public License version 2 for more details (a copy is included
+ * in the LICENSE file that accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License
+ * version 2 along with this program; If not, see
+ * http://www.gnu.org/licenses/gpl-2.0.html
+ *
+ * GPL HEADER END
+ */
+/*
+ * Copyright (c) 2008, 2010, Oracle and/or its affiliates. All rights reserved.
+ * Use is subject to license terms.
+ */
+/*
+ * This file is part of Lustre, http://www.lustre.org/
+ */
 
 #include <stdio.h>
 #include <unistd.h>
 #include <fcntl.h>
 #include <stdlib.h>
+#include <errno.h>
+#include <time.h>
 #include <sys/types.h>
 #include <sys/socket.h>
 #include <sys/sendfile.h>
 #include <sys/stat.h>
 #include <sys/socket.h>
 
-#include <liblustre.h>
-#include <lnet/lnetctl.h>
-#include <obd.h>
-#include <lustre_lib.h>
-#include <obd_lov.h>
-#include <lustre/liblustreapi.h>
+#include <lustre/lustreapi.h>
 
 #define syserr(str) { perror(str); exit(-1); }
 
@@ -23,12 +48,12 @@ int main(int argc, char *argv[])
        char *sfile, *tfile;
        struct stat stbuf;
        int size;
+       unsigned long bufsize = 1024 * 1024;
        int infd, outfd;
        int sd[2];
        int rc;
        char *buf;
        char cmd[1024];
-       int page_size = sysconf(_SC_PAGESIZE);
        loff_t pos;
 
        if (argc < 3) {
@@ -42,7 +67,8 @@ int main(int argc, char *argv[])
        if (stat(sfile, &stbuf) < 0) {
                if (errno == ENOENT) {
                        /* assume doing non-object file testing */
-                       infd = open(sfile, O_LOV_DELAY_CREATE|O_CREAT|O_RDWR,
+                       infd = open(sfile,
+                                   O_LOV_DELAY_CREATE | O_CREAT | O_RDWR,
                                    0644);
                        if (infd < 0)
                                syserr("open source file:");
@@ -63,7 +89,7 @@ int main(int argc, char *argv[])
                exit(-1);
        }
 
-       outfd = open(tfile, O_WRONLY|O_TRUNC|O_CREAT, 0666);
+       outfd = open(tfile, O_WRONLY | O_TRUNC | O_CREAT, 0666);
        if (outfd < 0)
                syserr("open dest file:");
 
@@ -71,34 +97,48 @@ int main(int argc, char *argv[])
        if (rc < 0)
                syserr("socketpair");
 
+       rc = fcntl(sd[0], F_SETFL, O_NONBLOCK);
+       if (rc < 0)
+               syserr("fcntl");
+
+       rc = setsockopt(sd[0], SOL_SOCKET, SO_SNDBUF,
+                       &bufsize, sizeof(bufsize));
+       if (rc)
+               syserr("setsockopt");
+
+       srandom(time(NULL));
+
        pos = 0;
        while (size > 0) {
                int rc2;
                size_t seg_size;
 
-               seg_size = (size < page_size) ? size : (random() % size + 1);
-               if (seg_size > 4 * page_size)
-                       seg_size = 4 * page_size;
-               rc = sendfile(sd[0], infd, &pos, seg_size);
-               if (rc < 0)
-                       syserr("sendfile:");
-
-               size -= seg_size;
-               if (size == 0)
-                       close(sd[0]);
-
-               buf = malloc(seg_size);
-               rc = read(sd[1], buf, seg_size);
-               if (rc != seg_size)
-                       syserr("read from socket:");
-
-               rc2 = write(outfd, buf, rc);
-               if (rc2 != rc)
-                       syserr("write dest file error:");
-               free(buf);
+               seg_size = random() % bufsize + 1;
+               if (seg_size > size)
+                       seg_size = size;
+
+               while (seg_size) {
+                       rc = sendfile(sd[0], infd, &pos, seg_size);
+                       if (rc < 0)
+                               syserr("sendfile:");
+
+                       seg_size -= rc;
+                       size -= rc;
+                       if (size == 0)
+                               close(sd[0]);
+
+                       buf = malloc(rc);
+                       if (read(sd[1], buf, rc) < 0)
+                               syserr("read from socket:");
+
+                       rc2 = write(outfd, buf, rc);
+                       if (rc2 != rc)
+                               syserr("write dest file error:");
+                       free(buf);
+               }
        }
        close(sd[1]), close(infd), close(outfd);
 
-       sprintf(cmd, "cmp %s %s\n", sfile, tfile);
+       snprintf(cmd, sizeof(cmd), "cmp %s %s\n", sfile, tfile);
        return system(cmd);
 }