added disk support to the nbd server

[palacios.git] / misc / network_servers / v3_nbd / raw.cc

diff --git a/misc/network_servers/v3_nbd/raw.cc b/misc/network_servers/v3_nbd/raw.cc
index 668850f..04228ae 100644 (file)
--- a/misc/network_servers/v3_nbd/raw.cc
+++ b/misc/network_servers/v3_nbd/raw.cc
@@ -27,22 +27,49 @@ raw_disk::raw_disk(string & filename) : v3_disk(filename){
 
 
 off_t raw_disk::get_capacity() {
-    
-    return 0;
-}
+        struct stat f_stats;
+
+    stat(this->filename.c_str(), &f_stats);
 
+    return f_stats.st_size;
+}
 
 
 unsigned int raw_disk::read(unsigned char * buf, off_t offset, int length) {
-
-    return 0;
+    int total_bytes = length;
+    int bytes_read = 0;
+    
+    fseeko(this->f, offset, SEEK_SET);
+
+    while (bytes_read < total_bytes) {
+       int tmp_bytes = fread(buf, 1, length - bytes_read, this->f);
+       
+       if (tmp_bytes == 0) {
+           break;
+       }
+       bytes_read += tmp_bytes;
+    }
+
+    return bytes_read;
 }
 
 
-
 unsigned int raw_disk::write(unsigned char * buf, off_t offset, int length) {
-    
-    return 0;
+    int total_bytes = length;
+    int bytes_written = 0;
+
+    fseeko(this->f, offset, SEEK_SET);
+
+    while (bytes_written < total_bytes) {
+       int tmp_bytes = fwrite(buf, 1, length - bytes_written, this->f);
+
+       if (tmp_bytes == 0) {
+           break;
+       }
+       bytes_written += tmp_bytes;
+    }
+
+    return bytes_written;
 }