r279: * Mention -persist in nbd-client manpage

[nbd.git] / nbd-client.c

diff --git a/nbd-client.c b/nbd-client.c
index f4836c5..8771d40 100644 (file)
--- a/nbd-client.c
+++ b/nbd-client.c
@@ -160,6 +160,7 @@ int main(int argc, char *argv[]) {
        char *hostname, *nbddev;
        int swap=0;
        int cont=0;
+       int timeout=0;
        u64 size64;
        u32 flags;
 
@@ -222,7 +223,7 @@ int main(int argc, char *argv[]) {
          err("Can not open NBD: %m");
        ++argv; --argc; /* skip device */
 
-       if (argc>2) goto errmsg;
+       if (argc>3) goto errmsg;
        if (argc!=0) {
                if(strncmp(argv[0], "-swap", 5)==0) {
                        swap=1;
@@ -235,6 +236,17 @@ int main(int argc, char *argv[]) {
                        ++argv;--argc;
                }
        }
+       if(argc!=0) {
+               if(strncmp(argv[0], "-timeout", 8)==0) {
+#ifdef NBD_SET_TIMEOUT
+                       timeout=strtol(++argv, NULL, 0);
+                       ++argv;--argc--;
+#else
+                       fprintf(stderr, "This nbd-client does not support -timeout yet; please recompile on a more recent kernel");
+                       exit(EXIT_FAILURE);
+#endif
+               }
+       }
        argv=NULL; argc=0; /* don't use it later suddenly */
 
        negotiate(sock, &size64, &flags);
@@ -249,6 +261,12 @@ int main(int argc, char *argv[]) {
                exit(0);
 #endif
 
+#ifdef NBD_SET_TIMEOUT
+       if(ioctl(nbd, NBD_SET_TIMEOUT, &timeout)<0) {
+               fprintf(stderr, "Setting timeout failed: %m");
+               exit(EXIT_FAILURE);
+       }
+#endif
        do {
                if (ioctl(nbd, NBD_DO_IT) < 0) {
                        fprintf(stderr, "Kernel call returned: %m");