Implement support for flush, fua and rotational.
[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 #define __be32 u32
39 #define __be64 u64
40 #include "nbd.h"
41
42 #if NBD_LFS==1
43 /* /usr/include/features.h (included from /usr/include/sys/types.h)
44    defines this when _GNU_SOURCE is defined
45  */
46 #ifndef _LARGEFILE_SOURCE
47 #define _LARGEFILE_SOURCE
48 #endif
49 #define _FILE_OFFSET_BITS 64
50 #endif
51
52 u64 cliserv_magic = 0x00420281861253LL;
53 u64 opts_magic = 0x49484156454F5054LL;
54 #define INIT_PASSWD "NBDMAGIC"
55
56 #define INFO(a) do { } while(0)
57
58 void setmysockopt(int sock) {
59         int size = 1;
60 #if 0
61         if (setsockopt(sock, SOL_SOCKET, SO_SNDBUF, &size, sizeof(int)) < 0)
62                  INFO("(no sockopt/1: %m)");
63 #endif
64 #ifdef  IPPROTO_TCP
65         size = 1;
66         if (setsockopt(sock, IPPROTO_TCP, TCP_NODELAY, &size, sizeof(int)) < 0)
67                  INFO("(no sockopt/2: %m)");
68 #endif
69 #if 0
70         size = 1024;
71         if (setsockopt(sock, IPPROTO_TCP, TCP_MAXSEG, &size, sizeof(int)) < 0)
72                  INFO("(no sockopt/3: %m)");
73 #endif
74 }
75
76 #ifndef G_GNUC_NORETURN
77 #if __GNUC__ > 2 || (__GNUC__ == 2 && __GNUC_MINOR__ > 4)
78 #define G_GNUC_NORETURN __attribute__((__noreturn__))
79 #else
80 #define G_GNUC_NORETURN
81 #endif
82 #endif
83
84 void err_nonfatal(const char *s) {
85         char s1[150], *s2;
86
87         strncpy(s1, s, sizeof(s1));
88         if ((s2 = strstr(s, "%m"))) {
89                 strcpy(s1 + (s2 - s), strerror(errno));
90                 s2 += 2;
91                 strcpy(s1 + strlen(s1), s2);
92         }
93 #ifndef sun
94         /* Solaris doesn't have %h in syslog */
95         else if ((s2 = strstr(s, "%h"))) {
96                 strcpy(s1 + (s2 - s), hstrerror(h_errno));
97                 s2 += 2;
98                 strcpy(s1 + strlen(s1), s2);
99         }
100 #endif
101
102         s1[sizeof(s1)-1] = '\0';
103 #ifdef ISSERVER
104         syslog(LOG_ERR, "%s", s1);
105         syslog(LOG_ERR, "Exiting.");
106 #endif
107         fprintf(stderr, "Error: %s\nExiting.\n", s1);
108 }
109
110 void err(const char *s) G_GNUC_NORETURN;
111
112 void err(const char *s) {
113         err_nonfatal(s);
114         exit(EXIT_FAILURE);
115 }
116
117 void logging(void) {
118 #ifdef ISSERVER
119         openlog(MY_NAME, LOG_PID, LOG_DAEMON);
120 #endif
121         setvbuf(stdout, NULL, _IONBF, 0);
122         setvbuf(stderr, NULL, _IONBF, 0);
123 }
124
125 #ifdef WORDS_BIGENDIAN
126 u64 ntohll(u64 a) {
127         return a;
128 }
129 #else
130 u64 ntohll(u64 a) {
131         u32 lo = a & 0xffffffff;
132         u32 hi = a >> 32U;
133         lo = ntohl(lo);
134         hi = ntohl(hi);
135         return ((u64) lo) << 32U | hi;
136 }
137 #endif
138 #define htonll ntohll
139
140 #define NBD_DEFAULT_PORT        "10809" /* Port on which named exports are
141                                          * served */
142
143 /* Options that the client can select to the server */
144 #define NBD_OPT_EXPORT_NAME     (1 << 0)        /* Client wants to select a named export (is followed by length and name of export) */