bug fix

[IRC.git] / Robust / src / Runtime / DSTM / interface / sockpool.h

diff --git a/Robust/src/Runtime/DSTM/interface/sockpool.h b/Robust/src/Runtime/DSTM/interface/sockpool.h
index c85d7da826a9bdde5bcf5d46fbf3863da0279f9f..8a9a64aa26096c075d2ad8cc37efdc11c00bcd7d 100644 (file)
--- a/Robust/src/Runtime/DSTM/interface/sockpool.h
+++ b/Robust/src/Runtime/DSTM/interface/sockpool.h
@@ -2,6 +2,7 @@
 #define _SOCKPOOL_H_
 
 #include "dstm.h"
+#include "ip.h"
 
 int test_and_set(volatile unsigned int *addr);
 void UnLock(volatile unsigned int *addr);
@@ -19,9 +20,11 @@ typedef struct sockPoolHashTable {
     volatile unsigned int mylock;
 } sockPoolHashTable_t;
 
+void addSockWithLock(sockPoolHashTable_t *sockhash, socknode_t *ptr);
 sockPoolHashTable_t *createSockPool(sockPoolHashTable_t *, unsigned int);
 int getSock(sockPoolHashTable_t *, unsigned int);
 int getSock2(sockPoolHashTable_t *, unsigned int);
+int getSock2WithLock(sockPoolHashTable_t *h, unsigned int); 
 int getSockWithLock(sockPoolHashTable_t *, unsigned int);
 void freeSock(sockPoolHashTable_t *, unsigned int, int);
 void freeSockWithLock(sockPoolHashTable_t *, unsigned int, int);