sys_futex.h 2.24 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you 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
18 19 20 21
// bthread - A M:N threading library to make applications more concurrent.

// Date: Tue Jul 10 17:40:58 CST 2012

gejun's avatar
gejun committed
22 23
#ifndef BTHREAD_SYS_FUTEX_H
#define BTHREAD_SYS_FUTEX_H
gejun's avatar
gejun committed
24

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

namespace bthread {

gejun's avatar
gejun committed
34 35 36
#ifndef FUTEX_PRIVATE_FLAG
#define FUTEX_PRIVATE_FLAG 128
#endif
gejun's avatar
gejun committed
37 38 39

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

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

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

}  // namespace bthread

gejun's avatar
gejun committed
56 57 58 59
#elif defined(OS_MACOSX)

namespace bthread {

60 61 62 63 64
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
65 66 67 68 69 70 71

}  // namespace bthread

#else
#error "Unsupported OS"
#endif

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