Compare commits
36 Commits
Author | SHA1 | Date | |
---|---|---|---|
70632d975d | |||
d210a39249 | |||
e86ff63f91 | |||
3454a1965a | |||
c41fa2e98f | |||
a012ca4003 | |||
b74e5aa4ee | |||
44ac15ab78 | |||
b1b652b943 | |||
530a482402 | |||
f9023cf0ab | |||
0e91ea6786 | |||
63c6226a2a | |||
48a49c2a2d | |||
43c28b749d | |||
241767df13 | |||
b2baba9237 | |||
3e36ec0984 | |||
5be89e4858 | |||
a48d9cb739 | |||
a7206cbd02 | |||
1450a6adfe | |||
5fa9bd7b49 | |||
c7eb233fc7 | |||
20c8b659d1 | |||
4ea9c2b740 | |||
48205b2fc9 | |||
f5fe7bec4a | |||
c567b88d5d | |||
fe884541fc | |||
8973bdb2af | |||
4c4e65b373 | |||
7bed7f039f | |||
8ddea0aa46 | |||
cb6cc65113 | |||
4de42ca227 |
1
.gitignore
vendored
1
.gitignore
vendored
@ -5,6 +5,7 @@
|
|||||||
.dub
|
.dub
|
||||||
__test__*__
|
__test__*__
|
||||||
__test__*__.core
|
__test__*__.core
|
||||||
|
/tanya-test-library
|
||||||
|
|
||||||
/docs/
|
/docs/
|
||||||
/docs.json
|
/docs.json
|
||||||
|
@ -2,12 +2,15 @@ sudo: false
|
|||||||
|
|
||||||
os:
|
os:
|
||||||
- linux
|
- linux
|
||||||
|
- osx
|
||||||
|
|
||||||
language: d
|
language: d
|
||||||
|
|
||||||
d:
|
d:
|
||||||
|
- dmd-2.073.0
|
||||||
- dmd-2.072.2
|
- dmd-2.072.2
|
||||||
- dmd-2.071.2
|
- dmd-2.071.2
|
||||||
|
- dmd-2.070.2
|
||||||
|
|
||||||
env:
|
env:
|
||||||
matrix:
|
matrix:
|
||||||
|
33
README.md
33
README.md
@ -13,48 +13,45 @@ Garbage Collector heap. Everything in the library is usable in @nogc code.
|
|||||||
Tanya extends Phobos functionality and provides alternative implementations for
|
Tanya extends Phobos functionality and provides alternative implementations for
|
||||||
data structures and utilities that depend on the Garbage Collector in Phobos.
|
data structures and utilities that depend on the Garbage Collector in Phobos.
|
||||||
|
|
||||||
|
* [Bug tracker](https://issues.caraus.io/projects/tanya)
|
||||||
|
* [Documentation](https://docs.caraus.io/tanya)
|
||||||
|
|
||||||
## Overview
|
## Overview
|
||||||
|
|
||||||
Tanya consists of the following packages:
|
Tanya consists of the following packages:
|
||||||
|
|
||||||
* `async`: Event loop (epoll, kqueue and IOCP).
|
* `async`: Event loop (epoll, kqueue and IOCP).
|
||||||
* `container`: Queue, Vector, Singly linked list, buffers.
|
* `container`: Queue, Vector, Singly linked list, buffers.
|
||||||
* `crypto`: Work in progress TLS implementation.
|
* `math`: Arbitrary precision integer and a set of functions.
|
||||||
* `math`: Multiple precision integer and a set of functions.
|
|
||||||
* `memory`: Tools for manual memory management (allocator, reference counting,
|
* `memory`: Tools for manual memory management (allocator, reference counting,
|
||||||
helper functions).
|
helper functions).
|
||||||
* `network`: URL-Parsing, sockets.
|
* `network`: URL-Parsing, sockets.
|
||||||
|
|
||||||
### Supported compilers
|
### Supported compilers
|
||||||
|
|
||||||
|
* dmd 2.073.0
|
||||||
* dmd 2.072.2
|
* dmd 2.072.2
|
||||||
* dmd 2.071.2
|
* dmd 2.071.2
|
||||||
|
* dmd 2.070.2
|
||||||
|
|
||||||
### Current status
|
### Current status
|
||||||
|
|
||||||
The library is currently under development, but some parts of it can already be
|
The library is currently under development, but the API is becoming gradually
|
||||||
used.
|
stable.
|
||||||
|
|
||||||
`network` and `async` exist for quite some time and could be better tested than
|
`container`s are being extended to support ranges. Also following modules are
|
||||||
other components.
|
coming soon:
|
||||||
|
* UTF-8 string.
|
||||||
|
* Hash table.
|
||||||
|
|
||||||
Containers were newly reworked and the API won't change significantly, but will
|
`math` package contains an arbitrary precision integer implementation that
|
||||||
be only extended. The same is true for the `memory` package.
|
needs more test cases, better performance and some additional features
|
||||||
|
(constructing from a string and an ubyte array, and converting it back).
|
||||||
`math` package contains an arbitrary precision integer implementation that has
|
|
||||||
a stable API (that mostly consists of operator overloads), but still needs
|
|
||||||
testing and work on its performance.
|
|
||||||
|
|
||||||
I'm currently mostly working on `crypto` that is not a complete cryptographic
|
|
||||||
suite, but contains (will contain) algorithm implementations required by TLS.
|
|
||||||
|
|
||||||
### Further characteristics
|
### Further characteristics
|
||||||
|
|
||||||
* Tanya is a native D library.
|
* Tanya is a native D library.
|
||||||
|
|
||||||
* Documentation and usage examples can be found in the source code.
|
|
||||||
Online documentation will be published soon.
|
|
||||||
|
|
||||||
* Tanya is cross-platform. The development happens on a 64-bit Linux, but it
|
* Tanya is cross-platform. The development happens on a 64-bit Linux, but it
|
||||||
is being tested on Windows and FreeBSD as well.
|
is being tested on Windows and FreeBSD as well.
|
||||||
|
|
||||||
|
2
dub.json
2
dub.json
@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "tanya",
|
"name": "tanya",
|
||||||
"description": "D library with event loop",
|
"description": "General purpose, @nogc library",
|
||||||
"license": "MPL-2.0",
|
"license": "MPL-2.0",
|
||||||
"copyright": "(c) Eugene Wissner <info@caraus.de>",
|
"copyright": "(c) Eugene Wissner <info@caraus.de>",
|
||||||
"authors": [
|
"authors": [
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async.event.epoll;
|
module tanya.async.event.epoll;
|
||||||
|
|
||||||
@ -18,6 +18,7 @@ import tanya.async.event.selector;
|
|||||||
import tanya.async.loop;
|
import tanya.async.loop;
|
||||||
import tanya.async.transport;
|
import tanya.async.transport;
|
||||||
import tanya.async.watcher;
|
import tanya.async.watcher;
|
||||||
|
import tanya.container.vector;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
import tanya.memory.mmappool;
|
import tanya.memory.mmappool;
|
||||||
import tanya.network.socket;
|
import tanya.network.socket;
|
||||||
@ -33,10 +34,10 @@ extern (C) nothrow @nogc
|
|||||||
int epoll_wait (int epfd, epoll_event *events, int maxevents, int timeout);
|
int epoll_wait (int epfd, epoll_event *events, int maxevents, int timeout);
|
||||||
}
|
}
|
||||||
|
|
||||||
class EpollLoop : SelectorLoop
|
final class EpollLoop : SelectorLoop
|
||||||
{
|
{
|
||||||
protected int fd;
|
protected int fd;
|
||||||
private epoll_event[] events;
|
private Vector!epoll_event events;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Initializes the loop.
|
* Initializes the loop.
|
||||||
@ -45,18 +46,17 @@ class EpollLoop : SelectorLoop
|
|||||||
{
|
{
|
||||||
if ((fd = epoll_create1(EPOLL_CLOEXEC)) < 0)
|
if ((fd = epoll_create1(EPOLL_CLOEXEC)) < 0)
|
||||||
{
|
{
|
||||||
throw MmapPool.instance.make!BadLoopException("epoll initialization failed");
|
throw defaultAllocator.make!BadLoopException("epoll initialization failed");
|
||||||
}
|
}
|
||||||
super();
|
super();
|
||||||
MmapPool.instance.resizeArray(events, maxEvents);
|
events = Vector!epoll_event(maxEvents, MmapPool.instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Free loop internals.
|
* Frees loop internals.
|
||||||
*/
|
*/
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(events);
|
|
||||||
close(fd);
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -70,14 +70,9 @@ class EpollLoop : SelectorLoop
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
protected override bool reify(ConnectionWatcher watcher,
|
protected override bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc
|
EventMask events) @nogc
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(watcher !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
{
|
||||||
int op = EPOLL_CTL_DEL;
|
int op = EPOLL_CTL_DEL;
|
||||||
epoll_event ev;
|
epoll_event ev;
|
||||||
@ -110,7 +105,7 @@ class EpollLoop : SelectorLoop
|
|||||||
{
|
{
|
||||||
// Don't block
|
// Don't block
|
||||||
immutable timeout = cast(immutable int) blockTime.total!"msecs";
|
immutable timeout = cast(immutable int) blockTime.total!"msecs";
|
||||||
auto eventCount = epoll_wait(fd, events.ptr, maxEvents, timeout);
|
auto eventCount = epoll_wait(fd, events.get().ptr, maxEvents, timeout);
|
||||||
|
|
||||||
if (eventCount < 0)
|
if (eventCount < 0)
|
||||||
{
|
{
|
||||||
@ -123,29 +118,30 @@ class EpollLoop : SelectorLoop
|
|||||||
|
|
||||||
for (auto i = 0; i < eventCount; ++i)
|
for (auto i = 0; i < eventCount; ++i)
|
||||||
{
|
{
|
||||||
auto io = cast(IOWatcher) connections[events[i].data.fd];
|
auto transport = cast(StreamTransport) connections[events[i].data.fd];
|
||||||
|
|
||||||
if (io is null)
|
if (transport is null)
|
||||||
{
|
{
|
||||||
acceptConnections(connections[events[i].data.fd]);
|
auto connection = cast(ConnectionWatcher) connections[events[i].data.fd];
|
||||||
|
assert(connection !is null);
|
||||||
|
|
||||||
|
acceptConnections(connection);
|
||||||
}
|
}
|
||||||
else if (events[i].events & EPOLLERR)
|
else if (events[i].events & EPOLLERR)
|
||||||
{
|
{
|
||||||
kill(io, null);
|
kill(transport);
|
||||||
|
continue;
|
||||||
}
|
}
|
||||||
else if (events[i].events & (EPOLLIN | EPOLLPRI | EPOLLHUP))
|
else if (events[i].events & (EPOLLIN | EPOLLPRI | EPOLLHUP))
|
||||||
{
|
{
|
||||||
auto transport = cast(SelectorStreamTransport) io.transport;
|
|
||||||
assert(transport !is null);
|
|
||||||
|
|
||||||
SocketException exception;
|
SocketException exception;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
ptrdiff_t received;
|
ptrdiff_t received;
|
||||||
do
|
do
|
||||||
{
|
{
|
||||||
received = transport.socket.receive(io.output[]);
|
received = transport.socket.receive(transport.output[]);
|
||||||
io.output += received;
|
transport.output += received;
|
||||||
}
|
}
|
||||||
while (received);
|
while (received);
|
||||||
}
|
}
|
||||||
@ -155,18 +151,16 @@ class EpollLoop : SelectorLoop
|
|||||||
}
|
}
|
||||||
if (transport.socket.disconnected)
|
if (transport.socket.disconnected)
|
||||||
{
|
{
|
||||||
kill(io, exception);
|
kill(transport, exception);
|
||||||
|
continue;
|
||||||
}
|
}
|
||||||
else if (io.output.length)
|
else if (transport.output.length)
|
||||||
{
|
{
|
||||||
swapPendings.enqueue(io);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else if (events[i].events & EPOLLOUT)
|
if (events[i].events & EPOLLOUT)
|
||||||
{
|
{
|
||||||
auto transport = cast(SelectorStreamTransport) io.transport;
|
|
||||||
assert(transport !is null);
|
|
||||||
|
|
||||||
transport.writeReady = true;
|
transport.writeReady = true;
|
||||||
if (transport.input.length)
|
if (transport.input.length)
|
||||||
{
|
{
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async.event.iocp;
|
module tanya.async.event.iocp;
|
||||||
|
|
||||||
@ -26,32 +26,68 @@ import core.sys.windows.winbase;
|
|||||||
import core.sys.windows.windef;
|
import core.sys.windows.windef;
|
||||||
import core.sys.windows.winsock2;
|
import core.sys.windows.winsock2;
|
||||||
|
|
||||||
class IOCPStreamTransport : StreamTransport
|
/**
|
||||||
|
* Transport for stream sockets.
|
||||||
|
*/
|
||||||
|
final class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
|
||||||
{
|
{
|
||||||
private OverlappedConnectedSocket socket_;
|
private SocketException exception;
|
||||||
|
|
||||||
|
private ReadBuffer!ubyte output;
|
||||||
|
|
||||||
private WriteBuffer!ubyte input;
|
private WriteBuffer!ubyte input;
|
||||||
|
|
||||||
|
private Protocol protocol_;
|
||||||
|
|
||||||
|
private bool closing;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Creates new completion port transport.
|
* Creates new completion port transport.
|
||||||
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* socket = Socket.
|
* socket = Socket.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE socket !is null)
|
||||||
*/
|
*/
|
||||||
this(OverlappedConnectedSocket socket) @nogc
|
this(OverlappedConnectedSocket socket) @nogc
|
||||||
in
|
{
|
||||||
|
super(socket);
|
||||||
|
output = ReadBuffer!ubyte(8192, 1024, MmapPool.instance);
|
||||||
|
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
||||||
|
active = true;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Socket.
|
||||||
|
*
|
||||||
|
* Postcondition: $(D_INLINECODE socket !is null)
|
||||||
|
*/
|
||||||
|
override @property OverlappedConnectedSocket socket() pure nothrow @safe @nogc
|
||||||
|
out (socket)
|
||||||
{
|
{
|
||||||
assert(socket !is null);
|
assert(socket !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
socket_ = socket;
|
return cast(OverlappedConnectedSocket) socket_;
|
||||||
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@property inout(OverlappedConnectedSocket) socket()
|
/**
|
||||||
inout pure nothrow @safe @nogc
|
* Returns $(D_PARAM true) if the transport is closing or closed.
|
||||||
|
*/
|
||||||
|
bool isClosing() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return socket_;
|
return closing;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Close the transport.
|
||||||
|
*
|
||||||
|
* Buffered data will be flushed. No more data will be received.
|
||||||
|
*/
|
||||||
|
void close() pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
closing = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -62,8 +98,48 @@ class IOCPStreamTransport : StreamTransport
|
|||||||
*/
|
*/
|
||||||
void write(ubyte[] data) @nogc
|
void write(ubyte[] data) @nogc
|
||||||
{
|
{
|
||||||
immutable empty = input.length == 0;
|
|
||||||
input ~= data;
|
input ~= data;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Application protocol.
|
||||||
|
*/
|
||||||
|
@property Protocol protocol() pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return protocol_;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Switches the protocol.
|
||||||
|
*
|
||||||
|
* The protocol is deallocated by the event loop, it should currently be
|
||||||
|
* allocated with $(D_PSYMBOL MmapPool).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* protocol = Application protocol.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE protocol !is null)
|
||||||
|
*/
|
||||||
|
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(protocol !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
protocol_ = protocol;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Invokes the watcher callback.
|
||||||
|
*/
|
||||||
|
override void invoke() @nogc
|
||||||
|
{
|
||||||
|
if (output.length)
|
||||||
|
{
|
||||||
|
immutable empty = input.length == 0;
|
||||||
|
protocol.received(output[0 .. $]);
|
||||||
|
output.clear();
|
||||||
if (empty)
|
if (empty)
|
||||||
{
|
{
|
||||||
SocketState overlapped;
|
SocketState overlapped;
|
||||||
@ -79,9 +155,17 @@ class IOCPStreamTransport : StreamTransport
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
protocol.disconnected(exception);
|
||||||
|
MmapPool.instance.dispose(protocol_);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
active = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
class IOCPLoop : Loop
|
final class IOCPLoop : Loop
|
||||||
{
|
{
|
||||||
protected HANDLE completionPort;
|
protected HANDLE completionPort;
|
||||||
|
|
||||||
@ -112,7 +196,7 @@ class IOCPLoop : Loop
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
override protected bool reify(ConnectionWatcher watcher,
|
override protected bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc
|
EventMask events) @nogc
|
||||||
{
|
{
|
||||||
@ -145,10 +229,7 @@ class IOCPLoop : Loop
|
|||||||
if (!(oldEvents & Event.read) && (events & Event.read)
|
if (!(oldEvents & Event.read) && (events & Event.read)
|
||||||
|| !(oldEvents & Event.write) && (events & Event.write))
|
|| !(oldEvents & Event.write) && (events & Event.write))
|
||||||
{
|
{
|
||||||
auto io = cast(IOWatcher) watcher;
|
auto transport = cast(StreamTransport) watcher;
|
||||||
assert(io !is null);
|
|
||||||
|
|
||||||
auto transport = cast(IOCPStreamTransport) io.transport;
|
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
|
|
||||||
if (CreateIoCompletionPort(cast(HANDLE) transport.socket.handle,
|
if (CreateIoCompletionPort(cast(HANDLE) transport.socket.handle,
|
||||||
@ -165,7 +246,7 @@ class IOCPLoop : Loop
|
|||||||
try
|
try
|
||||||
{
|
{
|
||||||
overlapped = MmapPool.instance.make!SocketState;
|
overlapped = MmapPool.instance.make!SocketState;
|
||||||
transport.socket.beginReceive(io.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
@ -178,6 +259,20 @@ class IOCPLoop : Loop
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private void kill(StreamTransport transport,
|
||||||
|
SocketException exception = null) @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(transport !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
transport.socket.shutdown();
|
||||||
|
defaultAllocator.dispose(transport.socket);
|
||||||
|
transport.exception = exception;
|
||||||
|
pendings.enqueue(transport);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Does the actual polling.
|
* Does the actual polling.
|
||||||
*/
|
*/
|
||||||
@ -215,29 +310,26 @@ class IOCPLoop : Loop
|
|||||||
assert(listener !is null);
|
assert(listener !is null);
|
||||||
|
|
||||||
auto socket = listener.endAccept(overlapped);
|
auto socket = listener.endAccept(overlapped);
|
||||||
auto transport = MmapPool.instance.make!IOCPStreamTransport(socket);
|
auto transport = MmapPool.instance.make!StreamTransport(socket);
|
||||||
auto io = MmapPool.instance.make!IOWatcher(transport, connection.protocol);
|
|
||||||
|
|
||||||
connection.incoming.enqueue(io);
|
connection.incoming.enqueue(transport);
|
||||||
|
|
||||||
reify(io, EventMask(Event.none), EventMask(Event.read, Event.write));
|
reify(transport, EventMask(Event.none), EventMask(Event.read, Event.write));
|
||||||
|
|
||||||
swapPendings.enqueue(connection);
|
pendings.enqueue(connection);
|
||||||
listener.beginAccept(overlapped);
|
listener.beginAccept(overlapped);
|
||||||
break;
|
break;
|
||||||
case OverlappedSocketEvent.read:
|
case OverlappedSocketEvent.read:
|
||||||
auto io = cast(IOWatcher) (cast(void*) key);
|
auto transport = cast(StreamTransport) (cast(void*) key);
|
||||||
assert(io !is null);
|
assert(transport !is null);
|
||||||
if (!io.active)
|
|
||||||
|
if (!transport.active)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(io);
|
MmapPool.instance.dispose(transport);
|
||||||
MmapPool.instance.dispose(overlapped);
|
MmapPool.instance.dispose(overlapped);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto transport = cast(IOCPStreamTransport) io.transport;
|
|
||||||
assert(transport !is null);
|
|
||||||
|
|
||||||
int received;
|
int received;
|
||||||
SocketException exception;
|
SocketException exception;
|
||||||
try
|
try
|
||||||
@ -250,38 +342,39 @@ class IOCPLoop : Loop
|
|||||||
}
|
}
|
||||||
if (transport.socket.disconnected)
|
if (transport.socket.disconnected)
|
||||||
{
|
{
|
||||||
// We want to get one last notification to destroy the watcher
|
// We want to get one last notification to destroy the watcher.
|
||||||
transport.socket.beginReceive(io.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
kill(io, exception);
|
kill(transport, exception);
|
||||||
}
|
}
|
||||||
else if (received > 0)
|
else if (received > 0)
|
||||||
{
|
{
|
||||||
immutable full = io.output.free == received;
|
immutable full = transport.output.free == received;
|
||||||
|
|
||||||
io.output += received;
|
transport.output += received;
|
||||||
// Receive was interrupted because the buffer is full. We have to continue
|
// Receive was interrupted because the buffer is full. We have to continue.
|
||||||
if (full)
|
if (full)
|
||||||
{
|
{
|
||||||
transport.socket.beginReceive(io.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
}
|
}
|
||||||
swapPendings.enqueue(io);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
case OverlappedSocketEvent.write:
|
case OverlappedSocketEvent.write:
|
||||||
auto io = cast(IOWatcher) (cast(void*) key);
|
auto transport = cast(StreamTransport) (cast(void*) key);
|
||||||
assert(io !is null);
|
|
||||||
|
|
||||||
auto transport = cast(IOCPStreamTransport) io.transport;
|
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
|
|
||||||
transport.input += transport.socket.endSend(overlapped);
|
transport.input += transport.socket.endSend(overlapped);
|
||||||
if (transport.input.length)
|
if (transport.input.length > 0)
|
||||||
{
|
{
|
||||||
transport.socket.beginSend(transport.input[], overlapped);
|
transport.socket.beginSend(transport.input[], overlapped);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
transport.socket.beginReceive(io.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
|
if (transport.isClosing())
|
||||||
|
{
|
||||||
|
kill(transport);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
|
@ -3,7 +3,7 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
@ -41,8 +41,19 @@ else version (DragonFlyBSD)
|
|||||||
|
|
||||||
version (MacBSD):
|
version (MacBSD):
|
||||||
|
|
||||||
import core.stdc.stdint; // intptr_t, uintptr_t
|
import core.stdc.errno;
|
||||||
import core.sys.posix.time; // timespec
|
import core.sys.posix.time; // timespec
|
||||||
|
import core.sys.posix.unistd;
|
||||||
|
import core.time;
|
||||||
|
import std.algorithm.comparison;
|
||||||
|
import tanya.async.event.selector;
|
||||||
|
import tanya.async.loop;
|
||||||
|
import tanya.async.transport;
|
||||||
|
import tanya.async.watcher;
|
||||||
|
import tanya.container.vector;
|
||||||
|
import tanya.memory;
|
||||||
|
import tanya.memory.mmappool;
|
||||||
|
import tanya.network.socket;
|
||||||
|
|
||||||
void EV_SET(kevent_t* kevp, typeof(kevent_t.tupleof) args) pure nothrow @nogc
|
void EV_SET(kevent_t* kevp, typeof(kevent_t.tupleof) args) pure nothrow @nogc
|
||||||
{
|
{
|
||||||
@ -102,32 +113,19 @@ extern(C) int kevent(int kq, const kevent_t *changelist, int nchanges,
|
|||||||
kevent_t *eventlist, int nevents, const timespec *timeout)
|
kevent_t *eventlist, int nevents, const timespec *timeout)
|
||||||
nothrow @nogc;
|
nothrow @nogc;
|
||||||
|
|
||||||
import tanya.async.event.selector;
|
final class KqueueLoop : SelectorLoop
|
||||||
import tanya.async.loop;
|
|
||||||
import tanya.async.transport;
|
|
||||||
import tanya.async.watcher;
|
|
||||||
import tanya.memory;
|
|
||||||
import tanya.memory.mmappool;
|
|
||||||
import tanya.network.socket;
|
|
||||||
import core.stdc.errno;
|
|
||||||
import core.sys.posix.unistd;
|
|
||||||
import core.sys.posix.sys.time;
|
|
||||||
import core.time;
|
|
||||||
import std.algorithm.comparison;
|
|
||||||
|
|
||||||
class KqueueLoop : SelectorLoop
|
|
||||||
{
|
{
|
||||||
protected int fd;
|
protected int fd;
|
||||||
private kevent_t[] events;
|
private Vector!kevent_t events;
|
||||||
private kevent_t[] changes;
|
private Vector!kevent_t changes;
|
||||||
private size_t changeCount;
|
private size_t changeCount;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Maximal event count can be got at a time
|
* Returns: Maximal event count can be got at a time
|
||||||
* (should be supported by the backend).
|
* (should be supported by the backend).
|
||||||
*/
|
*/
|
||||||
override protected @property inout(uint) maxEvents()
|
override protected @property uint maxEvents()
|
||||||
inout const pure nothrow @safe @nogc
|
const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return cast(uint) events.length;
|
return cast(uint) events.length;
|
||||||
}
|
}
|
||||||
@ -138,19 +136,18 @@ class KqueueLoop : SelectorLoop
|
|||||||
|
|
||||||
if ((fd = kqueue()) == -1)
|
if ((fd = kqueue()) == -1)
|
||||||
{
|
{
|
||||||
throw MmapPool.instance.make!BadLoopException("epoll initialization failed");
|
throw make!BadLoopException(defaultAllocator,
|
||||||
|
"kqueue initialization failed");
|
||||||
}
|
}
|
||||||
MmapPool.instance.resizeArray(events, 64);
|
events = Vector!kevent_t(64, MmapPool.instance);
|
||||||
MmapPool.instance.resizeArray(changes, 64);
|
changes = Vector!kevent_t(64, MmapPool.instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Free loop internals.
|
* Frees loop internals.
|
||||||
*/
|
*/
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(events);
|
|
||||||
MmapPool.instance.dispose(changes);
|
|
||||||
close(fd);
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -158,7 +155,7 @@ class KqueueLoop : SelectorLoop
|
|||||||
{
|
{
|
||||||
if (changes.length <= changeCount)
|
if (changes.length <= changeCount)
|
||||||
{
|
{
|
||||||
MmapPool.instance.resizeArray(changes, changeCount + maxEvents);
|
changes.length = changeCount + maxEvents;
|
||||||
}
|
}
|
||||||
EV_SET(&changes[changeCount],
|
EV_SET(&changes[changeCount],
|
||||||
cast(ulong) socket,
|
cast(ulong) socket,
|
||||||
@ -180,7 +177,7 @@ class KqueueLoop : SelectorLoop
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
override protected bool reify(ConnectionWatcher watcher,
|
override protected bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc
|
EventMask events) @nogc
|
||||||
{
|
{
|
||||||
@ -216,10 +213,15 @@ class KqueueLoop : SelectorLoop
|
|||||||
|
|
||||||
if (changeCount > maxEvents)
|
if (changeCount > maxEvents)
|
||||||
{
|
{
|
||||||
MmapPool.instance.resizeArray(events, changes.length);
|
events.length = changes.length;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto eventCount = kevent(fd, changes.ptr, cast(int) changeCount, events.ptr, maxEvents, &ts);
|
auto eventCount = kevent(fd,
|
||||||
|
changes.get().ptr,
|
||||||
|
cast(int) changeCount,
|
||||||
|
events.get().ptr,
|
||||||
|
maxEvents,
|
||||||
|
&ts);
|
||||||
changeCount = 0;
|
changeCount = 0;
|
||||||
|
|
||||||
if (eventCount < 0)
|
if (eventCount < 0)
|
||||||
@ -235,29 +237,29 @@ class KqueueLoop : SelectorLoop
|
|||||||
{
|
{
|
||||||
assert(connections.length > events[i].ident);
|
assert(connections.length > events[i].ident);
|
||||||
|
|
||||||
IOWatcher io = cast(IOWatcher) connections[events[i].ident];
|
auto transport = cast(StreamTransport) connections[events[i].ident];
|
||||||
// If it is a ConnectionWatcher. Accept connections.
|
// If it is a ConnectionWatcher. Accept connections.
|
||||||
if (io is null)
|
if (transport is null)
|
||||||
{
|
{
|
||||||
acceptConnections(connections[events[i].ident]);
|
auto connection = cast(ConnectionWatcher) connections[events[i].ident];
|
||||||
|
assert(connection !is null);
|
||||||
|
|
||||||
|
acceptConnections(connection);
|
||||||
}
|
}
|
||||||
else if (events[i].flags & EV_ERROR)
|
else if (events[i].flags & EV_ERROR)
|
||||||
{
|
{
|
||||||
kill(io, null);
|
kill(transport);
|
||||||
}
|
}
|
||||||
else if (events[i].filter == EVFILT_READ)
|
else if (events[i].filter == EVFILT_READ)
|
||||||
{
|
{
|
||||||
auto transport = cast(SelectorStreamTransport) io.transport;
|
|
||||||
assert(transport !is null);
|
|
||||||
|
|
||||||
SocketException exception;
|
SocketException exception;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
ptrdiff_t received;
|
ptrdiff_t received;
|
||||||
do
|
do
|
||||||
{
|
{
|
||||||
received = transport.socket.receive(io.output[]);
|
received = transport.socket.receive(transport.output[]);
|
||||||
io.output += received;
|
transport.output += received;
|
||||||
}
|
}
|
||||||
while (received);
|
while (received);
|
||||||
}
|
}
|
||||||
@ -267,18 +269,15 @@ class KqueueLoop : SelectorLoop
|
|||||||
}
|
}
|
||||||
if (transport.socket.disconnected)
|
if (transport.socket.disconnected)
|
||||||
{
|
{
|
||||||
kill(io, exception);
|
kill(transport, exception);
|
||||||
}
|
}
|
||||||
else if (io.output.length)
|
else if (transport.output.length)
|
||||||
{
|
{
|
||||||
swapPendings.enqueue(io);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else if (events[i].filter == EVFILT_WRITE)
|
else if (events[i].filter == EVFILT_WRITE)
|
||||||
{
|
{
|
||||||
auto transport = cast(SelectorStreamTransport) io.transport;
|
|
||||||
assert(transport !is null);
|
|
||||||
|
|
||||||
transport.writeReady = true;
|
transport.writeReady = true;
|
||||||
if (transport.input.length)
|
if (transport.input.length)
|
||||||
{
|
{
|
||||||
@ -307,9 +306,9 @@ class KqueueLoop : SelectorLoop
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation could be successfully
|
* Returns: $(D_KEYWORD true) if the operation could be successfully
|
||||||
* completed or scheduled, $(D_KEYWORD false) otherwise (the
|
* completed or scheduled, $(D_KEYWORD false) otherwise (the
|
||||||
* transport is be destroyed then).
|
* transport will be destroyed then).
|
||||||
*/
|
*/
|
||||||
protected override bool feed(SelectorStreamTransport transport,
|
protected override bool feed(StreamTransport transport,
|
||||||
SocketException exception = null) @nogc
|
SocketException exception = null) @nogc
|
||||||
{
|
{
|
||||||
if (!super.feed(transport, exception))
|
if (!super.feed(transport, exception))
|
||||||
|
@ -3,36 +3,41 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async.event.selector;
|
module tanya.async.event.selector;
|
||||||
|
|
||||||
version (Posix):
|
version (Posix):
|
||||||
|
|
||||||
import tanya.async.loop;
|
import tanya.async.loop;
|
||||||
|
import tanya.async.protocol;
|
||||||
import tanya.async.transport;
|
import tanya.async.transport;
|
||||||
import tanya.async.watcher;
|
import tanya.async.watcher;
|
||||||
import tanya.container.buffer;
|
import tanya.container.buffer;
|
||||||
|
import tanya.container.vector;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
import tanya.memory.mmappool;
|
import tanya.memory.mmappool;
|
||||||
import tanya.network.socket;
|
import tanya.network.socket;
|
||||||
import core.sys.posix.netinet.in_;
|
|
||||||
import core.stdc.errno;
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Transport for stream sockets.
|
* Transport for stream sockets.
|
||||||
*/
|
*/
|
||||||
class SelectorStreamTransport : StreamTransport
|
package class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
|
||||||
{
|
{
|
||||||
private ConnectedSocket socket_;
|
private SelectorLoop loop;
|
||||||
|
|
||||||
|
private SocketException exception;
|
||||||
|
|
||||||
|
package ReadBuffer!ubyte output;
|
||||||
|
|
||||||
/// Input buffer.
|
|
||||||
package WriteBuffer!ubyte input;
|
package WriteBuffer!ubyte input;
|
||||||
|
|
||||||
private SelectorLoop loop;
|
private Protocol protocol_;
|
||||||
|
|
||||||
|
private bool closing;
|
||||||
|
|
||||||
/// Received notification that the underlying socket is write-ready.
|
/// Received notification that the underlying socket is write-ready.
|
||||||
package bool writeReady;
|
package bool writeReady;
|
||||||
@ -41,20 +46,117 @@ class SelectorStreamTransport : StreamTransport
|
|||||||
* Params:
|
* Params:
|
||||||
* loop = Event loop.
|
* loop = Event loop.
|
||||||
* socket = Socket.
|
* socket = Socket.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE loop !is null && socket !is null)
|
||||||
*/
|
*/
|
||||||
this(SelectorLoop loop, ConnectedSocket socket) @nogc
|
this(SelectorLoop loop, ConnectedSocket socket) @nogc
|
||||||
|
in
|
||||||
{
|
{
|
||||||
socket_ = socket;
|
assert(loop !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
super(socket);
|
||||||
this.loop = loop;
|
this.loop = loop;
|
||||||
|
output = ReadBuffer!ubyte(8192, 1024, MmapPool.instance);
|
||||||
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
||||||
|
active = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Transport socket.
|
* Returns: Socket.
|
||||||
|
*
|
||||||
|
* Postcondition: $(D_INLINECODE socket !is null)
|
||||||
*/
|
*/
|
||||||
inout(ConnectedSocket) socket() inout pure nothrow @safe @nogc
|
override @property ConnectedSocket socket() pure nothrow @safe @nogc
|
||||||
|
out (socket)
|
||||||
{
|
{
|
||||||
return socket_;
|
assert(socket !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return cast(ConnectedSocket) socket_;
|
||||||
|
}
|
||||||
|
|
||||||
|
private @property void socket(ConnectedSocket socket) pure nothrow @safe @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(socket !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
socket_ = socket;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Application protocol.
|
||||||
|
*/
|
||||||
|
@property Protocol protocol() pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return protocol_;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Switches the protocol.
|
||||||
|
*
|
||||||
|
* The protocol is deallocated by the event loop, it should currently be
|
||||||
|
* allocated with $(D_PSYMBOL MmapPool).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* protocol = Application protocol.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE protocol !is null)
|
||||||
|
*/
|
||||||
|
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(protocol !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
protocol_ = protocol;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns $(D_PARAM true) if the transport is closing or closed.
|
||||||
|
*/
|
||||||
|
bool isClosing() const pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return closing;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Close the transport.
|
||||||
|
*
|
||||||
|
* Buffered data will be flushed. No more data will be received.
|
||||||
|
*/
|
||||||
|
void close() @nogc
|
||||||
|
{
|
||||||
|
closing = true;
|
||||||
|
loop.reify(this, EventMask(Event.read, Event.write), EventMask(Event.write));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Invokes the watcher callback.
|
||||||
|
*/
|
||||||
|
override void invoke() @nogc
|
||||||
|
{
|
||||||
|
if (output.length)
|
||||||
|
{
|
||||||
|
protocol.received(output[0 .. $]);
|
||||||
|
output.clear();
|
||||||
|
if (isClosing() && input.length == 0)
|
||||||
|
{
|
||||||
|
loop.kill(this);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
protocol.disconnected(exception);
|
||||||
|
MmapPool.instance.dispose(protocol_);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
active = false;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -103,28 +205,60 @@ class SelectorStreamTransport : StreamTransport
|
|||||||
abstract class SelectorLoop : Loop
|
abstract class SelectorLoop : Loop
|
||||||
{
|
{
|
||||||
/// Pending connections.
|
/// Pending connections.
|
||||||
protected ConnectionWatcher[] connections;
|
protected Vector!SocketWatcher connections;
|
||||||
|
|
||||||
this() @nogc
|
this() @nogc
|
||||||
{
|
{
|
||||||
super();
|
super();
|
||||||
MmapPool.instance.resizeArray(connections, maxEvents);
|
connections = Vector!SocketWatcher(maxEvents, MmapPool.instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
foreach (ref connection; connections)
|
foreach (ref connection; connections)
|
||||||
{
|
{
|
||||||
// We want to free only IOWatchers. ConnectionWatcher are created by the
|
// We want to free only the transports. ConnectionWatcher are created by the
|
||||||
// user and should be freed by himself.
|
// user and should be freed by himself.
|
||||||
auto io = cast(IOWatcher) connection;
|
if (cast(StreamTransport) connection !is null)
|
||||||
if (io !is null)
|
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(io);
|
MmapPool.instance.dispose(connection);
|
||||||
connection = null;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
MmapPool.instance.dispose(connections);
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Should be called if the backend configuration changes.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* watcher = Watcher.
|
||||||
|
* oldEvents = The events were already set.
|
||||||
|
* events = The events should be set.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
|
*/
|
||||||
|
override abstract protected bool reify(SocketWatcher watcher,
|
||||||
|
EventMask oldEvents,
|
||||||
|
EventMask events) @nogc;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Kills the watcher and closes the connection.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* transport = Transport.
|
||||||
|
* exception = Occurred exception.
|
||||||
|
*/
|
||||||
|
protected void kill(StreamTransport transport,
|
||||||
|
SocketException exception = null) @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(transport !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
transport.socket.shutdown();
|
||||||
|
defaultAllocator.dispose(transport.socket);
|
||||||
|
transport.exception = exception;
|
||||||
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -139,8 +273,13 @@ abstract class SelectorLoop : Loop
|
|||||||
* completed or scheduled, $(D_KEYWORD false) otherwise (the
|
* completed or scheduled, $(D_KEYWORD false) otherwise (the
|
||||||
* transport will be destroyed then).
|
* transport will be destroyed then).
|
||||||
*/
|
*/
|
||||||
protected bool feed(SelectorStreamTransport transport,
|
protected bool feed(StreamTransport transport,
|
||||||
SocketException exception = null) @nogc
|
SocketException exception = null) @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(transport !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
{
|
{
|
||||||
while (transport.input.length && transport.writeReady)
|
while (transport.input.length && transport.writeReady)
|
||||||
{
|
{
|
||||||
@ -164,12 +303,13 @@ abstract class SelectorLoop : Loop
|
|||||||
}
|
}
|
||||||
if (exception !is null)
|
if (exception !is null)
|
||||||
{
|
{
|
||||||
auto watcher = cast(IOWatcher) connections[transport.socket.handle];
|
kill(transport, exception);
|
||||||
assert(watcher !is null);
|
|
||||||
|
|
||||||
kill(watcher, exception);
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
if (transport.input.length == 0 && transport.isClosing())
|
||||||
|
{
|
||||||
|
kill(transport);
|
||||||
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -188,7 +328,7 @@ abstract class SelectorLoop : Loop
|
|||||||
|
|
||||||
if (connections.length <= watcher.socket)
|
if (connections.length <= watcher.socket)
|
||||||
{
|
{
|
||||||
MmapPool.instance.resizeArray(connections, watcher.socket.handle + maxEvents / 2);
|
connections.length = watcher.socket.handle + maxEvents / 2;
|
||||||
}
|
}
|
||||||
connections[watcher.socket.handle] = watcher;
|
connections[watcher.socket.handle] = watcher;
|
||||||
|
|
||||||
@ -225,35 +365,33 @@ abstract class SelectorLoop : Loop
|
|||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
IOWatcher io;
|
StreamTransport transport;
|
||||||
auto transport = MmapPool.instance.make!SelectorStreamTransport(this, client);
|
|
||||||
|
|
||||||
if (connections.length > client.handle)
|
if (connections.length > client.handle)
|
||||||
{
|
{
|
||||||
io = cast(IOWatcher) connections[client.handle];
|
transport = cast(StreamTransport) connections[client.handle];
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
MmapPool.instance.resizeArray(connections, client.handle + maxEvents / 2);
|
connections.length = client.handle + maxEvents / 2;
|
||||||
}
|
}
|
||||||
if (io is null)
|
if (transport is null)
|
||||||
{
|
{
|
||||||
io = MmapPool.instance.make!IOWatcher(transport,
|
transport = MmapPool.instance.make!StreamTransport(this, client);
|
||||||
connection.protocol);
|
connections[client.handle] = transport;
|
||||||
connections[client.handle] = io;
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
io(transport, connection.protocol);
|
transport.socket = client;
|
||||||
}
|
}
|
||||||
|
|
||||||
reify(io, EventMask(Event.none), EventMask(Event.read, Event.write));
|
reify(transport, EventMask(Event.none), EventMask(Event.read, Event.write));
|
||||||
connection.incoming.enqueue(io);
|
connection.incoming.enqueue(transport);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!connection.incoming.empty)
|
if (!connection.incoming.empty)
|
||||||
{
|
{
|
||||||
swapPendings.enqueue(connection);
|
pendings.enqueue(connection);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async.iocp;
|
module tanya.async.iocp;
|
||||||
|
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*
|
*
|
||||||
* ---
|
* ---
|
||||||
* import tanya.async;
|
* import tanya.async;
|
||||||
@ -17,7 +17,7 @@
|
|||||||
* {
|
* {
|
||||||
* private DuplexTransport transport;
|
* private DuplexTransport transport;
|
||||||
*
|
*
|
||||||
* void received(ubyte[] data) @nogc
|
* void received(in ubyte[] data) @nogc
|
||||||
* {
|
* {
|
||||||
* transport.write(data);
|
* transport.write(data);
|
||||||
* }
|
* }
|
||||||
@ -27,7 +27,7 @@
|
|||||||
* this.transport = transport;
|
* this.transport = transport;
|
||||||
* }
|
* }
|
||||||
*
|
*
|
||||||
* void disconnected(SocketException e = null) @nogc
|
* void disconnected(SocketException e) @nogc
|
||||||
* {
|
* {
|
||||||
* }
|
* }
|
||||||
* }
|
* }
|
||||||
@ -68,7 +68,6 @@ import core.time;
|
|||||||
import std.algorithm.iteration;
|
import std.algorithm.iteration;
|
||||||
import std.algorithm.mutation;
|
import std.algorithm.mutation;
|
||||||
import std.typecons;
|
import std.typecons;
|
||||||
import tanya.async.protocol;
|
|
||||||
import tanya.async.transport;
|
import tanya.async.transport;
|
||||||
import tanya.async.watcher;
|
import tanya.async.watcher;
|
||||||
import tanya.container.buffer;
|
import tanya.container.buffer;
|
||||||
@ -130,18 +129,17 @@ alias EventMask = BitFlags!Event;
|
|||||||
*/
|
*/
|
||||||
abstract class Loop
|
abstract class Loop
|
||||||
{
|
{
|
||||||
/// Pending watchers.
|
private bool done;
|
||||||
protected Queue!Watcher* pendings;
|
|
||||||
|
|
||||||
/// Ditto.
|
/// Pending watchers.
|
||||||
protected Queue!Watcher* swapPendings;
|
protected Queue!Watcher pendings;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Maximal event count can be got at a time
|
* Returns: Maximal event count can be got at a time
|
||||||
* (should be supported by the backend).
|
* (should be supported by the backend).
|
||||||
*/
|
*/
|
||||||
protected @property inout(uint) maxEvents()
|
protected @property uint maxEvents()
|
||||||
inout const pure nothrow @safe @nogc
|
const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return 128U;
|
return 128U;
|
||||||
}
|
}
|
||||||
@ -151,8 +149,7 @@ abstract class Loop
|
|||||||
*/
|
*/
|
||||||
this() @nogc
|
this() @nogc
|
||||||
{
|
{
|
||||||
pendings = MmapPool.instance.make!(Queue!Watcher)(MmapPool.instance);
|
pendings = Queue!Watcher(MmapPool.instance);
|
||||||
swapPendings = MmapPool.instance.make!(Queue!Watcher)(MmapPool.instance);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -160,17 +157,10 @@ abstract class Loop
|
|||||||
*/
|
*/
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
foreach (w; *pendings)
|
foreach (w; pendings)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(w);
|
MmapPool.instance.dispose(w);
|
||||||
}
|
}
|
||||||
MmapPool.instance.dispose(pendings);
|
|
||||||
|
|
||||||
foreach (w; *swapPendings)
|
|
||||||
{
|
|
||||||
MmapPool.instance.dispose(w);
|
|
||||||
}
|
|
||||||
MmapPool.instance.dispose(swapPendings);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -178,19 +168,18 @@ abstract class Loop
|
|||||||
*/
|
*/
|
||||||
void run() @nogc
|
void run() @nogc
|
||||||
{
|
{
|
||||||
done_ = false;
|
done = false;
|
||||||
do
|
do
|
||||||
{
|
{
|
||||||
poll();
|
poll();
|
||||||
|
|
||||||
// Invoke pendings
|
// Invoke pendings
|
||||||
foreach (ref w; *swapPendings)
|
foreach (ref w; pendings)
|
||||||
{
|
{
|
||||||
w.invoke();
|
w.invoke();
|
||||||
}
|
}
|
||||||
swap(pendings, swapPendings);
|
|
||||||
}
|
}
|
||||||
while (!done_);
|
while (!done);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -198,7 +187,7 @@ abstract class Loop
|
|||||||
*/
|
*/
|
||||||
void unloop() @safe pure nothrow @nogc
|
void unloop() @safe pure nothrow @nogc
|
||||||
{
|
{
|
||||||
done_ = true;
|
done = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -214,6 +203,7 @@ abstract class Loop
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
watcher.active = true;
|
watcher.active = true;
|
||||||
|
|
||||||
reify(watcher, EventMask(Event.none), EventMask(Event.accept));
|
reify(watcher, EventMask(Event.none), EventMask(Event.accept));
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -244,7 +234,7 @@ abstract class Loop
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
abstract protected bool reify(ConnectionWatcher watcher,
|
abstract protected bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc;
|
EventMask events) @nogc;
|
||||||
|
|
||||||
@ -255,7 +245,7 @@ abstract class Loop
|
|||||||
inout @safe pure nothrow @nogc
|
inout @safe pure nothrow @nogc
|
||||||
{
|
{
|
||||||
// Don't block if we have to do.
|
// Don't block if we have to do.
|
||||||
return swapPendings.empty ? blockTime_ : Duration.zero;
|
return pendings.empty ? blockTime_ : Duration.zero;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -276,26 +266,11 @@ abstract class Loop
|
|||||||
blockTime_ = blockTime;
|
blockTime_ = blockTime;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Kills the watcher and closes the connection.
|
|
||||||
*/
|
|
||||||
protected void kill(IOWatcher watcher, SocketException exception) @nogc
|
|
||||||
{
|
|
||||||
watcher.socket.shutdown();
|
|
||||||
defaultAllocator.dispose(watcher.socket);
|
|
||||||
MmapPool.instance.dispose(watcher.transport);
|
|
||||||
watcher.exception = exception;
|
|
||||||
swapPendings.enqueue(watcher);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Does the actual polling.
|
* Does the actual polling.
|
||||||
*/
|
*/
|
||||||
abstract protected void poll() @nogc;
|
abstract protected void poll() @nogc;
|
||||||
|
|
||||||
/// Whether the event loop should be stopped.
|
|
||||||
private bool done_;
|
|
||||||
|
|
||||||
/// Maximal block time.
|
/// Maximal block time.
|
||||||
protected Duration blockTime_ = 1.dur!"minutes";
|
protected Duration blockTime_ = 1.dur!"minutes";
|
||||||
}
|
}
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async;
|
module tanya.async;
|
||||||
|
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async.protocol;
|
module tanya.async.protocol;
|
||||||
|
|
||||||
@ -22,7 +22,7 @@ interface Protocol
|
|||||||
* Params:
|
* Params:
|
||||||
* data = Read data.
|
* data = Read data.
|
||||||
*/
|
*/
|
||||||
void received(ubyte[] data) @nogc;
|
void received(in ubyte[] data) @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Called when a connection is made.
|
* Called when a connection is made.
|
||||||
@ -39,7 +39,7 @@ interface Protocol
|
|||||||
* exception = $(D_PSYMBOL Exception) if an error caused
|
* exception = $(D_PSYMBOL Exception) if an error caused
|
||||||
* the disconnect, $(D_KEYWORD null) otherwise.
|
* the disconnect, $(D_KEYWORD null) otherwise.
|
||||||
*/
|
*/
|
||||||
void disconnected(SocketException exception = null) @nogc;
|
void disconnected(SocketException exception) @nogc;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -3,13 +3,14 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async.transport;
|
module tanya.async.transport;
|
||||||
|
|
||||||
|
import tanya.async.protocol;
|
||||||
import tanya.network.socket;
|
import tanya.network.socket;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -45,6 +46,46 @@ interface WriteTransport : Transport
|
|||||||
*/
|
*/
|
||||||
interface DuplexTransport : ReadTransport, WriteTransport
|
interface DuplexTransport : ReadTransport, WriteTransport
|
||||||
{
|
{
|
||||||
|
/**
|
||||||
|
* Returns: Application protocol.
|
||||||
|
*
|
||||||
|
* Postcondition: $(D_INLINECODE protocol !is null)
|
||||||
|
*/
|
||||||
|
@property Protocol protocol() pure nothrow @safe @nogc
|
||||||
|
out (protocol)
|
||||||
|
{
|
||||||
|
assert(protocol !is null);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Switches the protocol.
|
||||||
|
*
|
||||||
|
* The protocol is deallocated by the event loop, it should currently be
|
||||||
|
* allocated with $(D_PSYMBOL MmapPool).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* protocol = Application protocol.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE protocol !is null)
|
||||||
|
*/
|
||||||
|
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(protocol !is null);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns $(D_PARAM true) if the transport is closing or closed.
|
||||||
|
*/
|
||||||
|
bool isClosing() const pure nothrow @safe @nogc;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Close the transport.
|
||||||
|
*
|
||||||
|
* Buffered data will be flushed. No more data will be received.
|
||||||
|
*/
|
||||||
|
void close() @nogc;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -52,12 +93,8 @@ interface DuplexTransport : ReadTransport, WriteTransport
|
|||||||
*/
|
*/
|
||||||
interface SocketTransport : Transport
|
interface SocketTransport : Transport
|
||||||
{
|
{
|
||||||
@property inout(Socket) socket() inout pure nothrow @safe @nogc;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Represents a connection-oriented socket transport.
|
* Returns: Socket.
|
||||||
*/
|
*/
|
||||||
package interface StreamTransport : DuplexTransport, SocketTransport
|
@property Socket socket() pure nothrow @safe @nogc;
|
||||||
{
|
|
||||||
}
|
}
|
||||||
|
@ -3,10 +3,10 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.async.watcher;
|
module tanya.async.watcher;
|
||||||
|
|
||||||
@ -21,15 +21,6 @@ import tanya.memory;
|
|||||||
import tanya.memory.mmappool;
|
import tanya.memory.mmappool;
|
||||||
import tanya.network.socket;
|
import tanya.network.socket;
|
||||||
|
|
||||||
version (Windows)
|
|
||||||
{
|
|
||||||
import core.sys.windows.basetyps;
|
|
||||||
import core.sys.windows.mswsock;
|
|
||||||
import core.sys.windows.winbase;
|
|
||||||
import core.sys.windows.windef;
|
|
||||||
import core.sys.windows.winsock2;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* A watcher is an opaque structure that you allocate and register to record
|
* A watcher is an opaque structure that you allocate and register to record
|
||||||
* your interest in some event.
|
* your interest in some event.
|
||||||
@ -45,15 +36,48 @@ abstract class Watcher
|
|||||||
void invoke() @nogc;
|
void invoke() @nogc;
|
||||||
}
|
}
|
||||||
|
|
||||||
class ConnectionWatcher : Watcher
|
/**
|
||||||
|
* Socket watcher.
|
||||||
|
*/
|
||||||
|
abstract class SocketWatcher : Watcher
|
||||||
{
|
{
|
||||||
/// Watched socket.
|
/// Watched socket.
|
||||||
private Socket socket_;
|
protected Socket socket_;
|
||||||
|
|
||||||
/// Protocol factory.
|
/**
|
||||||
protected Protocol delegate() @nogc protocolFactory;
|
* Params:
|
||||||
|
* socket = Socket.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE socket !is null)
|
||||||
|
*/
|
||||||
|
this(Socket socket) pure nothrow @safe @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(socket !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
socket_ = socket;
|
||||||
|
}
|
||||||
|
|
||||||
package Queue!IOWatcher incoming;
|
/**
|
||||||
|
* Returns: Socket.
|
||||||
|
*/
|
||||||
|
@property Socket socket() pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return socket_;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Connection watcher.
|
||||||
|
*/
|
||||||
|
class ConnectionWatcher : SocketWatcher
|
||||||
|
{
|
||||||
|
/// Incoming connection queue.
|
||||||
|
Queue!DuplexTransport incoming;
|
||||||
|
|
||||||
|
private Protocol delegate() @nogc protocolFactory;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
@ -61,20 +85,8 @@ class ConnectionWatcher : Watcher
|
|||||||
*/
|
*/
|
||||||
this(Socket socket) @nogc
|
this(Socket socket) @nogc
|
||||||
{
|
{
|
||||||
socket_ = socket;
|
super(socket);
|
||||||
}
|
incoming = Queue!DuplexTransport(MmapPool.instance);
|
||||||
|
|
||||||
/// Ditto.
|
|
||||||
protected this() pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
}
|
|
||||||
|
|
||||||
~this() @nogc
|
|
||||||
{
|
|
||||||
foreach (w; incoming)
|
|
||||||
{
|
|
||||||
MmapPool.instance.dispose(w);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -87,159 +99,19 @@ class ConnectionWatcher : Watcher
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Socket.
|
* Invokes new connection callback.
|
||||||
*/
|
*/
|
||||||
@property inout(Socket) socket() inout pure nothrow @nogc
|
override void invoke() @nogc
|
||||||
{
|
|
||||||
return socket_;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: New protocol instance.
|
|
||||||
*/
|
|
||||||
@property Protocol protocol() @nogc
|
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(protocolFactory !is null, "Protocol isn't set.");
|
assert(protocolFactory !is null, "Protocol isn't set.");
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
return protocolFactory();
|
foreach (transport; incoming)
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Invokes new connection callback.
|
|
||||||
*/
|
|
||||||
override void invoke() @nogc
|
|
||||||
{
|
{
|
||||||
foreach (io; incoming)
|
transport.protocol = protocolFactory();
|
||||||
{
|
transport.protocol.connected(transport);
|
||||||
io.protocol.connected(cast(DuplexTransport) io.transport);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Contains a pending watcher with the invoked events or a transport can be
|
|
||||||
* read from.
|
|
||||||
*/
|
|
||||||
class IOWatcher : ConnectionWatcher
|
|
||||||
{
|
|
||||||
/// If an exception was thrown the transport should be already invalid.
|
|
||||||
private union
|
|
||||||
{
|
|
||||||
StreamTransport transport_;
|
|
||||||
SocketException exception_;
|
|
||||||
}
|
|
||||||
|
|
||||||
private Protocol protocol_;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Underlying output buffer.
|
|
||||||
*/
|
|
||||||
package ReadBuffer!ubyte output;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* transport = Transport.
|
|
||||||
* protocol = New instance of the application protocol.
|
|
||||||
*/
|
|
||||||
this(StreamTransport transport, Protocol protocol) @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(transport !is null);
|
|
||||||
assert(protocol !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
super();
|
|
||||||
transport_ = transport;
|
|
||||||
protocol_ = protocol;
|
|
||||||
output = ReadBuffer!ubyte(8192, 1024, MmapPool.instance);
|
|
||||||
active = true;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Destroys the watcher.
|
|
||||||
*/
|
|
||||||
~this() @nogc
|
|
||||||
{
|
|
||||||
MmapPool.instance.dispose(protocol_);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Assigns a transport.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* transport = Transport.
|
|
||||||
* protocol = Application protocol.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD this).
|
|
||||||
*/
|
|
||||||
IOWatcher opCall(StreamTransport transport, Protocol protocol)
|
|
||||||
pure nothrow @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(transport !is null);
|
|
||||||
assert(protocol !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
transport_ = transport;
|
|
||||||
protocol_ = protocol;
|
|
||||||
active = true;
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Transport used by this watcher.
|
|
||||||
*/
|
|
||||||
@property inout(StreamTransport) transport() inout pure nothrow @nogc
|
|
||||||
{
|
|
||||||
return transport_;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Sets an exception occurred during a read/write operation.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* exception = Thrown exception.
|
|
||||||
*/
|
|
||||||
@property void exception(SocketException exception) pure nothrow @nogc
|
|
||||||
{
|
|
||||||
exception_ = exception;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Application protocol.
|
|
||||||
*/
|
|
||||||
override @property Protocol protocol() pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return protocol_;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Socket.
|
|
||||||
*/
|
|
||||||
override @property inout(Socket) socket() inout pure nothrow @nogc
|
|
||||||
{
|
|
||||||
return transport.socket;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Invokes the watcher callback.
|
|
||||||
*/
|
|
||||||
override void invoke() @nogc
|
|
||||||
{
|
|
||||||
if (output.length)
|
|
||||||
{
|
|
||||||
protocol.received(output[0..$]);
|
|
||||||
output.clear();
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
protocol.disconnected(exception_);
|
|
||||||
active = false;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -3,7 +3,7 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
@ -261,7 +261,6 @@ struct ReadBuffer(T = ubyte)
|
|||||||
allocator.reallocate(buf, (cap + blockSize) * T.sizeof);
|
allocator.reallocate(buf, (cap + blockSize) * T.sizeof);
|
||||||
buffer_ = cast(T[]) buf;
|
buffer_ = cast(T[]) buf;
|
||||||
}();
|
}();
|
||||||
buffer_[cap .. $] = T.init;
|
|
||||||
}
|
}
|
||||||
ring = length_;
|
ring = length_;
|
||||||
return buffer_[length_ .. $];
|
return buffer_[length_ .. $];
|
||||||
@ -337,33 +336,25 @@ struct WriteBuffer(T = ubyte)
|
|||||||
* size = Initial buffer size and the size by which the buffer will
|
* size = Initial buffer size and the size by which the buffer will
|
||||||
* grow.
|
* grow.
|
||||||
* allocator = Allocator.
|
* allocator = Allocator.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE size > 0 && allocator !is null)
|
||||||
*/
|
*/
|
||||||
this(in size_t size, shared Allocator allocator = defaultAllocator) @trusted
|
this(in size_t size, shared Allocator allocator = defaultAllocator) @trusted
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(size > 0);
|
assert(size > 0);
|
||||||
|
assert(allocator !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
blockSize = size;
|
blockSize = size;
|
||||||
ring = size - 1;
|
ring = size - 1;
|
||||||
this(allocator);
|
allocator_ = allocator;
|
||||||
buffer_ = cast(T[]) allocator_.allocate(size * T.sizeof);
|
buffer_ = cast(T[]) allocator_.allocate(size * T.sizeof);
|
||||||
}
|
}
|
||||||
|
|
||||||
@disable this();
|
@disable this();
|
||||||
|
|
||||||
/// Ditto.
|
|
||||||
this(shared Allocator allocator)
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(allocator !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
allocator_ = allocator;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Deallocates the internal buffer.
|
* Deallocates the internal buffer.
|
||||||
*/
|
*/
|
||||||
|
@ -5,41 +5,18 @@
|
|||||||
/*
|
/*
|
||||||
* Internal package used by containers that rely on entries/nodes.
|
* Internal package used by containers that rely on entries/nodes.
|
||||||
*
|
*
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.container.entry;
|
module tanya.container.entry;
|
||||||
|
|
||||||
version (unittest)
|
package struct SEntry(T)
|
||||||
{
|
|
||||||
package struct ConstEqualsStruct
|
|
||||||
{
|
|
||||||
int opEquals(typeof(this) that) const @nogc
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
package struct MutableEqualsStruct
|
|
||||||
{
|
|
||||||
int opEquals(typeof(this) that) @nogc
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
package struct NoEqualsStruct
|
|
||||||
{
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
package struct Entry(T)
|
|
||||||
{
|
{
|
||||||
/// Item content.
|
/// Item content.
|
||||||
T content;
|
T content;
|
||||||
|
|
||||||
/// Next item.
|
/// Next item.
|
||||||
Entry* next;
|
SEntry* next;
|
||||||
}
|
}
|
||||||
|
@ -3,16 +3,77 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.container.list;
|
module tanya.container.list;
|
||||||
|
|
||||||
|
import std.algorithm.comparison;
|
||||||
|
import std.algorithm.searching;
|
||||||
|
import std.traits;
|
||||||
import tanya.container.entry;
|
import tanya.container.entry;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
|
|
||||||
|
private struct Range(E)
|
||||||
|
if (__traits(isSame, TemplateOf!E, SEntry))
|
||||||
|
{
|
||||||
|
private alias T = typeof(E.content);
|
||||||
|
|
||||||
|
private E* head;
|
||||||
|
|
||||||
|
private this(E* head)
|
||||||
|
{
|
||||||
|
this.head = head;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property Range save()
|
||||||
|
{
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property size_t length() const
|
||||||
|
{
|
||||||
|
return count(opIndex());
|
||||||
|
}
|
||||||
|
|
||||||
|
@property bool empty() const
|
||||||
|
{
|
||||||
|
return head is null;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property ref inout(T) front() inout
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(!empty);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return head.content;
|
||||||
|
}
|
||||||
|
|
||||||
|
void popFront()
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(!empty);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
head = head.next;
|
||||||
|
}
|
||||||
|
|
||||||
|
Range opIndex()
|
||||||
|
{
|
||||||
|
return typeof(return)(head);
|
||||||
|
}
|
||||||
|
|
||||||
|
Range!(const E) opIndex() const
|
||||||
|
{
|
||||||
|
return typeof(return)(head);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Singly-linked list.
|
* Singly-linked list.
|
||||||
*
|
*
|
||||||
@ -21,6 +82,11 @@ import tanya.memory;
|
|||||||
*/
|
*/
|
||||||
struct SList(T)
|
struct SList(T)
|
||||||
{
|
{
|
||||||
|
private alias Entry = SEntry!T;
|
||||||
|
|
||||||
|
// 0th element of the list.
|
||||||
|
private Entry* head;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Removes all elements from the list.
|
* Removes all elements from the list.
|
||||||
*/
|
*/
|
||||||
@ -36,7 +102,7 @@ struct SList(T)
|
|||||||
{
|
{
|
||||||
while (!empty)
|
while (!empty)
|
||||||
{
|
{
|
||||||
popFront();
|
removeFront();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -61,7 +127,7 @@ struct SList(T)
|
|||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
return first.next.content;
|
return head.content;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -72,11 +138,11 @@ struct SList(T)
|
|||||||
*/
|
*/
|
||||||
void insertFront(ref T x)
|
void insertFront(ref T x)
|
||||||
{
|
{
|
||||||
auto temp = allocator.make!(Entry!T);
|
auto temp = allocator.make!Entry;
|
||||||
|
|
||||||
temp.content = x;
|
temp.content = x;
|
||||||
temp.next = first.next;
|
temp.next = head;
|
||||||
first.next = temp;
|
head = temp;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
@ -99,30 +165,98 @@ struct SList(T)
|
|||||||
assert(l.front == 9);
|
assert(l.front == 9);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: How many elements are in the list.
|
||||||
|
*/
|
||||||
|
@property size_t length() const
|
||||||
|
{
|
||||||
|
return count(opIndex());
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
SList!int l;
|
||||||
|
|
||||||
|
l.insertFront(8);
|
||||||
|
l.insertFront(9);
|
||||||
|
assert(l.length == 2);
|
||||||
|
l.removeFront();
|
||||||
|
assert(l.length == 1);
|
||||||
|
l.removeFront();
|
||||||
|
assert(l.length == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Comparison for equality.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* that = The list to compare with.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD true) if the lists are equal, $(D_KEYWORD false)
|
||||||
|
* otherwise.
|
||||||
|
*/
|
||||||
|
bool opEquals()(auto ref typeof(this) that) @trusted
|
||||||
|
{
|
||||||
|
return equal(opIndex(), that[]);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
bool opEquals()(in auto ref typeof(this) that) const @trusted
|
||||||
|
{
|
||||||
|
return equal(opIndex(), that[]);
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
SList!int l1, l2;
|
||||||
|
|
||||||
|
l1.insertFront(8);
|
||||||
|
l1.insertFront(9);
|
||||||
|
l2.insertFront(8);
|
||||||
|
l2.insertFront(10);
|
||||||
|
assert(l1 != l2);
|
||||||
|
|
||||||
|
l1.removeFront();
|
||||||
|
assert(l1 != l2);
|
||||||
|
|
||||||
|
l2.removeFront();
|
||||||
|
assert(l1 == l2);
|
||||||
|
|
||||||
|
l1.removeFront();
|
||||||
|
assert(l1 != l2);
|
||||||
|
|
||||||
|
l2.removeFront();
|
||||||
|
assert(l1 == l2);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: $(D_KEYWORD true) if the list is empty.
|
* Returns: $(D_KEYWORD true) if the list is empty.
|
||||||
*/
|
*/
|
||||||
@property bool empty() const
|
@property bool empty() const
|
||||||
{
|
{
|
||||||
return first.next is null;
|
return head is null;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns the first element and moves to the next one.
|
* Returns the first element and moves to the next one.
|
||||||
*
|
*
|
||||||
* Returns: The first element.
|
* Returns: The first element.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE !empty)
|
||||||
*/
|
*/
|
||||||
void popFront()
|
void removeFront()
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(!empty);
|
assert(!empty);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
auto n = first.next.next;
|
auto n = head.next;
|
||||||
|
|
||||||
allocator.dispose(first.next);
|
allocator.dispose(head);
|
||||||
first.next = n;
|
head = n;
|
||||||
}
|
}
|
||||||
|
|
||||||
///
|
///
|
||||||
@ -133,14 +267,16 @@ struct SList(T)
|
|||||||
l.insertFront(8);
|
l.insertFront(8);
|
||||||
l.insertFront(9);
|
l.insertFront(9);
|
||||||
assert(l.front == 9);
|
assert(l.front == 9);
|
||||||
l.popFront();
|
l.removeFront();
|
||||||
assert(l.front == 8);
|
assert(l.front == 8);
|
||||||
|
l.removeFront();
|
||||||
|
assert(l.empty);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Removes $(D_PARAM howMany) elements from the list.
|
* Removes $(D_PARAM howMany) elements from the list.
|
||||||
*
|
*
|
||||||
* Unlike $(D_PSYMBOL popFront()), this method doesn't fail, if it could not
|
* Unlike $(D_PSYMBOL removeFront()), this method doesn't fail, if it could not
|
||||||
* remove $(D_PARAM howMany) elements. Instead, if $(D_PARAM howMany) is
|
* remove $(D_PARAM howMany) elements. Instead, if $(D_PARAM howMany) is
|
||||||
* greater than the list length, all elements are removed.
|
* greater than the list length, all elements are removed.
|
||||||
*
|
*
|
||||||
@ -149,12 +285,17 @@ struct SList(T)
|
|||||||
*
|
*
|
||||||
* Returns: The number of elements removed.
|
* Returns: The number of elements removed.
|
||||||
*/
|
*/
|
||||||
size_t removeFront(in size_t howMany = 1)
|
size_t removeFront(in size_t howMany)
|
||||||
|
out (removed)
|
||||||
|
{
|
||||||
|
assert(removed <= howMany);
|
||||||
|
}
|
||||||
|
body
|
||||||
{
|
{
|
||||||
size_t i;
|
size_t i;
|
||||||
for (; i < howMany && !empty; ++i)
|
for (; i < howMany && !empty; ++i)
|
||||||
{
|
{
|
||||||
popFront();
|
removeFront();
|
||||||
}
|
}
|
||||||
return i;
|
return i;
|
||||||
}
|
}
|
||||||
@ -181,13 +322,15 @@ struct SList(T)
|
|||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* dg = $(D_KEYWORD foreach) body.
|
* dg = $(D_KEYWORD foreach) body.
|
||||||
|
*
|
||||||
|
* Returns: The value returned from $(D_PARAM dg).
|
||||||
*/
|
*/
|
||||||
int opApply(scope int delegate(ref size_t i, ref T) dg)
|
int opApply(scope int delegate(ref size_t i, ref T) @nogc dg)
|
||||||
{
|
{
|
||||||
int result;
|
int result;
|
||||||
size_t i;
|
size_t i;
|
||||||
|
|
||||||
for (auto pos = first.next; pos; pos = pos.next, ++i)
|
for (auto pos = head; pos; pos = pos.next, ++i)
|
||||||
{
|
{
|
||||||
result = dg(i, pos.content);
|
result = dg(i, pos.content);
|
||||||
|
|
||||||
@ -200,11 +343,11 @@ struct SList(T)
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
int opApply(scope int delegate(ref T) dg)
|
int opApply(scope int delegate(ref T) @nogc dg)
|
||||||
{
|
{
|
||||||
int result;
|
int result;
|
||||||
|
|
||||||
for (auto pos = first.next; pos; pos = pos.next)
|
for (auto pos = head; pos; pos = pos.next)
|
||||||
{
|
{
|
||||||
result = dg(pos.content);
|
result = dg(pos.content);
|
||||||
|
|
||||||
@ -232,8 +375,15 @@ struct SList(T)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// 0th element of the list.
|
Range!Entry opIndex()
|
||||||
private Entry!T first;
|
{
|
||||||
|
return typeof(return)(head);
|
||||||
|
}
|
||||||
|
|
||||||
|
Range!(const Entry) opIndex() const
|
||||||
|
{
|
||||||
|
return typeof(return)(head);
|
||||||
|
}
|
||||||
|
|
||||||
mixin DefaultAllocator;
|
mixin DefaultAllocator;
|
||||||
}
|
}
|
||||||
@ -257,7 +407,7 @@ unittest
|
|||||||
assert(i == 3);
|
assert(i == 3);
|
||||||
}
|
}
|
||||||
|
|
||||||
private unittest
|
unittest
|
||||||
{
|
{
|
||||||
interface Stuff
|
interface Stuff
|
||||||
{
|
{
|
||||||
|
@ -3,7 +3,7 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
|
@ -3,13 +3,14 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
*/
|
*/
|
||||||
module tanya.container.queue;
|
module tanya.container.queue;
|
||||||
|
|
||||||
|
import core.exception;
|
||||||
import std.traits;
|
import std.traits;
|
||||||
import std.algorithm.mutation;
|
import std.algorithm.mutation;
|
||||||
import tanya.container.entry;
|
import tanya.container.entry;
|
||||||
@ -34,18 +35,6 @@ struct Queue(T)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Removes all elements from the queue.
|
|
||||||
*/
|
|
||||||
deprecated
|
|
||||||
void clear()
|
|
||||||
{
|
|
||||||
while (!empty)
|
|
||||||
{
|
|
||||||
dequeue();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns how many elements are in the queue. It iterates through the queue
|
* Returns how many elements are in the queue. It iterates through the queue
|
||||||
* to count the elements.
|
* to count the elements.
|
||||||
@ -55,7 +44,7 @@ struct Queue(T)
|
|||||||
size_t length() const
|
size_t length() const
|
||||||
{
|
{
|
||||||
size_t len;
|
size_t len;
|
||||||
for (const(Entry!T)* i = first.next; i !is null; i = i.next)
|
for (const(SEntry!T)* i = first; i !is null; i = i.next)
|
||||||
{
|
{
|
||||||
++len;
|
++len;
|
||||||
}
|
}
|
||||||
@ -83,84 +72,27 @@ struct Queue(T)
|
|||||||
assert(q.length == 0);
|
assert(q.length == 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
version (D_Ddoc)
|
private void enqueueEntry(ref SEntry!T* entry)
|
||||||
{
|
{
|
||||||
/**
|
if (empty)
|
||||||
* Compares two queues. Checks if all elements of the both queues are equal.
|
|
||||||
*
|
|
||||||
* Returns: Whether $(D_KEYWORD this) and $(D_PARAM that) are equal.
|
|
||||||
*/
|
|
||||||
deprecated
|
|
||||||
int opEquals(ref typeof(this) that);
|
|
||||||
|
|
||||||
/// Ditto.
|
|
||||||
deprecated
|
|
||||||
int opEquals(typeof(this) that);
|
|
||||||
}
|
|
||||||
else static if (!hasMember!(T, "opEquals")
|
|
||||||
|| (functionAttributes!(T.opEquals) & FunctionAttribute.const_))
|
|
||||||
{
|
{
|
||||||
deprecated
|
first = rear = entry;
|
||||||
bool opEquals(in ref typeof(this) that) const
|
|
||||||
{
|
|
||||||
const(Entry!T)* i = first.next;
|
|
||||||
const(Entry!T)* j = that.first.next;
|
|
||||||
while (i !is null && j !is null)
|
|
||||||
{
|
|
||||||
if (i.content != j.content)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
i = i.next;
|
|
||||||
j = j.next;
|
|
||||||
}
|
|
||||||
return i is null && j is null;
|
|
||||||
}
|
|
||||||
|
|
||||||
deprecated
|
|
||||||
bool opEquals(in typeof(this) that) const
|
|
||||||
{
|
|
||||||
return opEquals(that);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
deprecated
|
rear.next = entry;
|
||||||
bool opEquals(ref typeof(this) that)
|
rear = rear.next;
|
||||||
{
|
|
||||||
Entry!T* i = first.next;
|
|
||||||
Entry!T* j = that.first.next;
|
|
||||||
while (i !is null && j !is null)
|
|
||||||
{
|
|
||||||
if (i.content != j.content)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
i = i.next;
|
|
||||||
j = j.next;
|
|
||||||
}
|
|
||||||
return i is null && j is null;
|
|
||||||
}
|
|
||||||
|
|
||||||
deprecated
|
|
||||||
bool opEquals(typeof(this) that)
|
|
||||||
{
|
|
||||||
return opEquals(that);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
private SEntry!T* allocateEntry()
|
||||||
* Returns: First element.
|
|
||||||
*/
|
|
||||||
deprecated("Use dequeue instead.")
|
|
||||||
@property ref inout(T) front() inout
|
|
||||||
in
|
|
||||||
{
|
{
|
||||||
assert(!empty);
|
auto temp = cast(SEntry!T*) allocator.allocate(SEntry!T.sizeof);
|
||||||
|
if (temp is null)
|
||||||
|
{
|
||||||
|
onOutOfMemoryError();
|
||||||
}
|
}
|
||||||
body
|
return temp;
|
||||||
{
|
|
||||||
return first.next.content;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -168,43 +100,36 @@ struct Queue(T)
|
|||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* x = New element.
|
* x = New element.
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD this).
|
|
||||||
*/
|
*/
|
||||||
ref typeof(this) enqueue(ref T x)
|
void enqueue(ref T x)
|
||||||
{
|
{
|
||||||
auto temp = allocator.make!(Entry!T)(x);
|
auto temp = allocateEntry();
|
||||||
if (empty)
|
|
||||||
{
|
*temp = SEntry!T.init;
|
||||||
first.next = rear = temp;
|
temp.content = x;
|
||||||
}
|
|
||||||
else
|
enqueueEntry(temp);
|
||||||
{
|
|
||||||
rear.next = temp;
|
|
||||||
rear = rear.next;
|
|
||||||
}
|
|
||||||
return this;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
ref typeof(this) enqueue(T x)
|
void enqueue(T x)
|
||||||
{
|
{
|
||||||
return enqueue(x);
|
auto temp = allocateEntry();
|
||||||
|
|
||||||
|
moveEmplace(x, (*temp).content);
|
||||||
|
(*temp).next = null;
|
||||||
|
|
||||||
|
enqueueEntry(temp);
|
||||||
}
|
}
|
||||||
|
|
||||||
deprecated("Use enqueue instead.")
|
|
||||||
alias insert = enqueue;
|
|
||||||
|
|
||||||
deprecated("Use enqueue instead.")
|
|
||||||
alias insertBack = enqueue;
|
|
||||||
|
|
||||||
///
|
///
|
||||||
unittest
|
unittest
|
||||||
{
|
{
|
||||||
Queue!int q;
|
Queue!int q;
|
||||||
|
|
||||||
assert(q.empty);
|
assert(q.empty);
|
||||||
q.enqueue(8).enqueue(9);
|
q.enqueue(8);
|
||||||
|
q.enqueue(9);
|
||||||
assert(q.dequeue() == 8);
|
assert(q.dequeue() == 8);
|
||||||
assert(q.dequeue() == 9);
|
assert(q.dequeue() == 9);
|
||||||
}
|
}
|
||||||
@ -214,7 +139,7 @@ struct Queue(T)
|
|||||||
*/
|
*/
|
||||||
@property bool empty() const
|
@property bool empty() const
|
||||||
{
|
{
|
||||||
return first.next is null;
|
return first is null;
|
||||||
}
|
}
|
||||||
|
|
||||||
///
|
///
|
||||||
@ -237,27 +162,24 @@ struct Queue(T)
|
|||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(!empty);
|
assert(!empty);
|
||||||
assert(allocator !is null);
|
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
auto n = first.next.next;
|
auto n = first.next;
|
||||||
T ret = move(first.next.content);
|
T ret = move(first.content);
|
||||||
|
|
||||||
dispose(allocator, first.next);
|
allocator.dispose(first);
|
||||||
first.next = n;
|
first = n;
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
deprecated("Use dequeue instead.")
|
|
||||||
alias popFront = dequeue;
|
|
||||||
|
|
||||||
///
|
///
|
||||||
unittest
|
unittest
|
||||||
{
|
{
|
||||||
Queue!int q;
|
Queue!int q;
|
||||||
|
|
||||||
q.enqueue(8).enqueue(9);
|
q.enqueue(8);
|
||||||
|
q.enqueue(9);
|
||||||
assert(q.dequeue() == 8);
|
assert(q.dequeue() == 8);
|
||||||
assert(q.dequeue() == 9);
|
assert(q.dequeue() == 9);
|
||||||
}
|
}
|
||||||
@ -268,6 +190,8 @@ struct Queue(T)
|
|||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* dg = $(D_KEYWORD foreach) body.
|
* dg = $(D_KEYWORD foreach) body.
|
||||||
|
*
|
||||||
|
* Returns: The value returned from $(D_PARAM dg).
|
||||||
*/
|
*/
|
||||||
int opApply(scope int delegate(ref size_t i, ref T) @nogc dg)
|
int opApply(scope int delegate(ref size_t i, ref T) @nogc dg)
|
||||||
{
|
{
|
||||||
@ -306,7 +230,9 @@ struct Queue(T)
|
|||||||
Queue!int q;
|
Queue!int q;
|
||||||
|
|
||||||
size_t j;
|
size_t j;
|
||||||
q.enqueue(5).enqueue(4).enqueue(9);
|
q.enqueue(5);
|
||||||
|
q.enqueue(4);
|
||||||
|
q.enqueue(9);
|
||||||
foreach (i, e; q)
|
foreach (i, e; q)
|
||||||
{
|
{
|
||||||
assert(i != 2 || e == 9);
|
assert(i != 2 || e == 9);
|
||||||
@ -318,7 +244,9 @@ struct Queue(T)
|
|||||||
assert(q.empty);
|
assert(q.empty);
|
||||||
|
|
||||||
j = 0;
|
j = 0;
|
||||||
q.enqueue(5).enqueue(4).enqueue(9);
|
q.enqueue(5);
|
||||||
|
q.enqueue(4);
|
||||||
|
q.enqueue(9);
|
||||||
foreach (e; q)
|
foreach (e; q)
|
||||||
{
|
{
|
||||||
assert(j != 2 || e == 9);
|
assert(j != 2 || e == 9);
|
||||||
@ -330,11 +258,8 @@ struct Queue(T)
|
|||||||
assert(q.empty);
|
assert(q.empty);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// The first element of the list.
|
private SEntry!T* first;
|
||||||
private Entry!T first;
|
private SEntry!T* rear;
|
||||||
|
|
||||||
/// The last element of the list.
|
|
||||||
private Entry!T* rear;
|
|
||||||
|
|
||||||
mixin DefaultAllocator;
|
mixin DefaultAllocator;
|
||||||
}
|
}
|
||||||
@ -347,7 +272,8 @@ unittest
|
|||||||
q.enqueue(5);
|
q.enqueue(5);
|
||||||
assert(!q.empty);
|
assert(!q.empty);
|
||||||
|
|
||||||
q.enqueue(4).enqueue(9);
|
q.enqueue(4);
|
||||||
|
q.enqueue(9);
|
||||||
|
|
||||||
assert(q.dequeue() == 5);
|
assert(q.dequeue() == 5);
|
||||||
|
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -1,510 +0,0 @@
|
|||||||
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Copyright: Eugene Wissner 2016.
|
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
|
||||||
*/
|
|
||||||
module tanya.crypto.bit;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Wrapper that allows bit manipulation on $(D_KEYWORD ubyte[]) array.
|
|
||||||
*/
|
|
||||||
struct BitVector
|
|
||||||
{
|
|
||||||
protected ubyte[] vector;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* array = Array should be manipulated on.
|
|
||||||
*/
|
|
||||||
this(inout(ubyte[]) array) inout pure nothrow @safe @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(array.length <= size_t.max / 8);
|
|
||||||
assert(array !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
vector = array;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
ubyte[5] array1 = [234, 3, 252, 10, 18];
|
|
||||||
ubyte[3] array2 = [65, 13, 173];
|
|
||||||
auto bits = BitVector(array1);
|
|
||||||
|
|
||||||
assert(bits[] is array1);
|
|
||||||
assert(bits[] !is array2);
|
|
||||||
|
|
||||||
bits = BitVector(array2);
|
|
||||||
assert(bits[] is array2);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Number of bits in the vector.
|
|
||||||
*/
|
|
||||||
@property inout(size_t) length() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return vector.length * 8;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Ditto.
|
|
||||||
inout(size_t) opDollar() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return vector.length * 8;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
// [01000001, 00001101, 10101101]
|
|
||||||
ubyte[3] arr = [65, 13, 173];
|
|
||||||
auto bits = BitVector(arr);
|
|
||||||
|
|
||||||
assert(bits.length == 24);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* bit = Bit position.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD true) if the bit on position $(D_PARAM bit) is set,
|
|
||||||
* $(D_KEYWORD false) if not set.
|
|
||||||
*/
|
|
||||||
inout(bool) opIndex(size_t bit) inout const pure nothrow @safe @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(bit / 8 <= vector.length);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
return (vector[bit / 8] & (0x80 >> (bit % 8))) != 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
// [01000001, 00001101, 10101101]
|
|
||||||
ubyte[3] arr = [65, 13, 173];
|
|
||||||
auto bits = BitVector(arr);
|
|
||||||
|
|
||||||
assert(!bits[0]);
|
|
||||||
assert(bits[1]);
|
|
||||||
assert(bits[7]);
|
|
||||||
assert(!bits[8]);
|
|
||||||
assert(!bits[11]);
|
|
||||||
assert(bits[12]);
|
|
||||||
assert(bits[20]);
|
|
||||||
assert(bits[23]);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Underlying array.
|
|
||||||
*/
|
|
||||||
inout(ubyte[]) opIndex() inout pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return vector;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
// [01000001, 00001101, 10101101]
|
|
||||||
ubyte[3] arr = [65, 13, 173];
|
|
||||||
auto bits = BitVector(arr);
|
|
||||||
|
|
||||||
assert(bits[] is arr);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* value = $(D_KEYWORD true) if the bit should be set,
|
|
||||||
* $(D_KEYWORD false) if cleared.
|
|
||||||
* bit = Bit position.
|
|
||||||
*
|
|
||||||
* Returns: $(D_PSYMBOL this).
|
|
||||||
*/
|
|
||||||
bool opIndexAssign(bool value, size_t bit) pure nothrow @safe @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(bit / 8 <= vector.length);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
if (value)
|
|
||||||
{
|
|
||||||
vector[bit / 8] |= (0x80 >> (bit % 8));
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
vector[bit / 8] &= ~(0x80 >> (bit % 8));
|
|
||||||
}
|
|
||||||
return value;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
// [01000001, 00001101, 10101101]
|
|
||||||
ubyte[3] arr = [65, 13, 173];
|
|
||||||
auto bits = BitVector(arr);
|
|
||||||
|
|
||||||
bits[5] = bits[6] = true;
|
|
||||||
assert(bits[][0] == 71);
|
|
||||||
|
|
||||||
bits[14] = true;
|
|
||||||
bits[15] = false;
|
|
||||||
assert(bits[][1] == 14);
|
|
||||||
|
|
||||||
bits[16] = bits[23] = false;
|
|
||||||
assert(bits[][2] == 44);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Copies bits from $(D_PARAM vector) into this $(D_PSYMBOL BitVector).
|
|
||||||
*
|
|
||||||
* The array that should be assigned, can be smaller (but not larger) than
|
|
||||||
* the underlying array of this $(D_PSYMBOL BitVector), leading zeros will
|
|
||||||
* be added in this case to the left.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* vector = $(D_KEYWORD ubyte[]) array not larger than
|
|
||||||
* `$(D_PSYMBOL length) / 8`.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD this).
|
|
||||||
*/
|
|
||||||
BitVector opAssign(ubyte[] vector) pure nothrow @safe @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(vector.length <= this.vector.length);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
immutable delta = this.vector.length - vector.length;
|
|
||||||
if (delta > 0)
|
|
||||||
{
|
|
||||||
this.vector[0..delta] = 0;
|
|
||||||
}
|
|
||||||
this.vector[delta..$] = vector[0..$];
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
ubyte[5] array1 = [234, 3, 252, 10, 18];
|
|
||||||
ubyte[3] array2 = [65, 13, 173];
|
|
||||||
auto bits = BitVector(array1);
|
|
||||||
|
|
||||||
bits = array2;
|
|
||||||
assert(bits[][0] == 0);
|
|
||||||
assert(bits[][1] == 0);
|
|
||||||
assert(bits[][2] == 65);
|
|
||||||
assert(bits[][3] == 13);
|
|
||||||
assert(bits[][4] == 173);
|
|
||||||
|
|
||||||
bits = array2[0..2];
|
|
||||||
assert(bits[][0] == 0);
|
|
||||||
assert(bits[][1] == 0);
|
|
||||||
assert(bits[][2] == 0);
|
|
||||||
assert(bits[][3] == 65);
|
|
||||||
assert(bits[][4] == 13);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Support for bitwise operations.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* that = Another bit vector.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD this).
|
|
||||||
*/
|
|
||||||
BitVector opOpAssign(string op)(BitVector that) pure nothrow @safe @nogc
|
|
||||||
if ((op == "^") || (op == "|") || (op == "&"))
|
|
||||||
{
|
|
||||||
return opOpAssign(op)(that.vector);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Ditto.
|
|
||||||
BitVector opOpAssign(string op)(ubyte[] that) pure nothrow @safe @nogc
|
|
||||||
if ((op == "^") || (op == "|") || (op == "&"))
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(that.length <= vector.length);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
for (int i = cast(int) vector.length - 1; i >= 0; --i)
|
|
||||||
{
|
|
||||||
mixin("vector[i] " ~ op ~ "= " ~ "that[i];");
|
|
||||||
}
|
|
||||||
immutable delta = vector.length - that.length;
|
|
||||||
if (delta)
|
|
||||||
{
|
|
||||||
static if (op == "&")
|
|
||||||
{
|
|
||||||
vector[0..delta] = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
// [01000001, 00001101, 10101101]
|
|
||||||
ubyte[3] array1 = [65, 13, 173];
|
|
||||||
ubyte[3] array2 = [0b01010010, 0b10111110, 0b10111110];
|
|
||||||
auto bits = BitVector(array1);
|
|
||||||
|
|
||||||
bits |= array2;
|
|
||||||
assert(bits[][0] == 0b01010011);
|
|
||||||
assert(bits[][1] == 0b10111111);
|
|
||||||
assert(bits[][2] == 0b10111111);
|
|
||||||
|
|
||||||
bits &= array2;
|
|
||||||
assert(bits[][0] == array2[0]);
|
|
||||||
assert(bits[][1] == array2[1]);
|
|
||||||
assert(bits[][2] == array2[2]);
|
|
||||||
|
|
||||||
bits ^= array2;
|
|
||||||
assert(bits[][0] == 0);
|
|
||||||
assert(bits[][1] == 0);
|
|
||||||
assert(bits[][2] == 0);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Support for shift operations.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* n = Number of bits.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD this).
|
|
||||||
*/
|
|
||||||
BitVector opOpAssign(string op)(in size_t n) pure nothrow @safe @nogc
|
|
||||||
if ((op == "<<") || (op == ">>"))
|
|
||||||
|
|
||||||
{
|
|
||||||
if (n >= length)
|
|
||||||
{
|
|
||||||
vector[0..$] = 0;
|
|
||||||
}
|
|
||||||
else if (n != 0)
|
|
||||||
{
|
|
||||||
immutable bit = n % 8, step = n / 8;
|
|
||||||
immutable delta = 8 - bit;
|
|
||||||
size_t i, j;
|
|
||||||
|
|
||||||
static if (op == "<<")
|
|
||||||
{
|
|
||||||
for (j = step; j < vector.length - 1; ++i)
|
|
||||||
{
|
|
||||||
vector[i] = cast(ubyte)((vector[j] << bit)
|
|
||||||
| vector[++j] >> delta);
|
|
||||||
}
|
|
||||||
vector[i] = cast(ubyte)(vector[j] << bit);
|
|
||||||
vector[$ - step ..$] = 0;
|
|
||||||
}
|
|
||||||
else static if (op == ">>")
|
|
||||||
{
|
|
||||||
for (i = vector.length - 1, j = i - step; j > 0; --i)
|
|
||||||
{
|
|
||||||
vector[i] = cast(ubyte)((vector[j] >> bit)
|
|
||||||
| vector[--j] << delta);
|
|
||||||
}
|
|
||||||
vector[i] = cast(ubyte)(vector[j] >> bit);
|
|
||||||
vector[0..step] = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
nothrow @safe @nogc unittest
|
|
||||||
{
|
|
||||||
ubyte[4] arr = [0b10111110, 0b11110010, 0b01010010, 0b01010011];
|
|
||||||
auto bits = BitVector(arr);
|
|
||||||
|
|
||||||
bits <<= 0;
|
|
||||||
assert(bits[][0] == 0b10111110 && bits[][1] == 0b11110010
|
|
||||||
&& bits[][2] == 0b01010010 && bits[][3] == 0b01010011);
|
|
||||||
|
|
||||||
bits <<= 2;
|
|
||||||
assert(bits[][0] == 0b11111011 && bits[][1] == 0b11001001
|
|
||||||
&& bits[][2] == 0b01001001 && bits[][3] == 0b01001100);
|
|
||||||
|
|
||||||
bits <<= 4;
|
|
||||||
assert(bits[][0] == 0b10111100 && bits[][1] == 0b10010100
|
|
||||||
&& bits[][2] == 0b10010100 && bits[][3] == 0b11000000);
|
|
||||||
|
|
||||||
bits <<= 8;
|
|
||||||
assert(bits[][0] == 0b10010100 && bits[][1] == 0b10010100
|
|
||||||
&& bits[][2] == 0b11000000 && bits[][3] == 0b00000000);
|
|
||||||
|
|
||||||
bits <<= 7;
|
|
||||||
assert(bits[][0] == 0b01001010 && bits[][1] == 0b01100000
|
|
||||||
&& bits[][2] == 0b00000000 && bits[][3] == 0b00000000);
|
|
||||||
|
|
||||||
bits <<= 25;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00000000 && bits[][3] == 0b00000000);
|
|
||||||
|
|
||||||
arr = [0b00110011, 0b11001100, 0b11111111, 0b01010101];
|
|
||||||
bits <<= 24;
|
|
||||||
assert(bits[][0] == 0b01010101 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00000000 && bits[][3] == 0b00000000);
|
|
||||||
|
|
||||||
arr[1] = 0b11001100;
|
|
||||||
arr[2] = 0b11111111;
|
|
||||||
arr[3] = 0b01010101;
|
|
||||||
bits <<= 12;
|
|
||||||
assert(bits[][0] == 0b11001111 && bits[][1] == 0b11110101
|
|
||||||
&& bits[][2] == 0b01010000 && bits[][3] == 0b00000000);
|
|
||||||
|
|
||||||
bits <<= 100;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00000000 && bits[][3] == 0b00000000);
|
|
||||||
|
|
||||||
arr = [0b10111110, 0b11110010, 0b01010010, 0b01010011];
|
|
||||||
bits >>= 0;
|
|
||||||
assert(bits[][0] == 0b10111110 && bits[][1] == 0b11110010
|
|
||||||
&& bits[][2] == 0b01010010 && bits[][3] == 0b01010011);
|
|
||||||
|
|
||||||
bits >>= 2;
|
|
||||||
assert(bits[][0] == 0b00101111 && bits[][1] == 0b10111100
|
|
||||||
&& bits[][2] == 0b10010100 && bits[][3] == 0b10010100);
|
|
||||||
|
|
||||||
bits >>= 4;
|
|
||||||
assert(bits[][0] == 0b00000010 && bits[][1] == 0b11111011
|
|
||||||
&& bits[][2] == 0b11001001 && bits[][3] == 0b01001001);
|
|
||||||
|
|
||||||
bits >>= 8;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000010
|
|
||||||
&& bits[][2] == 0b11111011 && bits[][3] == 0b11001001);
|
|
||||||
|
|
||||||
bits >>= 7;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00000101 && bits[][3] == 0b11110111);
|
|
||||||
|
|
||||||
bits >>= 25;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00000000 && bits[][3] == 0b00000000);
|
|
||||||
|
|
||||||
arr = [0b00110011, 0b11001100, 0b11111111, 0b01010101];
|
|
||||||
bits >>= 24;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00000000 && bits[][3] == 0b00110011);
|
|
||||||
|
|
||||||
arr[1] = 0b11001100;
|
|
||||||
arr[2] = 0b11111111;
|
|
||||||
arr[3] = 0b01010101;
|
|
||||||
bits >>= 12;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00001100 && bits[][3] == 0b11001111);
|
|
||||||
|
|
||||||
bits >>= 100;
|
|
||||||
assert(bits[][0] == 0b00000000 && bits[][1] == 0b00000000
|
|
||||||
&& bits[][2] == 0b00000000 && bits[][3] == 0b00000000);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Negates all bits.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD this).
|
|
||||||
*/
|
|
||||||
BitVector opUnary(string op)() pure nothrow @safe @nogc
|
|
||||||
if (op == "~")
|
|
||||||
{
|
|
||||||
foreach (ref b; vector)
|
|
||||||
{
|
|
||||||
b = ~b;
|
|
||||||
}
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
// [01000001, 00001101, 10101101]
|
|
||||||
ubyte[3] arr = [65, 13, 173];
|
|
||||||
auto bits = BitVector(arr);
|
|
||||||
|
|
||||||
~bits;
|
|
||||||
assert(bits[][0] == 0b10111110);
|
|
||||||
assert(bits[][1] == 0b11110010);
|
|
||||||
assert(bits[][2] == 0b01010010);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Iterates through all bits.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* dg = $(D_KEYWORD foreach) delegate.
|
|
||||||
*
|
|
||||||
* Returns: By $(D_PARAM dg) returned value.
|
|
||||||
*/
|
|
||||||
int opApply(int delegate(size_t, bool) dg)
|
|
||||||
{
|
|
||||||
int result;
|
|
||||||
foreach (i, ref v; vector)
|
|
||||||
{
|
|
||||||
foreach (c; 0..8)
|
|
||||||
{
|
|
||||||
result = dg(i * 8 + c, (v & (0x80 >> c)) != 0);
|
|
||||||
if (result)
|
|
||||||
{
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Ditto.
|
|
||||||
int opApply(int delegate(bool) dg)
|
|
||||||
{
|
|
||||||
int result;
|
|
||||||
foreach (ref v; vector)
|
|
||||||
{
|
|
||||||
foreach (c; 0..8)
|
|
||||||
{
|
|
||||||
result = dg((v & (0x80 >> c)) != 0);
|
|
||||||
if (result)
|
|
||||||
{
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
ubyte[2] arr = [0b01000001, 0b00001101];
|
|
||||||
auto bits = BitVector(arr);
|
|
||||||
size_t c;
|
|
||||||
|
|
||||||
foreach (i, v; bits)
|
|
||||||
{
|
|
||||||
assert(i == c);
|
|
||||||
if (i == 1 || i == 7 || i == 15 || i == 13 || i == 12)
|
|
||||||
{
|
|
||||||
assert(v);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
assert(!v);
|
|
||||||
}
|
|
||||||
++c;
|
|
||||||
}
|
|
||||||
assert(c == 16);
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,607 +0,0 @@
|
|||||||
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Copyright: Eugene Wissner 2016.
|
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
|
||||||
*/
|
|
||||||
module tanya.crypto.des;
|
|
||||||
|
|
||||||
import tanya.crypto.bit;
|
|
||||||
import tanya.crypto.symmetric;
|
|
||||||
|
|
||||||
/// Initial permutation table.
|
|
||||||
private immutable ubyte[64] ipTable = [58, 50, 42, 34, 26, 18, 10, 2,
|
|
||||||
60, 52, 44, 36, 28, 20, 12, 4,
|
|
||||||
62, 54, 46, 38, 30, 22, 14, 6,
|
|
||||||
64, 56, 48, 40, 32, 24, 16, 8,
|
|
||||||
57, 49, 41, 33, 25, 17, 9, 1,
|
|
||||||
59, 51, 43, 35, 27, 19, 11, 3,
|
|
||||||
61, 53, 45, 37, 29, 21, 13, 5,
|
|
||||||
63, 55, 47, 39, 31, 23, 15, 7];
|
|
||||||
|
|
||||||
/// Final permutation table.
|
|
||||||
private immutable ubyte[64] fpTable = [40, 8, 48, 16, 56, 24, 64, 32,
|
|
||||||
39, 7, 47, 15, 55, 23, 63, 31,
|
|
||||||
38, 6, 46, 14, 54, 22, 62, 30,
|
|
||||||
37, 5, 45, 13, 53, 21, 61, 29,
|
|
||||||
36, 4, 44, 12, 52, 20, 60, 28,
|
|
||||||
35, 3, 43, 11, 51, 19, 59, 27,
|
|
||||||
34, 2, 42, 10, 50, 18, 58, 26,
|
|
||||||
33, 1, 41, 9, 49, 17, 57, 25];
|
|
||||||
|
|
||||||
/// Key permutation table 1.
|
|
||||||
private immutable ubyte[64] pc1Table = [57, 49, 41, 33, 25, 17, 9, 1,
|
|
||||||
58, 50, 42, 34, 26, 18, 10, 2,
|
|
||||||
59, 51, 43, 35, 27, 19, 11, 3,
|
|
||||||
60, 52, 44, 36, 63, 55, 47, 39,
|
|
||||||
31, 23, 15, 7, 62, 54, 46, 38,
|
|
||||||
30, 22, 14, 6, 61, 53, 45, 37,
|
|
||||||
29, 21, 13, 5, 28, 20, 12, 4];
|
|
||||||
|
|
||||||
/// Key permutation table 2.
|
|
||||||
private immutable ubyte[48] pc2Table = [14, 17, 11, 24, 1, 5, 3, 28,
|
|
||||||
15, 6, 21, 10, 23, 19, 12, 4,
|
|
||||||
26, 8, 16, 7, 27, 20, 13, 2,
|
|
||||||
41, 52, 31, 37, 47, 55, 30, 40,
|
|
||||||
51, 45, 33, 48, 44, 49, 39, 56,
|
|
||||||
34, 53, 46, 42, 50, 36, 29, 32];
|
|
||||||
|
|
||||||
/// Expansion table.
|
|
||||||
private immutable ubyte[48] expansionTable = [32, 1, 2, 3, 4, 5, 4, 5,
|
|
||||||
6, 7, 8, 9, 8, 9, 10, 11,
|
|
||||||
12, 13, 12, 13, 14, 15, 16, 17,
|
|
||||||
16, 17, 18, 19, 20, 21, 20, 21,
|
|
||||||
22, 23, 24, 25, 24, 25, 26, 27,
|
|
||||||
28, 29, 28, 29, 30, 31, 32, 1];
|
|
||||||
|
|
||||||
/// Final input block permutation.
|
|
||||||
private immutable ubyte[32] pTable = [16, 7, 20, 21, 29, 12, 28, 17,
|
|
||||||
1, 15, 23, 26, 5, 18, 31, 10,
|
|
||||||
2, 8, 24, 14, 32, 27, 3, 9,
|
|
||||||
19, 13, 30, 6, 22, 11, 4, 25];
|
|
||||||
|
|
||||||
/// The (in)famous S-boxes.
|
|
||||||
private immutable ubyte[64][8] sBox = [[
|
|
||||||
14, 0, 4, 15, 13, 7, 1, 4, 2, 14, 15, 2, 11, 13, 8, 1,
|
|
||||||
3, 10, 10, 6, 6, 12, 12, 11, 5, 9, 9, 5, 0, 3, 7, 8,
|
|
||||||
4, 15, 1, 12, 14, 8, 8, 2, 13, 4, 6, 9, 2, 1, 11, 7,
|
|
||||||
15, 5, 12, 11, 9, 3, 7, 14, 3, 10, 10, 0, 5, 6, 0, 13,
|
|
||||||
],[
|
|
||||||
15, 3, 1, 13, 8, 4, 14, 7, 6, 15, 11, 2, 3, 8, 4, 14,
|
|
||||||
9, 12, 7, 0, 2, 1, 13, 10, 12, 6, 0, 9, 5, 11, 10, 5,
|
|
||||||
0, 13, 14, 8, 7, 10, 11, 1, 10, 3, 4, 15, 13, 4, 1, 2,
|
|
||||||
5, 11, 8, 6, 12, 7, 6, 12, 9, 0, 3, 5, 2, 14, 15, 9,
|
|
||||||
],[
|
|
||||||
10, 13, 0, 7, 9, 0, 14, 9, 6, 3, 3, 4, 15, 6, 5, 10,
|
|
||||||
1, 2, 13, 8, 12, 5, 7, 14, 11, 12, 4, 11, 2, 15, 8, 1,
|
|
||||||
13, 1, 6, 10, 4, 13, 9, 0, 8, 6, 15, 9, 3, 8, 0, 7,
|
|
||||||
11, 4, 1, 15, 2, 14, 12, 3, 5, 11, 10, 5, 14, 2, 7, 12,
|
|
||||||
],[
|
|
||||||
7, 13, 13, 8, 14, 11, 3, 5, 0, 6, 6, 15, 9, 0, 10, 3,
|
|
||||||
1, 4, 2, 7, 8, 2, 5, 12, 11, 1, 12, 10, 4, 14, 15, 9,
|
|
||||||
10, 3, 6, 15, 9, 0, 0, 6, 12, 10, 11, 1, 7, 13, 13, 8,
|
|
||||||
15, 9, 1, 4, 3, 5, 14, 11, 5, 12, 2, 7, 8, 2, 4, 14,
|
|
||||||
],[
|
|
||||||
2, 14, 12, 11, 4, 2, 1, 12, 7, 4, 10, 7, 11, 13, 6, 1,
|
|
||||||
8, 5, 5, 0, 3, 15, 15, 10, 13, 3, 0, 9, 14, 8, 9, 6,
|
|
||||||
4, 11, 2, 8, 1, 12, 11, 7, 10, 1, 13, 14, 7, 2, 8, 13,
|
|
||||||
15, 6, 9, 15, 12, 0, 5, 9, 6, 10, 3, 4, 0, 5, 14, 3,
|
|
||||||
],[
|
|
||||||
12, 10, 1, 15, 10, 4, 15, 2, 9, 7, 2, 12, 6, 9, 8, 5,
|
|
||||||
0, 6, 13, 1, 3, 13, 4, 14, 14, 0, 7, 11, 5, 3, 11, 8,
|
|
||||||
9, 4, 14, 3, 15, 2, 5, 12, 2, 9, 8, 5, 12, 15, 3, 10,
|
|
||||||
7, 11, 0, 14, 4, 1, 10, 7, 1, 6, 13, 0, 11, 8, 6, 13,
|
|
||||||
],[
|
|
||||||
4, 13, 11, 0, 2, 11, 14, 7, 15, 4, 0, 9, 8, 1, 13, 10,
|
|
||||||
3, 14, 12, 3, 9, 5, 7, 12, 5, 2, 10, 15, 6, 8, 1, 6,
|
|
||||||
1, 6, 4, 11, 11, 13, 13, 8, 12, 1, 3, 4, 7, 10, 14, 7,
|
|
||||||
10, 9, 15, 5, 6, 0, 8, 15, 0, 14, 5, 2, 9, 3, 2, 12,
|
|
||||||
],[
|
|
||||||
13, 1, 2, 15, 8, 13, 4, 8, 6, 10, 15, 3, 11, 7, 1, 4,
|
|
||||||
10, 12, 9, 5, 3, 6, 14, 11, 5, 0, 0, 14, 12, 9, 7, 2,
|
|
||||||
7, 2, 11, 1, 4, 14, 1, 7, 9, 4, 12, 10, 14, 8, 2, 13,
|
|
||||||
0, 15, 6, 12, 10, 9, 13, 0, 15, 3, 3, 5, 5, 6, 8, 11,
|
|
||||||
]];
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Data Encryption Standard.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* L = Number of keys.
|
|
||||||
*/
|
|
||||||
class DES(ushort L = 1) : BlockCipher
|
|
||||||
if (L == 1)
|
|
||||||
{
|
|
||||||
mixin FixedBlockSize!8;
|
|
||||||
mixin KeyLength!8;
|
|
||||||
|
|
||||||
private enum expansionBlockSize = 6;
|
|
||||||
private enum pc1KeyLength = 7;
|
|
||||||
private enum subkeyLength = 6;
|
|
||||||
private ubyte[] key_;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* key = Key.
|
|
||||||
*/
|
|
||||||
@property void key(ubyte[] key) pure nothrow @safe @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(key.length >= minKeyLength);
|
|
||||||
assert(key.length <= maxKeyLength);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
key_ = key;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Encrypts a block.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* plain = Plain text, input.
|
|
||||||
* cipher = Cipher text, output.
|
|
||||||
*/
|
|
||||||
void encrypt(in ubyte[] plain, ubyte[] cipher)
|
|
||||||
nothrow
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(plain.length == blockSize);
|
|
||||||
assert(cipher.length == blockSize);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
operateBlock!(Direction.encryption)(plain, cipher);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Decrypts a block.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* cipher = Cipher text, input.
|
|
||||||
* plain = Plain text, output.
|
|
||||||
*/
|
|
||||||
void decrypt(in ubyte[] cipher, ubyte[] plain)
|
|
||||||
nothrow
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(plain.length == blockSize);
|
|
||||||
assert(cipher.length == blockSize);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
operateBlock!(Direction.decryption)(cipher, plain);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void operateBlock(Direction D)(in ubyte[] source, ref ubyte[] target)
|
|
||||||
{
|
|
||||||
ubyte[blockSize_] ipBlock;
|
|
||||||
ubyte[expansionBlockSize] expansionBlock;
|
|
||||||
ubyte[4] substitutionBlock;
|
|
||||||
ubyte[4] pBoxTarget;
|
|
||||||
ubyte[pc1KeyLength] pc1Key;
|
|
||||||
ubyte[subkeyLength] subkey;
|
|
||||||
|
|
||||||
// Initial permutation
|
|
||||||
permute(source, ipBlock, ipTable, blockSize);
|
|
||||||
|
|
||||||
// Key schedule computation
|
|
||||||
permute(key_, pc1Key, pc1Table, pc1KeyLength);
|
|
||||||
|
|
||||||
// Feistel function
|
|
||||||
for (ubyte round; round < 16; ++round)
|
|
||||||
{
|
|
||||||
auto bitVector = BitVector(expansionBlock);
|
|
||||||
/* Expansion. This permutation only looks at the first 4 bytes (32
|
|
||||||
bits of ipBlock); 16 of these are repeated in expansion table.*/
|
|
||||||
permute(BitVector(ipBlock[4..$]), bitVector, expansionTable, 6);
|
|
||||||
|
|
||||||
// Key mixing
|
|
||||||
static if (D == Direction.encryption)
|
|
||||||
{
|
|
||||||
rotateLeft(pc1Key);
|
|
||||||
if (!(round <= 1 || round == 8 || round == 15))
|
|
||||||
{
|
|
||||||
// Rotate twice.
|
|
||||||
rotateLeft(pc1Key);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
permute(pc1Key, subkey, pc2Table, subkeyLength);
|
|
||||||
static if (D == Direction.decryption)
|
|
||||||
{
|
|
||||||
rotateRight(pc1Key);
|
|
||||||
if (!(round >= 14 || round == 7 || round == 0))
|
|
||||||
{
|
|
||||||
// Rotate twice.
|
|
||||||
rotateRight(pc1Key);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bitVector ^= subkey;
|
|
||||||
|
|
||||||
// Substitution; copy from updated expansion block to ciphertext block
|
|
||||||
substitutionBlock[0] = cast(ubyte) (sBox[0][(expansionBlock[0] & 0xfc ) >> 2] << 4);
|
|
||||||
substitutionBlock[0] |= sBox[1][(expansionBlock[0] & 0x03) << 4 | (expansionBlock[1] & 0xf0) >> 4];
|
|
||||||
substitutionBlock[1] = cast(ubyte) (sBox[2][(expansionBlock[1] & 0x0f) << 2 | (expansionBlock[2] & 0xc0) >> 6] << 4);
|
|
||||||
substitutionBlock[1] |= sBox[3][(expansionBlock[2] & 0x3f)];
|
|
||||||
substitutionBlock[2] = cast(ubyte) (sBox[4][(expansionBlock[3] & 0xfc) >> 2 ] << 4);
|
|
||||||
substitutionBlock[2] |= sBox[5][(expansionBlock[3] & 0x03) << 4 | (expansionBlock[4] & 0xf0) >> 4];
|
|
||||||
substitutionBlock[3] = cast(ubyte) (sBox[6][(expansionBlock[4] & 0x0F) << 2 | (expansionBlock[5] & 0xc0) >> 6] << 4);
|
|
||||||
substitutionBlock[3] |= sBox[7][(expansionBlock[5] & 0x3f)];
|
|
||||||
|
|
||||||
// Permutation
|
|
||||||
bitVector = BitVector(substitutionBlock);
|
|
||||||
permute(bitVector, pBoxTarget, pTable, blockSize / 2);
|
|
||||||
|
|
||||||
// Swap the halves.
|
|
||||||
substitutionBlock = ipBlock[0..4];
|
|
||||||
ipBlock[0..4] = ipBlock[4..$];
|
|
||||||
|
|
||||||
bitVector ^= pBoxTarget;
|
|
||||||
ipBlock[4..$] = substitutionBlock;
|
|
||||||
}
|
|
||||||
|
|
||||||
substitutionBlock = ipBlock[0..4];
|
|
||||||
ipBlock[0..4] = ipBlock[4..$];
|
|
||||||
ipBlock[4..$] = substitutionBlock;
|
|
||||||
|
|
||||||
// Final permutaion (undo initial permuation).
|
|
||||||
permute(ipBlock, target, fpTable, blockSize);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Performs the left rotation operation on the key.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* key = The key to rotate.
|
|
||||||
*/
|
|
||||||
private void rotateLeft(ref ubyte[7] key) const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
immutable carryLeft = (key[0] & 0x80) >> 3;
|
|
||||||
|
|
||||||
key[0] = cast(ubyte) ((key[0] << 1) | ((key[1] & 0x80) >> 7));
|
|
||||||
key[1] = cast(ubyte) ((key[1] << 1) | ((key[2] & 0x80) >> 7));
|
|
||||||
key[2] = cast(ubyte) ((key[2] << 1) | ((key[3] & 0x80) >> 7));
|
|
||||||
|
|
||||||
immutable carryRight = (key[3] & 0x08) >> 3;
|
|
||||||
key[3] = cast(ubyte) ((((key[3] << 1) | ((key[4] & 0x80) >> 7)) & ~0x10) | carryLeft);
|
|
||||||
|
|
||||||
key[4] = cast(ubyte) ((key[4] << 1) | ((key[5] & 0x80) >> 7));
|
|
||||||
key[5] = cast(ubyte) ((key[5] << 1) | ((key[6] & 0x80) >> 7));
|
|
||||||
key[6] = cast(ubyte) ((key[6] << 1) | carryRight);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Performs the right rotation operation on the key.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* key = The key to rotate.
|
|
||||||
*/
|
|
||||||
private void rotateRight(ref ubyte[7] key) const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
immutable carryRight = (key[6] & 0x01) << 3;
|
|
||||||
|
|
||||||
key[6] = cast(ubyte) ((key[6] >> 1) | ((key[5] & 0x01) << 7));
|
|
||||||
key[5] = cast(ubyte) ((key[5] >> 1) | ((key[4] & 0x01) << 7));
|
|
||||||
key[4] = cast(ubyte) ((key[4] >> 1) | ((key[3] & 0x01) << 7));
|
|
||||||
|
|
||||||
immutable carryLeft = (key[3] & 0x10) << 3;
|
|
||||||
key[3] = cast(ubyte) ((((key[3] >> 1) | ((key[2] & 0x01) << 7)) & ~0x08) | carryRight);
|
|
||||||
|
|
||||||
key[2] = cast(ubyte) ((key[2] >> 1) | ((key[1] & 0x01) << 7));
|
|
||||||
key[1] = cast(ubyte) ((key[1] >> 1) | ((key[0] & 0x01) << 7));
|
|
||||||
key[0] = cast(ubyte) ((key[0] >> 1) | carryLeft);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void permute(in ubyte[] source, ubyte[] target, immutable(ubyte[]) permuteTable, size_t length)
|
|
||||||
const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
const sourceVector = const BitVector(source);
|
|
||||||
auto targetVector = BitVector(target);
|
|
||||||
|
|
||||||
permute(sourceVector, targetVector, permuteTable, length);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void permute(in BitVector source, ubyte[] target, immutable(ubyte[]) permuteTable, size_t length)
|
|
||||||
const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
auto targetVector = BitVector(target);
|
|
||||||
|
|
||||||
permute(source, targetVector, permuteTable, length);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void permute(in BitVector source, ref BitVector target, immutable(ubyte[]) permuteTable, size_t length)
|
|
||||||
const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
for (uint i; i < length * 8; ++i)
|
|
||||||
{
|
|
||||||
target[i] = source[permuteTable[i] - 1];
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
version (unittest)
|
|
||||||
{
|
|
||||||
import std.typecons;
|
|
||||||
|
|
||||||
/* Test vectors for DES. Source:
|
|
||||||
"Validating the Correctness of Hardware
|
|
||||||
Implementations of the NBS Data Encryption Standard"
|
|
||||||
NBS Special Publication 500-20, 1980. Appendix B */
|
|
||||||
|
|
||||||
// Initial and reverse Permutation and Expansion tests. Encrypt.
|
|
||||||
ubyte[8][64] desTestVectors1 = [
|
|
||||||
[0x95, 0xf8, 0xa5, 0xe5, 0xdd, 0x31, 0xd9, 0x00],
|
|
||||||
[0xdd, 0x7f, 0x12, 0x1c, 0xa5, 0x01, 0x56, 0x19],
|
|
||||||
[0x2e, 0x86, 0x53, 0x10, 0x4f, 0x38, 0x34, 0xea],
|
|
||||||
[0x4b, 0xd3, 0x88, 0xff, 0x6c, 0xd8, 0x1d, 0x4f],
|
|
||||||
[0x20, 0xb9, 0xe7, 0x67, 0xb2, 0xfb, 0x14, 0x56],
|
|
||||||
[0x55, 0x57, 0x93, 0x80, 0xd7, 0x71, 0x38, 0xef],
|
|
||||||
[0x6c, 0xc5, 0xde, 0xfa, 0xaf, 0x04, 0x51, 0x2f],
|
|
||||||
[0x0d, 0x9f, 0x27, 0x9b, 0xa5, 0xd8, 0x72, 0x60],
|
|
||||||
[0xd9, 0x03, 0x1b, 0x02, 0x71, 0xbd, 0x5a, 0x0a],
|
|
||||||
[0x42, 0x42, 0x50, 0xb3, 0x7c, 0x3d, 0xd9, 0x51],
|
|
||||||
[0xb8, 0x06, 0x1b, 0x7e, 0xcd, 0x9a, 0x21, 0xe5],
|
|
||||||
[0xf1, 0x5d, 0x0f, 0x28, 0x6b, 0x65, 0xbd, 0x28],
|
|
||||||
[0xad, 0xd0, 0xcc, 0x8d, 0x6e, 0x5d, 0xeb, 0xa1],
|
|
||||||
[0xe6, 0xd5, 0xf8, 0x27, 0x52, 0xad, 0x63, 0xd1],
|
|
||||||
[0xec, 0xbf, 0xe3, 0xbd, 0x3f, 0x59, 0x1a, 0x5e],
|
|
||||||
[0xf3, 0x56, 0x83, 0x43, 0x79, 0xd1, 0x65, 0xcd],
|
|
||||||
[0x2b, 0x9f, 0x98, 0x2f, 0x20, 0x03, 0x7f, 0xa9],
|
|
||||||
[0x88, 0x9d, 0xe0, 0x68, 0xa1, 0x6f, 0x0b, 0xe6],
|
|
||||||
[0xe1, 0x9e, 0x27, 0x5d, 0x84, 0x6a, 0x12, 0x98],
|
|
||||||
[0x32, 0x9a, 0x8e, 0xd5, 0x23, 0xd7, 0x1a, 0xec],
|
|
||||||
[0xe7, 0xfc, 0xe2, 0x25, 0x57, 0xd2, 0x3c, 0x97],
|
|
||||||
[0x12, 0xa9, 0xf5, 0x81, 0x7f, 0xf2, 0xd6, 0x5d],
|
|
||||||
[0xa4, 0x84, 0xc3, 0xad, 0x38, 0xdc, 0x9c, 0x19],
|
|
||||||
[0xfb, 0xe0, 0x0a, 0x8a, 0x1e, 0xf8, 0xad, 0x72],
|
|
||||||
[0x75, 0x0d, 0x07, 0x94, 0x07, 0x52, 0x13, 0x63],
|
|
||||||
[0x64, 0xfe, 0xed, 0x9c, 0x72, 0x4c, 0x2f, 0xaf],
|
|
||||||
[0xf0, 0x2b, 0x26, 0x3b, 0x32, 0x8e, 0x2b, 0x60],
|
|
||||||
[0x9d, 0x64, 0x55, 0x5a, 0x9a, 0x10, 0xb8, 0x52],
|
|
||||||
[0xd1, 0x06, 0xff, 0x0b, 0xed, 0x52, 0x55, 0xd7],
|
|
||||||
[0xe1, 0x65, 0x2c, 0x6b, 0x13, 0x8c, 0x64, 0xa5],
|
|
||||||
[0xe4, 0x28, 0x58, 0x11, 0x86, 0xec, 0x8f, 0x46],
|
|
||||||
[0xae, 0xb5, 0xf5, 0xed, 0xe2, 0x2d, 0x1a, 0x36],
|
|
||||||
[0xe9, 0x43, 0xd7, 0x56, 0x8a, 0xec, 0x0c, 0x5c],
|
|
||||||
[0xdf, 0x98, 0xc8, 0x27, 0x6f, 0x54, 0xb0, 0x4b],
|
|
||||||
[0xb1, 0x60, 0xe4, 0x68, 0x0f, 0x6c, 0x69, 0x6f],
|
|
||||||
[0xfa, 0x07, 0x52, 0xb0, 0x7d, 0x9c, 0x4a, 0xb8],
|
|
||||||
[0xca, 0x3a, 0x2b, 0x03, 0x6d, 0xbc, 0x85, 0x02],
|
|
||||||
[0x5e, 0x09, 0x05, 0x51, 0x7b, 0xb5, 0x9b, 0xcf],
|
|
||||||
[0x81, 0x4e, 0xeb, 0x3b, 0x91, 0xd9, 0x07, 0x26],
|
|
||||||
[0x4d, 0x49, 0xdb, 0x15, 0x32, 0x91, 0x9c, 0x9f],
|
|
||||||
[0x25, 0xeb, 0x5f, 0xc3, 0xf8, 0xcf, 0x06, 0x21],
|
|
||||||
[0xab, 0x6a, 0x20, 0xc0, 0x62, 0x0d, 0x1c, 0x6f],
|
|
||||||
[0x79, 0xe9, 0x0d, 0xbc, 0x98, 0xf9, 0x2c, 0xca],
|
|
||||||
[0x86, 0x6e, 0xce, 0xdd, 0x80, 0x72, 0xbb, 0x0e],
|
|
||||||
[0x8b, 0x54, 0x53, 0x6f, 0x2f, 0x3e, 0x64, 0xa8],
|
|
||||||
[0xea, 0x51, 0xd3, 0x97, 0x55, 0x95, 0xb8, 0x6b],
|
|
||||||
[0xca, 0xff, 0xc6, 0xac, 0x45, 0x42, 0xde, 0x31],
|
|
||||||
[0x8d, 0xd4, 0x5a, 0x2d, 0xdf, 0x90, 0x79, 0x6c],
|
|
||||||
[0x10, 0x29, 0xd5, 0x5e, 0x88, 0x0e, 0xc2, 0xd0],
|
|
||||||
[0x5d, 0x86, 0xcb, 0x23, 0x63, 0x9d, 0xbe, 0xa9],
|
|
||||||
[0x1d, 0x1c, 0xa8, 0x53, 0xae, 0x7c, 0x0c, 0x5f],
|
|
||||||
[0xce, 0x33, 0x23, 0x29, 0x24, 0x8f, 0x32, 0x28],
|
|
||||||
[0x84, 0x05, 0xd1, 0xab, 0xe2, 0x4f, 0xb9, 0x42],
|
|
||||||
[0xe6, 0x43, 0xd7, 0x80, 0x90, 0xca, 0x42, 0x07],
|
|
||||||
[0x48, 0x22, 0x1b, 0x99, 0x37, 0x74, 0x8a, 0x23],
|
|
||||||
[0xdd, 0x7c, 0x0b, 0xbd, 0x61, 0xfa, 0xfd, 0x54],
|
|
||||||
[0x2f, 0xbc, 0x29, 0x1a, 0x57, 0x0d, 0xb5, 0xc4],
|
|
||||||
[0xe0, 0x7c, 0x30, 0xd7, 0xe4, 0xe2, 0x6e, 0x12],
|
|
||||||
[0x09, 0x53, 0xe2, 0x25, 0x8e, 0x8e, 0x90, 0xa1],
|
|
||||||
[0x5b, 0x71, 0x1b, 0xc4, 0xce, 0xeb, 0xf2, 0xee],
|
|
||||||
[0xcc, 0x08, 0x3f, 0x1e, 0x6d, 0x9e, 0x85, 0xf6],
|
|
||||||
[0xd2, 0xfd, 0x88, 0x67, 0xd5, 0x0d, 0x2d, 0xfe],
|
|
||||||
[0x06, 0xe7, 0xea, 0x22, 0xce, 0x92, 0x70, 0x8f],
|
|
||||||
[0x16, 0x6b, 0x40, 0xb4, 0x4a, 0xba, 0x4b, 0xd6],
|
|
||||||
];
|
|
||||||
|
|
||||||
// Key Permutation test. Encrypt.
|
|
||||||
// Test of right-shifts. Decrypt.
|
|
||||||
ubyte[8][56] desTestVectors2 = [
|
|
||||||
[0x95, 0xa8, 0xd7, 0x28, 0x13, 0xda, 0xa9, 0x4d],
|
|
||||||
[0x0e, 0xec, 0x14, 0x87, 0xdd, 0x8c, 0x26, 0xd5],
|
|
||||||
[0x7a, 0xd1, 0x6f, 0xfb, 0x79, 0xc4, 0x59, 0x26],
|
|
||||||
[0xd3, 0x74, 0x62, 0x94, 0xca, 0x6a, 0x6c, 0xf3],
|
|
||||||
[0x80, 0x9f, 0x5f, 0x87, 0x3c, 0x1f, 0xd7, 0x61],
|
|
||||||
[0xc0, 0x2f, 0xaf, 0xfe, 0xc9, 0x89, 0xd1, 0xfc],
|
|
||||||
[0x46, 0x15, 0xaa, 0x1d, 0x33, 0xe7, 0x2f, 0x10],
|
|
||||||
[0x20, 0x55, 0x12, 0x33, 0x50, 0xc0, 0x08, 0x58],
|
|
||||||
[0xdf, 0x3b, 0x99, 0xd6, 0x57, 0x73, 0x97, 0xc8],
|
|
||||||
[0x31, 0xfe, 0x17, 0x36, 0x9b, 0x52, 0x88, 0xc9],
|
|
||||||
[0xdf, 0xdd, 0x3c, 0xc6, 0x4d, 0xae, 0x16, 0x42],
|
|
||||||
[0x17, 0x8c, 0x83, 0xce, 0x2b, 0x39, 0x9d, 0x94],
|
|
||||||
[0x50, 0xf6, 0x36, 0x32, 0x4a, 0x9b, 0x7f, 0x80],
|
|
||||||
[0xa8, 0x46, 0x8e, 0xe3, 0xbc, 0x18, 0xf0, 0x6d],
|
|
||||||
[0xa2, 0xdc, 0x9e, 0x92, 0xfd, 0x3c, 0xde, 0x92],
|
|
||||||
[0xca, 0xc0, 0x9f, 0x79, 0x7d, 0x03, 0x12, 0x87],
|
|
||||||
[0x90, 0xba, 0x68, 0x0b, 0x22, 0xae, 0xb5, 0x25],
|
|
||||||
[0xce, 0x7a, 0x24, 0xf3, 0x50, 0xe2, 0x80, 0xb6],
|
|
||||||
[0x88, 0x2b, 0xff, 0x0a, 0xa0, 0x1a, 0x0b, 0x87],
|
|
||||||
[0x25, 0x61, 0x02, 0x88, 0x92, 0x45, 0x11, 0xc2],
|
|
||||||
[0xc7, 0x15, 0x16, 0xc2, 0x9c, 0x75, 0xd1, 0x70],
|
|
||||||
[0x51, 0x99, 0xc2, 0x9a, 0x52, 0xc9, 0xf0, 0x59],
|
|
||||||
[0xc2, 0x2f, 0x0a, 0x29, 0x4a, 0x71, 0xf2, 0x9f],
|
|
||||||
[0xee, 0x37, 0x14, 0x83, 0x71, 0x4c, 0x02, 0xea],
|
|
||||||
[0xa8, 0x1f, 0xbd, 0x44, 0x8f, 0x9e, 0x52, 0x2f],
|
|
||||||
[0x4f, 0x64, 0x4c, 0x92, 0xe1, 0x92, 0xdf, 0xed],
|
|
||||||
[0x1a, 0xfa, 0x9a, 0x66, 0xa6, 0xdf, 0x92, 0xae],
|
|
||||||
[0xb3, 0xc1, 0xcc, 0x71, 0x5c, 0xb8, 0x79, 0xd8],
|
|
||||||
[0x19, 0xd0, 0x32, 0xe6, 0x4a, 0xb0, 0xbd, 0x8b],
|
|
||||||
[0x3c, 0xfa, 0xa7, 0xa7, 0xdc, 0x87, 0x20, 0xdc],
|
|
||||||
[0xb7, 0x26, 0x5f, 0x7f, 0x44, 0x7a, 0xc6, 0xf3],
|
|
||||||
[0x9d, 0xb7, 0x3b, 0x3c, 0x0d, 0x16, 0x3f, 0x54],
|
|
||||||
[0x81, 0x81, 0xb6, 0x5b, 0xab, 0xf4, 0xa9, 0x75],
|
|
||||||
[0x93, 0xc9, 0xb6, 0x40, 0x42, 0xea, 0xa2, 0x40],
|
|
||||||
[0x55, 0x70, 0x53, 0x08, 0x29, 0x70, 0x55, 0x92],
|
|
||||||
[0x86, 0x38, 0x80, 0x9e, 0x87, 0x87, 0x87, 0xa0],
|
|
||||||
[0x41, 0xb9, 0xa7, 0x9a, 0xf7, 0x9a, 0xc2, 0x08],
|
|
||||||
[0x7a, 0x9b, 0xe4, 0x2f, 0x20, 0x09, 0xa8, 0x92],
|
|
||||||
[0x29, 0x03, 0x8d, 0x56, 0xba, 0x6d, 0x27, 0x45],
|
|
||||||
[0x54, 0x95, 0xc6, 0xab, 0xf1, 0xe5, 0xdf, 0x51],
|
|
||||||
[0xae, 0x13, 0xdb, 0xd5, 0x61, 0x48, 0x89, 0x33],
|
|
||||||
[0x02, 0x4d, 0x1f, 0xfa, 0x89, 0x04, 0xe3, 0x89],
|
|
||||||
[0xd1, 0x39, 0x97, 0x12, 0xf9, 0x9b, 0xf0, 0x2e],
|
|
||||||
[0x14, 0xc1, 0xd7, 0xc1, 0xcf, 0xfe, 0xc7, 0x9e],
|
|
||||||
[0x1d, 0xe5, 0x27, 0x9d, 0xae, 0x3b, 0xed, 0x6f],
|
|
||||||
[0xe9, 0x41, 0xa3, 0x3f, 0x85, 0x50, 0x13, 0x03],
|
|
||||||
[0xda, 0x99, 0xdb, 0xbc, 0x9a, 0x03, 0xf3, 0x79],
|
|
||||||
[0xb7, 0xfc, 0x92, 0xf9, 0x1d, 0x8e, 0x92, 0xe9],
|
|
||||||
[0xae, 0x8e, 0x5c, 0xaa, 0x3c, 0xa0, 0x4e, 0x85],
|
|
||||||
[0x9c, 0xc6, 0x2d, 0xf4, 0x3b, 0x6e, 0xed, 0x74],
|
|
||||||
[0xd8, 0x63, 0xdb, 0xb5, 0xc5, 0x9a, 0x91, 0xa0],
|
|
||||||
[0xa1, 0xab, 0x21, 0x90, 0x54, 0x5b, 0x91, 0xd7],
|
|
||||||
[0x08, 0x75, 0x04, 0x1e, 0x64, 0xc5, 0x70, 0xf7],
|
|
||||||
[0x5a, 0x59, 0x45, 0x28, 0xbe, 0xbe, 0xf1, 0xcc],
|
|
||||||
[0xfc, 0xdb, 0x32, 0x91, 0xde, 0x21, 0xf0, 0xc0],
|
|
||||||
[0x86, 0x9e, 0xfd, 0x7f, 0x9f, 0x26, 0x5a, 0x09],
|
|
||||||
];
|
|
||||||
|
|
||||||
// Data permutation test. Encrypt.
|
|
||||||
ubyte[8][2][32] desTestVectors3 = [
|
|
||||||
[[0x10, 0x46, 0x91, 0x34, 0x89, 0x98, 0x01, 0x31], [0x88, 0xd5, 0x5e, 0x54, 0xf5, 0x4c, 0x97, 0xb4]],
|
|
||||||
[[0x10, 0x07, 0x10, 0x34, 0x89, 0x98, 0x80, 0x20], [0x0c, 0x0c, 0xc0, 0x0c, 0x83, 0xea, 0x48, 0xfd]],
|
|
||||||
[[0x10, 0x07, 0x10, 0x34, 0xc8, 0x98, 0x01, 0x20], [0x83, 0xbc, 0x8e, 0xf3, 0xa6, 0x57, 0x01, 0x83]],
|
|
||||||
[[0x10, 0x46, 0x10, 0x34, 0x89, 0x98, 0x80, 0x20], [0xdf, 0x72, 0x5d, 0xca, 0xd9, 0x4e, 0xa2, 0xe9]],
|
|
||||||
[[0x10, 0x86, 0x91, 0x15, 0x19, 0x19, 0x01, 0x01], [0xe6, 0x52, 0xb5, 0x3b, 0x55, 0x0b, 0xe8, 0xb0]],
|
|
||||||
[[0x10, 0x86, 0x91, 0x15, 0x19, 0x58, 0x01, 0x01], [0xaf, 0x52, 0x71, 0x20, 0xc4, 0x85, 0xcb, 0xb0]],
|
|
||||||
[[0x51, 0x07, 0xb0, 0x15, 0x19, 0x58, 0x01, 0x01], [0x0f, 0x04, 0xce, 0x39, 0x3d, 0xb9, 0x26, 0xd5]],
|
|
||||||
[[0x10, 0x07, 0xb0, 0x15, 0x19, 0x19, 0x01, 0x01], [0xc9, 0xf0, 0x0f, 0xfc, 0x74, 0x07, 0x90, 0x67]],
|
|
||||||
[[0x31, 0x07, 0x91, 0x54, 0x98, 0x08, 0x01, 0x01], [0x7c, 0xfd, 0x82, 0xa5, 0x93, 0x25, 0x2b, 0x4e]],
|
|
||||||
[[0x31, 0x07, 0x91, 0x94, 0x98, 0x08, 0x01, 0x01], [0xcb, 0x49, 0xa2, 0xf9, 0xe9, 0x13, 0x63, 0xe3]],
|
|
||||||
[[0x10, 0x07, 0x91, 0x15, 0xb9, 0x08, 0x01, 0x40], [0x00, 0xb5, 0x88, 0xbe, 0x70, 0xd2, 0x3f, 0x56]],
|
|
||||||
[[0x31, 0x07, 0x91, 0x15, 0x98, 0x08, 0x01, 0x40], [0x40, 0x6a, 0x9a, 0x6a, 0xb4, 0x33, 0x99, 0xae]],
|
|
||||||
[[0x10, 0x07, 0xd0, 0x15, 0x89, 0x98, 0x01, 0x01], [0x6c, 0xb7, 0x73, 0x61, 0x1d, 0xca, 0x9a, 0xda]],
|
|
||||||
[[0x91, 0x07, 0x91, 0x15, 0x89, 0x98, 0x01, 0x01], [0x67, 0xfd, 0x21, 0xc1, 0x7d, 0xbb, 0x5d, 0x70]],
|
|
||||||
[[0x91, 0x07, 0xd0, 0x15, 0x89, 0x19, 0x01, 0x01], [0x95, 0x92, 0xcb, 0x41, 0x10, 0x43, 0x07, 0x87]],
|
|
||||||
[[0x10, 0x07, 0xd0, 0x15, 0x98, 0x98, 0x01, 0x20], [0xa6, 0xb7, 0xff, 0x68, 0xa3, 0x18, 0xdd, 0xd3]],
|
|
||||||
[[0x10, 0x07, 0x94, 0x04, 0x98, 0x19, 0x01, 0x01], [0x4d, 0x10, 0x21, 0x96, 0xc9, 0x14, 0xca, 0x16]],
|
|
||||||
[[0x01, 0x07, 0x91, 0x04, 0x91, 0x19, 0x04, 0x01], [0x2d, 0xfa, 0x9f, 0x45, 0x73, 0x59, 0x49, 0x65]],
|
|
||||||
[[0x01, 0x07, 0x91, 0x04, 0x91, 0x19, 0x01, 0x01], [0xb4, 0x66, 0x04, 0x81, 0x6c, 0x0e, 0x07, 0x74]],
|
|
||||||
[[0x01, 0x07, 0x94, 0x04, 0x91, 0x19, 0x04, 0x01], [0x6e, 0x7e, 0x62, 0x21, 0xa4, 0xf3, 0x4e, 0x87]],
|
|
||||||
[[0x19, 0x07, 0x92, 0x10, 0x98, 0x1a, 0x01, 0x01], [0xaa, 0x85, 0xe7, 0x46, 0x43, 0x23, 0x31, 0x99]],
|
|
||||||
[[0x10, 0x07, 0x91, 0x19, 0x98, 0x19, 0x08, 0x01], [0x2e, 0x5a, 0x19, 0xdb, 0x4d, 0x19, 0x62, 0xd6]],
|
|
||||||
[[0x10, 0x07, 0x91, 0x19, 0x98, 0x1a, 0x08, 0x01], [0x23, 0xa8, 0x66, 0xa8, 0x09, 0xd3, 0x08, 0x94]],
|
|
||||||
[[0x10, 0x07, 0x92, 0x10, 0x98, 0x19, 0x01, 0x01], [0xd8, 0x12, 0xd9, 0x61, 0xf0, 0x17, 0xd3, 0x20]],
|
|
||||||
[[0x10, 0x07, 0x91, 0x15, 0x98, 0x19, 0x01, 0x0b], [0x05, 0x56, 0x05, 0x81, 0x6e, 0x58, 0x60, 0x8f]],
|
|
||||||
[[0x10, 0x04, 0x80, 0x15, 0x98, 0x19, 0x01, 0x01], [0xab, 0xd8, 0x8e, 0x8b, 0x1b, 0x77, 0x16, 0xf1]],
|
|
||||||
[[0x10, 0x04, 0x80, 0x15, 0x98, 0x19, 0x01, 0x02], [0x53, 0x7a, 0xc9, 0x5b, 0xe6, 0x9d, 0xa1, 0xe1]],
|
|
||||||
[[0x10, 0x04, 0x80, 0x15, 0x98, 0x19, 0x01, 0x08], [0xae, 0xd0, 0xf6, 0xae, 0x3c, 0x25, 0xcd, 0xd8]],
|
|
||||||
[[0x10, 0x02, 0x91, 0x14, 0x98, 0x10, 0x01, 0x04], [0xb3, 0xe3, 0x5a, 0x5e, 0xe5, 0x3e, 0x7b, 0x8d]],
|
|
||||||
[[0x10, 0x02, 0x91, 0x15, 0x98, 0x19, 0x01, 0x04], [0x61, 0xc7, 0x9c, 0x71, 0x92, 0x1a, 0x2e, 0xf8]],
|
|
||||||
[[0x10, 0x02, 0x91, 0x15, 0x98, 0x10, 0x02, 0x01], [0xe2, 0xf5, 0x72, 0x8f, 0x09, 0x95, 0x01, 0x3c]],
|
|
||||||
[[0x10, 0x02, 0x91, 0x16, 0x98, 0x10, 0x01, 0x01], [0x1a, 0xea, 0xc3, 0x9a, 0x61, 0xf0, 0xa4, 0x64]],
|
|
||||||
];
|
|
||||||
|
|
||||||
// S-Box test. Encrypt.
|
|
||||||
ubyte[8][3][19] desTestVectors4 = [
|
|
||||||
[[0x7c, 0xa1, 0x10, 0x45, 0x4a, 0x1a, 0x6e, 0x57], [0x01, 0xa1, 0xd6, 0xd0, 0x39, 0x77, 0x67, 0x42],
|
|
||||||
[0x69, 0x0f, 0x5b, 0x0d, 0x9a, 0x26, 0x93, 0x9b]],
|
|
||||||
[[0x01, 0x31, 0xd9, 0x61, 0x9d, 0xc1, 0x37, 0x6e], [0x5c, 0xd5, 0x4c, 0xa8, 0x3d, 0xef, 0x57, 0xda],
|
|
||||||
[0x7a, 0x38, 0x9d, 0x10, 0x35, 0x4b, 0xd2, 0x71]],
|
|
||||||
[[0x07, 0xa1, 0x13, 0x3e, 0x4a, 0x0b, 0x26, 0x86], [0x02, 0x48, 0xd4, 0x38, 0x06, 0xf6, 0x71, 0x72],
|
|
||||||
[0x86, 0x8e, 0xbb, 0x51, 0xca, 0xb4, 0x59, 0x9a]],
|
|
||||||
[[0x38, 0x49, 0x67, 0x4c, 0x26, 0x02, 0x31, 0x9e], [0x51, 0x45, 0x4b, 0x58, 0x2d, 0xdf, 0x44, 0x0a],
|
|
||||||
[0x71, 0x78, 0x87, 0x6e, 0x01, 0xf1, 0x9b, 0x2a]],
|
|
||||||
[[0x04, 0xb9, 0x15, 0xba, 0x43, 0xfe, 0xb5, 0xb6], [0x42, 0xfd, 0x44, 0x30, 0x59, 0x57, 0x7f, 0xa2],
|
|
||||||
[0xaf, 0x37, 0xfb, 0x42, 0x1f, 0x8c, 0x40, 0x95]],
|
|
||||||
[[0x01, 0x13, 0xb9, 0x70, 0xfd, 0x34, 0xf2, 0xce], [0x05, 0x9b, 0x5e, 0x08, 0x51, 0xcf, 0x14, 0x3a],
|
|
||||||
[0x86, 0xa5, 0x60, 0xf1, 0x0e, 0xc6, 0xd8, 0x5b]],
|
|
||||||
[[0x01, 0x70, 0xf1, 0x75, 0x46, 0x8f, 0xb5, 0xe6], [0x07, 0x56, 0xd8, 0xe0, 0x77, 0x47, 0x61, 0xd2],
|
|
||||||
[0x0c, 0xd3, 0xda, 0x02, 0x00, 0x21, 0xdc, 0x09]],
|
|
||||||
[[0x43, 0x29, 0x7f, 0xad, 0x38, 0xe3, 0x73, 0xfe], [0x76, 0x25, 0x14, 0xb8, 0x29, 0xbf, 0x48, 0x6a],
|
|
||||||
[0xea, 0x67, 0x6b, 0x2c, 0xb7, 0xdb, 0x2b, 0x7a]],
|
|
||||||
[[0x07, 0xa7, 0x13, 0x70, 0x45, 0xda, 0x2a, 0x16], [0x3b, 0xdd, 0x11, 0x90, 0x49, 0x37, 0x28, 0x02],
|
|
||||||
[0xdf, 0xd6, 0x4a, 0x81, 0x5c, 0xaf, 0x1a, 0x0f]],
|
|
||||||
[[0x04, 0x68, 0x91, 0x04, 0xc2, 0xfd, 0x3b, 0x2f], [0x26, 0x95, 0x5f, 0x68, 0x35, 0xaf, 0x60, 0x9a],
|
|
||||||
[0x5c, 0x51, 0x3c, 0x9c, 0x48, 0x86, 0xc0, 0x88]],
|
|
||||||
[[0x37, 0xd0, 0x6b, 0xb5, 0x16, 0xcb, 0x75, 0x46], [0x16, 0x4d, 0x5e, 0x40, 0x4f, 0x27, 0x52, 0x32],
|
|
||||||
[0x0a, 0x2a, 0xee, 0xae, 0x3f, 0xf4, 0xab, 0x77]],
|
|
||||||
[[0x1f, 0x08, 0x26, 0x0d, 0x1a, 0xc2, 0x46, 0x5e], [0x6b, 0x05, 0x6e, 0x18, 0x75, 0x9f, 0x5c, 0xca],
|
|
||||||
[0xef, 0x1b, 0xf0, 0x3e, 0x5d, 0xfa, 0x57, 0x5a]],
|
|
||||||
[[0x58, 0x40, 0x23, 0x64, 0x1a, 0xba, 0x61, 0x76], [0x00, 0x4b, 0xd6, 0xef, 0x09, 0x17, 0x60, 0x62],
|
|
||||||
[0x88, 0xbf, 0x0d, 0xb6, 0xd7, 0x0d, 0xee, 0x56]],
|
|
||||||
[[0x02, 0x58, 0x16, 0x16, 0x46, 0x29, 0xb0, 0x07], [0x48, 0x0d, 0x39, 0x00, 0x6e, 0xe7, 0x62, 0xf2],
|
|
||||||
[0xa1, 0xf9, 0x91, 0x55, 0x41, 0x02, 0x0b, 0x56]],
|
|
||||||
[[0x49, 0x79, 0x3e, 0xbc, 0x79, 0xb3, 0x25, 0x8f], [0x43, 0x75, 0x40, 0xc8, 0x69, 0x8f, 0x3c, 0xfa],
|
|
||||||
[0x6f, 0xbf, 0x1c, 0xaf, 0xcf, 0xfd, 0x05, 0x56]],
|
|
||||||
[[0x4f, 0xb0, 0x5e, 0x15, 0x15, 0xab, 0x73, 0xa7], [0x07, 0x2d, 0x43, 0xa0, 0x77, 0x07, 0x52, 0x92],
|
|
||||||
[0x2f, 0x22, 0xe4, 0x9b, 0xab, 0x7c, 0xa1, 0xac]],
|
|
||||||
[[0x49, 0xe9, 0x5d, 0x6d, 0x4c, 0xa2, 0x29, 0xbf], [0x02, 0xfe, 0x55, 0x77, 0x81, 0x17, 0xf1, 0x2a],
|
|
||||||
[0x5a, 0x6b, 0x61, 0x2c, 0xc2, 0x6c, 0xce, 0x4a]],
|
|
||||||
[[0x01, 0x83, 0x10, 0xdc, 0x40, 0x9b, 0x26, 0xd6], [0x1d, 0x9d, 0x5c, 0x50, 0x18, 0xf7, 0x28, 0xc2],
|
|
||||||
[0x5f, 0x4c, 0x03, 0x8e, 0xd1, 0x2b, 0x2e, 0x41]],
|
|
||||||
[[0x1c, 0x58, 0x7f, 0x1c, 0x13, 0x92, 0x4f, 0xef], [0x30, 0x55, 0x32, 0x28, 0x6d, 0x6f, 0x29, 0x5a],
|
|
||||||
[0x63, 0xfa, 0xc0, 0xd0, 0x34, 0xd9, 0xf7, 0x93]],
|
|
||||||
];
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
auto des = scoped!(DES!1);
|
|
||||||
ubyte[8] key = [0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01];
|
|
||||||
ubyte[8] plain = [0x80, 0, 0, 0, 0, 0, 0, 0];
|
|
||||||
ubyte[8] cipher;
|
|
||||||
|
|
||||||
des.key = key;
|
|
||||||
foreach (ubyte i; 0..64)
|
|
||||||
{
|
|
||||||
if (i != 0)
|
|
||||||
{
|
|
||||||
plain[i / 8] = i % 8 ? plain[i / 8] >> 1 : 0x80;
|
|
||||||
if (i % 8 == 0)
|
|
||||||
{
|
|
||||||
plain[i / 8 - 1] = 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// Initial Permutation and Expansion test.
|
|
||||||
des.encrypt(plain, cipher);
|
|
||||||
assert(cipher == desTestVectors1[i]);
|
|
||||||
|
|
||||||
// Inverse Permutation and Expansion test.
|
|
||||||
des.encrypt(cipher, cipher);
|
|
||||||
assert(cipher == plain);
|
|
||||||
}
|
|
||||||
|
|
||||||
plain[0..$] = 0;
|
|
||||||
foreach (ubyte i; 0..56)
|
|
||||||
{
|
|
||||||
key[i / 7] = i % 7 ? key[i / 7] >> 1 : 0x80;
|
|
||||||
if (i % 7 == 0 && i != 0)
|
|
||||||
{
|
|
||||||
key[i / 7 - 1] = 0x01;
|
|
||||||
}
|
|
||||||
des.key = key;
|
|
||||||
|
|
||||||
// Initial Permutation and Expansion test.
|
|
||||||
des.encrypt(plain, cipher);
|
|
||||||
assert(cipher == desTestVectors2[i]);
|
|
||||||
|
|
||||||
// Test of right-shifts in Decryption.
|
|
||||||
des.decrypt(desTestVectors2[i], cipher);
|
|
||||||
assert(cipher == plain);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Data permutation test.
|
|
||||||
plain[0..$] = 0;
|
|
||||||
foreach (i; desTestVectors3)
|
|
||||||
{
|
|
||||||
des.key = i[0];
|
|
||||||
des.encrypt(plain, cipher);
|
|
||||||
assert(cipher == i[1]);
|
|
||||||
}
|
|
||||||
|
|
||||||
// S-Box test.
|
|
||||||
foreach (i; desTestVectors4)
|
|
||||||
{
|
|
||||||
des.key = i[0];
|
|
||||||
des.encrypt(i[1], cipher);
|
|
||||||
assert(cipher == i[2]);
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,279 +0,0 @@
|
|||||||
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Block cipher modes of operation.
|
|
||||||
*
|
|
||||||
* Copyright: Eugene Wissner 2016.
|
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
|
||||||
*/
|
|
||||||
module tanya.crypto.mode;
|
|
||||||
|
|
||||||
import tanya.memory;
|
|
||||||
import std.algorithm.iteration;
|
|
||||||
import std.typecons;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Supported padding mode.
|
|
||||||
*
|
|
||||||
* See_Also:
|
|
||||||
* $(D_PSYMBOL pad)
|
|
||||||
*/
|
|
||||||
enum PaddingMode
|
|
||||||
{
|
|
||||||
zero,
|
|
||||||
pkcs7,
|
|
||||||
ansiX923,
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* input = Sequence that should be padded.
|
|
||||||
* mode = Padding mode.
|
|
||||||
* blockSize = Block size.
|
|
||||||
* allocator = Allocator was used to allocate $(D_PARAM input).
|
|
||||||
*
|
|
||||||
* Returns: The function modifies the initial array and returns it.
|
|
||||||
*
|
|
||||||
* See_Also:
|
|
||||||
* $(D_PSYMBOL PaddingMode)
|
|
||||||
*/
|
|
||||||
ubyte[] pad(ref ubyte[] input,
|
|
||||||
in PaddingMode mode,
|
|
||||||
in ushort blockSize,
|
|
||||||
shared Allocator allocator = defaultAllocator)
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(blockSize > 0 && blockSize <= 256);
|
|
||||||
assert(blockSize % 64 == 0);
|
|
||||||
assert(input.length > 0);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
immutable rest = cast(ubyte) input.length % blockSize;
|
|
||||||
immutable size_t lastBlock = input.length - (rest > 0 ? rest : blockSize);
|
|
||||||
immutable needed = cast(ubyte) (rest > 0 ? blockSize - rest : 0);
|
|
||||||
|
|
||||||
final switch (mode) with (PaddingMode)
|
|
||||||
{
|
|
||||||
case zero:
|
|
||||||
allocator.resizeArray(input, input.length + needed);
|
|
||||||
break;
|
|
||||||
case pkcs7:
|
|
||||||
if (needed)
|
|
||||||
{
|
|
||||||
allocator.resizeArray(input, input.length + needed);
|
|
||||||
input[input.length - needed ..$].each!((ref e) => e = needed);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
allocator.resizeArray(input, input.length + blockSize);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
case ansiX923:
|
|
||||||
allocator.resizeArray(input, input.length + (needed ? needed : blockSize));
|
|
||||||
input[$ - 1] = needed;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
return input;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
{ // Zeros
|
|
||||||
auto input = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
|
|
||||||
pad(input, PaddingMode.zero, 64);
|
|
||||||
assert(input.length == 64);
|
|
||||||
|
|
||||||
pad(input, PaddingMode.zero, 64);
|
|
||||||
assert(input.length == 64);
|
|
||||||
assert(input[63] == 0);
|
|
||||||
|
|
||||||
defaultAllocator.dispose(input);
|
|
||||||
}
|
|
||||||
{ // PKCS#7
|
|
||||||
auto input = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
for (ubyte i; i < 40; ++i)
|
|
||||||
{
|
|
||||||
input[i] = i;
|
|
||||||
}
|
|
||||||
|
|
||||||
pad(input, PaddingMode.pkcs7, 64);
|
|
||||||
assert(input.length == 64);
|
|
||||||
for (ubyte i; i < 64; ++i)
|
|
||||||
{
|
|
||||||
if (i >= 40 && i < 50)
|
|
||||||
{
|
|
||||||
assert(input[i] == 0);
|
|
||||||
}
|
|
||||||
else if (i >= 50)
|
|
||||||
{
|
|
||||||
assert(input[i] == 14);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
assert(input[i] == i);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pad(input, PaddingMode.pkcs7, 64);
|
|
||||||
assert(input.length == 128);
|
|
||||||
for (ubyte i; i < 128; ++i)
|
|
||||||
{
|
|
||||||
if (i >= 64 || (i >= 40 && i < 50))
|
|
||||||
{
|
|
||||||
assert(input[i] == 0);
|
|
||||||
}
|
|
||||||
else if (i >= 50 && i < 64)
|
|
||||||
{
|
|
||||||
assert(input[i] == 14);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
assert(input[i] == i);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
defaultAllocator.dispose(input);
|
|
||||||
}
|
|
||||||
{ // ANSI X.923
|
|
||||||
auto input = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
for (ubyte i; i < 40; ++i)
|
|
||||||
{
|
|
||||||
input[i] = i;
|
|
||||||
}
|
|
||||||
|
|
||||||
pad(input, PaddingMode.ansiX923, 64);
|
|
||||||
assert(input.length == 64);
|
|
||||||
for (ubyte i; i < 64; ++i)
|
|
||||||
{
|
|
||||||
if (i < 40)
|
|
||||||
{
|
|
||||||
assert(input[i] == i);
|
|
||||||
}
|
|
||||||
else if (i == 63)
|
|
||||||
{
|
|
||||||
assert(input[i] == 14);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
assert(input[i] == 0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pad(input, PaddingMode.pkcs7, 64);
|
|
||||||
assert(input.length == 128);
|
|
||||||
for (ubyte i = 0; i < 128; ++i)
|
|
||||||
{
|
|
||||||
if (i < 40)
|
|
||||||
{
|
|
||||||
assert(input[i] == i);
|
|
||||||
}
|
|
||||||
else if (i == 63)
|
|
||||||
{
|
|
||||||
assert(input[i] == 14);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
assert(input[i] == 0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
defaultAllocator.dispose(input);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* input = Sequence that should be padded.
|
|
||||||
* mode = Padding mode.
|
|
||||||
* blockSize = Block size.
|
|
||||||
* allocator = Allocator was used to allocate $(D_PARAM input).
|
|
||||||
*
|
|
||||||
* Returns: The function modifies the initial array and returns it.
|
|
||||||
*
|
|
||||||
* See_Also:
|
|
||||||
* $(D_PSYMBOL pad)
|
|
||||||
*/
|
|
||||||
ref ubyte[] unpad(ref ubyte[] input,
|
|
||||||
in PaddingMode mode,
|
|
||||||
in ushort blockSize,
|
|
||||||
shared Allocator allocator = defaultAllocator)
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(input.length != 0);
|
|
||||||
assert(input.length % 64 == 0);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
final switch (mode) with (PaddingMode)
|
|
||||||
{
|
|
||||||
case zero:
|
|
||||||
break;
|
|
||||||
case pkcs7:
|
|
||||||
case ansiX923:
|
|
||||||
immutable last = input[$ - 1];
|
|
||||||
|
|
||||||
allocator.resizeArray(input, input.length - (last ? last : blockSize));
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
return input;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
{ // Zeros
|
|
||||||
auto input = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
auto inputDup = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
|
|
||||||
pad(input, PaddingMode.zero, 64);
|
|
||||||
pad(inputDup, PaddingMode.zero, 64);
|
|
||||||
|
|
||||||
unpad(input, PaddingMode.zero, 64);
|
|
||||||
assert(input == inputDup);
|
|
||||||
|
|
||||||
defaultAllocator.dispose(input);
|
|
||||||
defaultAllocator.dispose(inputDup);
|
|
||||||
|
|
||||||
}
|
|
||||||
{ // PKCS#7
|
|
||||||
auto input = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
auto inputDup = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
for (ubyte i; i < 40; ++i)
|
|
||||||
{
|
|
||||||
input[i] = i;
|
|
||||||
inputDup[i] = i;
|
|
||||||
}
|
|
||||||
|
|
||||||
pad(input, PaddingMode.pkcs7, 64);
|
|
||||||
unpad(input, PaddingMode.pkcs7, 64);
|
|
||||||
assert(input == inputDup);
|
|
||||||
|
|
||||||
defaultAllocator.dispose(input);
|
|
||||||
defaultAllocator.dispose(inputDup);
|
|
||||||
}
|
|
||||||
{ // ANSI X.923
|
|
||||||
auto input = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
auto inputDup = defaultAllocator.makeArray!ubyte(50);
|
|
||||||
for (ubyte i; i < 40; ++i)
|
|
||||||
{
|
|
||||||
input[i] = i;
|
|
||||||
inputDup[i] = i;
|
|
||||||
}
|
|
||||||
|
|
||||||
pad(input, PaddingMode.pkcs7, 64);
|
|
||||||
unpad(input, PaddingMode.pkcs7, 64);
|
|
||||||
assert(input == inputDup);
|
|
||||||
|
|
||||||
defaultAllocator.dispose(input);
|
|
||||||
defaultAllocator.dispose(inputDup);
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,16 +0,0 @@
|
|||||||
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Copyright: Eugene Wissner 2016.
|
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
|
||||||
*/
|
|
||||||
module tanya.crypto;
|
|
||||||
|
|
||||||
public import tanya.crypto.bit;
|
|
||||||
public import tanya.crypto.des;
|
|
||||||
public import tanya.crypto.mode;
|
|
||||||
public import tanya.crypto.symmetric;
|
|
@ -1,177 +0,0 @@
|
|||||||
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Interfaces for implementing secret key algorithms.
|
|
||||||
*
|
|
||||||
* Copyright: Eugene Wissner 2016.
|
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:belka@caraus.de, Eugene Wissner)
|
|
||||||
*/
|
|
||||||
module tanya.crypto.symmetric;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Implemented by secret key algorithms.
|
|
||||||
*/
|
|
||||||
interface SymmetricCipher
|
|
||||||
{
|
|
||||||
/**
|
|
||||||
* Returns: Key length.
|
|
||||||
*/
|
|
||||||
@property inout(uint) keyLength() inout const pure nothrow @safe @nogc;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Minimum key length.
|
|
||||||
*/
|
|
||||||
@property inout(uint) minKeyLength() inout const pure nothrow @safe @nogc;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Maximum key length.
|
|
||||||
*/
|
|
||||||
@property inout(uint) maxKeyLength() inout const pure nothrow @safe @nogc;
|
|
||||||
|
|
||||||
/// Cipher direction.
|
|
||||||
protected enum Direction : ushort
|
|
||||||
{
|
|
||||||
encryption,
|
|
||||||
decryption,
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* key = Key.
|
|
||||||
*/
|
|
||||||
@property void key(ubyte[] key) pure nothrow @safe @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(key.length >= minKeyLength);
|
|
||||||
assert(key.length <= maxKeyLength);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Implemented by block ciphers.
|
|
||||||
*/
|
|
||||||
interface BlockCipher : SymmetricCipher
|
|
||||||
{
|
|
||||||
/**
|
|
||||||
* Returns: Block size.
|
|
||||||
*/
|
|
||||||
@property inout(uint) blockSize() inout const pure nothrow @safe @nogc;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Encrypts a block.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* plain = Plain text, input.
|
|
||||||
* cipher = Cipher text, output.
|
|
||||||
*/
|
|
||||||
void encrypt(in ubyte[] plain, ubyte[] cipher)
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(plain.length == blockSize);
|
|
||||||
assert(cipher.length == blockSize);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Decrypts a block.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* cipher = Cipher text, input.
|
|
||||||
* plain = Plain text, output.
|
|
||||||
*/
|
|
||||||
void decrypt(in ubyte[] cipher, ubyte[] plain)
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(plain.length == blockSize);
|
|
||||||
assert(cipher.length == blockSize);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Mixed in by algorithms with fixed block size.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* N = Block size.
|
|
||||||
*/
|
|
||||||
mixin template FixedBlockSize(uint N)
|
|
||||||
if (N != 0)
|
|
||||||
{
|
|
||||||
private enum uint blockSize_ = N;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Fixed block size.
|
|
||||||
*/
|
|
||||||
final @property inout(uint) blockSize() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return blockSize_;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Mixed in by symmetric algorithms.
|
|
||||||
* If $(D_PARAM Min) equals $(D_PARAM Max) fixed key length is assumed.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* Min = Minimum key length.
|
|
||||||
* Max = Maximum key length.
|
|
||||||
*/
|
|
||||||
mixin template KeyLength(uint Min, uint Max = Min)
|
|
||||||
if (Min != 0 && Max != 0)
|
|
||||||
{
|
|
||||||
static if (Min == Max)
|
|
||||||
{
|
|
||||||
private enum uint keyLength_ = Min;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Key length.
|
|
||||||
*/
|
|
||||||
final @property inout(uint) keyLength() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return keyLength_;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Minimum key length.
|
|
||||||
*/
|
|
||||||
final @property inout(uint) minKeyLength() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return keyLength_;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Maximum key length.
|
|
||||||
*/
|
|
||||||
final @property inout(uint) maxKeyLength() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return keyLength_;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else static if (Min < Max)
|
|
||||||
{
|
|
||||||
private enum uint minKeyLength_ = Min;
|
|
||||||
private enum uint maxKeyLength_ = Max;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Minimum key length.
|
|
||||||
*/
|
|
||||||
final @property inout(uint) minKeyLength() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return minKeyLength_;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Maximum key length.
|
|
||||||
*/
|
|
||||||
final @property inout(uint) maxKeyLength() inout const pure nothrow @safe @nogc
|
|
||||||
{
|
|
||||||
return maxKeyLength_;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
static assert(false, "Max should be larger or equal to Min");
|
|
||||||
}
|
|
||||||
}
|
|
@ -3,7 +3,7 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
@ -52,7 +52,10 @@ interface Allocator
|
|||||||
bool reallocate(ref void[] p, in size_t size) shared nothrow @nogc;
|
bool reallocate(ref void[] p, in size_t size) shared nothrow @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Expands a memory block in place.
|
* Reallocates a memory block in place if possible or returns
|
||||||
|
* $(D_KEYWORD false). This function cannot be used to allocate or
|
||||||
|
* deallocate memory, so if $(D_PARAM p) is $(D_KEYWORD null) or
|
||||||
|
* $(D_PARAM size) is `0`, it should return $(D_KEYWORD false).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* p = A pointer to the memory block.
|
* p = A pointer to the memory block.
|
||||||
@ -60,68 +63,5 @@ interface Allocator
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if successful, $(D_KEYWORD false) otherwise.
|
* Returns: $(D_KEYWORD true) if successful, $(D_KEYWORD false) otherwise.
|
||||||
*/
|
*/
|
||||||
bool expand(ref void[] p, in size_t size) shared nothrow @nogc;
|
bool reallocateInPlace(ref void[] p, in size_t size) shared nothrow @nogc;
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* The mixin generates common methods for classes and structs using
|
|
||||||
* allocators. It provides a protected member, constructor and a read-only property,
|
|
||||||
* that checks if an allocator was already set and sets it to the default
|
|
||||||
* one, if not (useful for structs which don't have a default constructor).
|
|
||||||
*/
|
|
||||||
mixin template DefaultAllocator()
|
|
||||||
{
|
|
||||||
/// Allocator.
|
|
||||||
protected shared Allocator allocator_;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Params:
|
|
||||||
* allocator = The allocator should be used.
|
|
||||||
*/
|
|
||||||
this(shared Allocator allocator)
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(allocator !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
this.allocator_ = allocator;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* This property checks if the allocator was set in the constructor
|
|
||||||
* and sets it to the default one, if not.
|
|
||||||
*
|
|
||||||
* Returns: Used allocator.
|
|
||||||
*
|
|
||||||
* Postcondition: $(D_INLINECODE allocator_ !is null)
|
|
||||||
*/
|
|
||||||
protected @property shared(Allocator) allocator() nothrow @safe @nogc
|
|
||||||
out (allocator)
|
|
||||||
{
|
|
||||||
assert(allocator !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
if (allocator_ is null)
|
|
||||||
{
|
|
||||||
allocator_ = defaultAllocator;
|
|
||||||
}
|
|
||||||
return allocator_;
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Ditto.
|
|
||||||
@property shared(Allocator) allocator() const nothrow @trusted @nogc
|
|
||||||
out (allocator)
|
|
||||||
{
|
|
||||||
assert(allocator !is null);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
if (allocator_ is null)
|
|
||||||
{
|
|
||||||
return defaultAllocator;
|
|
||||||
}
|
|
||||||
return cast(shared Allocator) allocator_;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
@ -3,7 +3,7 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
@ -50,6 +50,21 @@ else version (Windows)
|
|||||||
*/
|
*/
|
||||||
final class MmapPool : Allocator
|
final class MmapPool : Allocator
|
||||||
{
|
{
|
||||||
|
invariant
|
||||||
|
{
|
||||||
|
for (auto r = &head; *r !is null; r = &((*r).next))
|
||||||
|
{
|
||||||
|
auto block = cast(Block) (cast(void*) *r + RegionEntry.sizeof);
|
||||||
|
do
|
||||||
|
{
|
||||||
|
assert(block.prev is null || block.prev.next is block);
|
||||||
|
assert(block.next is null || block.next.prev is block);
|
||||||
|
assert(block.region is *r);
|
||||||
|
}
|
||||||
|
while ((block = block.next) !is null);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Allocates $(D_PARAM size) bytes of memory.
|
* Allocates $(D_PARAM size) bytes of memory.
|
||||||
*
|
*
|
||||||
@ -156,7 +171,7 @@ final class MmapPool : Allocator
|
|||||||
*/
|
*/
|
||||||
bool deallocate(void[] p) shared nothrow @nogc
|
bool deallocate(void[] p) shared nothrow @nogc
|
||||||
{
|
{
|
||||||
if (p is null)
|
if (p.ptr is null)
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -216,7 +231,10 @@ final class MmapPool : Allocator
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Expands a memory block in place.
|
* Reallocates a memory block in place if possible or returns
|
||||||
|
* $(D_KEYWORD false). This function cannot be used to allocate or
|
||||||
|
* deallocate memory, so if $(D_PARAM p) is $(D_KEYWORD null) or
|
||||||
|
* $(D_PARAM size) is `0`, it should return $(D_KEYWORD false).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* p = A pointer to the memory block.
|
* p = A pointer to the memory block.
|
||||||
@ -224,16 +242,18 @@ final class MmapPool : Allocator
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if successful, $(D_KEYWORD false) otherwise.
|
* Returns: $(D_KEYWORD true) if successful, $(D_KEYWORD false) otherwise.
|
||||||
*/
|
*/
|
||||||
bool expand(ref void[] p, in size_t size) shared nothrow @nogc
|
bool reallocateInPlace(ref void[] p, in size_t size) shared nothrow @nogc
|
||||||
{
|
{
|
||||||
if (size <= p.length)
|
if (p is null || size == 0)
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
if (p is null)
|
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
if (size <= p.length)
|
||||||
|
{
|
||||||
|
// Leave the block as is.
|
||||||
|
p = p.ptr[0 .. size];
|
||||||
|
return true;
|
||||||
|
}
|
||||||
Block block1 = cast(Block) (p.ptr - BlockEntry.sizeof);
|
Block block1 = cast(Block) (p.ptr - BlockEntry.sizeof);
|
||||||
|
|
||||||
if (block1.size >= size)
|
if (block1.size >= size)
|
||||||
@ -243,7 +263,7 @@ final class MmapPool : Allocator
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
immutable dataSize = addAlignment(size);
|
immutable dataSize = addAlignment(size);
|
||||||
immutable delta = dataSize - p.length;
|
immutable delta = dataSize - addAlignment(p.length);
|
||||||
|
|
||||||
if (block1.next is null
|
if (block1.next is null
|
||||||
|| !block1.next.free
|
|| !block1.next.free
|
||||||
@ -255,21 +275,17 @@ final class MmapPool : Allocator
|
|||||||
}
|
}
|
||||||
if (block1.next.size >= delta + alignment_)
|
if (block1.next.size >= delta + alignment_)
|
||||||
{
|
{
|
||||||
// We should move the start position of the next block. The order may be
|
// Move size from block2 to block1.
|
||||||
// important because the old block and the new one can overlap.
|
block1.next.size = block1.next.size - delta;
|
||||||
auto block2 = cast(Block) (p.ptr + dataSize);
|
|
||||||
block2.size = block1.next.size - delta;
|
|
||||||
block2.free = true;
|
|
||||||
block2.region = block1.region;
|
|
||||||
block2.next = block1.next.next;
|
|
||||||
block2.prev = block1;
|
|
||||||
|
|
||||||
block1.size = block1.size + delta;
|
block1.size = block1.size + delta;
|
||||||
|
|
||||||
|
auto block2 = cast(Block) (p.ptr + dataSize);
|
||||||
if (block1.next.next !is null)
|
if (block1.next.next !is null)
|
||||||
{
|
{
|
||||||
block1.next.next.prev = block2;
|
block1.next.next.prev = block2;
|
||||||
}
|
}
|
||||||
|
// block1.next and block2 can overlap.
|
||||||
|
memmove(cast(void*) block2, cast(void*) block1.next, BlockEntry.sizeof);
|
||||||
block1.next = block2;
|
block1.next = block2;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
@ -287,22 +303,22 @@ final class MmapPool : Allocator
|
|||||||
nothrow unittest
|
nothrow unittest
|
||||||
{
|
{
|
||||||
void[] p;
|
void[] p;
|
||||||
assert(!MmapPool.instance.expand(p, 5));
|
assert(!MmapPool.instance.reallocateInPlace(p, 5));
|
||||||
assert(p is null);
|
assert(p is null);
|
||||||
|
|
||||||
p = MmapPool.instance.allocate(1);
|
p = MmapPool.instance.allocate(1);
|
||||||
auto orig = p.ptr;
|
auto orig = p.ptr;
|
||||||
|
|
||||||
assert(MmapPool.instance.expand(p, 2));
|
assert(MmapPool.instance.reallocateInPlace(p, 2));
|
||||||
assert(p.length == 2);
|
assert(p.length == 2);
|
||||||
assert(p.ptr == orig);
|
assert(p.ptr == orig);
|
||||||
|
|
||||||
assert(MmapPool.instance.expand(p, 4));
|
assert(MmapPool.instance.reallocateInPlace(p, 4));
|
||||||
assert(p.length == 4);
|
assert(p.length == 4);
|
||||||
assert(p.ptr == orig);
|
assert(p.ptr == orig);
|
||||||
|
|
||||||
assert(MmapPool.instance.expand(p, 2));
|
assert(MmapPool.instance.reallocateInPlace(p, 2));
|
||||||
assert(p.length == 4);
|
assert(p.length == 2);
|
||||||
assert(p.ptr == orig);
|
assert(p.ptr == orig);
|
||||||
|
|
||||||
MmapPool.instance.deallocate(p);
|
MmapPool.instance.deallocate(p);
|
||||||
@ -328,17 +344,12 @@ final class MmapPool : Allocator
|
|||||||
}
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
else if (size <= p.length)
|
else if (reallocateInPlace(p, size))
|
||||||
{
|
|
||||||
// Leave the block as is.
|
|
||||||
p = p.ptr[0 .. size];
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
else if (expand(p, size))
|
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
// Can't extend, allocate a new block, copy and delete the previous.
|
// Can't reallocate in place, allocate a new block,
|
||||||
|
// copy and delete the previous one.
|
||||||
void[] reallocP = allocate(size);
|
void[] reallocP = allocate(size);
|
||||||
if (reallocP is null)
|
if (reallocP is null)
|
||||||
{
|
{
|
||||||
@ -574,3 +585,61 @@ final class MmapPool : Allocator
|
|||||||
}
|
}
|
||||||
private alias Block = shared BlockEntry*;
|
private alias Block = shared BlockEntry*;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// A lot of allocations/deallocations, but it is the minimum caused a
|
||||||
|
// segmentation fault because MmapPool reallocateInPlace moves a block wrong.
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
auto a = MmapPool.instance.allocate(16);
|
||||||
|
auto d = MmapPool.instance.allocate(16);
|
||||||
|
auto b = MmapPool.instance.allocate(16);
|
||||||
|
auto e = MmapPool.instance.allocate(16);
|
||||||
|
auto c = MmapPool.instance.allocate(16);
|
||||||
|
auto f = MmapPool.instance.allocate(16);
|
||||||
|
|
||||||
|
MmapPool.instance.deallocate(a);
|
||||||
|
MmapPool.instance.deallocate(b);
|
||||||
|
MmapPool.instance.deallocate(c);
|
||||||
|
|
||||||
|
a = MmapPool.instance.allocate(50);
|
||||||
|
MmapPool.instance.reallocateInPlace(a, 64);
|
||||||
|
MmapPool.instance.deallocate(a);
|
||||||
|
|
||||||
|
a = MmapPool.instance.allocate(1);
|
||||||
|
auto tmp1 = MmapPool.instance.allocate(1);
|
||||||
|
auto h1 = MmapPool.instance.allocate(1);
|
||||||
|
auto tmp2 = cast(ubyte[]) MmapPool.instance.allocate(1);
|
||||||
|
|
||||||
|
auto h2 = MmapPool.instance.allocate(2);
|
||||||
|
tmp1 = MmapPool.instance.allocate(1);
|
||||||
|
MmapPool.instance.deallocate(h2);
|
||||||
|
MmapPool.instance.deallocate(h1);
|
||||||
|
|
||||||
|
h2 = MmapPool.instance.allocate(2);
|
||||||
|
h1 = MmapPool.instance.allocate(1);
|
||||||
|
MmapPool.instance.deallocate(h2);
|
||||||
|
|
||||||
|
auto rep = cast(void[]) tmp2;
|
||||||
|
MmapPool.instance.reallocate(rep, tmp1.length);
|
||||||
|
tmp2 = cast(ubyte[]) rep;
|
||||||
|
|
||||||
|
MmapPool.instance.reallocate(tmp1, 9);
|
||||||
|
|
||||||
|
rep = cast(void[]) tmp2;
|
||||||
|
MmapPool.instance.reallocate(rep, tmp1.length);
|
||||||
|
tmp2 = cast(ubyte[]) rep;
|
||||||
|
MmapPool.instance.reallocate(tmp1, 17);
|
||||||
|
|
||||||
|
tmp2[$ - 1] = 0;
|
||||||
|
|
||||||
|
MmapPool.instance.deallocate(tmp1);
|
||||||
|
|
||||||
|
b = MmapPool.instance.allocate(16);
|
||||||
|
|
||||||
|
MmapPool.instance.deallocate(h1);
|
||||||
|
MmapPool.instance.deallocate(a);
|
||||||
|
MmapPool.instance.deallocate(b);
|
||||||
|
MmapPool.instance.deallocate(d);
|
||||||
|
MmapPool.instance.deallocate(e);
|
||||||
|
MmapPool.instance.deallocate(f);
|
||||||
|
}
|
||||||
|
@ -3,7 +3,7 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copyright: Eugene Wissner 2016.
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
||||||
* Mozilla Public License, v. 2.0).
|
* Mozilla Public License, v. 2.0).
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
||||||
@ -14,7 +14,69 @@ import core.exception;
|
|||||||
public import std.experimental.allocator : make, makeArray;
|
public import std.experimental.allocator : make, makeArray;
|
||||||
import std.traits;
|
import std.traits;
|
||||||
public import tanya.memory.allocator;
|
public import tanya.memory.allocator;
|
||||||
public import tanya.memory.types;
|
|
||||||
|
/**
|
||||||
|
* The mixin generates common methods for classes and structs using
|
||||||
|
* allocators. It provides a protected member, constructor and a read-only property,
|
||||||
|
* that checks if an allocator was already set and sets it to the default
|
||||||
|
* one, if not (useful for structs which don't have a default constructor).
|
||||||
|
*/
|
||||||
|
mixin template DefaultAllocator()
|
||||||
|
{
|
||||||
|
/// Allocator.
|
||||||
|
protected shared Allocator allocator_;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Params:
|
||||||
|
* allocator = The allocator should be used.
|
||||||
|
*/
|
||||||
|
this(shared Allocator allocator)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.allocator_ = allocator;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This property checks if the allocator was set in the constructor
|
||||||
|
* and sets it to the default one, if not.
|
||||||
|
*
|
||||||
|
* Returns: Used allocator.
|
||||||
|
*
|
||||||
|
* Postcondition: $(D_INLINECODE allocator_ !is null)
|
||||||
|
*/
|
||||||
|
protected @property shared(Allocator) allocator() nothrow @safe @nogc
|
||||||
|
out (allocator)
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
if (allocator_ is null)
|
||||||
|
{
|
||||||
|
allocator_ = defaultAllocator;
|
||||||
|
}
|
||||||
|
return allocator_;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
@property shared(Allocator) allocator() const nothrow @trusted @nogc
|
||||||
|
out (allocator)
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
if (allocator_ is null)
|
||||||
|
{
|
||||||
|
return defaultAllocator;
|
||||||
|
}
|
||||||
|
return cast(shared Allocator) allocator_;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// From druntime
|
// From druntime
|
||||||
private extern (C) void _d_monitordelete(Object h, bool det) nothrow @nogc;
|
private extern (C) void _d_monitordelete(Object h, bool det) nothrow @nogc;
|
||||||
@ -28,11 +90,21 @@ shared static this() nothrow @trusted @nogc
|
|||||||
}
|
}
|
||||||
|
|
||||||
@property ref shared(Allocator) defaultAllocator() nothrow @safe @nogc
|
@property ref shared(Allocator) defaultAllocator() nothrow @safe @nogc
|
||||||
|
out (allocator)
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
{
|
{
|
||||||
return allocator;
|
return allocator;
|
||||||
}
|
}
|
||||||
|
|
||||||
@property void defaultAllocator(shared(Allocator) allocator) nothrow @safe @nogc
|
@property void defaultAllocator(shared(Allocator) allocator) nothrow @safe @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
{
|
{
|
||||||
.allocator = allocator;
|
.allocator = allocator;
|
||||||
}
|
}
|
||||||
@ -220,15 +292,26 @@ void dispose(T)(shared Allocator allocator, auto ref T p)
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
void dispose(T)(shared Allocator allocator, auto ref T[] array)
|
void dispose(T)(shared Allocator allocator, auto ref T[] p)
|
||||||
{
|
{
|
||||||
static if (hasElaborateDestructor!(typeof(array[0])))
|
static if (hasElaborateDestructor!(typeof(p[0])))
|
||||||
{
|
{
|
||||||
foreach (ref e; array)
|
import std.algorithm.iteration;
|
||||||
|
p.each!(e => destroy(e));
|
||||||
|
}
|
||||||
|
() @trusted { allocator.deallocate(p); }();
|
||||||
|
p = null;
|
||||||
|
}
|
||||||
|
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
struct S
|
||||||
|
{
|
||||||
|
~this()
|
||||||
{
|
{
|
||||||
destroy(e);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
() @trusted { allocator.deallocate(array); }();
|
auto p = cast(S[]) defaultAllocator.allocate(S.sizeof);
|
||||||
array = null;
|
|
||||||
|
defaultAllocator.dispose(p);
|
||||||
}
|
}
|
||||||
|
@ -1,49 +0,0 @@
|
|||||||
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Templates that generate values.
|
|
||||||
*
|
|
||||||
* Copyright: Eugene Wissner 2016.
|
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
|
||||||
*/
|
|
||||||
module tanya.meta.gen;
|
|
||||||
|
|
||||||
import std.traits;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Initializer list.
|
|
||||||
*
|
|
||||||
* Generates a static array with elements from $(D_PARAM args). All elements
|
|
||||||
* should have the same type. It can be used in constructors which accept a
|
|
||||||
* list of the elements of the same type in the situations where variadic
|
|
||||||
* functions and templates can't be used.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* Args = Argument type.
|
|
||||||
* args = Arguments.
|
|
||||||
*/
|
|
||||||
enum IL(Args...)(Args args)
|
|
||||||
if (Args.length > 0)
|
|
||||||
{
|
|
||||||
alias BaseType = typeof(args[0]);
|
|
||||||
|
|
||||||
BaseType[args.length] result;
|
|
||||||
|
|
||||||
foreach (i, a; args)
|
|
||||||
{
|
|
||||||
static assert(isImplicitlyConvertible!(typeof(a), BaseType));
|
|
||||||
result[i] = a;
|
|
||||||
}
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
unittest
|
|
||||||
{
|
|
||||||
static assert(IL(1, 5, 8).length == 3);
|
|
||||||
static assert(IL(1, 5, 8).sizeof == 3 * int.sizeof);
|
|
||||||
}
|
|
@ -1,15 +0,0 @@
|
|||||||
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Metaprogramming.
|
|
||||||
*
|
|
||||||
* Copyright: Eugene Wissner 2016.
|
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
|
||||||
*/
|
|
||||||
module tanya.meta;
|
|
||||||
|
|
||||||
public import tanya.meta.gen;
|
|
@ -255,7 +255,7 @@ else version (Windows)
|
|||||||
|
|
||||||
overlapped.handle = cast(HANDLE) handle_;
|
overlapped.handle = cast(HANDLE) handle_;
|
||||||
overlapped.event = OverlappedSocketEvent.read;
|
overlapped.event = OverlappedSocketEvent.read;
|
||||||
overlapped.buffer.len = buffer.length;
|
overlapped.buffer.len = cast(ULONG) buffer.length;
|
||||||
overlapped.buffer.buf = cast(char*) buffer.ptr;
|
overlapped.buffer.buf = cast(char*) buffer.ptr;
|
||||||
|
|
||||||
auto result = WSARecv(handle_,
|
auto result = WSARecv(handle_,
|
||||||
@ -326,7 +326,7 @@ else version (Windows)
|
|||||||
{
|
{
|
||||||
overlapped.handle = cast(HANDLE) handle_;
|
overlapped.handle = cast(HANDLE) handle_;
|
||||||
overlapped.event = OverlappedSocketEvent.write;
|
overlapped.event = OverlappedSocketEvent.write;
|
||||||
overlapped.buffer.len = buffer.length;
|
overlapped.buffer.len = cast(ULONG) buffer.length;
|
||||||
overlapped.buffer.buf = cast(char*) buffer.ptr;
|
overlapped.buffer.buf = cast(char*) buffer.ptr;
|
||||||
|
|
||||||
auto result = WSASend(handle_,
|
auto result = WSASend(handle_,
|
||||||
|
Reference in New Issue
Block a user