tdbUtil.c 1.2 KB
Newer Older
H
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

H
Hongze Cheng 已提交
16
#include "tdbInt.h"
H
Hongze Cheng 已提交
17

H
Hongze Cheng 已提交
18
int tdbGnrtFileID(const char *fname, uint8_t *fileid, bool unique) {
H
Hongze Cheng 已提交
19 20 21 22 23 24 25 26 27 28
  struct stat statbuf;

  if (stat(fname, &statbuf) < 0) {
    return -1;
  }

  memset(fileid, 0, TDB_FILE_ID_LEN);

  ((uint64_t *)fileid)[0] = (uint64_t)statbuf.st_ino;
  ((uint64_t *)fileid)[1] = (uint64_t)statbuf.st_dev;
H
Hongze Cheng 已提交
29 30 31
  if (unique) {
    ((uint64_t *)fileid)[2] = rand();
  }
H
Hongze Cheng 已提交
32 33 34 35

  return 0;
}

H
Hongze Cheng 已提交
36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52
int tdbCheckFileAccess(const char *pathname, int mode) {
  int flags = 0;

  if (mode & TDB_F_OK) {
    flags |= F_OK;
  }

  if (mode & TDB_R_OK) {
    flags |= R_OK;
  }

  if (mode & TDB_W_OK) {
    flags |= W_OK;
  }

  return access(pathname, flags);
}