test.cpp

// Copyright 2011-2023:
//   GobySoft, LLC (2013-)
//   Massachusetts Institute of Technology (2007-2014)
//   Community contributors (see AUTHORS file)
// File authors:
//   Toby Schneider <toby@gobysoft.org>
//   Chris Murphy <cmurphy@aphysci.com>
//
//
// This file is part of the Dynamic Compact Control Language Library
// ("DCCL").
//
// DCCL is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 2.1 of the License, or
// (at your option) any later version.
//
// DCCL 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with DCCL.  If not, see <http://www.gnu.org/licenses/>.
// tests custom message codec
// tests cryptography
 
#include "../../codec.h"
#include "test.pb.h"
 
#include "../../binary.h"
using namespace dccl::test;
 
using dccl::operator<<;
using dccl::Bitset;
 
namespace dccl
{
namespace test
{
class CustomCodec : public dccl::TypedFixedFieldCodec<CustomMsg>
{
  private:
    unsigned size() override { return (part() == dccl::HEAD) ? 0 : A_SIZE + B_SIZE; }
    Bitset encode() override { return Bitset(size()); }
 
    Bitset encode(const CustomMsg& msg) override
    {
        if (part() == dccl::HEAD)
        {
            return encode();
        }
        else
        {
            Bitset a(A_SIZE, static_cast<unsigned long>(msg.a()));
            Bitset b(B_SIZE, static_cast<unsigned long>(msg.b()));
 
            dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "a: " << a << std::endl;
            dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "b: " << b << std::endl;
 
            a.append(b);
            return a;
        }
    }
 
    CustomMsg decode(Bitset* bits) override
    {
        if (part() == dccl::HEAD)
        {
            throw dccl::NullValueException();
        }
        else
        {
            Bitset a = *bits;
            a.resize(A_SIZE);
            Bitset b = *bits;
            b >>= A_SIZE;
            b.resize(B_SIZE);
 
            CustomMsg msg;
            msg.set_a(a.to_ulong());
            msg.set_b(b.to_ulong());
            return msg;
        }
    }
 
    void validate() override {}
 
    enum
    {
        A_SIZE = 32
    };
    enum
    {
        B_SIZE = 1
    };
};
 
class Int32RepeatedCodec : public dccl::RepeatedTypedFieldCodec<dccl::int32>
{
  private:
    enum
    {
        REPEAT_STORAGE_BITS = 4
    };
    enum
    {
        MAX_REPEAT_SIZE = 1 << REPEAT_STORAGE_BITS
    }; // 2^4
 
    dccl::int32 max() { return FieldCodecBase::dccl_field_options().max(); }
    dccl::int32 min() { return FieldCodecBase::dccl_field_options().min(); }
    dccl::int32 max_repeat() { return FieldCodecBase::dccl_field_options().max_repeat(); }
 
    Bitset encode_repeated(const std::vector<dccl::int32>& wire_values) override
    {
        Bitset value_bits;
        int repeat_size =
            static_cast<int>(wire_values.size()) > max_repeat() ? max_repeat() : wire_values.size();
 
        dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "repeat size is " << repeat_size << std::endl;
 
        for (int i = 0, n = repeat_size; i < n; ++i)
        {
            dccl::int32 wire_value = wire_values[i];
            wire_value -= min();
            value_bits.append(Bitset(singular_size(), static_cast<unsigned long>(wire_value)));
        }
 
        dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << value_bits << std::endl;
        dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << Bitset(REPEAT_STORAGE_BITS, repeat_size) << std::endl;
        Bitset out(REPEAT_STORAGE_BITS, repeat_size);
        out.append(value_bits);
        return out;
    }
 
    std::vector<dccl::int32> decode_repeated(Bitset* bits) override
    {
        int repeat_size = bits->to_ulong();
        dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "repeat size is " << repeat_size << std::endl;
        // grabs more bits to add to the MSBs of `bits`
        bits->get_more_bits(repeat_size * singular_size());
 
        Bitset value_bits = *bits;
        value_bits >>= REPEAT_STORAGE_BITS;
 
        std::vector<dccl::int32> out;
        for (int i = 0; i < repeat_size; ++i)
        {
            dccl::int32 value = value_bits.to_ulong() & ((1 << singular_size()) - 1);
            value += min();
            out.push_back(value);
            value_bits >>= singular_size();
        }
        return out;
    }
 
