1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
|
/*
* This file is part of the Distributed Network Block Device 3
*
* Copyright(c) 2019 Frederic Robra <frederic@robra.org>
* Parts copyright 2011-2012 Johann Latocha <johann@latocha.de>
*
* This file may be licensed under the terms of of the
* GNU General Public License Version 2 (the ``GPL'').
*
* Software distributed under the License is distributed
* on an ``AS IS'' basis, WITHOUT WARRANTY OF ANY KIND, either
* express or implied. See the GPL for the specific language
* governing rights and limitations.
*
* You should have received a copy of the GPL along with this
* program. If not, go to http://www.gnu.org/licenses/gpl.html
* or write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*
*/
#ifndef DNBD_H_
#define DNBD_H_
#include <linux/blk-mq.h>
#include <linux/types.h>
#include <linux/refcount.h>
#include <linux/blkdev.h>
#define KERNEL_MODULE
#include "types.h"
#include "serialize.h"
#define NUMBER_CONNECTIONS 2 // power of 2
#define DEBUG
extern struct workqueue_struct *dnbd3_wq;
/**
* struct dnbd3_server - defining a server for dnbd3
* @host: host of this server
* @rtts: last four round trip time measurements in microseconds
* @protocol_version: dnbd3 protocol version of this server
* @failures: how many times the server was unreachable
*/
struct dnbd3_server {
dnbd3_host_t host;
uint64_t rtts[4];
uint16_t protocol_version;
uint8_t failures;
};
/**
* struct dnbd3_sock - defining a socket for dnbd3
* @sock_nr: nr of this socket
* @device: the dnbd3_device this socket belongs to
* @server: the server this socket is connected to, 'NULL' if not connected
*
*/
struct dnbd3_sock {
uint8_t sock_nr;
struct dnbd3_device *device;
struct dnbd3_server *server;
struct mutex tx_lock;
struct socket *sock;
bool panic;//, discover, panic_count;
uint32_t cookie;
struct work_struct keepalive_worker;
struct work_struct receive_worker;
struct request *pending;
uint16_t receive_command;
};
struct dnbd3_device {
int minor;
struct blk_mq_tag_set tag_set;
struct list_head list;
// block
struct gendisk *disk;
// sysfs
struct kobject kobj;
struct mutex device_lock;
// network
struct dnbd3_sock socks[NUMBER_CONNECTIONS];
char *imgname;
struct dnbd3_server initial_server;
struct dnbd3_server alt_servers[NUMBER_SERVERS]; // array of alt servers
int new_servers_num; // number of new alt servers that are waiting to be copied to above array
dnbd3_server_entry_t new_servers[NUMBER_SERVERS]; // pending new alt servers
bool update_available;
bool use_server_provided_alts;
uint16_t rid;
uint64_t reported_size;
struct work_struct panic_worker;
struct work_struct discovery_worker;
uint8_t discovery_count;
struct timer_list timer;
uint8_t timer_count;
};
struct dnbd3_cmd {
struct dnbd3_device *dnbd3;
struct mutex lock;
uint32_t cookie;
blk_status_t status;
bool requed;
};
#define _print_sock(level, sock, fmt, ...) \
do { \
if ((sock)->server->host.type == HOST_IP4) { \
printk(level "dnbd%d/%d %pI4:%d: " fmt "\n", (sock)->device->minor, (sock)->sock_nr, (sock)->server->host.addr, (sock)->server->host.port, ## __VA_ARGS__); \
} else { \
printk(level "dnbd%d/%d %pI6:%d: " fmt "\n", (sock)->device->minor, (sock)->sock_nr, (sock)->server->host.addr, (sock)->server->host.port, ## __VA_ARGS__); \
} \
} while (0)
#define _print_server(level, dev, server, fmt, ...) \
do { \
if ((server)->host.type == HOST_IP4) { \
printk(level "dnbd%d: " fmt " %pI4:%d\n", (dev)->minor, ## __VA_ARGS__, (server)->host.addr, (server)->host.port); \
} else { \
printk(level "dnbd%d: " fmt " %pI6:%d\n", (dev)->minor, ## __VA_ARGS__, (server)->host.addr, (server)->host.port); \
} \
} while (0)
#ifdef DEBUG
#define debug(fmt, ...) \
printk(KERN_DEBUG "dnbd: " fmt "\n", ## __VA_ARGS__)
#define debug_dev(dev, fmt, ...) \
printk(KERN_DEBUG "dnbd%d: " fmt "\n", (dev)->minor, ## __VA_ARGS__)
#define debug_sock(sock, fmt, ...) \
_print_sock(KERN_DEBUG, sock, fmt, ## __VA_ARGS__)
#define debug_server(dev, server, fmt, ...) \
_print_server(KERN_DEBUG, dev, server, fmt, ## __VA_ARGS__)
#else
#define debug(fmt, ...)
#define debug_dev(dev, fmt, ...)
#define debug_sock(sock, fmt, ...)
#define debug_server(dev, server, fmt, ...)
#endif
#define info(fmt, ...) \
printk(KERN_INFO "dnbd: " fmt "\n", ## __VA_ARGS__)
#define info_dev(dev, fmt, ...) \
printk(KERN_INFO "dnbd%d: " fmt "\n", (dev)->minor, ## __VA_ARGS__)
#define info_sock(sock, fmt, ...) \
_print_sock(KERN_INFO, sock, fmt, ## __VA_ARGS__)
#define info_server(dev, server, fmt, ...) \
_print_server(KERN_INFO, dev, server, fmt, ## __VA_ARGS__)
#define warn(fmt, ...) \
printk(KERN_WARNING "dnbd: " fmt "\n", ## __VA_ARGS__)
#define warn_dev(dev, fmt, ...) \
printk(KERN_WARNING "dnbd%d: " fmt "\n", (dev)->minor, ## __VA_ARGS__)
#define warn_sock(sock, fmt, ...) \
_print_sock(KERN_WARNING, sock, fmt, ## __VA_ARGS__)
#define warn_server(dev, server, fmt, ...) \
_print_server(KERN_WARNING, dev, server, fmt, ## __VA_ARGS__)
#define error(fmt, ...) \
printk(KERN_ERR "dnbd: " fmt "\n", ## __VA_ARGS__)
#define error_dev(dev, fmt, ...) \
printk(KERN_ERR "dnbd%d: " fmt "\n", (dev)->minor, ## __VA_ARGS__)
#define error_sock(sock, fmt, ...) \
_print_sock(KERN_ERR, sock, fmt, ## __VA_ARGS__)
#define error_server(dev, server, fmt, ...) \
_print_server(KERN_ERR, dev, server, fmt, ## __VA_ARGS__)
#endif /* DNBD_H_ */
|
