Initial commit

src/Clientset.cpp

@@ -0,0 +1,153 @@
+/*
+ *  Copyright 2016 Bjango Pty Ltd. All rights reserved.
+ *  Copyright 2010 William Tisäter. All rights reserved.
+ * 
+ *  Redistribution and use in source and binary forms, with or without
+ *  modification, are permitted provided that the following conditions are met:
+ *
+ *    1.  Redistributions of source code must retain the above copyright
+ *        notice, this list of conditions and the following disclaimer.
+ *
+ *    2.  Redistributions in binary form must reproduce the above copyright
+ *        notice, this list of conditions and the following disclaimer in the
+ *        documentation and/or other materials provided with the distribution.
+ *
+ *    3.  The name of the copyright holder may not be used to endorse or promote
+ *        products derived from this software without specific prior written
+ *        permission.
+ *
+ *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER ``AS IS'' AND ANY
+ *  EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ *  WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ *  DISCLAIMED. IN NO EVENT SHALL WILLIAM TISÄTER BE LIABLE FOR ANY
+ *  DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ *  (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ *  ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ *  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ *  SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ */
+
+#include <vector>
+#include <sstream>
+#include <errno.h>
+#include <fstream>
+#include <iostream>
+#include <string.h>
+#include <unistd.h>
+#include <sys/stat.h>
+
+#include "Utility.h"
+#include "Clientset.h"
+#include <syslog.h>
+
+using namespace std;
+
+void ClientSet::authenticate(string uuid)
+{
+	authenticatedClients.push_back(uuid);
+	save_cache();
+}
+
+int ClientSet::is_authenticated(string _duuid)
+{
+	for (vector<string>::const_iterator search = authenticatedClients.begin(); search != authenticatedClients.end(); ++search)
+	{
+		if (*search == _duuid)
+		{
+			return 1;
+		}
+	}
+	
+	return 0;
+}
+
+void ClientSet::clear_cache(void)
+{
+	if (this->cache_dir.length())
+	{
+		stringstream path;
+		string cache_file = "clients.dat";
+		path << this->cache_dir << cache_file;
+
+		if (check_file_exist(path.str()))
+		{
+			if (unlink(path.str().c_str()) == 0)
+			{
+				cout << "Successfully cleared all sessions." << endl;
+			}
+			else
+			{
+				cout << "Could not clear sessions in '" << path.str() << "': " << strerror(errno) << endl;
+			}
+		}
+	}
+}
+
+void ClientSet::save_cache(void)
+{
+	if (this->cache_dir.length())
+	{
+		stringstream path;
+		string cache_file = "clients.dat";
+		path << this->cache_dir << cache_file;
+		
+		ofstream out(path.str().c_str());
+		chmod(path.str().c_str(), 0600);
+		
+		if (!out)
+		{
+			cout << "Could not create file '" << path.str() << "': " << strerror(errno) << endl;
+			return;
+		}
+		for (vector<string>::const_iterator search = authenticatedClients.begin(); search != authenticatedClients.end(); ++search)
+		{
+			string uuid = *search;
+			out << uuid << endl;
+		}
+		
+		out.close();
+	}
+}
+
+void ClientSet::read_cache(const std::string & _cache_dir)
+{
+	string line;
+	stringstream path;
+	vector<string> array;
+	string cache_file = "clients.dat";
+	
+	this->cache_dir = _cache_dir;
+	path << this->cache_dir << cache_file;
+	
+	ifstream cache(path.str().c_str());
+	
+	if (cache.good())
+	{
+		while (getline(cache, line))
+		{
+			if (line.length())
+			{
+				array = explode(line, ":");
+				
+				if(array.size() > 1){
+					if (array.size() < 3) continue;
+				
+					if(to_int(array[2]) == 1)
+						authenticatedClients.push_back(array[1]);
+				} else {
+					authenticatedClients.push_back(line);
+				}
+			}
+		}
+	}
+	else
+	{
+		// Ignore no such file errors, we will create the file upon save
+		if (errno != ENOENT)
+			cout << "Could not read cache file '" << path.str() << "': " << strerror(errno) << endl;
+	}
+
+	cache.close();
+}