    unsigned size_repeated(const std::vector<dccl::int32>& field_values) override
    {
        return REPEAT_STORAGE_BITS + field_values.size() * singular_size();
    }
 
    unsigned singular_size() { return dccl::ceil_log2((max() - min()) + 1); }
 
    unsigned max_size_repeated() override
    {
        return REPEAT_STORAGE_BITS + max_repeat() * singular_size();
    }
 
    unsigned min_size_repeated() override { return REPEAT_STORAGE_BITS; }
 
    void validate() override
    {
        FieldCodecBase::require(FieldCodecBase::dccl_field_options().has_min(),
                                "missing (dccl.field).min");
        FieldCodecBase::require(FieldCodecBase::dccl_field_options().has_max(),
                                "missing (dccl.field).max");
        FieldCodecBase::require(
            FieldCodecBase::dccl_field_options().max_repeat() < MAX_REPEAT_SIZE,
            "(dccl.field).max_repeat must be less than " +
                std::to_string(static_cast<int>(MAX_REPEAT_SIZE)));
    }
};
 
} // namespace test
} // namespace dccl
 
int main(int argc, char* argv[])
{
    bool verbose = false;
    for (int i = 1; i < argc; ++i)
    {
        if (argv[i] && argv[i][0] == '-' && argv[i][1] == 'v' && argv[i][2] == '\0')
            verbose = true;
    }
 
    dccl::dlog.connect(verbose ? dccl::logger::ALL : dccl::logger::WARN_PLUS, &std::cerr);
 
    dccl::Codec codec;
    codec.manager().add<dccl::test::CustomCodec>("custom_codec");
    codec.manager().add<dccl::test::Int32RepeatedCodec>("int32_test_codec");
 
    codec.set_crypto_passphrase("my_passphrase!");
 
    CustomMsg msg_in1;
 
    msg_in1.set_a(10);
    msg_in1.set_b(true);
 
    if (dccl::dlog.is(dccl::logger::INFO))
        codec.info(msg_in1.GetDescriptor(), &dccl::dlog);
 
    
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "Message in:\n" << msg_in1.DebugString() << std::endl;
    codec.load(msg_in1.GetDescriptor());
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "Try encode..." << std::endl;
    std::string bytes1;
    codec.encode(&bytes1, msg_in1);
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "... got bytes (hex): " << dccl::hex_encode(bytes1) << std::endl;
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "Try decode..." << std::endl;
    std::shared_ptr<google::protobuf::Message> msg_out1 =
        codec.decode<std::shared_ptr<google::protobuf::Message>>(bytes1);
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "... got Message out:\n" << msg_out1->DebugString() << std::endl;
    assert(msg_in1.SerializeAsString() == msg_out1->SerializeAsString());
 
    CustomMsg2 msg_in2, msg_out2;
 
    msg_in2.mutable_msg()->CopyFrom(msg_in1);
    msg_in2.add_c(30);
    msg_in2.add_c(2);
 
    if (dccl::dlog.is(dccl::logger::INFO))
    {
        codec.info(msg_in2.GetDescriptor(), &dccl::dlog);
    }
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "Message in:\n" << msg_in2.DebugString() << std::endl;
    codec.load(msg_in2.GetDescriptor());
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "Try encode..." << std::endl;
    std::string bytes2;
    codec.encode(&bytes2, msg_in2);
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "... got bytes (hex): " << dccl::hex_encode(bytes2) << std::endl;
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "Try decode..." << std::endl;
    codec.decode(bytes2, &msg_out2);
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "... got Message out:\n" << msg_out2.DebugString() << std::endl;
    assert(msg_in2.SerializeAsString() == msg_out2.SerializeAsString());
 
    dccl::dlog.is(dccl::logger::INFO) && dccl::dlog << "all tests passed" << std::endl;
}