iocp 0.0.6

A Windows IOCP library for Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60

#![feature(box_syntax)]
#![allow(unstable)]
extern crate iocp;
extern crate libc;

use iocp::{IoCompletionPort, CompletionStatus};
use std::{ptr, mem};
use std::sync::TaskPool;
use std::os::num_cpus;
use std::old_io::timer::sleep;
use std::time::duration::Duration;
use std::rand::random;

fn main() {
	let iocp = IoCompletionPort::new(0).unwrap();
	
	let threads = num_cpus() * 2;
	let taskpool = TaskPool::new(threads);
	
	for i in range(0, threads) {
		let iocp_clone = iocp.clone();
		taskpool.execute(move || {
			loop {
				sleep(Duration::milliseconds(100 * i as i64));
				let status = iocp_clone.get_queued(libc::INFINITE).unwrap();
				println!("Dequeued: {} from {} with {} {:p}", status.completion_key, i, status.byte_count, status.overlapped);
				
				// We re-box all this stuff so it gets freed
				let overlapped: Box<libc::OVERLAPPED> = unsafe { mem::transmute(status.overlapped) };
				let internal: Box<u32> = unsafe { mem::transmute(overlapped.Internal) };
				let internal_high: Box<u32> = unsafe { mem::transmute(overlapped.InternalHigh) };
				
				println!("Overlapped: {} {} {} {} {:p}", internal, internal_high, overlapped.Offset, overlapped.OffsetHigh, overlapped.hEvent);
				
				sleep(Duration::milliseconds(500));
			}
		});
	}
	
	loop {
		let internal = box 3u32;
		let internal_high = box 4u32;
		// Transmute the boxes so they're on the heap but don't disappear
		let overlapped = box libc::OVERLAPPED {
			Internal: unsafe { mem::transmute(internal) },
			InternalHigh: unsafe { mem::transmute(internal_high) },
			Offset: 200,
			OffsetHigh: 300,
			hEvent: ptr::null_mut()
		};
		let status = CompletionStatus {
			byte_count: 100,
			completion_key: random(),
			overlapped: unsafe { mem::transmute(overlapped) }
		};
		println!("Queued: {}", status.completion_key);
		iocp.post_queued(status).unwrap();
		sleep(Duration::milliseconds(100));
	}
}