Skip to content
user.cpp 9.14 KiB
Newer Older
jan.koester's avatar
jan.koester committed
/*******************************************************************************
 * Copyright (c) 2025, Jan Koester jan.koester@gmx.net
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 * Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 * 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.
 * Neither the name of the <organization> nor the
 *      names of its contributors may be used to endorse or promote products
 *      derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "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 <COPYRIGHT HOLDER> 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.
 *******************************************************************************/

jan.koester's avatar
jan.koester committed
#include "types.h"
jan.koester's avatar
jan.koester committed
#include "user.h"

jan.koester's avatar
jan.koester committed
namespace authdb {
    inline const char *getUserData(const AuthData *dat,AuthData::Record **varclass,const char *varname){
        if(*varclass && (*varclass)->data ){
            return (*varclass)->data;
jan.koester's avatar
jan.koester committed
        }else{
            for(const AuthData *curdat=dat; curdat; curdat=curdat->next()){
                if(curdat->Data && strcmp(curdat->Data->fieldname,varname)==0){
                    *varclass=curdat->Data;
                    return curdat->Data->data;
jan.koester's avatar
jan.koester committed
                }
            }
        }
        std::string err;
        err=varname;
        err+=" data not found !";
        throw(authdb::AuthBackendError(err.c_str()));
jan.koester's avatar
jan.koester committed
    }

