]>
Commit | Line | Data |
---|---|---|
7c673cae FG |
1 | // -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*- |
2 | // vim: ts=8 sw=2 smarttab | |
3 | /* | |
4 | * Ceph - scalable distributed file system | |
5 | * | |
6 | * Copyright (C) 2015 XSky <haomai@xsky.com> | |
7 | * | |
8 | * Author: Haomai Wang <haomaiwang@gmail.com> | |
9 | * | |
10 | * This is free software; you can redistribute it and/or | |
11 | * modify it under the terms of the GNU Lesser General Public | |
12 | * License version 2.1, as published by the Free Software | |
13 | * Foundation. See file COPYING. | |
14 | * | |
15 | */ | |
16 | ||
17 | #ifndef CEPH_COMMON_EVENT_SOCKET_H | |
18 | #define CEPH_COMMON_EVENT_SOCKET_H | |
19 | ||
7c673cae | 20 | #include <unistd.h> |
11fdf7f2 | 21 | #if defined(__FreeBSD__) || defined(__APPLE__) |
7c673cae | 22 | #include <errno.h> |
31f18b77 FG |
23 | #endif |
24 | #include "include/event_type.h" | |
7c673cae FG |
25 | |
26 | class EventSocket { | |
27 | int socket; | |
28 | int type; | |
29 | ||
30 | public: | |
31 | EventSocket(): socket(-1), type(EVENT_SOCKET_TYPE_NONE) {} | |
32 | bool is_valid() const { return socket != -1; } | |
33 | int init(int fd, int t) { | |
34 | switch (t) { | |
35 | case EVENT_SOCKET_TYPE_PIPE: | |
36 | #ifdef HAVE_EVENTFD | |
37 | case EVENT_SOCKET_TYPE_EVENTFD: | |
38 | #endif | |
39 | { | |
40 | socket = fd; | |
41 | type = t; | |
42 | return 0; | |
43 | } | |
44 | } | |
45 | return -EINVAL; | |
46 | } | |
47 | int notify() { | |
48 | int ret; | |
49 | switch (type) { | |
50 | case EVENT_SOCKET_TYPE_PIPE: | |
51 | { | |
52 | char buf[1]; | |
53 | buf[0] = 'i'; | |
54 | ret = write(socket, buf, 1); | |
55 | if (ret < 0) | |
56 | ret = -errno; | |
57 | else | |
58 | ret = 0; | |
59 | break; | |
60 | } | |
61 | #ifdef HAVE_EVENTFD | |
62 | case EVENT_SOCKET_TYPE_EVENTFD: | |
63 | { | |
64 | uint64_t value = 1; | |
65 | ret = write(socket, &value, sizeof (value)); | |
66 | if (ret < 0) | |
67 | ret = -errno; | |
68 | else | |
69 | ret = 0; | |
70 | break; | |
71 | } | |
72 | #endif | |
73 | default: | |
74 | { | |
75 | ret = -1; | |
76 | break; | |
77 | } | |
78 | } | |
79 | return ret; | |
80 | } | |
81 | }; | |
82 | ||
83 | #endif |