blob: 0558b61ca8d7308d86197ba924e58d2862718a2e (
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
|
import io/dynstream
import io/posixstream
type
BufStream* = ref object of DynStream
source*: PosixStream
registerFun: proc(fd: int)
registered: bool
writeBuffer: string
method recvData*(s: BufStream, buffer: pointer, len: int): int =
s.source.recvData(buffer, len)
method sendData*(s: BufStream, buffer: pointer, len: int): int =
s.source.setBlocking(false)
block nobuf:
var n: int
if not s.registered:
try:
n = s.source.sendData(buffer, len)
if n == len:
break nobuf
except ErrorAgain:
discard
s.registerFun(s.source.fd)
s.registered = true
let olen = s.writeBuffer.len
s.writeBuffer.setLen(s.writeBuffer.len + len - n)
let buffer = cast[ptr UncheckedArray[uint8]](buffer)
copyMem(addr s.writeBuffer[olen], addr buffer[n], len - n)
s.source.setBlocking(true)
return len
method sclose*(s: BufStream) =
s.source.sclose()
proc flushWrite*(s: BufStream): bool =
s.source.setBlocking(false)
let n = s.source.sendData(s.writeBuffer)
s.source.setBlocking(true)
if n == s.writeBuffer.len:
s.writeBuffer = ""
s.registered = false
return true
s.writeBuffer = s.writeBuffer.substr(n)
return false
proc newBufStream*(ps: PosixStream, registerFun: proc(fd: int)): BufStream =
result = BufStream(
source: ps,
blocking: ps.blocking,
registerFun: registerFun
)
result.addStreamIface()
|