sys_futex.h 2.07 KB
Newer Older
gejun's avatar
gejun committed
1
// bthread - A M:N threading library to make applications more concurrent.
gejun's avatar
gejun committed
2
// Copyright (c) 2012 Baidu, Inc.
gejun's avatar
gejun committed
3 4 5 6 7 8 9 10 11 12 13 14
// 
// 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.
gejun's avatar
gejun committed
15 16 17 18

// Author: Ge,Jun (gejun@baidu.com)
// Date: Tue Jul 10 17:40:58 CST 2012

gejun's avatar
gejun committed
19 20
#ifndef BTHREAD_SYS_FUTEX_H
#define BTHREAD_SYS_FUTEX_H
gejun's avatar
gejun committed
21

gejun's avatar
gejun committed
22
#include "butil/build_config.h"         // OS_MACOSX
gejun's avatar
gejun committed
23 24
#include <unistd.h>                     // syscall
#include <time.h>                       // timespec
gejun's avatar
gejun committed
25 26
#if defined(OS_LINUX)
#include <syscall.h>                    // SYS_futex
gejun's avatar
gejun committed
27 28 29 30
#include <linux/futex.h>                // FUTEX_WAIT, FUTEX_WAKE

namespace bthread {

gejun's avatar
gejun committed
31 32 33
#ifndef FUTEX_PRIVATE_FLAG
#define FUTEX_PRIVATE_FLAG 128
#endif
gejun's avatar
gejun committed
34 35 36

inline int futex_wait_private(
    void* addr1, int expected, const timespec* timeout) {
gejun's avatar
gejun committed
37
    return syscall(SYS_futex, addr1, (FUTEX_WAIT | FUTEX_PRIVATE_FLAG),
gejun's avatar
gejun committed
38 39 40 41
                   expected, timeout, NULL, 0);
}

inline int futex_wake_private(void* addr1, int nwake) {
gejun's avatar
gejun committed
42
    return syscall(SYS_futex, addr1, (FUTEX_WAKE | FUTEX_PRIVATE_FLAG),
gejun's avatar
gejun committed
43 44 45 46
                   nwake, NULL, NULL, 0);
}

inline int futex_requeue_private(void* addr1, int nwake, void* addr2) {
gejun's avatar
gejun committed
47
    return syscall(SYS_futex, addr1, (FUTEX_REQUEUE | FUTEX_PRIVATE_FLAG),
gejun's avatar
gejun committed
48 49 50 51 52
                   nwake, NULL, addr2, 0);
}

}  // namespace bthread

gejun's avatar
gejun committed
53 54 55 56
#elif defined(OS_MACOSX)

namespace bthread {

57 58 59 60 61
int futex_wait_private(void* addr1, int expected, const timespec* timeout);

int futex_wake_private(void* addr1, int nwake);

int futex_requeue_private(void* addr1, int nwake, void* addr2);
gejun's avatar
gejun committed
62 63 64 65 66 67 68

}  // namespace bthread

#else
#error "Unsupported OS"
#endif

gejun's avatar
gejun committed
69
#endif // BTHREAD_SYS_FUTEX_H