cliserv.h

   1 /* This header file is shared by client & server. They really have
   2  * something to share...
   3  * */
   4
   5 /* Client/server protocol is as follows:
   6    Send INIT_PASSWD
   7    Send 64-bit cliserv_magic
   8    Send 64-bit size of exported device
   9    Send 128 bytes of zeros (reserved for future use)
  10  */
  11
  12 #include <errno.h>
  13 #include <string.h>
  14 #include <netdb.h>
  15 #include <netinet/tcp.h>
  16 #include <stdlib.h>
  17
  18 #if SIZEOF_UNSIGNED_SHORT_INT==4
  19 typedef unsigned short u32;
  20 #elif SIZEOF_UNSIGNED_INT==4
  21 typedef unsigned int u32;
  22 #elif SIZEOF_UNSIGNED_LONG_INT==4
  23 typedef unsigned long u32;
  24 #else
  25 #error I need at least some 32-bit type
  26 #endif
  27
  28 #if SIZEOF_UNSIGNED_INT==8
  29 typedef unsigned int u64;
  30 #elif SIZEOF_UNSIGNED_LONG_INT==8
  31 typedef unsigned long u64;
  32 #elif SIZEOF_UNSIGNED_LONG_LONG_INT==8
  33 typedef unsigned long long u64;
  34 #else
  35 #error I need at least some 64-bit type
  36 #endif
  37
  38 #define __be32 u32
  39 #define __be64 u64
  40 #include "nbd.h"
  41
  42 #if NBD_LFS==1
  43 #define _LARGEFILE_SOURCE
  44 #define _FILE_OFFSET_BITS 64
  45 #endif
  46
  47 u64 cliserv_magic = 0x00420281861253LL;
  48 u64 opts_magic = 0x49484156454F5054LL;
  49 #define INIT_PASSWD "NBDMAGIC"
  50
  51 #define INFO(a) do { } while(0)
  52
  53 void setmysockopt(int sock) {
  54         int size = 1;
  55 #if 0
  56         if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &size, sizeof(int)) < 0)
  57                  INFO("(no sockopt/1: %m)");
  58 #endif
  59 #ifdef  IPPROTO_TCP
  60         size = 1;
  61         if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, &size, sizeof(int)) < 0)
  62                  INFO("(no sockopt/2: %m)");
  63 #endif
  64 #if 0
  65         size = 1024;
  66         if (setsockopt(sock, IPPROTO_TCP, TCP_MAXSEG, &size, sizeof(int)) < 0)
  67                  INFO("(no sockopt/3: %m)");
  68 #endif
  69 }
  70
  71 #ifndef G_GNUC_NORETURN
  72 #if __GNUC__ > 2 || (__GNUC__ == 2 && __GNUC_MINOR__ > 4)
  73 #define G_GNUC_NORETURN __attribute__((__noreturn__))
  74 #else
  75 #define G_GNUC_NORETURN
  76 #endif
  77 #endif
  78
  79 void err(const char *s) G_GNUC_NORETURN;
  80
  81 void err(const char *s) {
  82         char s1[150], *s2;
  83
  84         strncpy(s1, s, sizeof(s1));
  85         if ((s2 = strstr(s, "%m"))) {
  86                 strcpy(s1 + (s2 - s), strerror(errno));
  87                 s2 += 2;
  88                 strcpy(s1 + strlen(s1), s2);
  89         }
  90 #ifndef sun
  91         /* Solaris doesn't have %h in syslog */
  92         else if ((s2 = strstr(s, "%h"))) {
  93                 strcpy(s1 + (s2 - s), hstrerror(h_errno));
  94                 s2 += 2;
  95                 strcpy(s1 + strlen(s1), s2);
  96         }
  97 #endif
  98
  99         s1[sizeof(s1)-1] = '\0';
 100 #ifdef ISSERVER
 101         syslog(LOG_ERR, "%s", s1);
 102         syslog(LOG_ERR, "Exiting.");
 103 #endif
 104         fprintf(stderr, "Error: %s\nExiting.\n", s1);
 105 }
 106
 107 void err(const char *s) G_GNUC_NORETURN;
 108
 109 void err(const char *s) {
 110         err_nonfatal(s);
 111         exit(EXIT_FAILURE);
 112 }
 113
 114 void logging(void) {
 115 #ifdef ISSERVER
 116         openlog(MY_NAME, LOG_PID, LOG_DAEMON);
 117 #endif
 118         setvbuf(stdout, NULL, _IONBF, 0);
 119         setvbuf(stderr, NULL, _IONBF, 0);
 120 }
 121
 122 #ifdef WORDS_BIGENDIAN
 123 u64 ntohll(u64 a) {
 124         return a;
 125 }
 126 #else
 127 u64 ntohll(u64 a) {
 128         u32 lo = a & 0xffffffff;
 129         u32 hi = a >> 32U;
 130         lo = ntohl(lo);
 131         hi = ntohl(hi);
 132         return ((u64) lo) << 32U | hi;
 133 }
 134 #endif
 135 #define htonll ntohll
 136
 137 /* Flags used between the client and server */
 138 #define NBD_FLAG_HAS_FLAGS      (1 << 0)        /* Flags are there */
 139 #define NBD_FLAG_READ_ONLY      (1 << 1)        /* Device is read-only */
 140
 141 #define NBD_DEFAULT_PORT        "10809" /* Port on which named exports are
 142                                          * served */
 143
 144 /* Options that the client can select to the server */
 145 #define NBD_OPT_EXPORT_NAME     (1 << 0)        /* Client wants to select a named export (is followed by length and name of export) */