implements avrt

Cargo.toml

@@ -12,6 +12,7 @@ include = ["src/**/*", "Cargo.toml", "LICENSE.md"]
 
 [dev-dependencies]
 advapi32-sys = { version = "0", path = "lib/advapi32" }
+avrt-sys = { version = "0", path = "lib/avrt" }
 bcrypt-sys = { version = "0", path = "lib/bcrypt" }
 comctl32-sys = { version = "0", path = "lib/comctl32" }
 comdlg32-sys = { version = "0", path = "lib/comdlg32" }

lib/avrt/src/lib.rs

@@ -1,9 +1,60 @@
 // Copyright © 2015, Peter Atashian
 // Licensed under the MIT License <LICENSE.md>
 //! FFI bindings to avrt.
-#![no_std]
-#![experimental]
+#![cfg(windows)]
 extern crate winapi;
-use winapi::*;
+
 extern "system" {
+    pub fn AvSetMmThreadCharacteristicsA(
+        TaskName: winapi::LPCSTR, TaskIndex: winapi::LPDWORD
+    ) -> winapi::HANDLE;
+
+    pub fn AvSetMmThreadCharacteristicsW(
+        TaskName: winapi::LPCSTR, TaskIndex: winapi::LPDWORD
+    ) -> winapi::HANDLE;
+
+    pub fn AvSetMmMaxThreadCharacteristicsA (
+        FirstTask: winapi::LPCSTR, SecondTask: winapi::LPCSTR, TaskIndex: winapi::LPDWORD
+    ) -> winapi::HANDLE;
+
+    pub fn AvSetMmMaxThreadCharacteristicsW (
+        FirstTask: winapi::LPCWSTR, SecondTask: winapi::LPCWSTR, TaskIndex: winapi::LPDWORD
+    ) -> winapi::HANDLE;
+
+    pub fn AvRevertMmThreadCharacteristics(avrt_handle: winapi::HANDLE) -> winapi::BOOL;
+
+    pub fn AvSetMmThreadPriority (
+        AvrtHandle: winapi::HANDLE, Priority: winapi::AVRT_PRIORITY) -> winapi::BOOL;
+
+    pub fn AvRtCreateThreadOrderingGroup (
+        Context: winapi::PHANDLE, Period: winapi::PLARGE_INTEGER,
+        ThreadOrderingGuid: * mut winapi::GUID, Timeout: winapi::PLARGE_INTEGER
+    ) -> winapi::BOOL;
+
+    pub fn AvRtCreateThreadOrderingGroupExA (
+        Context: winapi::PHANDLE, Period: winapi::PLARGE_INTEGER,
+        ThreadOrderingGuid: * mut winapi::GUID, Timeout: winapi::PLARGE_INTEGER,
+        TaskName: winapi::LPCSTR
+    ) -> winapi::BOOL;
+
+    pub fn AvRtCreateThreadOrderingGroupExW (
+        Context: winapi::PHANDLE, Period: winapi::PLARGE_INTEGER,
+        ThreadOrderingGuid: * mut winapi::GUID, Timeout: winapi::PLARGE_INTEGER,
+        TaskName: winapi::LPCWSTR
+    ) -> winapi::BOOL;
+
+    pub fn AvRtJoinThreadOrderingGroup (
+        Context: winapi::PHANDLE, ThreadOrderingGuid: * mut winapi::GUID,
+        Before: winapi::BOOL
+    ) -> winapi::BOOL;
+
+    pub fn AvRtWaitOnThreadOrderingGroup (Context: winapi::HANDLE) -> winapi::BOOL;
+
+    pub fn AvRtLeaveThreadOrderingGroup (Context: winapi::HANDLE) -> winapi::BOOL;
+
+    pub fn AvRtDeleteThreadOrderingGroup (Context: winapi::HANDLE) -> winapi::BOOL;
+
+    pub fn AvQuerySystemResponsiveness (
+        AvrtHandle: winapi::HANDLE, SystemResponsivenessValue: winapi::PULONG
+    ) -> winapi::BOOL;
 }

src/avrt.rs

