r274: Patch by Corey Minyard to set a device read-only at negotiate time
[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 <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 /* 2.6.18 and above use __be* rather than u* */
40 #define __be32 u32
41 #define __be64 u64
42 #include "nbd.h"
43 #endif
44 #ifdef NBD_H_LINUX
45 #include <linux/types.h>
46 #include <linux/nbd.h>
47 #endif
48
49 #if NBD_LFS==1
50 #define _LARGEFILE_SOURCE
51 #define _FILE_OFFSET_BITS 64
52 #endif
53
54 u64 cliserv_magic = 0x00420281861253LL;
55 #define INIT_PASSWD "NBDMAGIC"
56
57 #define INFO(a) do { } while(0)
58
59 void setmysockopt(int sock) {
60         int size = 1;
61 #if 0
62         if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &size, sizeof(int)) < 0)
63                  INFO("(no sockopt/1: %m)");
64 #endif
65 #ifdef  IPPROTO_TCP
66         size = 1;
67         if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, &size, sizeof(int)) < 0)
68                  INFO("(no sockopt/2: %m)");
69 #endif
70 #if 0
71         size = 1024;
72         if (setsockopt(sock, IPPROTO_TCP, TCP_MAXSEG, &size, sizeof(int)) < 0)
73                  INFO("(no sockopt/3: %m)");
74 #endif
75 }
76
77 void err(const char *s) {
78         const int maxlen = 150;
79         char s1[maxlen], *s2;
80
81         strncpy(s1, s, maxlen);
82         if ((s2 = strstr(s, "%m"))) {
83                 strcpy(s1 + (s2 - s), strerror(errno));
84                 s2 += 2;
85                 strcpy(s1 + strlen(s1), s2);
86         }
87 #ifndef sun
88         /* Solaris doesn't have %h in syslog */
89         else if ((s2 = strstr(s, "%h"))) {
90                 strcpy(s1 + (s2 - s), hstrerror(h_errno));
91                 s2 += 2;
92                 strcpy(s1 + strlen(s1), s2);
93         }
94 #endif
95
96         s1[maxlen-1] = '\0';
97 #ifdef ISSERVER
98         syslog(LOG_ERR, "%s", s1);
99 #else
100         fprintf(stderr, "Error: %s\n", s1);
101 #endif
102         exit(1);
103 }
104
105 void logging(void) {
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         return a;
116 }
117 #else
118 u64 ntohll(u64 a) {
119         u32 lo = a & 0xffffffff;
120         u32 hi = a >> 32U;
121         lo = ntohl(lo);
122         hi = ntohl(hi);
123         return ((u64) lo) << 32U | hi;
124 }
125 #endif
126 #define htonll ntohll
127
128 /* Flags used between the client and server */
129 #define NBD_FLAG_HAS_FLAGS      (1 << 0)        /* Flags are there */
130 #define NBD_FLAG_READ_ONLY      (1 << 1)        /* Device is read-only */