提交 77351055 编写于 作者: P Philipp Reisner

drbd: struct packet_info to hold information of decoded packets

Signed-off-by: NPhilipp Reisner <philipp.reisner@linbit.com>
Signed-off-by: NLars Ellenberg <lars.ellenberg@linbit.com>
上级 de0ff338
......@@ -48,6 +48,12 @@
#include "drbd_vli.h"
struct packet_info {
enum drbd_packet cmd;
int size;
int vnr;
};
enum finish_epoch {
FE_STILL_LIVE,
FE_DESTROYED,
......@@ -924,15 +930,15 @@ static int drbd_connect(struct drbd_conf *mdev)
return -1;
}
static bool decode_header(struct drbd_conf *mdev, struct p_header *h,
enum drbd_packet *cmd, unsigned int *packet_size)
static bool decode_header(struct drbd_conf *mdev, struct p_header *h, struct packet_info *pi)
{
if (h->h80.magic == cpu_to_be32(DRBD_MAGIC)) {
*cmd = be16_to_cpu(h->h80.command);
*packet_size = be16_to_cpu(h->h80.length);
pi->cmd = be16_to_cpu(h->h80.command);
pi->size = be16_to_cpu(h->h80.length);
} else if (h->h95.magic == cpu_to_be16(DRBD_MAGIC_BIG)) {
*cmd = be16_to_cpu(h->h95.command);
*packet_size = be32_to_cpu(h->h95.length) & 0x00ffffff;
pi->cmd = be16_to_cpu(h->h95.command);
pi->size = be32_to_cpu(h->h95.length) & 0x00ffffff;
pi->vnr = 0;
} else {
dev_err(DEV, "magic?? on data m: 0x%08x c: %d l: %d\n",
be32_to_cpu(h->h80.magic),
......@@ -943,8 +949,7 @@ static bool decode_header(struct drbd_conf *mdev, struct p_header *h,
return true;
}
static int drbd_recv_header(struct drbd_conf *mdev, enum drbd_packet *cmd,
unsigned int *packet_size)
static int drbd_recv_header(struct drbd_conf *mdev, struct packet_info *pi)
{
struct p_header *h = &mdev->tconn->data.rbuf.header;
int r;
......@@ -956,7 +961,7 @@ static int drbd_recv_header(struct drbd_conf *mdev, enum drbd_packet *cmd,
return false;
}
r = decode_header(mdev, h, cmd, packet_size);
r = decode_header(mdev, h, pi);
mdev->tconn->last_received = jiffies;
return r;
......@@ -3580,6 +3585,7 @@ static int receive_bitmap(struct drbd_conf *mdev, enum drbd_packet cmd,
int err;
int ok = false;
struct p_header *h = &mdev->tconn->data.rbuf.header;
struct packet_info pi;
drbd_bm_lock(mdev, "receive bitmap", BM_LOCKED_SET_ALLOWED);
/* you are supposed to send additional out-of-sync information
......@@ -3633,8 +3639,10 @@ static int receive_bitmap(struct drbd_conf *mdev, enum drbd_packet cmd,
goto out;
break;
}
if (!drbd_recv_header(mdev, &cmd, &data_size))
if (!drbd_recv_header(mdev, &pi))
goto out;
cmd = pi.cmd;
data_size = pi.size;
}
INFO_bm_xfer_stats(mdev, "receive", &c);
......@@ -3762,24 +3770,23 @@ static struct data_cmd drbd_cmd_handler[] = {
static void drbdd(struct drbd_conf *mdev)
{
struct p_header *header = &mdev->tconn->data.rbuf.header;
unsigned int packet_size;
enum drbd_packet cmd;
struct packet_info pi;
size_t shs; /* sub header size */
int rv;
while (get_t_state(&mdev->tconn->receiver) == RUNNING) {
drbd_thread_current_set_cpu(mdev, &mdev->tconn->receiver);
if (!drbd_recv_header(mdev, &cmd, &packet_size))
if (!drbd_recv_header(mdev, &pi))
goto err_out;
if (unlikely(cmd >= P_MAX_CMD || !drbd_cmd_handler[cmd].function)) {
dev_err(DEV, "unknown packet type %d, l: %d!\n", cmd, packet_size);
if (unlikely(pi.cmd >= P_MAX_CMD || !drbd_cmd_handler[pi.cmd].function)) {
dev_err(DEV, "unknown packet type %d, l: %d!\n", pi.cmd, pi.size);
goto err_out;
}
shs = drbd_cmd_handler[cmd].pkt_size - sizeof(struct p_header);
if (packet_size - shs > 0 && !drbd_cmd_handler[cmd].expect_payload) {
dev_err(DEV, "No payload expected %s l:%d\n", cmdname(cmd), packet_size);
shs = drbd_cmd_handler[pi.cmd].pkt_size - sizeof(struct p_header);
if (pi.size - shs > 0 && !drbd_cmd_handler[pi.cmd].expect_payload) {
dev_err(DEV, "No payload expected %s l:%d\n", cmdname(pi.cmd), pi.size);
goto err_out;
}
......@@ -3792,11 +3799,11 @@ static void drbdd(struct drbd_conf *mdev)
}
}
rv = drbd_cmd_handler[cmd].function(mdev, cmd, packet_size - shs);
rv = drbd_cmd_handler[pi.cmd].function(mdev, pi.cmd, pi.size - shs);
if (unlikely(!rv)) {
dev_err(DEV, "error receiving %s, l: %d!\n",
cmdname(cmd), packet_size);
cmdname(pi.cmd), pi.size);
goto err_out;
}
}
......@@ -3989,27 +3996,26 @@ static int drbd_do_handshake(struct drbd_conf *mdev)
/* ASSERT current == mdev->tconn->receiver ... */
struct p_handshake *p = &mdev->tconn->data.rbuf.handshake;
const int expect = sizeof(struct p_handshake) - sizeof(struct p_header80);
unsigned int length;
enum drbd_packet cmd;
struct packet_info pi;
int rv;
rv = drbd_send_handshake(mdev->tconn);
if (!rv)
return 0;
rv = drbd_recv_header(mdev, &cmd, &length);
rv = drbd_recv_header(mdev, &pi);
if (!rv)
return 0;
if (cmd != P_HAND_SHAKE) {
if (pi.cmd != P_HAND_SHAKE) {
dev_err(DEV, "expected HandShake packet, received: %s (0x%04x)\n",
cmdname(cmd), cmd);
cmdname(pi.cmd), pi.cmd);
return -1;
}
if (length != expect) {
if (pi.size != expect) {
dev_err(DEV, "expected HandShake length: %u, received: %u\n",
expect, length);
expect, pi.size);
return -1;
}
......@@ -4071,8 +4077,7 @@ static int drbd_do_auth(struct drbd_conf *mdev)
unsigned int key_len = strlen(mdev->tconn->net_conf->shared_secret);
unsigned int resp_size;
struct hash_desc desc;
enum drbd_packet cmd;
unsigned int length;
struct packet_info pi;
int rv;
desc.tfm = mdev->tconn->cram_hmac_tfm;
......@@ -4092,33 +4097,33 @@ static int drbd_do_auth(struct drbd_conf *mdev)
if (!rv)
goto fail;
rv = drbd_recv_header(mdev, &cmd, &length);
rv = drbd_recv_header(mdev, &pi);
if (!rv)
goto fail;
if (cmd != P_AUTH_CHALLENGE) {
if (pi.cmd != P_AUTH_CHALLENGE) {
dev_err(DEV, "expected AuthChallenge packet, received: %s (0x%04x)\n",
cmdname(cmd), cmd);
cmdname(pi.cmd), pi.cmd);
rv = 0;
goto fail;
}
if (length > CHALLENGE_LEN * 2) {
if (pi.size > CHALLENGE_LEN * 2) {
dev_err(DEV, "expected AuthChallenge payload too big.\n");
rv = -1;
goto fail;
}
peers_ch = kmalloc(length, GFP_NOIO);
peers_ch = kmalloc(pi.size, GFP_NOIO);
if (peers_ch == NULL) {
dev_err(DEV, "kmalloc of peers_ch failed\n");
rv = -1;
goto fail;
}
rv = drbd_recv(mdev->tconn, peers_ch, length);
rv = drbd_recv(mdev->tconn, peers_ch, pi.size);
if (rv != length) {
if (rv != pi.size) {
if (!signal_pending(current))
dev_warn(DEV, "short read AuthChallenge: l=%u\n", rv);
rv = 0;
......@@ -4134,7 +4139,7 @@ static int drbd_do_auth(struct drbd_conf *mdev)
}
sg_init_table(&sg, 1);
sg_set_buf(&sg, peers_ch, length);
sg_set_buf(&sg, peers_ch, pi.size);
rv = crypto_hash_digest(&desc, &sg, sg.length, response);
if (rv) {
......@@ -4147,18 +4152,18 @@ static int drbd_do_auth(struct drbd_conf *mdev)
if (!rv)
goto fail;
rv = drbd_recv_header(mdev, &cmd, &length);
rv = drbd_recv_header(mdev, &pi);
if (!rv)
goto fail;
if (cmd != P_AUTH_RESPONSE) {
if (pi.cmd != P_AUTH_RESPONSE) {
dev_err(DEV, "expected AuthResponse packet, received: %s (0x%04x)\n",
cmdname(cmd), cmd);
cmdname(pi.cmd), pi.cmd);
rv = 0;
goto fail;
}
if (length != resp_size) {
if (pi.size != resp_size) {
dev_err(DEV, "expected AuthResponse payload of wrong size\n");
rv = 0;
goto fail;
......@@ -4544,14 +4549,14 @@ int drbd_asender(struct drbd_thread *thi)
struct drbd_conf *mdev = thi->mdev;
struct p_header *h = &mdev->tconn->meta.rbuf.header;
struct asender_cmd *cmd = NULL;
struct packet_info pi;
int rv;
void *buf = h;
int received = 0;
int expect = sizeof(struct p_header);
int ping_timeout_active = 0;
int empty, pkt_size;
enum drbd_packet cmd_nr;
int empty;
sprintf(current->comm, "drbd%d_asender", mdev_to_minor(mdev));
......@@ -4640,25 +4645,25 @@ int drbd_asender(struct drbd_thread *thi)
}
if (received == expect && cmd == NULL) {
if (!decode_header(mdev, h, &cmd_nr, &pkt_size))
if (!decode_header(mdev, h, &pi))
goto reconnect;
cmd = get_asender_cmd(cmd_nr);
cmd = get_asender_cmd(pi.cmd);
if (unlikely(cmd == NULL)) {
dev_err(DEV, "unknown command %d on meta (l: %d)\n",
cmd_nr, pkt_size);
pi.cmd, pi.size);
goto disconnect;
}
expect = cmd->pkt_size;
if (pkt_size != expect - sizeof(struct p_header)) {
if (pi.size != expect - sizeof(struct p_header)) {
dev_err(DEV, "Wrong packet size on meta (c: %d, l: %d)\n",
cmd_nr, pkt_size);
pi.cmd, pi.size);
goto reconnect;
}
}
if (received == expect) {
mdev->tconn->last_received = jiffies;
D_ASSERT(cmd != NULL);
if (!cmd->process(mdev, cmd_nr))
if (!cmd->process(mdev, pi.cmd))
goto reconnect;
/* the idle_timeout (ping-int)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册