bytetree.cpp

Go to the documentation of this file.

#include <cstdlib>
#include <fstream>

#include "bytetree.hpp"
#include "leaf.hpp"
#include "node.hpp"

using namespace verifierUtils;


ByteTree::ByteTree()
{
}


ByteTree::~ByteTree()
{
}


bool ByteTree::isLeaf()
{
        return false;
}


bool ByteTree::isNode()
{
        return false;
}


std::string ByteTree::toString()
{
        std::string emptyString;
        return emptyString;
}


mpz_class ByteTree::toInteger()
{
        return 0;
}


std::vector<uint8_t> ByteTree::toVector()
{
        std::vector<uint8_t> emptyVector;
        return emptyVector;
}


std::vector<bool> ByteTree::toBoolArray()
{
        std::cout<<"ERROR: in ByteTree::toBoolArray(): trying to turn a"
                 <<" non-leaf into an array of booleans.\nAborting."
                 <<std::endl;
        prettyPrint("");
        std::cout<<std::endl;
        exit(1);
}


ByteTree * ByteTree::parseString(std::string &s)
{
        ByteTree * res;
        if (s.compare(0,2,"01") == 0)
                res = new Leaf(s);
        else if (s.compare(0,2,"00") == 0)
                res = new Node(s);
        else
        {
                std::cout<<"ERROR in ByteTree::parseString(): string "
                        "does not correspond to a bytetree! Here is the"
                        " problematic string: "<<s<<std::endl;
                exit(1);
        }         
        return res;
}


ByteTree * ByteTree::parseVector(std::vector<uint8_t> &v)
{
        ByteTree * res;
        if (v[0] == 1)
        {
                unsigned int l = v[1] * 0x1000000
                               + v[2] * 0x10000
                               + v[3] * 0x100
                               + v[4];
                std::vector<uint8_t> content(v.begin()+5,v.begin()+l+5);
                res = new Leaf(content);
                v.assign(v.begin()+l+5,v.end());
        }
        else if (v[0] == 0)
                res = new Node(v);
        else
        {
                std::cout<<"ERROR in ByteTree::parseVector(): vector "
                        "does not correspond to a bytetree! Here is the"
                        " problematic vector:\nv= ";
                for (unsigned int i=0; i<v.size(); i++)
                        std::cout<<v[i]<<", ";
                std::cout<<std::endl;
                exit(1);
        }
        return res;
}


ByteTree * ByteTree::parseFile(std::string path)
{
        std::vector<uint8_t> content;
        std::ifstream file(path);
        uint8_t c;
        while (file.good())
        {
                c = file.get();
                if (file.good())
                        content.push_back(c);
        }
        if (content.size() == 0)
                return NULL;
        else
                return parseVector(content);
}


ByteTree * ByteTree::string2ByteTree(std::string s)
{
        std::vector<uint8_t> content(s.size());
        for (unsigned int i=0; i<s.size(); i++)
                content[i] = s[i];
        return new Leaf(content);
}


unsigned int ByteTree::size()
{
        return 0;
}


void ByteTree::prettyPrint(std::string indent)
{
}


bool ByteTree::compare(std::string s)
{
        return true;
}


void ByteTree::addChild(ByteTree * bt)
{
        std::cout<<"ERROR: in ByteTree::addChild(): trying to access "
                "the children of a non-Node ByteTree. Aborting."
                 <<std::endl;
        prettyPrint("");
        std::cout<<std::endl;
        exit(1);
}


ByteTree * ByteTree::getChild(unsigned int i)
{
        std::cout<<"ERROR: in ByteTree::getChild(): trying to access "
                "the children of a non-Node ByteTree. Aborting."
                 <<std::endl;
        prettyPrint("");
        std::cout<<std::endl;
        exit(1);
}