    inline void setUserData(AuthData *dat,AuthData::Record **varclass,const char *varname,const char *value){
        AuthData::Record *target=*varclass;

        AuthData *curec=dat;

jan.koester's avatar
jan.koester committed
        uuid_t gid;
        uuid_copy(gid,dat->Data->uuid);

        if(!target){
            while(curec){
                if(curec->Data->type == authdb::UserData &&
jan.koester's avatar
jan.koester committed
                    uuid_compare(curec->Data->uuid,dat->Data->uuid)==0 &&
jan.koester's avatar
jan.koester committed
                    strcmp(curec->Data->fieldname,varname)==0
                ){

                    target=curec->Data;
                    target->datasize=0;
                    delete[] target->data;
                    target->data=nullptr;
                    break;
                }
                curec=curec->next();
            }
        }

        if(!target){
            target=new AuthData::Record;
jan.koester's avatar
jan.koester committed
            *varclass=dat->append(*target)->Data;
jan.koester's avatar
jan.koester committed
            target=*varclass;
        }

        target->start=0xFE;
jan.koester's avatar
jan.koester committed
        uuid_copy(target->uuid,gid);
        target->type=DataType::UserData;
        target->storage=StorageType::TextStorage;

        snprintf(target->fieldname,255,"%s",varname);
        target->datasize=strlen(value)+1;
        target->data= new char[target->datasize];
        strcpy(target->data,value);
    }

jan.koester's avatar
jan.koester committed
authdb::UserData::UserData(uuid_t id) : AuthData(id){
jan.koester's avatar
jan.koester committed
    username=nullptr;
    firstname=nullptr;
    lastname=nullptr;
    mail=nullptr;
    pwhash=nullptr;
    avatar=nullptr;
    uuid_copy(uid,id);
jan.koester's avatar
jan.koester committed
authdb::UserData::~UserData(){
}

void authdb::UserData::setFirstName(const char* src){
jan.koester's avatar
jan.koester committed
   setUserData(this,&firstname,"firstname",src);
jan.koester's avatar
jan.koester committed
const char *authdb::UserData::getFirstname(){
jan.koester's avatar
jan.koester committed
    return getUserData(this,&firstname,"firstname");
jan.koester's avatar
jan.koester committed
void authdb::UserData::setLastName(const char* src){
jan.koester's avatar
jan.koester committed
    setUserData(this,&lastname,"lastname",src);
jan.koester's avatar
jan.koester committed
const char * authdb::UserData::getLastname(){
jan.koester's avatar
jan.koester committed
    return getUserData(this,&lastname,"lastname");
jan.koester's avatar
jan.koester committed
void authdb::UserData::setMail(const char* src){
jan.koester's avatar
jan.koester committed
  setUserData(this,&mail,"mail",src);
jan.koester's avatar
jan.koester committed
const char *authdb::UserData::getMail(){
jan.koester's avatar
jan.koester committed
    return getUserData(this,&mail,"mail");
jan.koester's avatar
jan.koester committed
void authdb::UserData::setPwHash(const char* src){
jan.koester's avatar
jan.koester committed
    setUserData(this,&pwhash,"pwhash",src);
jan.koester's avatar
jan.koester committed
const char * authdb::UserData::getPwHash(){
jan.koester's avatar
jan.koester committed
    return getUserData(this,&pwhash,"pwhash");
jan.koester's avatar
jan.koester committed
void authdb::UserData::setUserName(const char* src){
jan.koester's avatar
jan.koester committed
    setUserData(this,&username,"username",src);
jan.koester's avatar
jan.koester committed
const char * authdb::UserData::getUsername(){
jan.koester's avatar
jan.koester committed
    return getUserData(this,&username,"username");
jan.koester's avatar
jan.koester committed
void authdb::UserData::setAvatar(const std::vector<char>& src){
jan.koester's avatar
jan.koester committed
   AuthData::Record *target=avatar;
jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed
   AuthData *curec=this;
jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed
   uuid_t gid;
   uuid_copy(gid,Data->uuid);

jan.koester's avatar
jan.koester committed
   if(!target){
        while(curec){
            if(curec->Data->type == authdb::UserData &&
                uuid_compare(curec->Data->uuid,uid)==0 &&
jan.koester's avatar
jan.koester committed
                strcmp(curec->Data->fieldname,"avatar")==0
             ){

                target=curec->Data;
                target->datasize=0;
                delete[] target->data;
                target->data=nullptr;
jan.koester's avatar
jan.koester committed
                break;
            }
jan.koester's avatar
jan.koester committed
            curec=curec->next();
jan.koester's avatar
jan.koester committed
        }
jan.koester's avatar
jan.koester committed
    }
jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed
    if(!target){
      AuthData::Record newrec;
jan.koester's avatar
jan.koester committed
      target=append(newrec)->Data;
jan.koester's avatar
jan.koester committed
    }
jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed
    target->type=DataType::UserData;
    target->storage=StorageType::BinaryStorage;

jan.koester's avatar
jan.koester committed
    uuid_copy(target->uuid,gid);

jan.koester's avatar
jan.koester committed
    snprintf(target->fieldname,255,"%s","avatar");
    target->datasize=src.size();
    target->data= new char[target->datasize];
    memcpy(target->data,src.data(),target->datasize);
jan.koester's avatar
jan.koester committed

    avatar=target;
}

jan.koester's avatar
jan.koester committed
void authdb::UserData::getAvatar(std::vector<char>& dest){
jan.koester's avatar
jan.koester committed
    if(avatar && avatar->data)
        std::copy(avatar->data,avatar->data+avatar->datasize,
jan.koester's avatar
jan.koester committed
                  std::back_inserter(dest));
    else
jan.koester's avatar
jan.koester committed
        throw(authdb::AuthBackendError("Avatar data nout found !"));
jan.koester's avatar
jan.koester committed
void authdb::User::create(AuthBackend &backend,class UserData* dat){
jan.koester's avatar
jan.koester committed

    int exi=0;
jan.koester's avatar
jan.koester committed

    if(exits(backend,&dat->uid,dat->getUsername(),exi)){
       switch(exi){
           case 1:
jan.koester's avatar
jan.koester committed
                throw authdb::AuthBackendError("Create User: Username already Exists aborting!");
           default:
jan.koester's avatar
jan.koester committed
                throw authdb::AuthBackendError("Create User: UUID already Exists aborting!");
jan.koester's avatar
jan.koester committed
    }
jan.koester's avatar
jan.koester committed

    size_t wd=backend.end();
jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed
    for(AuthData *curec=dat; curec; curec=curec->next()){
jan.koester's avatar
jan.koester committed
         curec->Data->type=UserData;
         backend.setPos(wd);
         backend.write((unsigned char*)curec->Data,sizeof(struct AuthData::Record));
jan.koester's avatar
jan.koester committed
         backend.write((unsigned char*)curec->Data->data,curec->Data->datasize);
jan.koester's avatar
jan.koester committed
         wd=backend.end();
jan.koester's avatar
jan.koester committed
    }
jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed
}

bool authdb::User::exits(AuthBackend& backend, uuid_t* uid, const char* name,int &ret){
    authdb::AuthData::Record cur;

    size_t rd=sizeof(authdb::AuthHeader),end=backend.end();

    while(rd<end){
        backend.setPos(rd);
        backend.read((unsigned char*)&cur,sizeof(AuthData::Record));
        rd=backend.getPos()+cur.datasize;
        if(strcmp(cur.fieldname,"username")==0 ){
            cur.data=new char[cur.datasize];
            backend.read((unsigned char*)cur.data,cur.datasize);
            if(name && strcmp(cur.data,name) == 0){
                delete[] cur.data;
                ret=1;
                return true;

            }else if(uid && uuid_compare(cur.uuid,*uid)==0){
                delete[] cur.data;
                ret=2;
                return true;
            }
            delete[] cur.data;
        }
    }
    ret=-1;
    return false;
}


void authdb::User::edit(AuthBackend &backend,class UserData& dat, size_t &pos){
jan.koester's avatar
jan.koester committed
}

jan.koester's avatar
jan.koester committed
void authdb::User::remove(AuthBackend &backend,const uuid_t uid){
jan.koester's avatar
jan.koester committed
    delRecord(backend,uid,DataType::UserData);
jan.koester's avatar
jan.koester committed
}
jan.koester's avatar
jan.koester committed

void authdb::User::info(AuthBackend &backend, class UserData& dat, size_t &pos){
jan.koester's avatar
jan.koester committed
    size_t end=backend.end();

    if(uuid_is_null(dat.uid))
jan.koester's avatar
jan.koester committed
        throw AuthBackendError("user info uid required!");

    while(pos<end){
jan.koester's avatar
jan.koester committed
        authdb::AuthData::Record *cur=new AuthData::Record;
        backend.setPos(pos);
        backend.read((unsigned char*)cur,sizeof(AuthData::Record));
jan.koester's avatar
jan.koester committed
        pos=backend.getPos()+cur->datasize;
        if(cur->type == UserData && uuid_compare(dat.uid,cur->uuid) == 0){
jan.koester's avatar
jan.koester committed

jan.koester's avatar
jan.koester committed
            if(!getRecord(backend,dat,DataType::UserData))
jan.koester's avatar
jan.koester committed
                continue;

jan.koester's avatar
jan.koester committed
            if(strcmp(dat.Data->fieldname,"username")==0){
                dat.username=dat.Data;
            }else if(strcmp(dat.Data->fieldname,"firstname")==0){
                dat.firstname=dat.Data;
            }else if(strcmp(dat.Data->fieldname,"lastname")==0){
                dat.lastname=dat.Data;
            }else if(strcmp(dat.Data->fieldname,"mail")==0){
                dat.mail=dat.Data;
            }else if(strcmp(dat.Data->fieldname,"pwhash")==0){
                dat.pwhash=dat.Data;
            }else if(strcmp(dat.Data->fieldname,"avatar")==0){
                dat.avatar=dat.Data;
jan.koester's avatar
jan.koester committed
            }
        }
        delete cur;
    }