Compiles

.vscode/c_cpp_properties.json

@@ -11,7 +11,8 @@
             "cStandard": "c17",
             "cppStandard": "gnu++20",
             "intelliSenseMode": "windows-gcc-x64",
-            "compileCommands": "./compile"
+            "compileCommands": "./compile",
+            "configurationProvider": "ms-vscode.cmake-tools"
         },
         {
             "name": "config",

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+   "cmake.configureOnOpen": false
+}

CommandList.cpp

@@ -17,11 +17,17 @@ namespace core {
       else {
          if(request.equals(""))
            return false;
-         request.split(delimiter);
+         request.split(delimiter, 2);
-         auto command = commands[request[0].str()];
+         request.reset();
-	      return command->processCommand(request, session);
+         try {
+            auto command = commands.at(request[0].str());
+ 	         return command->processCommand(request, session);
+         }
+         catch(...) {
+            return false;
+         }
       }
-      return false;
+      return true;
    }
 
    bool CommandList::grabInput(TCPSession &session, Command &command) {

ConsoleSession.cpp

@@ -33,6 +33,7 @@ namespace core {
          out << "Password: ";
          status = WAIT_PASSWORD;
          break;
+         
        case WAIT_PASSWORD:
          status = PROMPT;
          protocol(blank);

Socket.cpp

@@ -67,23 +67,23 @@ namespace core {
       if(event.events & EPOLLRDHUP) {
          readHangup = true;
          shutdown("hangup received");
-	 lock.unlock();
+      	lock.unlock();
          return false;
       }
 
       if(event.events & EPOLLIN) {
          coreutils::ZString zbuffer(buffer, length);
-	 receiveData(zbuffer);
+      	receiveData(zbuffer);
       }
 
       if(event.events & EPOLLWRNORM) {
-	 writeSocket();
+	      writeSocket();
       }
 
       if(event.events & EPOLLHUP) {
          shutdown();
-	 lock.unlock();
+	      lock.unlock();
-	 return false;
+	      return false;
       }
 
       lock.unlock();
@@ -115,6 +115,7 @@ namespace core {
 
       if((len = ::read(getDescriptor(), buffer.getData(), buffer.getLength())) >= 0) {
          coreutils::ZString zbuffer(buffer.getData(), len);
+//         coreutils::Log(coreutils::LOG_DEBUG_1) << zbuffer;
 	      onDataReceived(zbuffer);
       }
       else {
@@ -141,12 +142,12 @@ namespace core {
 
    void Socket::writeSocket() {
       if(fifo.size() > 0) {
-	 outlock.lock();
+      	 outlock.lock();
          ::write(descriptor, fifo.front().c_str(), fifo.front().length());
          fifo.pop();
-	 if(shutDown && !needsToWrite())
+	      if(shutDown && !needsToWrite())
-	   delete this;
+	        delete this;
-	 outlock.unlock();
+	      outlock.unlock();
       }
    }
 

Subscription.cpp

@@ -2,7 +2,15 @@
 
 namespace core {
 
-   Subscription::Subscription(std::string id, TCPSession &session) : id(id), owner(&session) {}
+   Subscription::Subscription(std::string id, TCPSession &session) : id(id), owner(session) {}
+
+   Subscription::~Subscription() {
+      std::stringstream out;
+      out << "cancel:" << id << std::endl;
+      for(auto subscriber : subscribers) {
+         subscriber->write(out.str());
+      }
+   }
 
    int Subscription::subscribe(TCPSession &session) {
       subscribers.push_back(&session);
@@ -12,7 +20,7 @@ namespace core {
    int Subscription::unsubscribe(TCPSession &session) {
       for(auto subscriber = subscribers.begin(); subscriber < subscribers.end(); ++subscriber) {
          if(*subscriber == &session) {
-            subscribers.erase(subscriber);
+            subscribers.erase(subscriber++);
             return 1;
          }
       }

Subscription.h

@@ -5,6 +5,7 @@
 #include "Command.h"
 #include "ZString.h"
 #include <vector>
+#include <string>
 
 namespace core {
 
@@ -12,6 +13,7 @@ namespace core {
 
     public:
       Subscription(std::string id, TCPSession &session);
+      ~Subscription();
 
       int subscribe(TCPSession &session);
       int unsubscribe(TCPSession &session);
@@ -22,7 +24,7 @@ namespace core {
       int processCommand(coreutils::ZString &request, TCPSession &session) override;
 
       std::string id;
-      TCPSession *owner;
+      TCPSession &owner;
 
       std::vector<TCPSession *> subscribers;
 

SubscriptionManager.cpp

@@ -6,49 +6,62 @@ namespace core {
    SubscriptionManager::SubscriptionManager() {}
 
    int SubscriptionManager::removeSessionSubscriptions(TCPSession &session) {
-      int count = 0;
+      int countSubscribed = 0;
-      for(auto subscription = subscriptions.begin(); subscription < subscriptions.end(); ++subscription) {
+      int countPublished = 0;
-         count += (*subscription)->unsubscribe(session);
+      for(auto subscription = subscriptions.begin(); subscription != subscriptions.end();) {
+         countSubscribed += (*subscription).second->unsubscribe(session);
+         if(&(*subscription).second->owner == &session) {
+            subscription = subscriptions.erase(subscription);
+            delete (*subscription).second;
+            ++countPublished;
+         } else
+           ++subscription;
+
       }
-      coreutils::Log(coreutils::LOG_DEBUG_2) << "Removed session from " << count << " subscriptions.";
+      coreutils::Log(coreutils::LOG_DEBUG_2) << "Removed session from " << countSubscribed << " subscription(s).";
-      return count;
+      coreutils::Log(coreutils::LOG_DEBUG_2) << "Cancelled " << countPublished << " channel(s) for session.";
+      return countSubscribed;
    }
 
    int SubscriptionManager::processCommand(coreutils::ZString &request, TCPSession &session) {
 
+      coreutils::Log(coreutils::LOG_DEBUG_2) << "Processing subscription request: " << request << ".";
+
       if(request[0].equals("publish")) {
-	      subscriptions.push_back(new Subscription(request[1].str(), session));
+         Subscription *newSubscription = new Subscription(request[1].str(), session);
+         subscriptions.insert(std::make_pair(request[1].str(), newSubscription));
+
 	      return 1;
       } else if(request[0].equals("catalog")) {
       	session.out << ":catalog:";
-	      for(auto subscription = subscriptions.begin(); subscription < subscriptions.end(); ++subscription) {
+         for(auto const& [key, subscription] : subscriptions) {
-     	      session.out << (*subscription)->id << "|";
+	  	      session.out << subscription->id << "|";
-            (*subscription)->processCommand(request, session);
+            subscription->processCommand(request, session);
      	      session.out << (";");
          }
  	      session.out << std::endl;
       	return 1;
       }
 
-	   for(auto subscription = subscriptions.begin(); subscription < subscriptions.end(); ++subscription) {
+      auto subscription = subscriptions[request[1].str()];
-	      if(request[1].equals((*subscription)->id)) {
+
-            if(request[0].equals("unpublish")) {
+	   if(request[1].equals(subscription->id)) {
-	            subscriptions.erase(subscription);
+         if(request[0].equals("unpublish")) {
-            } else if(request[0].equals("subscribe")) {
+	         subscriptions.erase(request[1].str());
-   	         (*subscription)->subscribe(session);
+         } else if(request[0].equals("subscribe")) {
-            } else if(request[0].equals("unsubscribe")) {
+   	      subscription->subscribe(session);
-	            (*subscription)->unsubscribe(session);
+         } else if(request[0].equals("unsubscribe")) {
-            } else if(request[0].equals("event")) {
+	         subscription->unsubscribe(session);
-               if((*subscription)->owner == &session) {
+         } else if(request[0].equals("event")) {
-                  std::stringstream out;
+            if(&subscription->owner == &session) {
-                  (*subscription)->process(request, out);
+               std::stringstream out;
-	               (*subscription)->event(out);
+               subscription->process(request, out);
-               }
+	            subscription->event(out);
-               else
-                 return 0;
             }
-            return 1;
+            else
+              return 0;
          }
+         return 1;
       }
       return 0;
    }

SubscriptionManager.h

@@ -6,6 +6,7 @@
 #include "Command.h"
 #include "ZString.h"
 #include <vector>
+#include <string>
 
 namespace core {
 
@@ -19,7 +20,7 @@ namespace core {
       int processCommand(coreutils::ZString &request, TCPSession &session) override;
 
     private:
-      std::vector<Subscription *> subscriptions;
+      std::map<std::string, Subscription *> subscriptions;
 
    };
 }

TCPServer.h

@@ -109,6 +109,12 @@ namespace core {
 
       void sendToAll(std::stringstream &out, TCPSession &sender);
 
+      ///
+      /// The Subscription Manager tracks all subscriptions on the server.
+      ///
+
+      SubscriptionManager subscriptions;
+
     protected:
 
       ///
@@ -136,7 +142,6 @@ namespace core {
 
       TCPSession * accept();
       std::mutex lock;
-      SubscriptionManager subscriptions;
 
    };
 

TCPSession.cpp

@@ -9,6 +9,7 @@ namespace core {
 
    TCPSession::~TCPSession() {
       server.removeFromSessionList(this);
+      server.subscriptions.removeSessionSubscriptions(*this);
    	coreutils::Log(coreutils::LOG_DEBUG_1) << "Terminating TCPSession level.";
    }
 
@@ -17,8 +18,10 @@ namespace core {
    }
 
    void TCPSession::protocol(coreutils::ZString &data) {
-      if(!server.commands.processRequest(data, *this)) {
+      if(data.getLength() != 0) {
-   	 coreutils::Log(coreutils::LOG_DEBUG_1) << "Received data could not be parsed: " << data.str();
+         if(!server.commands.processRequest(data, *this)) {
+      	 coreutils::Log(coreutils::LOG_DEBUG_1) << "Received data could not be parsed: " << data.str();
+         }
       }
    }
 
@@ -35,33 +38,33 @@ namespace core {
 
    void TCPSession::onDataReceived(coreutils::ZString &data) {
       if(data.getLength() > 0) {
-	 lineBuffer = (char *)realloc(lineBuffer, lineBufferSize + data.getLength());
+   	   lineBuffer = (char *)realloc(lineBuffer, lineBufferSize + data.getLength());
-	 memcpy(lineBuffer + lineBufferSize, data.getData(), data.getLength());
+	      memcpy(lineBuffer + lineBufferSize, data.getData(), data.getLength());
-	 lineBufferSize += data.getLength();
+	      lineBufferSize += data.getLength();
-	 while(lineBufferSize > 0) {
+	      while(lineBufferSize > 0) {
-	    if(blockSize == 0) {
+	         if(blockSize == 0) {
-	       lineLength = strcspn(lineBuffer, "\r\n");
+	            lineLength = strcspn(lineBuffer, "\r\n");
-	       if(lineLength == lineBufferSize)
+	            if(lineLength == lineBufferSize)
-		 break;
+		           break;
                coreutils::ZString zLine(lineBuffer, lineLength);
-	       onLineReceived(zLine);
+	            onLineReceived(zLine);
-	       if(lineBuffer[lineLength] == '\r')
+	            if(lineBuffer[lineLength] == '\r')
-		 ++lineLength;
+		           ++lineLength;
-	       if(lineBuffer[lineLength] == '\n')
+	            if(lineBuffer[lineLength] == '\n')
-		 ++lineLength;
+		           ++lineLength;
-	       lineBufferSize -= lineLength;
+	            lineBufferSize -= lineLength;
-	       if(lineBufferSize > 0)
+	            if(lineBufferSize > 0)
-		 memmove(lineBuffer, lineBuffer + lineLength, lineBufferSize);
+		           memmove(lineBuffer, lineBuffer + lineLength, lineBufferSize);
-	       lineBuffer = (char *)realloc(lineBuffer, lineBufferSize);
+	            lineBuffer = (char *)realloc(lineBuffer, lineBufferSize);
             } else if(lineBufferSize >= blockLength) {
-               coreutils::ZString zBlock(lineBuffer, blockLength);
+              coreutils::ZString zBlock(lineBuffer, blockLength);
-	       onBlockReceived(zBlock);
+	            onBlockReceived(zBlock);
-	       lineBufferSize -= blockLength;
+	            lineBufferSize -= blockLength;
-	       if(lineBufferSize > 0)
+	            if(lineBufferSize > 0)
-		 memmove(lineBuffer, lineBuffer + blockLength, lineBufferSize);
+		           memmove(lineBuffer, lineBuffer + blockLength, lineBufferSize);
-	       lineBuffer = (char *)realloc(lineBuffer, lineBufferSize);
+	            lineBuffer = (char *)realloc(lineBuffer, lineBufferSize);
-	    }
+	         }
-	 }
+	      }
       }
    }
 
@@ -84,7 +87,7 @@ namespace core {
 
    void TCPSession::send() {
       if(out.tellp() > 0)
-	write(out.str());
+     	  write(out.str());
       out.str("");
    }
 

compile

@@ -5,7 +5,7 @@ do
    filename="${file%.*}"
    list="$list $filename.o"
    echo -n "Compiling $filename..."
-   g++ -g -c -I../CoreUtils $file &
+   g++ -g -c -std=c++17 -I../CoreUtils $file &
    if [ $? = '0' ]
    then
     echo "OK"