Merge branch 'nxcomp'

1 files changed, 243 insertions, 0 deletions

diff --git a/nxcomp/Agent.h b/nxcomp/Agent.h
new file mode 100644
index 000000000..fac5acd43
--- /dev/null
+++ b/nxcomp/Agent.h
@@ -0,0 +1,243 @@
+/**************************************************************************/
+/*                                                                        */
+/* Copyright (c) 2001, 2010 NoMachine, http://www.nomachine.com/.         */
+/*                                                                        */
+/* NXCOMP, NX protocol compression and NX extensions to this software     */
+/* are copyright of NoMachine. Redistribution and use of the present      */
+/* software is allowed according to terms specified in the file LICENSE   */
+/* which comes in the source distribution.                                */
+/*                                                                        */
+/* Check http://www.nomachine.com/licensing.html for applicability.       */
+/*                                                                        */
+/* NX and NoMachine are trademarks of Medialogic S.p.A.                   */
+/*                                                                        */
+/* All rights reserved.                                                   */
+/*                                                                        */
+/**************************************************************************/
+
+#ifndef Agent_H
+#define Agent_H
+
+#include <unistd.h>
+
+#include <sys/time.h>
+#include <sys/types.h>
+#include <sys/select.h>
+
+#include "Misc.h"
+#include "Transport.h"
+#include "Proxy.h"
+
+extern Proxy *proxy;
+
+#define PANIC
+#define WARNING
+#undef  TEST
+#undef  DEBUG
+
+class Agent
+{
+  public:
+
+  //
+  // Must be created by passing the fake descriptor that
+  // will be used for simulating socket communication
+  // betwen the agent and the proxy. I/O will take place
+  // by copying data to the agent's read and write buf-
+  // fers.
+  //
+
+  Agent(int fd[2]);
+
+  ~Agent();
+
+  AgentTransport *getTransport() const
+  {
+    return transport_;
+  }
+
+  void saveReadMask(fd_set *readSet)
+  {
+    saveRead_ = *readSet;
+  }
+
+  void saveWriteMask(fd_set *writeSet)
+  {
+    saveWrite_ = *writeSet;
+  }
+
+  void clearReadMask(fd_set *readSet)
+  {
+    FD_CLR(remoteFd_, readSet);
+    FD_CLR(localFd_,  readSet);
+  }
+
+  void clearWriteMask(fd_set *writeSet)
+  {
+    FD_CLR(remoteFd_, writeSet);
+    FD_CLR(localFd_,  writeSet);
+  }
+
+  void setLocalRead(fd_set *readSet, int *result)
+  {
+    (*result)++;
+
+    FD_SET(localFd_, readSet);
+  }
+
+  void setRemoteRead(fd_set *readSet, int *result)
+  {
+    (*result)++;
+
+    FD_SET(remoteFd_, readSet);
+  }
+
+  void setRemoteWrite(fd_set *writeSet, int *result)
+  {
+    (*result)++;
+
+    FD_SET(remoteFd_, writeSet);
+  }
+
+  fd_set *getSavedReadMask()
+  {
+    return &saveRead_;
+  }
+
+  fd_set *getSavedWriteMask()
+  {
+    return &saveWrite_;
+  }
+
+  int getRemoteFd() const
+  {
+    return remoteFd_;
+  }
+
+  int getLocalFd() const
+  {
+    return localFd_;
+  }
+
+  int getProxyFd() const
+  {
+    return proxy -> getFd();
+  }
+
+  int isValid() const
+  {
+    return (transport_ != NULL);
+  }
+
+  int localReadable()
+  {
+    return (transport_ -> readable() != 0);
+  }
+
+  //
+  // Check if we can process more data from
+  // the agent descriptor and cache the result
+  // to avoid multiple calls. This must be
+  // always called before querying the other
+  // functions.
+  //
+
+  void saveChannelState()
+  {
+    canRead_ = (proxy != NULL ? proxy -> canRead(localFd_) : 0);
+  }
+
+  int remoteCanRead(const fd_set * const readSet)
+  {
+    #if defined(TEST) || defined(INFO)
+    *logofs << "Agent: remoteCanRead() is " <<
+               (FD_ISSET(remoteFd_, readSet) && transport_ -> dequeuable() != 0)
+            << " with FD_ISSET() " << (int) FD_ISSET(remoteFd_, readSet)
+            << " and dequeuable " << transport_ -> dequeuable()
+            << ".\n" << logofs_flush;
+    #endif
+
+    return (FD_ISSET(remoteFd_, readSet) &&
+                transport_ -> dequeuable() != 0);
+  }
+
+  int remoteCanWrite(const fd_set * const writeSet)
+  {
+    #if defined(TEST) || defined(INFO)
+    *logofs << "Agent: remoteCanWrite() is " <<
+               (FD_ISSET(remoteFd_, writeSet) && transport_ ->
+               queuable() != 0 && canRead_ == 1) << " with FD_ISSET() "
+            << (int) FD_ISSET(remoteFd_, writeSet) << " queueable "
+            << transport_ -> queuable() << " channel can read "
+            << canRead_ << ".\n" << logofs_flush;
+    #endif
+
+    return (FD_ISSET(remoteFd_, writeSet) &&
+                transport_ -> queuable() != 0 &&
+                    canRead_ == 1);
+  }
+
+  int localCanRead()
+  {
+    #if defined(TEST) || defined(INFO)
+    *logofs << "Agent: localCanRead() is " <<
+               (transport_ -> readable() != 0 && canRead_ == 1)
+            << " with readable " << transport_ -> readable()
+            << " channel can read " << canRead_ << ".\n"
+            << logofs_flush;
+    #endif
+
+    return (transport_ -> readable() != 0 &&
+                canRead_ == 1);
+  }
+
+  int proxyCanRead()
+  {
+    #if defined(TEST) || defined(INFO)
+    *logofs << "Agent: proxyCanRead() is " << proxy -> canRead()
+            << ".\n" << logofs_flush;
+    #endif
+
+    return (proxy -> canRead());
+  }
+
+  int proxyCanRead(const fd_set * const readSet)
+  {
+    #if defined(TEST) || defined(INFO)
+    *logofs << "Agent: proxyCanRead() is "
+            << ((int) FD_ISSET(proxy -> getFd(), readSet)
+            << ".\n" << logofs_flush;
+    #endif
+
+    return (FD_ISSET(proxy -> getFd(), readSet));
+  }
+
+  int enqueueData(const char *data, const int size) const
+  {
+    return transport_ -> enqueue(data, size);
+  }
+
+  int dequeueData(char *data, int size) const
+  {
+    return transport_ -> dequeue(data, size);
+  }
+
+  int dequeuableData() const
+  {
+    return transport_ -> dequeuable();
+  }
+
+  private:
+
+  int remoteFd_;
+  int localFd_;
+
+  fd_set saveRead_;
+  fd_set saveWrite_;
+
+  int canRead_;
+
+  AgentTransport *transport_;
+};
+
+#endif /* Agent_H */