/*
 * Copyright (C) 2023 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.
 */

#ifndef BERBERIS_GUEST_OS_PRIMITIVES_GET_TLS_H_
#define BERBERIS_GUEST_OS_PRIMITIVES_GET_TLS_H_

namespace berberis {

#if defined(__i386__)
#define GetTls()                              \
  ({                                          \
    void** __val;                             \
    __asm__("movl %%gs:0, %0" : "=r"(__val)); \
    __val;                                    \
  })
#elif defined(__x86_64__)
#define GetTls()                             \
  ({                                         \
    void** __val;                            \
    __asm__("mov %%fs:0, %0" : "=r"(__val)); \
    __val;                                   \
  })
#elif defined(__riscv)
#define GetTls()                        \
  ({                                    \
    void** __val;                       \
    __asm__("mv %0, tp" : "=r"(__val)); \
    __val;                              \
  })
#elif defined(__aarch64__)
#define GetTls()                                \
  ({                                            \
    void** __val;                               \
    __asm__("mrs %0, tpidr_el0" : "=r"(__val)); \
    __val;                                      \
  })
#else
#error unsupported architecture
#endif

}  // namespace berberis

#endif  // BERBERIS_GUEST_OS_PRIMITIVES_GET_TLS_H_