Subversion Repositories tpanel

/*

 * Copyright (C) 2020, 2021 by Andreas Theofilu <andreas@theosys.at>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 3 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301  USA

 */

#include "tprjresources.h"

#include "tbytearray.h"

#include "terror.h"

using std::string;

using std::vector;

TPrjResources::TPrjResources(vector<RESOURCE_LIST_T>& list)

        : mResources(list)

{

    DECL_TRACER("TPrjResources::TPrjResources(vector<RESOURCE_LIST_T>& list)");

}

TPrjResources::~TPrjResources()

{

    DECL_TRACER("TPrjResources::~TPrjResources()");

}

RESOURCE_LIST_T TPrjResources::findResourceType(const std::string& type)

{

    DECL_TRACER("TPrjResources::findResourceType(const std::string& type)");

    if (mResources.size() == 0)

        return RESOURCE_LIST_T();

    vector<RESOURCE_LIST_T>::iterator iter;

    for (iter = mResources.begin(); iter != mResources.end(); iter++)

    {

        if (iter->type.compare(type) == 0)

            return *iter;

    }

    return RESOURCE_LIST_T();

}

RESOURCE_T TPrjResources::findResource(int idx, const std::string& name)

{

    DECL_TRACER("TPrjResources::findResource(int idx, const std::string& name)");

    if (idx < 1 || mResources.size() < (size_t)idx)

    {

        MSG_ERROR("Invalid index " << idx << "!");

        return RESOURCE_T();

    }

    RESOURCE_LIST_T list = mResources[idx-1];

    if (list.type.empty())

    {

        MSG_ERROR("Resource list " << idx << " is empty!");

        return RESOURCE_T();

    }

    vector<RESOURCE_T>::iterator iter;

    for (iter = list.ressource.begin(); iter != list.ressource.end(); iter++)

    {

        MSG_DEBUG("Resource: " << iter->name);

        if (iter->name.compare(name) == 0)

            return *iter;

    }

    MSG_WARNING("Resource " << name << " not found!");

    return RESOURCE_T();

}

RESOURCE_T TPrjResources::findResource(const std::string& type, const std::string& name)

{

    DECL_TRACER("TPrjResources::findResource(const std::string& type, const std::string& name)");

    RESOURCE_LIST_T list = findResourceType(type);

    if (list.type.empty())

        return RESOURCE_T();

    vector<RESOURCE_T>::iterator iter;

    for (iter = list.ressource.begin(); iter != list.ressource.end(); iter++)

    {

        if (iter->name.compare(name) == 0)

            return *iter;

    }

    return RESOURCE_T();

}

RESOURCE_T TPrjResources::findResource(const std::string& name)

{

    DECL_TRACER("TPrjResources::findResource(const std::string& name)");

    if (mResources.size() == 0 || mResources[0].ressource.size() == 0)

        return RESOURCE_T();

    vector<RESOURCE_T> list = mResources[0].ressource;

    vector<RESOURCE_T>::iterator iter;

    for (iter = list.begin(); iter != list.end(); iter++)

    {

        if (iter->name.compare(name) == 0)

            return *iter;

    }

    return RESOURCE_T();

}

size_t TPrjResources::getResourceIndex(const string& type)

{

    DECL_TRACER("TPrjResources::getResourceIndex(const string& type)");

    if (mResources.size() == 0)

        return npos;

    vector<RESOURCE_LIST_T>::iterator iter;

    size_t idx = 0;

    for (iter = mResources.begin(); iter != mResources.end(); iter++)

    {

        idx++;

        if (iter->type.compare(type) == 0)

            return idx;

    }

    return npos;

}

bool TPrjResources::setResource(const string& name, const string& scheme, const string& host, const string& path, const string& file, const string& user, const string& pw, int refresh)

{

    DECL_TRACER("TPrjResources::setResource(const string& name, const string& scheme, const string& host, const string& path, const string& file, const string& user, const string& pw, int refresh)");

    if (mResources.size() == 0)

        return false;

    vector<RESOURCE_LIST_T>::iterator itRlist;

    for (itRlist = mResources.begin(); itRlist != mResources.end(); ++itRlist)

    {

        if (itRlist->type.compare("image") == 0)

            break;

    }

    if (itRlist == mResources.end())

    {

        MSG_ERROR("There was no resource type \"image\" found in the resources!");

        return false;

    }

    vector<RESOURCE_T> list = itRlist->ressource;

    vector<RESOURCE_T>::iterator iter;

    for (iter = list.begin(); iter != list.end(); iter++)

    {

        if (iter->name.compare(name) == 0)

        {

            RESOURCE_T res = *iter;

            if (!scheme.empty())

                res.protocol = scheme;

            if (!host.empty())

                res.host = host;

            if (!path.empty())

                res.path = path;

            if (!file.empty())

                res.file = file;

            if (!user.empty())

                res.user = user;

            if (!pw.empty())

                res.password = pw;

            if (refresh >= 0)

                res.refresh = refresh;

            list.erase(iter);

            list.push_back(res);

            itRlist->ressource.clear();

            itRlist->ressource = list;

            return true;

        }

    }

    return false;

}

bool TPrjResources::addResource(const string& name, const string& scheme, const string& host, const string& path, const string& file, const string& user, const string& pw, int refresh)

{

    DECL_TRACER("TPrjResources::addResource(const string& name, const string& scheme, const string& host, const string& path, const string& file, const string& user, const string& pw, int refresh)");

    vector<RESOURCE_LIST_T>::iterator itRlist;

    if (mResources.size() == 0)

    {

        RESOURCE_LIST_T rl;

        rl.type = "image";

        mResources.push_back(rl);

        itRlist = mResources.begin();

    }

    else

    {

        // Find the resource container

        for (itRlist = mResources.begin(); itRlist != mResources.end(); ++itRlist)

        {

            if (itRlist->type.compare("image") == 0)

                break;

        }

    }

    if (itRlist == mResources.end())

    {

        MSG_ERROR("There is no resouce container called \"image\"!");

        return false;

    }

    RESOURCE_T r;

    r.name = name;

    r.protocol = scheme;

    r.host = host;

    r.path = path;

    r.file = file;

    r.user = user;

    r.password = pw;

    r.refresh = refresh;

    // Make sure the resource does not already exist

    if (itRlist->ressource.size() == 0)

    {

        itRlist->ressource.push_back(r);

        return true;

    }

    vector<RESOURCE_T>::iterator iter;

    for (iter = itRlist->ressource.begin(); iter != itRlist->ressource.end(); iter++)

    {

        if (iter->name == name)

        {

            iter->protocol = scheme;

            iter->host = host;

            iter->path = path;

            iter->file = file;

            iter->user = user;

            iter->password = pw;

            iter->refresh = refresh;

            return true;

        }

    }

    itRlist->ressource.push_back(r);

    return true;

}

string TPrjResources::decryptPassword(const string& pw)

{

    DECL_TRACER("TPrjResources::decryptPassword(const string& pw)");

    string erg;

    size_t len;

    unsigned char *bytes = TByteArray::hexStrToByte(pw, &len);

    if (bytes == nullptr || (len % 16) != 0)

        return erg;

    // TODO: Implement DES-CBC algorithm to decrypt

    erg.assign(reinterpret_cast<char *>(bytes), len);

    return erg;

}