/*
 * Copyright (C) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef __H_TLV_H__
#define __H_TLV_H__

#include "common.h"

namespace Hdc {
#define TLV_HEAD_SIZE (sizeof(uint32_t) + sizeof(uint32_t))
#define TLV_VALUE_MAX_LEN (16 * 1024 * 1024)

class TlvBuf {
public:
    TlvBuf();
    // construct a empty TlvBuf object with valid tags
    explicit  TlvBuf(std::set<uint32_t> validtags);
    TlvBuf(const uint8_t *tlvs, const uint32_t size);
    // construct a TlvBuf object from a TLV buffer with valid tags
    TlvBuf(const uint8_t *tlvs, const uint32_t size, const std::set<uint32_t> validtags);
    ~TlvBuf();
public:
    void Clear(void);
    bool Append(const uint32_t tag, const string &val);
    bool Append(const uint32_t tag, const uint32_t len, const uint8_t *val);
    // calculate total size for net tlv size, not contain the size of field 'TLV::uint8_t *val'
    uint32_t GetBufSize(void) const;
    bool CopyToBuf(uint8_t *dst, const uint32_t size) const;
    // // the caller must free the memory pointed by the return value if not null
    bool FindTlv(const uint32_t tag, uint32_t &len, uint8_t *&val) const;
    // // the caller must free the memory pointed by the return value if not null
    bool FindTlv(const uint32_t tag, string &val) const;
    // // if return true, invalid_tags is empty, else the invalid tags will bring out by invalid_tags
    bool ContainInvalidTag(void) const;
    void Display(void) const;
private:
    // key is the tag
    std::map<uint32_t, std::vector<uint8_t>> mTlvMap;
    std::set<uint32_t> mValidTags;
};

}

#endif