Be less scary

[nbd.git] / nbd-server.c

diff --git a/nbd-server.c b/nbd-server.c
index 30b6181..2e3a673 100644 (file)
--- a/nbd-server.c
+++ b/nbd-server.c
@@ -96,6 +96,10 @@
 #define MY_NAME "nbd_server"
 #include "cliserv.h"
 
+#ifdef WITH_SDP
+#include <sdp_inet.h>
+#endif
+
 /** Default position of the config file */
 #ifndef SYSCONFDIR
 #define SYSCONFDIR "/etc"
@@ -515,6 +519,8 @@ SERVER* cmdline(int argc, char *argv[]) {
        if(nonspecial<2) {
                g_free(serve);
                serve=NULL;
+       } else {
+               do_oldstyle = TRUE;
        }
        if(do_output) {
                if(!serve) {
@@ -570,20 +576,20 @@ SERVER* dup_serve(SERVER *s) {
        SERVER *serve = NULL;
 
        serve=g_new0(SERVER, 1);
-       if (serve == NULL)
+       if(serve == NULL)
                return NULL;
 
-       if (s->exportname)
+       if(s->exportname)
                serve->exportname = g_strdup(s->exportname);
 
        serve->expected_size = s->expected_size;
 
-       if (s->listenaddr)
+       if(s->listenaddr)
                serve->listenaddr = g_strdup(s->listenaddr);
 
        serve->port = s->port;
 
-       if (s->authname)
+       if(s->authname)
                serve->authname = strdup(s->authname);
 
        serve->flags = s->flags;
@@ -591,11 +597,14 @@ SERVER* dup_serve(SERVER *s) {
        serve->socket_family = serve->socket_family;
        serve->cidrlen = s->cidrlen;
 
-       if (s->prerun)
+       if(s->prerun)
                serve->prerun = g_strdup(s->prerun);
 
-       if (s->postrun)
+       if(s->postrun)
                serve->postrun = g_strdup(s->postrun);
+       
+       if(s->servename)
+               serve->servename = g_strdup(s->servename);
 
        return serve;
 }
@@ -829,6 +838,10 @@ GArray* parse_cfile(gchar* f, GError** e) {
                                g_key_file_free(cfile);
                                return NULL;
                        }
+                       if(s.port && !do_oldstyle) {
+                               g_warning("A port was specified, but oldstyle exports were not requested. This may not do what you expect.");
+                               g_warning("Please read 'man 5 nbd-server' and search for oldstyle for more info");
+                       }
                } else {
                        s.virtstyle=VIRT_IPLIT;
                }
@@ -1223,12 +1236,19 @@ CLIENT* negotiate(int net, CLIENT *client, GArray* servers) {
 
        memset(zeros, '\0', sizeof(zeros));
        if(!client || !client->modern) {
+               /* common */
                if (write(net, INIT_PASSWD, 8) < 0) {
                        err_nonfatal("Negotiation failed: %m");
                        if(client)
                                exit(EXIT_FAILURE);
                }
-               magic = htonll(opts_magic);
+               if(!client || client->modern) {
+                       /* modern */
+                       magic = htonll(opts_magic);
+               } else {
+                       /* oldstyle */
+                       magic = htonll(cliserv_magic);
+               }
                if (write(net, &magic, sizeof(magic)) < 0) {
                        err_nonfatal("Negotiation failed: %m");
                        if(client)
@@ -1236,9 +1256,10 @@ CLIENT* negotiate(int net, CLIENT *client, GArray* servers) {
                }
        }
        if(!client) {
-               uint64_t reserved;
+               /* modern */
+               uint32_t reserved;
                uint32_t opt;
-               uint64_t namelen;
+               uint32_t namelen;
                char* name;
                int i;
 
@@ -1248,7 +1269,7 @@ CLIENT* negotiate(int net, CLIENT *client, GArray* servers) {
                read(net, &reserved, sizeof(reserved));
                read(net, &magic, sizeof(magic));
                magic = ntohll(magic);
-               if(magic != cliserv_magic) {
+               if(magic != opts_magic) {
                        close(net);
                        return NULL;
                }
@@ -1259,10 +1280,10 @@ CLIENT* negotiate(int net, CLIENT *client, GArray* servers) {
                        return NULL;
                }
                read(net, &namelen, sizeof(namelen));
-               namelen = ntohll(namelen);
+               namelen = ntohl(namelen);
                name = malloc(namelen+1);
-               name[namelen+1]=0;
-               read(net, &name, namelen);
+               name[namelen]=0;
+               read(net, name, namelen);
                for(i=0; i<servers->len; i++) {
                        SERVER* serve = &(g_array_index(servers, SERVER, i));
                        if(!strcmp(serve->servename, name)) {
@@ -1274,15 +1295,28 @@ CLIENT* negotiate(int net, CLIENT *client, GArray* servers) {
                                return client;
                        }
                }
+               return NULL;
        }
+       /* common */
        size_host = htonll((u64)(client->exportsize));
        if (write(net, &size_host, 8) < 0)
                err("Negotiation failed: %m");
        if (client->server->flags & F_READONLY)
                flags |= NBD_FLAG_READ_ONLY;
-       flags = htonl(flags);
-       if (write(client->net, &flags, 4) < 0)
-               err("Negotiation failed: %m");
+       if (!client->modern) {
+               /* oldstyle */
+               flags = htonl(flags);
+               if (write(client->net, &flags, 4) < 0)
+                       err("Negotiation failed: %m");
+       } else {
+               /* modern */
+               smallflags = (uint16_t)(flags & ~((uint16_t)0));
+               smallflags = htons(smallflags);
+               if (write(client->net, &smallflags, sizeof(smallflags)) < 0) {
+                       err("Negotiation failed: %m");
+               }
+       }
+       /* common */
        if (write(client->net, zeros, 124) < 0)
                err("Negotiation failed: %m");
        return NULL;
@@ -1660,13 +1694,14 @@ int serveloop(GArray* servers) {
        max=0;
        FD_ZERO(&mset);
        for(i=0;i<servers->len;i++) {
-               sock=(g_array_index(servers, SERVER, i)).socket;
-               FD_SET(sock, &mset);
-               max=sock>max?sock:max;
+               if((sock=(g_array_index(servers, SERVER, i)).socket)) {
+                       FD_SET(sock, &mset);
+                       max=sock>max?sock:max;
+               }
        }
        if(modernsock) {
                FD_SET(modernsock, &mset);
-               max=modernsock>max?sock:max;
+               max=modernsock>max?modernsock:max;
        }
        for(;;) {
                CLIENT *client = NULL;
@@ -1685,6 +1720,7 @@ int serveloop(GArray* servers) {
                                if(!client) {
                                        err_nonfatal("negotiation failed");
                                        close(net);
+                                       net=0;
                                }
                        }
                        for(i=0;i<servers->len && !net;i++) {
@@ -2061,8 +2097,11 @@ int main(int argc, char *argv[]) {
        }
     
        if(!servers || !servers->len) {
-               g_warning("Could not parse config file: %s", 
-                               err ? err->message : "Unknown error");
+               if(err && !(err->domain == g_quark_from_string("parse_cfile")
+                               && err->code == CFILE_NOTFOUND)) {
+                       g_warning("Could not parse config file: %s", 
+                                       err ? err->message : "Unknown error");
+               }
        }
        if(serve) {
                g_warning("Specifying an export on the command line is deprecated.");
@@ -2070,7 +2109,7 @@ int main(int argc, char *argv[]) {
        }
 
        if((!serve) && (!servers||!servers->len)) {
-               g_message("Nothing to do! Bye!");
+               g_message("No configured exports; quitting.");
                exit(EXIT_FAILURE);
        }
        daemonize(serve);