From 7c066f7d6d72154d0f5ef746b14190a5ad54421b Mon Sep 17 00:00:00 2001
From: Johann Latocha
Date: Mon, 6 Feb 2012 17:37:01 +0100
Subject: [ALL] Request latest available rid

---
 src/server/utils.c | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

(limited to 'src/server/utils.c')

diff --git a/src/server/utils.c b/src/server/utils.c
index 67d91c9..34da129 100644
--- a/src/server/utils.c
+++ b/src/server/utils.c
@@ -76,14 +76,23 @@ void dnbd3_reload_config(char* config_file_name)
 
 dnbd3_image_t* dnbd3_get_image(int vid, int rid)
 {
-    // TODO: find better data structure
+    int i, max = 0;
     dnbd3_image_t *result = NULL;
-    int i;
     for (i = 0; i < _num_images; ++i)
     {
-        if (_images[i].vid == vid && _images[i].rid == rid)
-            result = &_images[i];
-
+        if (rid != 0) // rid was specified
+        {
+            if (_images[i].vid == vid && _images[i].rid == rid)
+                result = &_images[i];
+        }
+        else // search max. rid available
+        {
+            if (_images[i].vid == vid && _images[i].rid > max)
+            {
+                result = &_images[i];
+                max = _images[i].rid;
+            }
+        }
     }
     return result;
 }
-- 
cgit v1.2.3-55-g7522