istatserverlinux/src/Clientset.cpp

/*
 *  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();
}