// Copyright (C) 2021 The Android Open Source Project
//
// 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.

#include <sys/syscall.h>

#include <sstream>

#include <ditto/logger.h>
#include <ditto/syscall.h>

namespace dittosuite {

Syscall& Syscall::GetSyscall() {
  static Syscall syscall;
  return syscall;
}

int Syscall::Access(const std::string& path_name, int mode) {
  return access(path_name.c_str(), mode);
}

int Syscall::Close(int fd) {
  return close(fd);
}

int Syscall::CloseDir(DIR* dirp) {
  return closedir(dirp);
}

int Syscall::FAdvise(int fd, int64_t offset, int64_t len, int advice) {
  return posix_fadvise64(fd, offset, len, advice);
}

int Syscall::FAllocate(int fd, int mode, int64_t offset, int64_t len) {
  return fallocate64(fd, mode, offset, len);
}

int Syscall::FTruncate(int fd, int64_t length) {
  return ftruncate64(fd, length);
}

int Syscall::FStat(int filedes, struct stat64* buf) {
  return fstat64(filedes, buf);
}

int Syscall::FSync(int fd) {
  return fsync(fd);
}

pid_t Syscall::GetTid() {
  long ret = syscall(SYS_gettid);
  if (ret == -1) {
    PLOGF("Error calling syscall(SYS_gettid)");
  }
  return ret;
}

int Syscall::Open(const std::string& path_name, int flags, int mode) {
  return open(path_name.c_str(), flags, mode);
}

DIR* Syscall::OpenDir(const std::string& name) {
  return opendir(name.c_str());
}

int64_t Syscall::Read(int fd, char* buf, int64_t count, int64_t offset) {
  return pread64(fd, buf, count, offset);
}

struct dirent* Syscall::ReadDir(DIR* dirp) {
  return readdir(dirp);
}

int64_t Syscall::ReadLink(const std::string& path_name, char* buf, int64_t bufsiz) {
  return readlink(path_name.c_str(), buf, bufsiz);
}

#ifndef __NR_sched_setattr

/* Define all the __NR_sched_setattr syscall numbers for every architecture */

#ifdef __x86_64__
#define __NR_sched_setattr 314
#endif

#ifdef __i386__
#define __NR_sched_setattr 351
#endif

#ifdef __arm__
#define __NR_sched_setattr 380
#endif

/* If none of the architecture above have been matched, then use the
 * asm-generic/unistd.h definition 274, which also matches the aarch64
 * definition of __NR_sched_setattr. */
#ifndef __NR_sched_setattr
#define __NR_sched_setattr 274
#endif

#else /* __NR_sched_setattr */

/* Make sure the __NR_sched_setattr syscall numbers are consistent with the
Linux implementation */

#if ((defined(__x86_64__) && __NR_sched_setattr != 314) || \
     (defined(__i386__) && __NR_sched_setattr != 351) ||   \
     (defined(__arm__) && __NR_sched_setattr != 380)) &&   \
    __NR_sched_setattr != 274 /* aarch64 and asm-generic/unistd.h */
#error "Wrong definition of __NR_sched_setattr"
#endif

#endif /* __NR_sched_setattr */

int Syscall::SchedSetattr(pid_t pid, const SchedAttr__& attr, unsigned int flags) {
  long ret = syscall(__NR_sched_setattr, pid, &attr, flags);
  if (ret == -1) {
    PLOGF("Error calling syscall(__NR_sched_setattr)");
  }
  return ret;
}

void Syscall::Sync() {
  return sync();
}

int Syscall::Unlink(const std::string& path_name) {
  return unlink(path_name.c_str());
}

int64_t Syscall::Write(int fd, char* buf, int64_t count, int64_t offset) {
  return pwrite64(fd, buf, count, offset);
}

int Syscall::LockMutex(pthread_mutex_t* mutex) {
  return pthread_mutex_lock(mutex);
}

int Syscall::UnlockMutex(pthread_mutex_t* mutex) {
  return pthread_mutex_unlock(mutex);
}

std::string to_string(const SchedAttr__& attr) {
  std::stringstream ss;
  ss << "size: " << attr.size << ", policy: " << attr.sched_policy
     << ", flags: " << attr.sched_flags << ", nice: " << attr.sched_nice
     << ", priority: " << attr.sched_priority << ", runtime: " << attr.sched_runtime
     << ", deadline: " << attr.sched_deadline << ", period: " << attr.sched_period;
  return ss.str();
}

}  // namespace dittosuite