Multiple tapdisk2 processes may use the same parent disk images (later used to

detect sharable memory pages). This patch establishes unique id for each disk
image opened by tapdisk2, and stores it in shared memory region, thus making it
available to the remaining tapdisk2s.

Signed-off-by: Grzegorz Milos <Grzegorz.Milos@citrix.com>

1 files changed, 20 insertions, 0 deletions

diff --git a/tools/memshr/interface.c b/tools/memshr/interface.c
index 2c2f45423b..ad14bbf2a0 100644
--- a/tools/memshr/interface.c
+++ b/tools/memshr/interface.c
@@ -106,3 +106,23 @@ void memshr_vbd_initialize(void)
     vbd_info.enabled = 1;
 }
 
+uint16_t memshr_vbd_image_get(char* file)
+{
+    uint16_t id;
+
+    if(pthread_mutex_lock(&SHARED_INFO->lock)) goto error_out;
+    id = shm_vbd_image_get(file, SHARED_INFO->vbd_images);
+    if(pthread_mutex_unlock(&SHARED_INFO->lock)) goto error_out;
+
+    return id;
+error_out:
+    return 0;    
+}
+
+void memshr_vbd_image_put(uint16_t memshr_id)
+{
+    if(pthread_mutex_lock(&SHARED_INFO->lock)) return;
+    shm_vbd_image_put(memshr_id, SHARED_INFO->vbd_images);
+    if(pthread_mutex_unlock(&SHARED_INFO->lock)) return;
+}
+