From afdef2fa176881f11ebaac09abf1d13d0502c945 Mon Sep 17 00:00:00 2001 From: Johann Latocha Date: Mon, 21 May 2012 18:41:20 +0200 Subject: [ALL] Some cleanup --- src/kernel/blk.c | 16 +++++++++------- src/kernel/net.c | 33 +++++++++++++++++---------------- src/kernel/net.h | 2 +- src/kernel/sysfs.c | 15 --------------- src/kernel/sysfs.h | 14 ++++++++++++++ 5 files changed, 41 insertions(+), 39 deletions(-) (limited to 'src/kernel') diff --git a/src/kernel/blk.c b/src/kernel/blk.c index 369627f..2f52cc3 100644 --- a/src/kernel/blk.c +++ b/src/kernel/blk.c @@ -99,6 +99,7 @@ struct block_device_operations dnbd3_blk_ops = int dnbd3_blk_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd, unsigned long arg) { + int result = 0; dnbd3_device_t *dev = bdev->bd_disk->private_data; struct request_queue *blk_queue = dev->disk->queue; dnbd3_ioctl_t *msg = kmalloc(sizeof(dnbd3_ioctl_t), GFP_KERNEL); @@ -108,33 +109,34 @@ int dnbd3_blk_ioctl(struct block_device *bdev, fmode_t mode, unsigned int cmd, u { case IOCTL_OPEN: strcpy(dev->cur_server.host, msg->host); - strcpy(dev->cur_server.port, msg->port); + strcpy(dev->cur_server.port, PORTSTR); dev->vid = msg->vid; dev->rid = msg->rid; blk_queue->backing_dev_info.ra_pages = (msg->read_ahead_kb * 1024)/ PAGE_CACHE_SIZE; - return dnbd3_net_connect(dev); + result = dnbd3_net_connect(dev); + break; case IOCTL_CLOSE: set_capacity(dev->disk, 0); - dnbd3_net_disconnect(dev); + result = dnbd3_net_disconnect(dev); break; case IOCTL_SWITCH: dnbd3_net_disconnect(dev); strcpy(dev->cur_server.host, msg->host); - return dnbd3_net_connect(dev); + result = dnbd3_net_connect(dev); + break; case BLKFLSBUF: break; default: - kfree(msg); - return -1; + result = -EIO; } kfree(msg); - return 0; + return result; } void dnbd3_blk_request(struct request_queue *q) diff --git a/src/kernel/net.c b/src/kernel/net.c index ad4884c..840282f 100644 --- a/src/kernel/net.c +++ b/src/kernel/net.c @@ -116,7 +116,7 @@ int dnbd3_net_connect(dnbd3_device_t *dev) return 0; } -void dnbd3_net_disconnect(dnbd3_device_t *dev) +int dnbd3_net_disconnect(dnbd3_device_t *dev) { printk("INFO: Disconnecting device %s\n", dev->disk->disk_name); @@ -150,6 +150,8 @@ void dnbd3_net_disconnect(dnbd3_device_t *dev) dev->cur_server.sock = NULL; memset(dev->cur_server.host, '\0', sizeof(dev->cur_server.host)); } + + return 0; } void dnbd3_net_heartbeat(unsigned long arg) @@ -157,6 +159,7 @@ void dnbd3_net_heartbeat(unsigned long arg) dnbd3_device_t *dev = (dnbd3_device_t *) arg; struct request *req = kmalloc(sizeof(struct request), GFP_ATOMIC); + // send keepalive if (req) { req->cmd_type = REQ_TYPE_SPECIAL; @@ -164,7 +167,12 @@ void dnbd3_net_heartbeat(unsigned long arg) list_add_tail(&req->queuelist, &dev->request_queue_send); wake_up(&dev->process_queue_send); } + else + { + printk("ERROR: Couldn't create keepalive request\n"); + } + // start discover dev->discover = 1; wake_up(&dev->process_queue_discover); @@ -255,7 +263,7 @@ int dnbd3_net_discover(void *data) if (kernel_sendmsg(sock, &msg, &iov, 1, sizeof(dnbd3_request)) <= 0) goto error; - // receive net replay + // receive net reply iov.iov_base = &dnbd3_reply; iov.iov_len = sizeof(dnbd3_reply); if (kernel_recvmsg(sock, &msg, &iov, 1, sizeof(dnbd3_reply), msg.msg_flags) <= 0) @@ -319,8 +327,7 @@ int dnbd3_net_discover(void *data) +dev->alt_servers[i].rtts[3] ) / 4; - - if ( best_rtt > dev->alt_servers[i].rtt) + if (best_rtt > dev->alt_servers[i].rtt) { best_rtt = dev->alt_servers[i].rtt; strcpy(best_server, current_server); @@ -346,7 +353,8 @@ int dnbd3_net_discover(void *data) continue; // take server with lowest rtt - if (ready && num > 1 && strcmp(dev->cur_server.host, best_server) && !kthread_should_stop() && dev->cur_server.rtt > best_rtt + RTT_THRESHOLD) + if (ready && num > 1 && strcmp(dev->cur_server.host, best_server) && !kthread_should_stop() + && dev->cur_server.rtt > best_rtt + RTT_THRESHOLD) { printk("INFO: Server %s on %s is faster (%lluus)\n", best_server, dev->disk->disk_name, best_rtt); kfree(buf); @@ -357,10 +365,6 @@ int dnbd3_net_discover(void *data) dnbd3_net_connect(dev); return 0; } - else - { - dev->cur_server.rtt = best_rtt; - } turn = (turn +1) % 4; if (turn == 3) @@ -382,6 +386,9 @@ int dnbd3_net_send(void *data) init_msghdr(msg); + dnbd3_request.vid = dev->vid; + dnbd3_request.rid = dev->rid; + set_user_nice(current, -20); while (!kthread_should_stop() || !list_empty(&dev->request_queue_send)) @@ -402,8 +409,6 @@ int dnbd3_net_send(void *data) { case REQ_TYPE_FS: dnbd3_request.cmd = CMD_GET_BLOCK; - dnbd3_request.vid = dev->vid; - dnbd3_request.rid = dev->rid; dnbd3_request.offset = blk_rq_pos(blk_request) << 9; // *512 dnbd3_request.size = blk_rq_bytes(blk_request); // bytes left to complete entire request break; @@ -413,13 +418,9 @@ int dnbd3_net_send(void *data) { case REQ_GET_FILESIZE: dnbd3_request.cmd = CMD_GET_SIZE; - dnbd3_request.vid = dev->vid; - dnbd3_request.rid = dev->rid; break; case REQ_GET_SERVERS: dnbd3_request.cmd = CMD_GET_SERVERS; - dnbd3_request.vid = dev->vid; - dnbd3_request.rid = dev->rid; break; } break; @@ -487,7 +488,7 @@ int dnbd3_net_receive(void *data) if (list_empty(&dev->request_queue_receive)) continue; - // receive net replay + // receive net reply iov.iov_base = &dnbd3_reply; iov.iov_len = sizeof(dnbd3_reply); if (kernel_recvmsg(dev->cur_server.sock, &msg, &iov, 1, sizeof(dnbd3_reply), msg.msg_flags) <= 0) diff --git a/src/kernel/net.h b/src/kernel/net.h index 3dfca41..64816bc 100644 --- a/src/kernel/net.h +++ b/src/kernel/net.h @@ -32,7 +32,7 @@ int dnbd3_net_connect(dnbd3_device_t *lo); -void dnbd3_net_disconnect(dnbd3_device_t *lo); +int dnbd3_net_disconnect(dnbd3_device_t *lo); int dnbd3_net_send(void *data); diff --git a/src/kernel/sysfs.c b/src/kernel/sysfs.c index fe4321e..a455bd2 100644 --- a/src/kernel/sysfs.c +++ b/src/kernel/sysfs.c @@ -23,21 +23,6 @@ #include "sysfs.h" #include "utils.h" -typedef struct -{ - struct attribute attr; - ssize_t (*show)(char *, dnbd3_device_t *); - ssize_t (*store)(const char *, size_t, dnbd3_device_t *); -} device_attr_t; - -typedef struct -{ - struct attribute attr; - ssize_t (*show)(char *, dnbd3_server_t *); - ssize_t (*store)(const char *, size_t, dnbd3_server_t *); -} server_attr_t; - - ssize_t show_cur_server_ip(char *buf, dnbd3_device_t *dev) { return sprintf(buf, "%s\n", dev->cur_server.host); diff --git a/src/kernel/sysfs.h b/src/kernel/sysfs.h index 09a053f..ab5ceb5 100644 --- a/src/kernel/sysfs.h +++ b/src/kernel/sysfs.h @@ -27,5 +27,19 @@ void dnbd3_sysfs_init(dnbd3_device_t *dev); void dnbd3_sysfs_exit(dnbd3_device_t *dev); +typedef struct +{ + struct attribute attr; + ssize_t (*show)(char *, dnbd3_device_t *); + ssize_t (*store)(const char *, size_t, dnbd3_device_t *); +} device_attr_t; + +typedef struct +{ + struct attribute attr; + ssize_t (*show)(char *, dnbd3_server_t *); + ssize_t (*store)(const char *, size_t, dnbd3_server_t *); +} server_attr_t; + #endif /* SYSFS_H_ */ -- cgit v1.2.3-55-g7522