blob: 3dba86f119b9af98e22d4b29ad755679f10d0ca8 (
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
|
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 reallyFlush*(s: BufStream) =
if s.writeBuffer.len > 0:
s.source.sendDataLoop(s.writeBuffer)
proc newBufStream*(ps: PosixStream; registerFun: proc(fd: int)): BufStream =
return BufStream(source: ps, blocking: ps.blocking, registerFun: registerFun)
|