New upstream version 8.1.0

client_module/source/common/net/message/session/locking/LockGrantedMsgEx.c

@@ -0,0 +1,79 @@
+#include <app/App.h>
+#include <components/AckManager.h>
+#include <common/toolkit/ackstore/AcknowledgmentStore.h>
+#include <common/toolkit/SocketTk.h>
+#include <common/net/msghelpers/MsgHelperAck.h>
+#include "LockGrantedMsgEx.h"
+
+const struct NetMessageOps LockGrantedMsgEx_Ops = {
+   .serializePayload = _NetMessage_serializeDummy,
+   .deserializePayload = LockGrantedMsgEx_deserializePayload,
+   .processIncoming = __LockGrantedMsgEx_processIncoming,
+   .getSupportedHeaderFeatureFlagsMask = NetMessage_getSupportedHeaderFeatureFlagsMask,
+};
+
+bool LockGrantedMsgEx_deserializePayload(NetMessage* this, DeserializeCtx* ctx)
+{
+   LockGrantedMsgEx* thisCast = (LockGrantedMsgEx*)this;
+
+   // lockAckID
+   if(!Serialization_deserializeStrAlign4(ctx,
+      &thisCast->lockAckIDLen, &thisCast->lockAckID) )
+      return false;
+
+   // ackID
+   if(!Serialization_deserializeStrAlign4(ctx, &thisCast->ackIDLen, &thisCast->ackID) )
+      return false;
+
+   // granterNodeID
+   if(!NumNodeID_deserialize(ctx, &thisCast->granterNodeID) )
+      return false;
+
+   return true;
+}
+
+bool __LockGrantedMsgEx_processIncoming(NetMessage* this, struct App* app,
+   fhgfs_sockaddr_in* fromAddr, struct Socket* sock, char* respBuf, size_t bufLen)
+{
+   const char* logContext = "LockGranted incoming";
+   Logger* log = App_getLogger(app);
+   AcknowledgmentStore* ackStore = App_getAckStore(app);
+   AckManager* ackManager = App_getAckManager(app);
+
+   LockGrantedMsgEx* thisCast = (LockGrantedMsgEx*)this;
+   bool gotLockWaiter;
+
+   #ifdef LOG_DEBUG_MESSAGES
+      const char* peer;
+
+      peer = fromAddr ?
+         SocketTk_ipaddrToStr(fromAddr->addr) : StringTk_strDup(Socket_getPeername(sock) );
+      LOG_DEBUG_FORMATTED(log, Log_DEBUG, logContext, "Received a AckMsg from: %s", peer);
+
+      kfree(peer);
+   #endif // LOG_DEBUG_MESSAGES
+
+   IGNORE_UNUSED_VARIABLE(log);
+   IGNORE_UNUSED_VARIABLE(logContext);
+
+   gotLockWaiter = AcknowledgmentStore_receivedAck(
+      ackStore, LockGrantedMsgEx_getLockAckID(thisCast) );
+
+   /* note: other than for standard acks, we send a ack repsonse to lock-grants only if someone was
+      still registered to wait for the lock-grant. (we do this to make our handling of interrupts
+      during lock waits easier, because we don't need lock canceling now.) */
+
+   if(gotLockWaiter)
+   { // lock waiter registered => send ack
+      const char* ackID = LockGrantedMsgEx_getAckID(thisCast);
+      NumNodeID granterNodeID = LockGrantedMsgEx_getGranterNodeID(thisCast);
+
+      MsgHelperAck_respondToAckRequest(app, LockGrantedMsgEx_getAckID(thisCast), fromAddr, sock,
+         respBuf, bufLen);
+
+      AckManager_addAckToQueue(ackManager, granterNodeID, ackID);
+   }
+
+   return true;
+}
+