platform/packages/wings/lib/src/wings_socket.dart
Tobe O 13e863f295 Add 'packages/wings/' from commit '361e7e5f4a3ec6463f0348bdc8eb0a3b7549340a'
git-subtree-dir: packages/wings
git-subtree-mainline: 61c716502b
git-subtree-split: 361e7e5f4a
2020-02-16 12:21:46 -05:00

130 lines
3.5 KiB
Dart

import 'dart:async';
import 'dart:io';
import 'dart:isolate';
import 'dart:typed_data';
import 'dart-ext:angel_wings';
int bindWingsIPv4ServerSocket(
String address,
int port,
bool shared,
int backlog,
bool v6Only,
SendPort sendPort) native 'Dart_WingsSocket_bindIPv4';
int bindWingsIPv6ServerSocket(
String address,
int port,
bool shared,
int backlog,
bool v6Only,
SendPort sendPort) native 'Dart_WingsSocket_bindIPv6';
String getWingsServerSocketAddress(int pointer)
native 'Dart_WingsSocket_getAddress';
int getWingsServerSocketPort(int pointer) native 'Dart_WingsSocket_getPort';
void writeToNativeSocket(int fd, Uint8List data)
native 'Dart_WingsSocket_write';
void closeNativeSocketDescriptor(int fd)
native 'Dart_WingsSocket_closeDescriptor';
SendPort wingsSocketListen(int pointer) native 'Dart_WingsSocket_listen';
void closeWingsSocket(int pointer, SendPort sendPort)
native 'Dart_WingsSocket_close';
SendPort wingsParseHttp() native 'Dart_WingsSocket_parseHttp';
class WingsClientSocket {
final int fileDescriptor;
final InternetAddress remoteAddress;
WingsClientSocket(this.fileDescriptor, this.remoteAddress);
}
class WingsSocket extends Stream<WingsClientSocket> {
final StreamController<WingsClientSocket> _ctrl = StreamController();
SendPort _acceptor;
InternetAddress _address;
final int _pointer;
final RawReceivePort _recv;
bool _open = true;
int _port;
WingsSocket._(this._pointer, this._recv) {
_acceptor = wingsSocketListen(_pointer);
_recv.handler = (h) {
if (!_ctrl.isClosed) {
_ctrl.add(
WingsClientSocket(h[0] as int, InternetAddress(h[1] as String)));
_acceptor.send([_recv.sendPort, _pointer]);
}
};
_acceptor.send([_recv.sendPort, _pointer]);
}
static Future<WingsSocket> bind(address, int port,
{bool shared = false, int backlog = 0, bool v6Only = false}) async {
var recv = RawReceivePort();
int ptr;
InternetAddress addr;
if (address is InternetAddress) {
addr = address;
} else if (address is String) {
var addrs = await InternetAddress.lookup(address);
if (addrs.isNotEmpty) {
addr = addrs[0];
} else {
throw StateError('Internet address lookup failed: $address');
}
} else {
throw ArgumentError.value(
address, 'address', 'must be an InternetAddress or String');
}
try {
if (addr.type == InternetAddressType.IPv6) {
ptr = bindWingsIPv6ServerSocket(
addr.address, port, shared, backlog, v6Only, recv.sendPort);
} else {
ptr = bindWingsIPv4ServerSocket(
addr.address, port, shared, backlog, v6Only, recv.sendPort);
}
return WingsSocket._(ptr, recv); //.._address = addr;
} catch (e) {
recv.close();
rethrow;
}
}
InternetAddress get address =>
_address ??= InternetAddress(getWingsServerSocketAddress(_pointer));
int get port => _port ??= getWingsServerSocketPort(_pointer);
@override
StreamSubscription<WingsClientSocket> listen(
void Function(WingsClientSocket event) onData,
{Function onError,
void Function() onDone,
bool cancelOnError}) {
return _ctrl.stream
.listen(onData, onError: onError, cancelOnError: cancelOnError);
}
Future<void> close(){
if (_open) {
_open = false;
closeWingsSocket(_pointer, _recv.sendPort);
_recv.close();
_ctrl.close();
}
return Future.value();
}
}