blob: df7e9632f057db7ae6c619e5791175c704a02376 (
plain) (
blame)
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
61
62
63
64
65
66
67
68
69
|
/* SPDX-License-Identifier: Apache-2.0 */
/*
* Copyright (C) Andrew Clayton
* Copyright (C) Timo Stark
* Copyright (C) F5, Inc.
*/
use unit_wasm::rusty::*;
use std::fs::File;
use std::ptr::{addr_of_mut, null_mut};
static mut CTX: luw_ctx_t = UWR_CTX_INITIALIZER();
static mut REQUEST_BUF: *mut u8 = null_mut();
static mut TOTAL_BYTES_WROTE: u64 = 0;
#[no_mangle]
pub unsafe extern "C" fn uwr_module_end_handler() {
uwr_free(REQUEST_BUF);
}
#[no_mangle]
pub unsafe extern "C" fn uwr_module_init_handler() {
REQUEST_BUF = uwr_malloc(uwr_mem_get_init_size());
}
#[no_mangle]
pub unsafe extern "C" fn uwr_response_end_handler() {
TOTAL_BYTES_WROTE = 0;
}
#[no_mangle]
pub extern "C" fn uwr_request_handler(addr: *mut u8) -> i32 {
let ctx: *mut luw_ctx_t = unsafe { addr_of_mut!(CTX) };
let mut f;
let bytes_wrote: isize;
let mut total = unsafe { TOTAL_BYTES_WROTE };
if total == 0 {
uwr_init_ctx(ctx, addr, 0);
uwr_set_req_buf(
ctx,
unsafe { addr_of_mut!(REQUEST_BUF) },
LUW_SRB_NONE,
);
f = File::create("/var/tmp/large-file.dat").unwrap();
} else {
f = File::options()
.append(true)
.open("/var/tmp/large-file.dat")
.unwrap();
}
bytes_wrote = uwr_mem_splice_file(addr, &mut f);
if bytes_wrote == -1 {
return -1;
}
total += bytes_wrote as u64;
if total == uwr_get_http_content_len(ctx) {
uwr_http_response_end();
} else {
unsafe { TOTAL_BYTES_WROTE = total };
}
return 0;
}
|
