osmo-st-network/IPAMuxer.st

"Copyright Header"

Object subclass: IPADemuxer [
    | socket |

    IPADemuxer class >> initOn: aSocket [
	^ (self new)
	    socket: aSocket;
	    yourself.
    ]

    next [
	"Return a tuple of stream and bytearray"

	| size stream data |

	size := socket nextUshort swap16.
	stream := socket nextByte.
	data := socket next: size.

	^ Array with: stream with: data.
    ]

    socket: aSocket [
	socket := aSocket.
    ]
]

Object subclass: IPAMuxer [
    | socket |

    IPADemuxer class >> initOn: aSocket [
	^ (self new)
	    socket: aSocket;
	    yourself.
    ]

    prepareNext: aData [
	"Write the data onto the stream"
	| stream size |

	aData size > 65536
	    ifTrue: [
	    self error: 'Too much data'.
	].

	stream := WriteStream on: (ByteArray new: aData size + 3).

	"put the size in ntohs"
	size := aData size swap16.
	stream nextPut: size bitAnd: 16rFF.
	stream nextPut: ((size bitShift: 8) bitAnd: 16rFF).

	"Copy the old data to the new stream"
	1 to: aData size do: [:byte | stream nextPut: aData at: byte].

	^ stream collection.
    ]

    nextPut: aData [
	socket nextPut: (self prepareNext: aData).
    ]

    socket: aSocket [
	socket := aSocket.
    ]
]