/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program 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.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#include "tdbInt.h"

void *tdbRealloc(void *ptr, size_t size) {
  void *nPtr;
  if ((ptr) == NULL || ((int *)(ptr))[-1] < (size)) {
    nPtr = tdbOsRealloc((ptr) ? (char *)(ptr) - sizeof(int) : NULL, (size) + sizeof(int));
    if (nPtr) {
      ((int *)nPtr)[0] = (size);
      nPtr = (char *)nPtr + sizeof(int);
    }
  } else {
    nPtr = (ptr);
  }
  return nPtr;
}

void tdbFree(void *p) {
  if (p) {
    tdbOsFree((char *)(p) - sizeof(int));
  }
}

int tdbGnrtFileID(const char *fname, uint8_t *fileid, bool unique) {
  int64_t stDev = 0, stIno = 0;

  if (taosDevInoFile(fname, &stDev, &stIno) < 0) {
    return -1;
  }

  memset(fileid, 0, TDB_FILE_ID_LEN);

  ((uint64_t *)fileid)[0] = stDev;
  ((uint64_t *)fileid)[1] = stIno;
  if (unique) {
    ((uint64_t *)fileid)[2] = taosRand();
  }

  return 0;
}

int tdbGetFileSize(tdb_fd_t fd, int szPage, SPgno *size) {
  int     ret;
  int64_t szBytes;

  ret = tdbOsFileSize(fd, &szBytes);
  if (ret < 0) {
    return -1;
  }

  *size = szBytes / szPage;
  return 0;
}