// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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.

#include <stdlib.h>
#include <stdio.h>
#include <iostream>
#include <limits.h>

#include "exec/read-write-util.h"
#include "testutil/gtest-util.h"

#include "common/names.h"

namespace impala {

template<typename T>
void TestBigEndian(T value) {
  uint8_t buffer[sizeof(T)];
  ReadWriteUtil::PutInt(buffer, value);
  EXPECT_EQ(value, ReadWriteUtil::GetInt<T>(buffer));
}

// Test put and get of big endian values
TEST(ReadWriteUtil, BigEndian) {
  TestBigEndian<uint16_t>(0);
  TestBigEndian<uint16_t>(0xff);
  TestBigEndian<uint16_t>(0xffff);

  TestBigEndian<uint32_t>(0);
  TestBigEndian<uint32_t>(0xff);
  TestBigEndian<uint32_t>(0xffff);
  TestBigEndian<uint32_t>(0xffffff);
  TestBigEndian<uint32_t>(0xffffffff);

  TestBigEndian<uint64_t>(0);
  TestBigEndian<uint64_t>(0xff);
  TestBigEndian<uint64_t>(0xffff);
  TestBigEndian<uint64_t>(0xffffff);
  TestBigEndian<uint64_t>(0xffffffff);
  TestBigEndian<uint64_t>(0xffffffffff);
  TestBigEndian<uint64_t>(0xffffffffffff);
  TestBigEndian<uint64_t>(0xffffffffffffff);
}

TEST(ReadWriteUtil, ZeroCompressedLongRequiredBytes) {
  // Small longs stored in 1 byte
  for (int64_t val = -112; val <= 127; val++) {
    EXPECT_EQ(1, ReadWriteUtil::VLongRequiredBytes(val));
  }
  // Small longs stored in 2 bytes
  for (int64_t val = -128; val < -112; val++) {
    EXPECT_EQ(2, ReadWriteUtil::VLongRequiredBytes(val));
  }
  // Positive longs stored in 3-9 bytes
  int64_t val = 0x7000ab00cd00ef00;
  for (int sh = 0; sh <= 6; sh++) {
    EXPECT_EQ(9 - sh, ReadWriteUtil::VLongRequiredBytes(val));
    val = val >> 8;
  }
  // Negative longs stored 3-9 bytes
  val = 0x8000ab00cd00ef00;
  for (int sh = 0; sh <= 6; sh++) {
    EXPECT_EQ(9 - sh, ReadWriteUtil::VLongRequiredBytes(val));
    val = val >> 8;
  }
  //Max/min long is stored in 9 bytes
  EXPECT_EQ(9, ReadWriteUtil::VLongRequiredBytes(0x7fffffffffffffff));
  EXPECT_EQ(9, ReadWriteUtil::VLongRequiredBytes(0x8000000000000000));
}

void TestPutGetZeroCompressedLong(int64_t val) {
  const int32_t BUFSZ = 9;
  uint8_t buffer[BUFSZ];
  int64_t read_val;
  int64_t num_bytes = ReadWriteUtil::PutVLong(val, buffer);
  int64_t read_bytes = ReadWriteUtil::GetVLong(buffer, &read_val, BUFSZ);
  EXPECT_EQ(read_bytes, num_bytes);
  EXPECT_EQ(read_val, val);
  // Out of bound access check, -1 should be returned because buffer size is passed
  // as 1 byte.
  if (read_bytes > 1) {
    read_bytes = ReadWriteUtil::GetVLong(buffer, &read_val, 1);
    EXPECT_EQ(read_bytes, -1);
  }
}

TEST(ReadWriteUtil, ZeroCompressedLong) {
  //1 byte longs
  for (int64_t val = -128; val <= 127; val++) {
    TestPutGetZeroCompressedLong(val);
  }
  //2+ byte positive longs
  int64_t val = 0x70100000200000ab;
  for (int sh = 0; sh <= 6; sh++) {
    TestPutGetZeroCompressedLong(val);
    val = val >> 8;
  }
  //2+ byte negative longs
  val = 0x80100000200000ab;
  for (int sh = 0; sh <= 6; sh++) {
    TestPutGetZeroCompressedLong(val);
    val = val >> 8;
  }
  //Max/min long
  TestPutGetZeroCompressedLong(0x7fffffffffffffff);
  TestPutGetZeroCompressedLong(0x8000000000000000);
}

}