1 files changed, 75 insertions, 0 deletions

diff --git a/src/lib/libc/include/cancel.h b/src/lib/libc/include/cancel.h
new file mode 100644
index 0000000000..4f4add471b
--- /dev/null
+++ b/src/lib/libc/include/cancel.h
@@ -0,0 +1,75 @@
+/*      $OpenBSD: cancel.h,v 1.1 2016/05/07 19:05:22 guenther Exp $ */
+/*
+ * Copyright (c) 2015 Philip Guenther <guenther@openbsd.org>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#ifndef _CANCEL_H_
+#define _CANCEL_H_
+
+#include <tib.h>
+#include "thread_private.h"
+
+__BEGIN_HIDDEN_DECLS
+/* process a cancel request at a cancel point */
+__dead void     _thread_canceled(void);
+__END_HIDDEN_DECLS
+
+#ifdef LIBC
+/*
+ * Redirect macros that would use the syscall to instead use our callback
+ */
+#define __get_tcb()     _thread_cb.tc_tcb()
+#endif
+
+#define PREP_CANCEL_POINT(tib)                                          \
+        int _cantcancel = (tib)->tib_cantcancel
+
+#define ENTER_CANCEL_POINT_INNER(tib, can_cancel, delay)                \
+        if (_cantcancel == 0) {                                         \
+                (tib)->tib_cancel_point = (delay) ?                     \
+                    CANCEL_POINT_DELAYED : CANCEL_POINT;                \
+                if (can_cancel) {                                       \
+                        __asm volatile("":::"memory");                  \
+                        if (__predict_false((tib)->tib_canceled))       \
+                                _thread_canceled();                     \
+                }                                                       \
+        }
+
+#define LEAVE_CANCEL_POINT_INNER(tib, can_cancel)                       \
+        if (_cantcancel == 0) {                                         \
+                (tib)->tib_cancel_point = 0;                            \
+                if (can_cancel) {                                       \
+                        __asm volatile("":::"memory");                  \
+                        if (__predict_false((tib)->tib_canceled))       \
+                                _thread_canceled();                     \
+                }                                                       \
+        }
+
+/*
+ * Enter or leave a cancelation point, optionally processing pending
+ * cancelation requests.  Note that ENTER_CANCEL_POINT opens a block
+ * and LEAVE_CANCEL_POINT must close that same block.
+ */
+#define ENTER_CANCEL_POINT(can_cancel)                                  \
+    {                                                                   \
+        struct tib *_tib = TIB_GET();                                   \
+        PREP_CANCEL_POINT(_tib);                                        \
+        ENTER_CANCEL_POINT_INNER(_tib, can_cancel, 0)
+
+#define LEAVE_CANCEL_POINT(can_cancel)                                  \
+        LEAVE_CANCEL_POINT_INNER(_tib, can_cancel);                     \
+    }
+
+#endif /* _CANCEL_H_ */