r3: Initial revision
[nbd.git] / 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 "config.h"
13 #include <errno.h>
14 #include <string.h>
15 #include <netdb.h>
16
17 #if SIZEOF_UNSIGNED_SHORT_INT==4
18 typedef unsigned short u32;
19 #elif SIZEOF_UNSIGNED_INT==4
20 typedef unsigned int u32;
21 #elif SIZEOF_UNSIGNED_LONG_INT==4
22 typedef unsigned long u32;
23 #else
24 #error I need at least some 32-bit type
25 #endif
26
27 #ifndef FS_32BIT
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 #endif  /* FS_32BIT */
38
39 #include "nbd.h"
40
41 #ifdef  FS_32BIT
42 struct { unsigned char m[8]; } cliserv_magic = { { 0x00, 0x00, 0x42, 0x02, 0x81, 0x86, 0x12, 0x53 } };
43 #else
44 u64 cliserv_magic = 0x00420281861253LL;
45 #endif
46 #define INIT_PASSWD "NBDMAGIC"
47
48 #define INFO(a) do { } while(0)
49
50 void setmysockopt(int sock)
51 {
52         int size = 1;
53 #if 0
54         if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &size, sizeof(int)) < 0)
55                  INFO("(no sockopt/1: %m)");
56 #endif
57 #ifdef  IPPROTO_TCP
58         size = 1;
59         if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, &size, sizeof(int)) < 0)
60                  INFO("(no sockopt/2: %m)");
61 #endif
62 #if 0
63         size = 1024;
64         if (setsockopt(sock, IPPROTO_TCP, TCP_MAXSEG, &size, sizeof(int)) < 0)
65                  INFO("(no sockopt/3: %m)");
66 #endif
67 }
68
69 void err(const char *s)
70 {
71         const int maxlen = 150;
72         char s1[maxlen], *s2;
73         int n = 0;
74
75         strncpy(s1, s, maxlen);
76         if (s2 = strstr(s, "%m")) {
77                 strcpy(s1 + (s2 - s), strerror(errno));
78                 s2 += 2;
79                 strcpy(s1 + strlen(s1), s2);
80         }
81 #ifndef sun
82         /* Solaris doesn't have %h in syslog */
83         else if (s2 = strstr(s, "%h")) {
84                 strcpy(s1 + (s2 - s), hstrerror(h_errno));
85                 s2 += 2;
86                 strcpy(s1 + strlen(s1), s2);
87         }
88 #endif
89
90         s1[maxlen-1] = '\0';
91 #ifdef ISSERVER
92         syslog(LOG_ERR, s1);
93 #else
94         fprintf(stderr, "Error: %s\n", s1);
95 #endif
96         exit(1);
97 }
98
99 void logging(void)
100 {
101 #ifdef ISSERVER
102         openlog(MY_NAME, LOG_PID, LOG_DAEMON);
103 #endif
104         setvbuf(stdout, NULL, _IONBF, 0);
105         setvbuf(stderr, NULL, _IONBF, 0);
106 }
107
108 #ifndef FS_32BIT
109 #ifdef WORDS_BIGENDIAN
110 u64 ntohll(u64 a)
111 {
112         return a;
113 }
114 #else
115 u64 ntohll(u64 a)
116 {
117         u32 lo = a & 0xffffffff;
118         u32 hi = a >> 32U;
119         lo = ntohl(lo);
120         hi = ntohl(hi);
121         return ((u64) lo) << 32U | hi;
122 }
123 #endif
124 #define htonll ntohll
125 #endif  /* FS_32BIT */