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 #ifdef NBD_H_LOCAL
  39 #include "nbd.h"
  40 #endif
  41 #ifdef NBD_H_LINUX
  42 #include <linux/nbd.h>
  43 #endif
  44
  45 #if NBD_LFS==1
  46 #define _LARGEFILE_SOURCE
  47 #define _FILE_OFFSET_BITS 64
  48 #endif
  49
  50 u64 cliserv_magic = 0x00420281861253LL;
  51 #define INIT_PASSWD "NBDMAGIC"
  52
  53 #define INFO(a) do { } while(0)
  54
  55 void setmysockopt(int sock)
  56 {
  57         int size = 1;
  58 #if 0
  59         if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &size, sizeof(int)) < 0)
  60                  INFO("(no sockopt/1: %m)");
  61 #endif
  62 #ifdef  IPPROTO_TCP
  63         size = 1;
  64         if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, &size, sizeof(int)) < 0)
  65                  INFO("(no sockopt/2: %m)");
  66 #endif
  67 #if 0
  68         size = 1024;
  69         if (setsockopt(sock, IPPROTO_TCP, TCP_MAXSEG, &size, sizeof(int)) < 0)
  70                  INFO("(no sockopt/3: %m)");
  71 #endif
  72 }
  73
  74 void err(const char *s)
  75 {
  76         const int maxlen = 150;
  77         char s1[maxlen], *s2;
  78         int n = 0;
  79
  80         strncpy(s1, s, maxlen);
  81         if (s2 = strstr(s, "%m")) {
  82                 strcpy(s1 + (s2 - s), strerror(errno));
  83                 s2 += 2;
  84                 strcpy(s1 + strlen(s1), s2);
  85         }
  86 #ifndef sun
  87         /* Solaris doesn't have %h in syslog */
  88         else if (s2 = strstr(s, "%h")) {
  89                 strcpy(s1 + (s2 - s), hstrerror(h_errno));
  90                 s2 += 2;
  91                 strcpy(s1 + strlen(s1), s2);
  92         }
  93 #endif
  94
  95         s1[maxlen-1] = '\0';
  96 #ifdef ISSERVER
  97         syslog(LOG_ERR, "%s", s1);
  98 #else
  99         fprintf(stderr, "Error: %s\n", s1);
 100 #endif
 101         exit(1);
 102 }
 103
 104 void logging(void)
 105 {
 106 #ifdef ISSERVER
 107         openlog(MY_NAME, LOG_PID, LOG_DAEMON);
 108 #endif
 109         setvbuf(stdout, NULL, _IONBF, 0);
 110         setvbuf(stderr, NULL, _IONBF, 0);
 111 }
 112
 113 #ifdef WORDS_BIGENDIAN
 114 u64 ntohll(u64 a)
 115 {
 116         return a;
 117 }
 118 #else
 119 u64 ntohll(u64 a)
 120 {
 121         u32 lo = a & 0xffffffff;
 122         u32 hi = a >> 32U;
 123         lo = ntohl(lo);
 124         hi = ntohl(hi);
 125         return ((u64) lo) << 32U | hi;
 126 }
 127 #endif
 128 #define htonll ntohll