@@ -0,0 +1,17 @@
+// Copyright © 2016, Baptiste AUBRY
+// Licensed under the MIT License <LICENSE.md>
+
+ENUM_NEG!{enum AVRT_PRIORITY {
+    AVRT_PRIORITY_VERYLOW = -2,
+    AVRT_PRIORITY_LOW,
+    AVRT_PRIORITY_NORMAL,
+    AVRT_PRIORITY_HIGH,
+    AVRT_PRIORITY_CRITICAL,
+}}
+
+ENUM!{enum AVRT_NOTIFICATION {
+    AVRT_PROCESS_SUSPEND = 0,
+    AVRT_PROCESS_RESUME,
+}}
+
+pub const THREAD_ORDER_GROUP_INFINITE_TIMEOUT: ::LONGLONG = -1;

src/lib.rs

@@ -27,6 +27,7 @@ pub use std::os::raw::{
 pub use activation::*;
 pub use audioclient::*;
 pub use audiosessiontypes::*;
+pub use avrt::*;
 pub use basetsd::*;
 pub use bcrypt::*;
 pub use cfg::*;
@@ -192,6 +193,7 @@ pub use xinput::*;
 pub mod activation;
 pub mod audioclient;
 pub mod audiosessiontypes;
+pub mod avrt;
 pub mod basetsd;
 pub mod bcrypt;
 pub mod cfg;

src/macros.rs

@@ -227,6 +227,33 @@ macro_rules! ENUM {
         ENUM!{@gen $name, $variant, $($rest)*}
     };
 }
+
+macro_rules! ENUM_NEG {
+    {enum $name:ident { $($variant:ident = $value:expr,)+ }} => {
+        #[repr(C)] #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
+        pub struct $name(pub i32);
+        $(pub const $variant: $name = $name($value);)+
+    };
+    {enum $name:ident { $variant:ident = $value:expr, $($rest:tt)* }} => {
+        #[repr(C)] #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
+        pub struct $name(pub i32);
+        pub const $variant: $name = $name($value);
+        ENUM_NEG!{@gen $name, $variant, $($rest)*}
+    };
+    {enum $name:ident { $variant:ident, $($rest:tt)* }} => {
+        ENUM_NEG!{enum $name { $variant = 0, $($rest)* }}
+    };
+    {@gen $name:ident, $base:ident,} => {};
+    {@gen $name:ident, $base:ident, $variant:ident = $value:expr, $($rest:tt)*} => {
+        pub const $variant: $name = $name($value);
+        ENUM_NEG!{@gen $name, $variant, $($rest)*}
+    };
+    {@gen $name:ident, $base:ident, $variant:ident, $($rest:tt)*} => {
+        pub const $variant: $name = $name($base.0 + 1i32);
+        ENUM_NEG!{@gen $name, $variant, $($rest)*}
+    };
+}
+
 macro_rules! FLAGS {
     {enum $name:ident { $($variant:ident = $value:expr,)+ }} => {
         #[repr(C)] #[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]

tests/avrt.rs

@@ -0,0 +1,27 @@
+// Copyright © 2016, Baptiste AUBRY
+// Licensed under the MIT License <LICENSE.md>
+#![cfg(windows)]
+extern crate avrt;
+extern crate winapi;
+
+use avrt::*;
+
+#[inline(never)] fn bb<T>(_: T) {}
+#[test] #[cfg(target_env = "msvc")]
+fn functions() {
+    bb(AvSetMmThreadCharacteristicsA);   
+    bb(AvSetMmThreadCharacteristicsW);
+    bb(AvSetMmMaxThreadCharacteristicsA);
+    bb(AvSetMmMaxThreadCharacteristicsW);
+    bb(AvRevertMmThreadCharacteristics);
+    bb(AvRtCreateThreadOrderingGroup);
+    bb(AvRtCreateThreadOrderingGroupExA);
+    bb(AvRtCreateThreadOrderingGroupExW);
+    bb(AvRtJoinThreadOrderingGroup);
+    bb(AvRtWaitOnThreadOrderingGroup);
+    bb(AvRtLeaveThreadOrderingGroup);
+    bb(AvRtDeleteThreadOrderingGroup);
+    bb(AvQuerySystemResponsiveness);
+
+    assert_eq!(winapi::THREAD_ORDER_GROUP_INFINITE_TIMEOUT, -1);
+}