Compare commits
106 Commits
Author | SHA1 | Date | |
---|---|---|---|
e79c75df81 | |||
a6dfb3a19e | |||
2af0db04bd | |||
2c9867c577 | |||
47b394d8c3 | |||
ede0107fd7 | |||
7d5dda1cba | |||
e5f83c22fa | |||
a4de1cc754 | |||
8d3cdb8862 | |||
ba1bd35d4a | |||
dfacabd88b | |||
aa306d9050 | |||
10019d7df9 | |||
ae36296ca6 | |||
56406fb593 | |||
ec9b2db4b9 | |||
f5d0c2af8f | |||
c62dc4063e | |||
3789853d98 | |||
f0d016bcde | |||
70e96c62b3 | |||
b723d763c8 | |||
508297f36f | |||
4b0134713c | |||
5b90286b70 | |||
8443f1b385 | |||
c9050c1a8e | |||
bdf87570e2 | |||
faa44b6704 | |||
278e851414 | |||
6f549df243 | |||
4633bcc680 | |||
dc39efd316 | |||
260937e4fb | |||
e17fff2881 | |||
bc32511254 | |||
f30972f948 | |||
ea33ca62c8 | |||
0f365758e1 | |||
2815b53a88 | |||
6c0588164a | |||
8ee1d647ce | |||
25791775e6 | |||
f013e2f1f4 | |||
ac3935d71b | |||
b1c217e272 | |||
d007aaa310 | |||
8aaf9e14be | |||
ae3e6b46f6 | |||
8687df1fbb | |||
ba0aff6737 | |||
a648e2120a | |||
bc61809050 | |||
8c42cbfd63 | |||
58664570f9 | |||
decb82f437 | |||
357c7e279d | |||
32e19c8b58 | |||
f5c6c5b483 | |||
ba2d086fb8 | |||
7a0241b484 | |||
36dad80e18 | |||
29d883150e | |||
e2bed0cfcb | |||
38afeac071 | |||
001c7c3e33 | |||
d4ab339feb | |||
8477312769 | |||
67f90e137d | |||
f264fd5597 | |||
9e75620f1b | |||
45825946c0 | |||
8afb552d59 | |||
e4091669f8 | |||
1cb9349226 | |||
06620dc5df | |||
708d95db49 | |||
85d9361bfb | |||
a6a6f496eb | |||
db12f03264 | |||
231aedb8ad | |||
c3b63ee40d | |||
6f405c5e08 | |||
16cf8478cf | |||
8915a0c7a7 | |||
e5c7edb72c | |||
64e0d666ed | |||
f2aac680c5 | |||
d629525a4b | |||
33d321f0d7 | |||
3d64d59ba9 | |||
4635835a99 | |||
8725ec5f20 | |||
9a4c8cea06 | |||
eb360bda38 | |||
4b1cd2cbfd | |||
cd944a61b7 | |||
47ef787353 | |||
6436ad49df | |||
e1964e47a5 | |||
43319e4e3a | |||
33dbf042c2 | |||
885fca9b5e | |||
074d027629 | |||
f4b90d8b51 |
9
.editorconfig
Normal file
9
.editorconfig
Normal file
@ -0,0 +1,9 @@
|
|||||||
|
root = true
|
||||||
|
|
||||||
|
[*]
|
||||||
|
end_of_line = lf
|
||||||
|
insert_final_newline = true
|
||||||
|
charset = utf-8
|
||||||
|
indent_style = space
|
||||||
|
indent_size = 4
|
||||||
|
trim_trailing_whitespace = true
|
3
.gitignore
vendored
3
.gitignore
vendored
@ -1,5 +1,6 @@
|
|||||||
# Binary
|
# Binary
|
||||||
*.[oa]
|
*.[oa]
|
||||||
|
*.exe
|
||||||
|
|
||||||
# D
|
# D
|
||||||
.dub
|
.dub
|
||||||
@ -9,3 +10,5 @@ __test__*__.core
|
|||||||
|
|
||||||
/docs/
|
/docs/
|
||||||
/docs.json
|
/docs.json
|
||||||
|
|
||||||
|
/*.lst
|
||||||
|
32
.travis.yml
32
.travis.yml
@ -1,20 +1,34 @@
|
|||||||
sudo: false
|
sudo: false
|
||||||
|
|
||||||
os:
|
os:
|
||||||
- linux
|
- linux
|
||||||
- osx
|
- osx
|
||||||
|
|
||||||
language: d
|
language: d
|
||||||
|
|
||||||
d:
|
d:
|
||||||
- dmd-2.074.0
|
- dmd-2.074.1
|
||||||
- dmd-2.073.2
|
- dmd-2.073.2
|
||||||
- dmd-2.072.2
|
- dmd-2.072.2
|
||||||
- dmd-2.071.2
|
- dmd-2.071.2
|
||||||
|
|
||||||
env:
|
env:
|
||||||
matrix:
|
matrix:
|
||||||
- ARCH=x86_64
|
- ARCH=x86_64
|
||||||
|
- ARCH=x86
|
||||||
|
|
||||||
|
addons:
|
||||||
|
apt:
|
||||||
|
packages:
|
||||||
|
- gcc-multilib
|
||||||
|
|
||||||
|
before_script:
|
||||||
|
- if [ "$PS1" = '(dmd-2.074.1)' ]; then
|
||||||
|
export UNITTEST="unittest-cov";
|
||||||
|
fi
|
||||||
|
|
||||||
script:
|
script:
|
||||||
- dub test --arch=$ARCH
|
- dub test -b ${UNITTEST:-unittest} --arch=$ARCH --compiler=$DC
|
||||||
|
|
||||||
|
after_success:
|
||||||
|
- test "$UNITTEST" = "unittest-cov" && bash <(curl -s https://codecov.io/bash)
|
||||||
|
5
CODE_OF_CONDUCT.md
Normal file
5
CODE_OF_CONDUCT.md
Normal file
@ -0,0 +1,5 @@
|
|||||||
|
# Contributor Code of Conduct
|
||||||
|
|
||||||
|
This project adheres to No Code of Conduct. We are all adults. We accept anyone's contributions. Nothing else matters.
|
||||||
|
|
||||||
|
For more information please visit the [No Code of Conduct](https://github.com/domgetter/NCoC) homepage.
|
64
CONTRIBUTING.md
Normal file
64
CONTRIBUTING.md
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
# Contributing
|
||||||
|
|
||||||
|
Tanya is a project in active development, therefore any help is appreciated. Thank you for considering contributing
|
||||||
|
to it, feel welcome.
|
||||||
|
These guidelines describe ways to get started.
|
||||||
|
|
||||||
|
|
||||||
|
## Ways to get involved
|
||||||
|
|
||||||
|
* **Reporting a problem**: [Report](https://issues.caraus.io/projects/tanya/issues) bugs and usage problems you
|
||||||
|
encounter.
|
||||||
|
* **Fixing issues**: [The bug tracker](https://issues.caraus.io/projects/tanya/issues) contains a list of issues you
|
||||||
|
can work on.
|
||||||
|
* **Documentation**: You can improve API documentation by correcting grammar errors, completing existing texts and
|
||||||
|
writing new ones, or providing usage examples.
|
||||||
|
* **Testing**: Test coverage is important for a library. Writing tests is not only helpful, but is also a great way
|
||||||
|
to get a feel for how tanya works.
|
||||||
|
|
||||||
|
|
||||||
|
## Opening an issue
|
||||||
|
|
||||||
|
If you have found a bug, an error, have some question, or suggestion, open in issue. Tanya uses an external
|
||||||
|
[bug tracker](https://issues.caraus.io/projects/tanya/issues). You should
|
||||||
|
[register](https://issues.caraus.io/account/register) before you can report your issue. There is also a list
|
||||||
|
of open issues that mirror the current development process and progress. If you're looking for a challenge, just
|
||||||
|
pick an issue you are interested in and start working on it. Fill free to comment on the issue to get more
|
||||||
|
information.
|
||||||
|
|
||||||
|
Some issues have a category assigned to it. Such issues belongs mostly to a larger part of the library that is
|
||||||
|
currently in development. The category specifies then the git branch development happens on. The remaining issues
|
||||||
|
can be fixed directly in master.
|
||||||
|
|
||||||
|
In the [roadmap](https://issues.caraus.io/projects/tanya/roadmap) you can find a list of issues that are planned
|
||||||
|
to be fixed till a specific release. Version numbers refer to the versions in the
|
||||||
|
[git repository](https://github.com/caraus-ecms/tanya/releases).
|
||||||
|
|
||||||
|
|
||||||
|
## Creating a pull request
|
||||||
|
|
||||||
|
I accept GitHub pull requests. Creating a pull request is like sending a patch with a suggested change.
|
||||||
|
First you have to [fork](https://guides.github.com/activities/forking/) the repository. Clone your fork locally
|
||||||
|
with `git clone` and create a new branch where you want to work, for example:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
git checkout -b bugfix-x
|
||||||
|
```
|
||||||
|
Commit your changes to your fork:
|
||||||
|
|
||||||
|
```shell
|
||||||
|
git commit -m "Fix Bug X"
|
||||||
|
git push -u origin bugfix-x
|
||||||
|
```
|
||||||
|
After that if you visit your fork on GitHub, GitHub will suggest to create pull request. Just follow the steps
|
||||||
|
described on GitHub to finish the process.
|
||||||
|
|
||||||
|
Please ensure that you fork is even with the upstream (original) repository. If not, you have to rebase your branch
|
||||||
|
on upstream master before submitting a pull request. See https://help.github.com/articles/syncing-a-fork/ for a
|
||||||
|
step-by-step guide.
|
||||||
|
|
||||||
|
|
||||||
|
## Questions and suggestions
|
||||||
|
|
||||||
|
* [Open an issue](https://issues.caraus.io/projects/tanya/issues)
|
||||||
|
* [Send an email](mailto:info@caraus.de)
|
149
README.md
149
README.md
@ -1,6 +1,8 @@
|
|||||||
# Tanya
|
# Tanya
|
||||||
|
|
||||||
[](https://travis-ci.org/caraus-ecms/tanya)
|
[](https://travis-ci.org/caraus-ecms/tanya)
|
||||||
|
[](https://ci.appveyor.com/project/belka-ew/tanya/branch/master)
|
||||||
|
[](https://codecov.io/gh/caraus-ecms/tanya)
|
||||||
[](https://code.dlang.org/packages/tanya)
|
[](https://code.dlang.org/packages/tanya)
|
||||||
[](https://code.dlang.org/packages/tanya)
|
[](https://code.dlang.org/packages/tanya)
|
||||||
[](https://raw.githubusercontent.com/caraus-ecms/tanya/master/LICENSE)
|
[](https://raw.githubusercontent.com/caraus-ecms/tanya/master/LICENSE)
|
||||||
@ -13,59 +15,148 @@ 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)
|
* [Bug tracker](https://issues.caraus.io/projects/tanya/issues)
|
||||||
* [Documentation](https://docs.caraus.io/tanya)
|
* [Documentation](https://docs.caraus.io/tanya)
|
||||||
|
* [Contribution guidelines](CONTRIBUTING.md)
|
||||||
|
|
||||||
## 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, Array, Singly and doubly linked lists, Buffers, UTF-8
|
||||||
|
string, Hash set.
|
||||||
|
* `format`: Formatting and conversion functions.
|
||||||
* `math`: Arbitrary precision integer and a set of functions.
|
* `math`: Arbitrary precision integer and a set of functions.
|
||||||
* `memory`: Tools for manual memory management (allocator, reference counting,
|
* `memory`: Tools for manual memory management (allocators, smart pointers).
|
||||||
helper functions).
|
* `network`: URL-Parsing, sockets, utilities.
|
||||||
* `network`: URL-Parsing, sockets.
|
* `os`: Platform-independent interfaces to operating system functionality.
|
||||||
|
|
||||||
|
## Basic usage
|
||||||
|
|
||||||
|
### Allocators
|
||||||
|
|
||||||
|
Memory management is done with allocators. Instead of using `new` to create an
|
||||||
|
instance of a class, an allocator is used:
|
||||||
|
|
||||||
|
```d
|
||||||
|
import tanya.memory;
|
||||||
|
|
||||||
|
class A
|
||||||
|
{
|
||||||
|
this(int arg)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
A a = defaultAllocator.make!A(5);
|
||||||
|
|
||||||
|
defaultAllocator.dispose(a);
|
||||||
|
```
|
||||||
|
|
||||||
|
As you can see, the user is responsible for deallocation, therefore `dispose`
|
||||||
|
is called at the end.
|
||||||
|
|
||||||
|
If you want to change the `defaultAllocator` to something different, you
|
||||||
|
probably want to do it at the program's beginning. Or you can invoke another
|
||||||
|
allocator directly. It is important to ensure that the object is destroyed
|
||||||
|
using the same allocator that was used to allocate it.
|
||||||
|
|
||||||
|
What if I get an allocated object from some function? The generic rule is: If
|
||||||
|
you haven't requested the memory yourself (with `make`), you don't need to free
|
||||||
|
it.
|
||||||
|
|
||||||
|
`tanya.memory.smartref` contains smart pointers, helpers that can take care of
|
||||||
|
a proper deallocation in some cases for you.
|
||||||
|
|
||||||
|
### Exceptions
|
||||||
|
|
||||||
|
Since exceptions are normal classes in D, they are allocated and dellocated the
|
||||||
|
same as described above, but:
|
||||||
|
|
||||||
|
1. The caller is **always** responsible for destroying a caught exception.
|
||||||
|
2. Exceptions are **always** allocated and should be always allocated with the
|
||||||
|
`defaultAllocator`.
|
||||||
|
|
||||||
|
```d
|
||||||
|
import tanya.memory;
|
||||||
|
|
||||||
|
void functionThatThrows()
|
||||||
|
{
|
||||||
|
throw defaultAlocator.make!Exception("An error occurred");
|
||||||
|
}
|
||||||
|
|
||||||
|
try
|
||||||
|
{
|
||||||
|
functionThatThrows()
|
||||||
|
}
|
||||||
|
catch (Exception e)
|
||||||
|
{
|
||||||
|
defaultAllocator.dispose(e);
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### Containers
|
||||||
|
|
||||||
|
Arrays are commonly used in programming. D's built-in arrays often rely on the
|
||||||
|
GC. It is inconvenient to change their size, reserve memory for future use and
|
||||||
|
so on. Containers can help here. The following example demonstrates how
|
||||||
|
`tanya.container.array.Array` can be used instead of `int[]`.
|
||||||
|
|
||||||
|
```d
|
||||||
|
import tanya.container.array;
|
||||||
|
|
||||||
|
Array!int arr;
|
||||||
|
|
||||||
|
// Reserve memory if I know that my container will contain approximately 15
|
||||||
|
// elements.
|
||||||
|
arr.reserve(15);
|
||||||
|
|
||||||
|
arr.insertBack(5); // Add one element.
|
||||||
|
arr.length = 10; // New elements are initialized to 0.
|
||||||
|
|
||||||
|
// Iterate over the first five elements.
|
||||||
|
foreach (el; arr[0 .. 5])
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
int i = arr[7]; // Access 7th element.
|
||||||
|
```
|
||||||
|
|
||||||
|
There are more containers in the `tanya.container` package.
|
||||||
|
|
||||||
|
## Development
|
||||||
|
|
||||||
### Supported compilers
|
### Supported compilers
|
||||||
|
|
||||||
| dmd |
|
| dmd |
|
||||||
|:-------:|
|
|:-------:|
|
||||||
| 2.074.0 |
|
| 2.074.1 |
|
||||||
| 2.073.2 |
|
| 2.073.2 |
|
||||||
| 2.072.2 |
|
| 2.072.2 |
|
||||||
| 2.071.2 |
|
| 2.071.2 |
|
||||||
|
|
||||||
### Current status
|
### Current status
|
||||||
|
|
||||||
The library is currently under development, but the API is becoming gradually
|
Following modules are under development:
|
||||||
stable.
|
|
||||||
|
|
||||||
Following modules are coming soon:
|
| Feature | Branch | Build status |
|
||||||
|
|----------|:---------:|-----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
|
||||||
|
| BitArray | bitvector | [](https://travis-ci.org/caraus-ecms/tanya) [](https://ci.appveyor.com/project/belka-ew/tanya/branch/bitvector) |
|
||||||
|
| TLS | crypto | [](https://travis-ci.org/caraus-ecms/tanya) [](https://ci.appveyor.com/project/belka-ew/tanya/branch/crypto) |
|
||||||
|
| File IO | io | [](https://travis-ci.org/caraus-ecms/tanya) [](https://ci.appveyor.com/project/belka-ew/tanya/branch/io) |
|
||||||
|
|
||||||
| Feature | Build status |
|
### Release management
|
||||||
|--------------|:-----------------------------------------------------------------------------------------------------------------------:|
|
|
||||||
| UTF-8 string | [](https://travis-ci.org/caraus-ecms/tanya) |
|
|
||||||
| BitVector | [](https://travis-ci.org/caraus-ecms/tanya) |
|
|
||||||
| Hash table | N/A |
|
|
||||||
|
|
||||||
`math` package contains an arbitrary precision integer implementation that
|
3-week release cycle.
|
||||||
needs more test cases, better performance and some additional features
|
|
||||||
(constructing from a string and an ubyte array, and converting it back).
|
|
||||||
|
|
||||||
### Further characteristics
|
Deprecated features are removed after one release (in approximately 6 weeks after deprecating).
|
||||||
|
|
||||||
* Tanya is a native D library.
|
## Further characteristics
|
||||||
|
|
||||||
|
* Tanya is a native D library without any external dependencies.
|
||||||
|
|
||||||
* 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.
|
||||||
|
|
||||||
* The library isn't thread-safe. Thread-safity should be added later.
|
* The library isn't thread-safe yet.
|
||||||
|
|
||||||
## Contributing
|
|
||||||
|
|
||||||
Since I'm mostly busy writing new code and implementing new features I would
|
|
||||||
appreciate, if anyone uses the library. It would help me to improve the
|
|
||||||
codebase and fix issues.
|
|
||||||
|
|
||||||
Feel free to contact me if you have any questions.
|
|
||||||
|
72
appveyor.yml
Normal file
72
appveyor.yml
Normal file
@ -0,0 +1,72 @@
|
|||||||
|
platform: x64
|
||||||
|
os: Visual Studio 2015
|
||||||
|
|
||||||
|
environment:
|
||||||
|
matrix:
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.074.1
|
||||||
|
arch: x64
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.074.1
|
||||||
|
arch: x86
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.073.2
|
||||||
|
arch: x64
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.073.2
|
||||||
|
arch: x86
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.072.2
|
||||||
|
arch: x64
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.072.2
|
||||||
|
arch: x86
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.071.2
|
||||||
|
arch: x64
|
||||||
|
- DC: dmd
|
||||||
|
DVersion: 2.071.2
|
||||||
|
arch: x86
|
||||||
|
|
||||||
|
skip_tags: true
|
||||||
|
|
||||||
|
install:
|
||||||
|
- ps: function SetUpDCompiler
|
||||||
|
{
|
||||||
|
$env:toolchain = "msvc";
|
||||||
|
$version = $env:DVersion;
|
||||||
|
Invoke-WebRequest "http://downloads.dlang.org/releases/2.x/$($version)/dmd.$($version).windows.7z" -OutFile "c:\dmd.7z";
|
||||||
|
echo "finished.";
|
||||||
|
pushd c:\\;
|
||||||
|
7z x dmd.7z > $null;
|
||||||
|
popd;
|
||||||
|
}
|
||||||
|
- ps: SetUpDCompiler
|
||||||
|
|
||||||
|
- ps: if($env:DVersion -eq "2.071.2"){
|
||||||
|
Invoke-WebRequest "http://code.dlang.org/files/dub-1.2.1-windows-x86.zip" -OutFile "dub.zip";
|
||||||
|
7z x dub.zip -odub > $null;
|
||||||
|
Move-Item "dub/dub.exe" "C:\dmd2\windows\bin"
|
||||||
|
}
|
||||||
|
|
||||||
|
before_build:
|
||||||
|
- ps: if($env:arch -eq "x86"){
|
||||||
|
$env:compilersetupargs = "x86";
|
||||||
|
$env:Darch = "x86";
|
||||||
|
}
|
||||||
|
elseif($env:arch -eq "x64"){
|
||||||
|
$env:compilersetupargs = "amd64";
|
||||||
|
$env:Darch = "x86_64";
|
||||||
|
}
|
||||||
|
- ps: $env:PATH += ";C:\dmd2\windows\bin;";
|
||||||
|
- call "C:\Program Files (x86)\Microsoft Visual Studio 14.0\VC\vcvarsall" %compilersetupargs%
|
||||||
|
|
||||||
|
build_script:
|
||||||
|
- echo dummy build script - dont remove me
|
||||||
|
|
||||||
|
test_script:
|
||||||
|
- echo %Darch%
|
||||||
|
- echo %PATH%
|
||||||
|
- 'dub --version'
|
||||||
|
- '%DC% --version'
|
||||||
|
- dub test -b unittest --arch=%Darch% --compiler=%DC%
|
4
codecov.yml
Normal file
4
codecov.yml
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
ignore:
|
||||||
|
- "source/tanya/async/event/iocp.d"
|
||||||
|
- "source/tanya/async/iocp.d"
|
||||||
|
- "source/tanya/memory/types.d"
|
2
dub.json
2
dub.json
@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "tanya",
|
"name": "tanya",
|
||||||
"description": "General purpose, @nogc library",
|
"description": "General purpose, @nogc library. Containers, networking, memory management, utilities",
|
||||||
"license": "MPL-2.0",
|
"license": "MPL-2.0",
|
||||||
"copyright": "(c) Eugene Wissner <info@caraus.de>",
|
"copyright": "(c) Eugene Wissner <info@caraus.de>",
|
||||||
"authors": [
|
"authors": [
|
||||||
|
@ -2,7 +2,9 @@
|
|||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
* 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/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/*
|
||||||
|
* Event loop implementation for Linux.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -10,7 +12,10 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.async.event.epoll;
|
module tanya.async.event.epoll;
|
||||||
|
|
||||||
version (linux):
|
version (D_Ddoc)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
else version (linux):
|
||||||
|
|
||||||
public import core.sys.linux.epoll;
|
public import core.sys.linux.epoll;
|
||||||
import tanya.async.protocol;
|
import tanya.async.protocol;
|
||||||
@ -18,7 +23,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.container.array;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
import tanya.memory.mmappool;
|
import tanya.memory.mmappool;
|
||||||
import tanya.network.socket;
|
import tanya.network.socket;
|
||||||
@ -29,153 +34,153 @@ import std.algorithm.comparison;
|
|||||||
|
|
||||||
extern (C) nothrow @nogc
|
extern (C) nothrow @nogc
|
||||||
{
|
{
|
||||||
int epoll_create1(int flags);
|
int epoll_create1(int flags);
|
||||||
int epoll_ctl (int epfd, int op, int fd, epoll_event *event);
|
int epoll_ctl (int epfd, int op, int fd, epoll_event *event);
|
||||||
int epoll_wait (int epfd, epoll_event *events, int maxevents, int timeout);
|
int epoll_wait (int epfd, epoll_event *events, int maxevents, int timeout);
|
||||||
}
|
}
|
||||||
|
|
||||||
final class EpollLoop : SelectorLoop
|
final class EpollLoop : SelectorLoop
|
||||||
{
|
{
|
||||||
protected int fd;
|
protected int fd;
|
||||||
private Vector!epoll_event events;
|
private Array!epoll_event events;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Initializes the loop.
|
* Initializes the loop.
|
||||||
*/
|
*/
|
||||||
this() @nogc
|
this() @nogc
|
||||||
{
|
{
|
||||||
if ((fd = epoll_create1(EPOLL_CLOEXEC)) < 0)
|
if ((fd = epoll_create1(EPOLL_CLOEXEC)) < 0)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!BadLoopException("epoll initialization failed");
|
throw defaultAllocator.make!BadLoopException("epoll initialization failed");
|
||||||
}
|
}
|
||||||
super();
|
super();
|
||||||
events = Vector!epoll_event(maxEvents, MmapPool.instance);
|
events = Array!epoll_event(maxEvents, MmapPool.instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Frees loop internals.
|
* Frees loop internals.
|
||||||
*/
|
*/
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
close(fd);
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Should be called if the backend configuration changes.
|
* Should be called if the backend configuration changes.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* watcher = Watcher.
|
* watcher = Watcher.
|
||||||
* oldEvents = The events were already set.
|
* oldEvents = The events were already set.
|
||||||
* events = The events should be set.
|
* events = The events should be set.
|
||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
protected override bool reify(SocketWatcher watcher,
|
protected override bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc
|
EventMask events) @nogc
|
||||||
{
|
{
|
||||||
int op = EPOLL_CTL_DEL;
|
int op = EPOLL_CTL_DEL;
|
||||||
epoll_event ev;
|
epoll_event ev;
|
||||||
|
|
||||||
if (events == oldEvents)
|
if (events == oldEvents)
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
if (events && oldEvents)
|
if (events && oldEvents)
|
||||||
{
|
{
|
||||||
op = EPOLL_CTL_MOD;
|
op = EPOLL_CTL_MOD;
|
||||||
}
|
}
|
||||||
else if (events && !oldEvents)
|
else if (events && !oldEvents)
|
||||||
{
|
{
|
||||||
op = EPOLL_CTL_ADD;
|
op = EPOLL_CTL_ADD;
|
||||||
}
|
}
|
||||||
|
|
||||||
ev.data.fd = watcher.socket.handle;
|
ev.data.fd = watcher.socket.handle;
|
||||||
ev.events = (events & (Event.read | Event.accept) ? EPOLLIN | EPOLLPRI : 0)
|
ev.events = (events & (Event.read | Event.accept) ? EPOLLIN | EPOLLPRI : 0)
|
||||||
| (events & Event.write ? EPOLLOUT : 0)
|
| (events & Event.write ? EPOLLOUT : 0)
|
||||||
| EPOLLET;
|
| EPOLLET;
|
||||||
|
|
||||||
return epoll_ctl(fd, op, watcher.socket.handle, &ev) == 0;
|
return epoll_ctl(fd, op, watcher.socket.handle, &ev) == 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Does the actual polling.
|
* Does the actual polling.
|
||||||
*/
|
*/
|
||||||
protected override void poll() @nogc
|
protected override void poll() @nogc
|
||||||
{
|
{
|
||||||
// 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.get().ptr, maxEvents, timeout);
|
auto eventCount = epoll_wait(fd, events.get().ptr, maxEvents, timeout);
|
||||||
|
|
||||||
if (eventCount < 0)
|
if (eventCount < 0)
|
||||||
{
|
{
|
||||||
if (errno != EINTR)
|
if (errno != EINTR)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!BadLoopException();
|
throw defaultAllocator.make!BadLoopException();
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (auto i = 0; i < eventCount; ++i)
|
for (auto i = 0; i < eventCount; ++i)
|
||||||
{
|
{
|
||||||
auto transport = cast(StreamTransport) connections[events[i].data.fd];
|
auto transport = cast(StreamTransport) connections[events[i].data.fd];
|
||||||
|
|
||||||
if (transport is null)
|
if (transport is null)
|
||||||
{
|
{
|
||||||
auto connection = cast(ConnectionWatcher) connections[events[i].data.fd];
|
auto connection = cast(ConnectionWatcher) connections[events[i].data.fd];
|
||||||
assert(connection !is null);
|
assert(connection !is null);
|
||||||
|
|
||||||
acceptConnections(connection);
|
acceptConnections(connection);
|
||||||
}
|
}
|
||||||
else if (events[i].events & EPOLLERR)
|
else if (events[i].events & EPOLLERR)
|
||||||
{
|
{
|
||||||
kill(transport);
|
kill(transport);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
else if (events[i].events & (EPOLLIN | EPOLLPRI | EPOLLHUP))
|
else if (events[i].events & (EPOLLIN | EPOLLPRI | EPOLLHUP))
|
||||||
{
|
{
|
||||||
SocketException exception;
|
SocketException exception;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
ptrdiff_t received;
|
ptrdiff_t received;
|
||||||
do
|
do
|
||||||
{
|
{
|
||||||
received = transport.socket.receive(transport.output[]);
|
received = transport.socket.receive(transport.output[]);
|
||||||
transport.output += received;
|
transport.output += received;
|
||||||
}
|
}
|
||||||
while (received);
|
while (received);
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
exception = e;
|
exception = e;
|
||||||
}
|
}
|
||||||
if (transport.socket.disconnected)
|
if (transport.socket.disconnected)
|
||||||
{
|
{
|
||||||
kill(transport, exception);
|
kill(transport, exception);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
else if (transport.output.length)
|
else if (transport.output.length)
|
||||||
{
|
{
|
||||||
pendings.enqueue(transport);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (events[i].events & EPOLLOUT)
|
if (events[i].events & EPOLLOUT)
|
||||||
{
|
{
|
||||||
transport.writeReady = true;
|
transport.writeReady = true;
|
||||||
if (transport.input.length)
|
if (transport.input.length)
|
||||||
{
|
{
|
||||||
feed(transport);
|
feed(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: The blocking time.
|
* Returns: The blocking time.
|
||||||
*/
|
*/
|
||||||
override protected @property inout(Duration) blockTime()
|
override protected @property inout(Duration) blockTime()
|
||||||
inout @safe pure nothrow
|
inout @safe pure nothrow
|
||||||
{
|
{
|
||||||
return min(super.blockTime, 1.dur!"seconds");
|
return min(super.blockTime, 1.dur!"seconds");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -2,7 +2,9 @@
|
|||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
* 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/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/*
|
||||||
|
* Event loop implementation for Windows.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -10,7 +12,10 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.async.event.iocp;
|
module tanya.async.event.iocp;
|
||||||
|
|
||||||
version (Windows):
|
version (D_Ddoc)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
else version (Windows):
|
||||||
|
|
||||||
import tanya.container.buffer;
|
import tanya.container.buffer;
|
||||||
import tanya.async.loop;
|
import tanya.async.loop;
|
||||||
@ -31,354 +36,354 @@ import core.sys.windows.winsock2;
|
|||||||
*/
|
*/
|
||||||
final class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
|
final class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
|
||||||
{
|
{
|
||||||
private SocketException exception;
|
private SocketException exception;
|
||||||
|
|
||||||
private ReadBuffer!ubyte output;
|
private ReadBuffer!ubyte output;
|
||||||
|
|
||||||
private WriteBuffer!ubyte input;
|
private WriteBuffer!ubyte input;
|
||||||
|
|
||||||
private Protocol protocol_;
|
private Protocol protocol_;
|
||||||
|
|
||||||
private bool closing;
|
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)
|
* Precondition: $(D_INLINECODE socket !is null)
|
||||||
*/
|
*/
|
||||||
this(OverlappedConnectedSocket socket) @nogc
|
this(OverlappedConnectedSocket socket) @nogc
|
||||||
{
|
{
|
||||||
super(socket);
|
super(socket);
|
||||||
output = ReadBuffer!ubyte(8192, 1024, MmapPool.instance);
|
output = ReadBuffer!ubyte(8192, 1024, MmapPool.instance);
|
||||||
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
||||||
active = true;
|
active = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Socket.
|
* Returns: Socket.
|
||||||
*
|
*
|
||||||
* Postcondition: $(D_INLINECODE socket !is null)
|
* Postcondition: $(D_INLINECODE socket !is null)
|
||||||
*/
|
*/
|
||||||
override @property OverlappedConnectedSocket socket() pure nothrow @safe @nogc
|
override @property OverlappedConnectedSocket socket() pure nothrow @safe @nogc
|
||||||
out (socket)
|
out (socket)
|
||||||
{
|
{
|
||||||
assert(socket !is null);
|
assert(socket !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
return cast(OverlappedConnectedSocket) socket_;
|
return cast(OverlappedConnectedSocket) socket_;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns $(D_PARAM true) if the transport is closing or closed.
|
* Returns $(D_PARAM true) if the transport is closing or closed.
|
||||||
*/
|
*/
|
||||||
bool isClosing() const pure nothrow @safe @nogc
|
bool isClosing() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return closing;
|
return closing;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Close the transport.
|
* Close the transport.
|
||||||
*
|
*
|
||||||
* Buffered data will be flushed. No more data will be received.
|
* Buffered data will be flushed. No more data will be received.
|
||||||
*/
|
*/
|
||||||
void close() pure nothrow @safe @nogc
|
void close() pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
closing = true;
|
closing = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Write some data to the transport.
|
* Write some data to the transport.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* data = Data to send.
|
* data = Data to send.
|
||||||
*/
|
*/
|
||||||
void write(ubyte[] data) @nogc
|
void write(ubyte[] data) @nogc
|
||||||
{
|
{
|
||||||
input ~= data;
|
input ~= data;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Application protocol.
|
* Returns: Application protocol.
|
||||||
*/
|
*/
|
||||||
@property Protocol protocol() pure nothrow @safe @nogc
|
@property Protocol protocol() pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return protocol_;
|
return protocol_;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Switches the protocol.
|
* Switches the protocol.
|
||||||
*
|
*
|
||||||
* The protocol is deallocated by the event loop, it should currently be
|
* The protocol is deallocated by the event loop, it should currently be
|
||||||
* allocated with $(D_PSYMBOL MmapPool).
|
* allocated with $(D_PSYMBOL MmapPool).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* protocol = Application protocol.
|
* protocol = Application protocol.
|
||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE protocol !is null)
|
* Precondition: $(D_INLINECODE protocol !is null)
|
||||||
*/
|
*/
|
||||||
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(protocol !is null);
|
assert(protocol !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
protocol_ = protocol;
|
protocol_ = protocol;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Invokes the watcher callback.
|
* Invokes the watcher callback.
|
||||||
*/
|
*/
|
||||||
override void invoke() @nogc
|
override void invoke() @nogc
|
||||||
{
|
{
|
||||||
if (output.length)
|
if (output.length)
|
||||||
{
|
{
|
||||||
immutable empty = input.length == 0;
|
immutable empty = input.length == 0;
|
||||||
protocol.received(output[0 .. $]);
|
protocol.received(output[0 .. $]);
|
||||||
output.clear();
|
output.clear();
|
||||||
if (empty)
|
if (empty)
|
||||||
{
|
{
|
||||||
SocketState overlapped;
|
SocketState overlapped;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
overlapped = MmapPool.instance.make!SocketState;
|
overlapped = MmapPool.instance.make!SocketState;
|
||||||
socket.beginSend(input[], overlapped);
|
socket.beginSend(input[], overlapped);
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(overlapped);
|
MmapPool.instance.dispose(overlapped);
|
||||||
MmapPool.instance.dispose(e);
|
MmapPool.instance.dispose(e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
protocol.disconnected(exception);
|
protocol.disconnected(exception);
|
||||||
MmapPool.instance.dispose(protocol_);
|
MmapPool.instance.dispose(protocol_);
|
||||||
defaultAllocator.dispose(exception);
|
defaultAllocator.dispose(exception);
|
||||||
active = false;
|
active = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
final class IOCPLoop : Loop
|
final class IOCPLoop : Loop
|
||||||
{
|
{
|
||||||
protected HANDLE completionPort;
|
protected HANDLE completionPort;
|
||||||
|
|
||||||
protected OVERLAPPED overlap;
|
protected OVERLAPPED overlap;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Initializes the loop.
|
* Initializes the loop.
|
||||||
*/
|
*/
|
||||||
this() @nogc
|
this() @nogc
|
||||||
{
|
{
|
||||||
super();
|
super();
|
||||||
|
|
||||||
completionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
|
completionPort = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, 0, 0);
|
||||||
if (!completionPort)
|
if (!completionPort)
|
||||||
{
|
{
|
||||||
throw make!BadLoopException(defaultAllocator,
|
throw make!BadLoopException(defaultAllocator,
|
||||||
"Creating completion port failed");
|
"Creating completion port failed");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Should be called if the backend configuration changes.
|
* Should be called if the backend configuration changes.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* watcher = Watcher.
|
* watcher = Watcher.
|
||||||
* oldEvents = The events were already set.
|
* oldEvents = The events were already set.
|
||||||
* events = The events should be set.
|
* events = The events should be set.
|
||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
override protected bool reify(SocketWatcher watcher,
|
override protected bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc
|
EventMask events) @nogc
|
||||||
{
|
{
|
||||||
SocketState overlapped;
|
SocketState overlapped;
|
||||||
if (!(oldEvents & Event.accept) && (events & Event.accept))
|
if (!(oldEvents & Event.accept) && (events & Event.accept))
|
||||||
{
|
{
|
||||||
auto socket = cast(OverlappedStreamSocket) watcher.socket;
|
auto socket = cast(OverlappedStreamSocket) watcher.socket;
|
||||||
assert(socket !is null);
|
assert(socket !is null);
|
||||||
|
|
||||||
if (CreateIoCompletionPort(cast(HANDLE) socket.handle,
|
if (CreateIoCompletionPort(cast(HANDLE) socket.handle,
|
||||||
completionPort,
|
completionPort,
|
||||||
cast(ULONG_PTR) (cast(void*) watcher),
|
cast(ULONG_PTR) (cast(void*) watcher),
|
||||||
0) !is completionPort)
|
0) !is completionPort)
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
overlapped = MmapPool.instance.make!SocketState;
|
overlapped = MmapPool.instance.make!SocketState;
|
||||||
socket.beginAccept(overlapped);
|
socket.beginAccept(overlapped);
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(overlapped);
|
MmapPool.instance.dispose(overlapped);
|
||||||
defaultAllocator.dispose(e);
|
defaultAllocator.dispose(e);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
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 transport = cast(StreamTransport) watcher;
|
auto transport = cast(StreamTransport) watcher;
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
|
|
||||||
if (CreateIoCompletionPort(cast(HANDLE) transport.socket.handle,
|
if (CreateIoCompletionPort(cast(HANDLE) transport.socket.handle,
|
||||||
completionPort,
|
completionPort,
|
||||||
cast(ULONG_PTR) (cast(void*) watcher),
|
cast(ULONG_PTR) (cast(void*) watcher),
|
||||||
0) !is completionPort)
|
0) !is completionPort)
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Begin to read
|
// Begin to read
|
||||||
if (!(oldEvents & Event.read) && (events & Event.read))
|
if (!(oldEvents & Event.read) && (events & Event.read))
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
overlapped = MmapPool.instance.make!SocketState;
|
overlapped = MmapPool.instance.make!SocketState;
|
||||||
transport.socket.beginReceive(transport.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(overlapped);
|
MmapPool.instance.dispose(overlapped);
|
||||||
defaultAllocator.dispose(e);
|
defaultAllocator.dispose(e);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
private void kill(StreamTransport transport,
|
private void kill(StreamTransport transport,
|
||||||
SocketException exception = null) @nogc
|
SocketException exception = null) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
transport.socket.shutdown();
|
transport.socket.shutdown();
|
||||||
defaultAllocator.dispose(transport.socket);
|
defaultAllocator.dispose(transport.socket);
|
||||||
transport.exception = exception;
|
transport.exception = exception;
|
||||||
pendings.enqueue(transport);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Does the actual polling.
|
* Does the actual polling.
|
||||||
*/
|
*/
|
||||||
override protected void poll() @nogc
|
override protected void poll() @nogc
|
||||||
{
|
{
|
||||||
DWORD lpNumberOfBytes;
|
DWORD lpNumberOfBytes;
|
||||||
ULONG_PTR key;
|
ULONG_PTR key;
|
||||||
LPOVERLAPPED overlap;
|
LPOVERLAPPED overlap;
|
||||||
immutable timeout = cast(immutable int) blockTime.total!"msecs";
|
immutable timeout = cast(immutable int) blockTime.total!"msecs";
|
||||||
|
|
||||||
auto result = GetQueuedCompletionStatus(completionPort,
|
auto result = GetQueuedCompletionStatus(completionPort,
|
||||||
&lpNumberOfBytes,
|
&lpNumberOfBytes,
|
||||||
&key,
|
&key,
|
||||||
&overlap,
|
&overlap,
|
||||||
timeout);
|
timeout);
|
||||||
if (result == FALSE && overlap == NULL)
|
if (result == FALSE && overlap == NULL)
|
||||||
{
|
{
|
||||||
return; // Timeout
|
return; // Timeout
|
||||||
}
|
}
|
||||||
|
|
||||||
auto overlapped = (cast(SocketState) ((cast(void*) overlap) - 8));
|
auto overlapped = (cast(SocketState) ((cast(void*) overlap) - 8));
|
||||||
assert(overlapped !is null);
|
assert(overlapped !is null);
|
||||||
scope (failure)
|
scope (failure)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(overlapped);
|
MmapPool.instance.dispose(overlapped);
|
||||||
}
|
}
|
||||||
|
|
||||||
switch (overlapped.event)
|
switch (overlapped.event)
|
||||||
{
|
{
|
||||||
case OverlappedSocketEvent.accept:
|
case OverlappedSocketEvent.accept:
|
||||||
auto connection = cast(ConnectionWatcher) (cast(void*) key);
|
auto connection = cast(ConnectionWatcher) (cast(void*) key);
|
||||||
assert(connection !is null);
|
assert(connection !is null);
|
||||||
|
|
||||||
auto listener = cast(OverlappedStreamSocket) connection.socket;
|
auto listener = cast(OverlappedStreamSocket) connection.socket;
|
||||||
assert(listener !is null);
|
assert(listener !is null);
|
||||||
|
|
||||||
auto socket = listener.endAccept(overlapped);
|
auto socket = listener.endAccept(overlapped);
|
||||||
auto transport = MmapPool.instance.make!StreamTransport(socket);
|
auto transport = MmapPool.instance.make!StreamTransport(socket);
|
||||||
|
|
||||||
connection.incoming.enqueue(transport);
|
connection.incoming.enqueue(transport);
|
||||||
|
|
||||||
reify(transport, EventMask(Event.none), EventMask(Event.read, Event.write));
|
reify(transport, EventMask(Event.none), EventMask(Event.read, Event.write));
|
||||||
|
|
||||||
pendings.enqueue(connection);
|
pendings.enqueue(connection);
|
||||||
listener.beginAccept(overlapped);
|
listener.beginAccept(overlapped);
|
||||||
break;
|
break;
|
||||||
case OverlappedSocketEvent.read:
|
case OverlappedSocketEvent.read:
|
||||||
auto transport = cast(StreamTransport) (cast(void*) key);
|
auto transport = cast(StreamTransport) (cast(void*) key);
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
|
|
||||||
if (!transport.active)
|
if (!transport.active)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(transport);
|
MmapPool.instance.dispose(transport);
|
||||||
MmapPool.instance.dispose(overlapped);
|
MmapPool.instance.dispose(overlapped);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
int received;
|
int received;
|
||||||
SocketException exception;
|
SocketException exception;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
received = transport.socket.endReceive(overlapped);
|
received = transport.socket.endReceive(overlapped);
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
exception = e;
|
exception = e;
|
||||||
}
|
}
|
||||||
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(transport.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
kill(transport, exception);
|
kill(transport, exception);
|
||||||
}
|
}
|
||||||
else if (received > 0)
|
else if (received > 0)
|
||||||
{
|
{
|
||||||
immutable full = transport.output.free == received;
|
immutable full = transport.output.free == received;
|
||||||
|
|
||||||
transport.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(transport.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
}
|
}
|
||||||
pendings.enqueue(transport);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
case OverlappedSocketEvent.write:
|
case OverlappedSocketEvent.write:
|
||||||
auto transport = cast(StreamTransport) (cast(void*) key);
|
auto transport = cast(StreamTransport) (cast(void*) key);
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
|
|
||||||
transport.input += transport.socket.endSend(overlapped);
|
transport.input += transport.socket.endSend(overlapped);
|
||||||
if (transport.input.length > 0)
|
if (transport.input.length > 0)
|
||||||
{
|
{
|
||||||
transport.socket.beginSend(transport.input[], overlapped);
|
transport.socket.beginSend(transport.input[], overlapped);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
transport.socket.beginReceive(transport.output[], overlapped);
|
transport.socket.beginReceive(transport.output[], overlapped);
|
||||||
if (transport.isClosing())
|
if (transport.isClosing())
|
||||||
{
|
{
|
||||||
kill(transport);
|
kill(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
assert(false, "Unknown event");
|
assert(false, "Unknown event");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -2,7 +2,9 @@
|
|||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
* 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/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/*
|
||||||
|
* Event loop implementation for *BSD.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -10,33 +12,36 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.async.event.kqueue;
|
module tanya.async.event.kqueue;
|
||||||
|
|
||||||
version (OSX)
|
version (D_Ddoc)
|
||||||
{
|
{
|
||||||
version = MacBSD;
|
}
|
||||||
|
else version (OSX)
|
||||||
|
{
|
||||||
|
version = MacBSD;
|
||||||
}
|
}
|
||||||
else version (iOS)
|
else version (iOS)
|
||||||
{
|
{
|
||||||
version = MacBSD;
|
version = MacBSD;
|
||||||
}
|
}
|
||||||
else version (TVOS)
|
else version (TVOS)
|
||||||
{
|
{
|
||||||
version = MacBSD;
|
version = MacBSD;
|
||||||
}
|
}
|
||||||
else version (WatchOS)
|
else version (WatchOS)
|
||||||
{
|
{
|
||||||
version = MacBSD;
|
version = MacBSD;
|
||||||
}
|
}
|
||||||
else version (FreeBSD)
|
else version (FreeBSD)
|
||||||
{
|
{
|
||||||
version = MacBSD;
|
version = MacBSD;
|
||||||
}
|
}
|
||||||
else version (OpenBSD)
|
else version (OpenBSD)
|
||||||
{
|
{
|
||||||
version = MacBSD;
|
version = MacBSD;
|
||||||
}
|
}
|
||||||
else version (DragonFlyBSD)
|
else version (DragonFlyBSD)
|
||||||
{
|
{
|
||||||
version = MacBSD;
|
version = MacBSD;
|
||||||
}
|
}
|
||||||
|
|
||||||
version (MacBSD):
|
version (MacBSD):
|
||||||
@ -50,62 +55,62 @@ 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.container.array;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
import tanya.memory.mmappool;
|
import tanya.memory.mmappool;
|
||||||
import tanya.network.socket;
|
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
|
||||||
{
|
{
|
||||||
*kevp = kevent_t(args);
|
*kevp = kevent_t(args);
|
||||||
}
|
}
|
||||||
|
|
||||||
enum : short
|
enum : short
|
||||||
{
|
{
|
||||||
EVFILT_READ = -1,
|
EVFILT_READ = -1,
|
||||||
EVFILT_WRITE = -2,
|
EVFILT_WRITE = -2,
|
||||||
EVFILT_AIO = -3, /* attached to aio requests */
|
EVFILT_AIO = -3, /* attached to aio requests */
|
||||||
EVFILT_VNODE = -4, /* attached to vnodes */
|
EVFILT_VNODE = -4, /* attached to vnodes */
|
||||||
EVFILT_PROC = -5, /* attached to struct proc */
|
EVFILT_PROC = -5, /* attached to struct proc */
|
||||||
EVFILT_SIGNAL = -6, /* attached to struct proc */
|
EVFILT_SIGNAL = -6, /* attached to struct proc */
|
||||||
EVFILT_TIMER = -7, /* timers */
|
EVFILT_TIMER = -7, /* timers */
|
||||||
EVFILT_MACHPORT = -8, /* Mach portsets */
|
EVFILT_MACHPORT = -8, /* Mach portsets */
|
||||||
EVFILT_FS = -9, /* filesystem events */
|
EVFILT_FS = -9, /* filesystem events */
|
||||||
EVFILT_USER = -10, /* User events */
|
EVFILT_USER = -10, /* User events */
|
||||||
EVFILT_VM = -12, /* virtual memory events */
|
EVFILT_VM = -12, /* virtual memory events */
|
||||||
EVFILT_SYSCOUNT = 11
|
EVFILT_SYSCOUNT = 11
|
||||||
}
|
}
|
||||||
|
|
||||||
struct kevent_t
|
struct kevent_t
|
||||||
{
|
{
|
||||||
uintptr_t ident; /* identifier for this event */
|
uintptr_t ident; /* identifier for this event */
|
||||||
short filter; /* filter for event */
|
short filter; /* filter for event */
|
||||||
ushort flags;
|
ushort flags;
|
||||||
uint fflags;
|
uint fflags;
|
||||||
intptr_t data;
|
intptr_t data;
|
||||||
void *udata; /* opaque user data identifier */
|
void *udata; /* opaque user data identifier */
|
||||||
}
|
}
|
||||||
|
|
||||||
enum
|
enum
|
||||||
{
|
{
|
||||||
/* actions */
|
/* actions */
|
||||||
EV_ADD = 0x0001, /* add event to kq (implies enable) */
|
EV_ADD = 0x0001, /* add event to kq (implies enable) */
|
||||||
EV_DELETE = 0x0002, /* delete event from kq */
|
EV_DELETE = 0x0002, /* delete event from kq */
|
||||||
EV_ENABLE = 0x0004, /* enable event */
|
EV_ENABLE = 0x0004, /* enable event */
|
||||||
EV_DISABLE = 0x0008, /* disable event (not reported) */
|
EV_DISABLE = 0x0008, /* disable event (not reported) */
|
||||||
|
|
||||||
/* flags */
|
/* flags */
|
||||||
EV_ONESHOT = 0x0010, /* only report one occurrence */
|
EV_ONESHOT = 0x0010, /* only report one occurrence */
|
||||||
EV_CLEAR = 0x0020, /* clear event state after reporting */
|
EV_CLEAR = 0x0020, /* clear event state after reporting */
|
||||||
EV_RECEIPT = 0x0040, /* force EV_ERROR on success, data=0 */
|
EV_RECEIPT = 0x0040, /* force EV_ERROR on success, data=0 */
|
||||||
EV_DISPATCH = 0x0080, /* disable event after reporting */
|
EV_DISPATCH = 0x0080, /* disable event after reporting */
|
||||||
|
|
||||||
EV_SYSFLAGS = 0xF000, /* reserved by system */
|
EV_SYSFLAGS = 0xF000, /* reserved by system */
|
||||||
EV_FLAG1 = 0x2000, /* filter-specific flag */
|
EV_FLAG1 = 0x2000, /* filter-specific flag */
|
||||||
|
|
||||||
/* returned values */
|
/* returned values */
|
||||||
EV_EOF = 0x8000, /* EOF detected */
|
EV_EOF = 0x8000, /* EOF detected */
|
||||||
EV_ERROR = 0x4000, /* error, data contains errno */
|
EV_ERROR = 0x4000, /* error, data contains errno */
|
||||||
}
|
}
|
||||||
|
|
||||||
extern(C) int kqueue() nothrow @nogc;
|
extern(C) int kqueue() nothrow @nogc;
|
||||||
@ -115,211 +120,211 @@ extern(C) int kevent(int kq, const kevent_t *changelist, int nchanges,
|
|||||||
|
|
||||||
final class KqueueLoop : SelectorLoop
|
final class KqueueLoop : SelectorLoop
|
||||||
{
|
{
|
||||||
protected int fd;
|
protected int fd;
|
||||||
private Vector!kevent_t events;
|
private Array!kevent_t events;
|
||||||
private Vector!kevent_t changes;
|
private Array!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 uint maxEvents()
|
override protected @property uint maxEvents()
|
||||||
const pure nothrow @safe @nogc
|
const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return cast(uint) events.length;
|
return cast(uint) events.length;
|
||||||
}
|
}
|
||||||
|
|
||||||
this() @nogc
|
this() @nogc
|
||||||
{
|
{
|
||||||
super();
|
super();
|
||||||
|
|
||||||
if ((fd = kqueue()) == -1)
|
if ((fd = kqueue()) == -1)
|
||||||
{
|
{
|
||||||
throw make!BadLoopException(defaultAllocator,
|
throw make!BadLoopException(defaultAllocator,
|
||||||
"kqueue initialization failed");
|
"kqueue initialization failed");
|
||||||
}
|
}
|
||||||
events = Vector!kevent_t(64, MmapPool.instance);
|
events = Array!kevent_t(64, MmapPool.instance);
|
||||||
changes = Vector!kevent_t(64, MmapPool.instance);
|
changes = Array!kevent_t(64, MmapPool.instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Frees loop internals.
|
* Frees loop internals.
|
||||||
*/
|
*/
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
close(fd);
|
close(fd);
|
||||||
}
|
}
|
||||||
|
|
||||||
private void set(socket_t socket, short filter, ushort flags) @nogc
|
private void set(SocketType socket, short filter, ushort flags) @nogc
|
||||||
{
|
{
|
||||||
if (changes.length <= changeCount)
|
if (changes.length <= changeCount)
|
||||||
{
|
{
|
||||||
changes.length = changeCount + maxEvents;
|
changes.length = changeCount + maxEvents;
|
||||||
}
|
}
|
||||||
EV_SET(&changes[changeCount],
|
EV_SET(&changes[changeCount],
|
||||||
cast(ulong) socket,
|
cast(ulong) socket,
|
||||||
filter,
|
filter,
|
||||||
flags,
|
flags,
|
||||||
0U,
|
0U,
|
||||||
0L,
|
0L,
|
||||||
null);
|
null);
|
||||||
++changeCount;
|
++changeCount;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Should be called if the backend configuration changes.
|
* Should be called if the backend configuration changes.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* watcher = Watcher.
|
* watcher = Watcher.
|
||||||
* oldEvents = The events were already set.
|
* oldEvents = The events were already set.
|
||||||
* events = The events should be set.
|
* events = The events should be set.
|
||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
override protected bool reify(SocketWatcher watcher,
|
override protected bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc
|
EventMask events) @nogc
|
||||||
{
|
{
|
||||||
if (events != oldEvents)
|
if (events != oldEvents)
|
||||||
{
|
{
|
||||||
if (oldEvents & Event.read || oldEvents & Event.accept)
|
if (oldEvents & Event.read || oldEvents & Event.accept)
|
||||||
{
|
{
|
||||||
set(watcher.socket.handle, EVFILT_READ, EV_DELETE);
|
set(watcher.socket.handle, EVFILT_READ, EV_DELETE);
|
||||||
}
|
}
|
||||||
if (oldEvents & Event.write)
|
if (oldEvents & Event.write)
|
||||||
{
|
{
|
||||||
set(watcher.socket.handle, EVFILT_WRITE, EV_DELETE);
|
set(watcher.socket.handle, EVFILT_WRITE, EV_DELETE);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (events & (Event.read | events & Event.accept))
|
if (events & (Event.read | events & Event.accept))
|
||||||
{
|
{
|
||||||
set(watcher.socket.handle, EVFILT_READ, EV_ADD | EV_ENABLE);
|
set(watcher.socket.handle, EVFILT_READ, EV_ADD | EV_ENABLE);
|
||||||
}
|
}
|
||||||
if (events & Event.write)
|
if (events & Event.write)
|
||||||
{
|
{
|
||||||
set(watcher.socket.handle, EVFILT_WRITE, EV_ADD | EV_DISPATCH);
|
set(watcher.socket.handle, EVFILT_WRITE, EV_ADD | EV_DISPATCH);
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Does the actual polling.
|
* Does the actual polling.
|
||||||
*/
|
*/
|
||||||
protected override void poll() @nogc
|
protected override void poll() @nogc
|
||||||
{
|
{
|
||||||
timespec ts;
|
timespec ts;
|
||||||
blockTime.split!("seconds", "nsecs")(ts.tv_sec, ts.tv_nsec);
|
blockTime.split!("seconds", "nsecs")(ts.tv_sec, ts.tv_nsec);
|
||||||
|
|
||||||
if (changeCount > maxEvents)
|
if (changeCount > maxEvents)
|
||||||
{
|
{
|
||||||
events.length = changes.length;
|
events.length = changes.length;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto eventCount = kevent(fd,
|
auto eventCount = kevent(fd,
|
||||||
changes.get().ptr,
|
changes.get().ptr,
|
||||||
cast(int) changeCount,
|
cast(int) changeCount,
|
||||||
events.get().ptr,
|
events.get().ptr,
|
||||||
maxEvents,
|
maxEvents,
|
||||||
&ts);
|
&ts);
|
||||||
changeCount = 0;
|
changeCount = 0;
|
||||||
|
|
||||||
if (eventCount < 0)
|
if (eventCount < 0)
|
||||||
{
|
{
|
||||||
if (errno != EINTR)
|
if (errno != EINTR)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!BadLoopException();
|
throw defaultAllocator.make!BadLoopException();
|
||||||
}
|
}
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int i; i < eventCount; ++i)
|
for (int i; i < eventCount; ++i)
|
||||||
{
|
{
|
||||||
assert(connections.length > events[i].ident);
|
assert(connections.length > events[i].ident);
|
||||||
|
|
||||||
auto transport = cast(StreamTransport) 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 (transport is null)
|
if (transport is null)
|
||||||
{
|
{
|
||||||
auto connection = cast(ConnectionWatcher) connections[events[i].ident];
|
auto connection = cast(ConnectionWatcher) connections[events[i].ident];
|
||||||
assert(connection !is null);
|
assert(connection !is null);
|
||||||
|
|
||||||
acceptConnections(connection);
|
acceptConnections(connection);
|
||||||
}
|
}
|
||||||
else if (events[i].flags & EV_ERROR)
|
else if (events[i].flags & EV_ERROR)
|
||||||
{
|
{
|
||||||
kill(transport);
|
kill(transport);
|
||||||
}
|
}
|
||||||
else if (events[i].filter == EVFILT_READ)
|
else if (events[i].filter == EVFILT_READ)
|
||||||
{
|
{
|
||||||
SocketException exception;
|
SocketException exception;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
ptrdiff_t received;
|
ptrdiff_t received;
|
||||||
do
|
do
|
||||||
{
|
{
|
||||||
received = transport.socket.receive(transport.output[]);
|
received = transport.socket.receive(transport.output[]);
|
||||||
transport.output += received;
|
transport.output += received;
|
||||||
}
|
}
|
||||||
while (received);
|
while (received);
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
exception = e;
|
exception = e;
|
||||||
}
|
}
|
||||||
if (transport.socket.disconnected)
|
if (transport.socket.disconnected)
|
||||||
{
|
{
|
||||||
kill(transport, exception);
|
kill(transport, exception);
|
||||||
}
|
}
|
||||||
else if (transport.output.length)
|
else if (transport.output.length)
|
||||||
{
|
{
|
||||||
pendings.enqueue(transport);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else if (events[i].filter == EVFILT_WRITE)
|
else if (events[i].filter == EVFILT_WRITE)
|
||||||
{
|
{
|
||||||
transport.writeReady = true;
|
transport.writeReady = true;
|
||||||
if (transport.input.length)
|
if (transport.input.length)
|
||||||
{
|
{
|
||||||
feed(transport);
|
feed(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: The blocking time.
|
* Returns: The blocking time.
|
||||||
*/
|
*/
|
||||||
override protected @property inout(Duration) blockTime()
|
override protected @property inout(Duration) blockTime()
|
||||||
inout @nogc @safe pure nothrow
|
inout @nogc @safe pure nothrow
|
||||||
{
|
{
|
||||||
return min(super.blockTime, 1.dur!"seconds");
|
return min(super.blockTime, 1.dur!"seconds");
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* If the transport couldn't send the data, the further sending should
|
* If the transport couldn't send the data, the further sending should
|
||||||
* be handled by the event loop.
|
* be handled by the event loop.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* transport = Transport.
|
* transport = Transport.
|
||||||
* exception = Exception thrown on sending.
|
* exception = Exception thrown on sending.
|
||||||
*
|
*
|
||||||
* 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 will be destroyed then).
|
* transport will be destroyed then).
|
||||||
*/
|
*/
|
||||||
protected override bool feed(StreamTransport 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))
|
||||||
{
|
{
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
if (!transport.writeReady)
|
if (!transport.writeReady)
|
||||||
{
|
{
|
||||||
set(transport.socket.handle, EVFILT_WRITE, EV_DISPATCH);
|
set(transport.socket.handle, EVFILT_WRITE, EV_DISPATCH);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -2,7 +2,9 @@
|
|||||||
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
* 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/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/*
|
||||||
|
* This module contains base implementations for reactor event loops.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -10,14 +12,17 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.async.event.selector;
|
module tanya.async.event.selector;
|
||||||
|
|
||||||
version (Posix):
|
version (D_Ddoc)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
else version (Posix):
|
||||||
|
|
||||||
import tanya.async.loop;
|
import tanya.async.loop;
|
||||||
import tanya.async.protocol;
|
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.container.array;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
import tanya.memory.mmappool;
|
import tanya.memory.mmappool;
|
||||||
import tanya.network.socket;
|
import tanya.network.socket;
|
||||||
@ -27,371 +32,374 @@ import tanya.network.socket;
|
|||||||
*/
|
*/
|
||||||
package class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
|
package class StreamTransport : SocketWatcher, DuplexTransport, SocketTransport
|
||||||
{
|
{
|
||||||
private SelectorLoop loop;
|
private SelectorLoop loop;
|
||||||
|
|
||||||
private SocketException exception;
|
private SocketException exception;
|
||||||
|
|
||||||
package ReadBuffer!ubyte output;
|
package ReadBuffer!ubyte output;
|
||||||
|
|
||||||
package WriteBuffer!ubyte input;
|
package WriteBuffer!ubyte input;
|
||||||
|
|
||||||
private Protocol protocol_;
|
private Protocol protocol_;
|
||||||
|
|
||||||
private bool closing;
|
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;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
* loop = Event loop.
|
* loop = Event loop.
|
||||||
* socket = Socket.
|
* socket = Socket.
|
||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE loop !is null && socket !is null)
|
* Precondition: $(D_INLINECODE loop !is null && socket !is null)
|
||||||
*/
|
*/
|
||||||
this(SelectorLoop loop, ConnectedSocket socket) @nogc
|
this(SelectorLoop loop, ConnectedSocket socket) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(loop !is null);
|
assert(loop !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
super(socket);
|
super(socket);
|
||||||
this.loop = loop;
|
this.loop = loop;
|
||||||
output = ReadBuffer!ubyte(8192, 1024, MmapPool.instance);
|
output = ReadBuffer!ubyte(8192, 1024, MmapPool.instance);
|
||||||
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
input = WriteBuffer!ubyte(8192, MmapPool.instance);
|
||||||
active = true;
|
active = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Socket.
|
* Returns: Socket.
|
||||||
*
|
*
|
||||||
* Postcondition: $(D_INLINECODE socket !is null)
|
* Postcondition: $(D_INLINECODE socket !is null)
|
||||||
*/
|
*/
|
||||||
override @property ConnectedSocket socket() pure nothrow @safe @nogc
|
override @property ConnectedSocket socket() pure nothrow @safe @nogc
|
||||||
out (socket)
|
out (socket)
|
||||||
{
|
{
|
||||||
assert(socket !is null);
|
assert(socket !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
return cast(ConnectedSocket) socket_;
|
return cast(ConnectedSocket) socket_;
|
||||||
}
|
}
|
||||||
|
|
||||||
private @property void socket(ConnectedSocket socket) pure nothrow @safe @nogc
|
private @property void socket(ConnectedSocket socket)
|
||||||
in
|
pure nothrow @safe @nogc
|
||||||
{
|
in
|
||||||
assert(socket !is null);
|
{
|
||||||
}
|
assert(socket !is null);
|
||||||
body
|
}
|
||||||
{
|
body
|
||||||
socket_ = socket;
|
{
|
||||||
}
|
socket_ = socket;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Application protocol.
|
* Returns: Application protocol.
|
||||||
*/
|
*/
|
||||||
@property Protocol protocol() pure nothrow @safe @nogc
|
@property Protocol protocol() pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return protocol_;
|
return protocol_;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Switches the protocol.
|
* Switches the protocol.
|
||||||
*
|
*
|
||||||
* The protocol is deallocated by the event loop, it should currently be
|
* The protocol is deallocated by the event loop, it should currently be
|
||||||
* allocated with $(D_PSYMBOL MmapPool).
|
* allocated with $(D_PSYMBOL MmapPool).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* protocol = Application protocol.
|
* protocol = Application protocol.
|
||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE protocol !is null)
|
* Precondition: $(D_INLINECODE protocol !is null)
|
||||||
*/
|
*/
|
||||||
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(protocol !is null);
|
assert(protocol !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
protocol_ = protocol;
|
protocol_ = protocol;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns $(D_PARAM true) if the transport is closing or closed.
|
* Returns $(D_PARAM true) if the transport is closing or closed.
|
||||||
*/
|
*/
|
||||||
bool isClosing() const pure nothrow @safe @nogc
|
bool isClosing() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return closing;
|
return closing;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Close the transport.
|
* Close the transport.
|
||||||
*
|
*
|
||||||
* Buffered data will be flushed. No more data will be received.
|
* Buffered data will be flushed. No more data will be received.
|
||||||
*/
|
*/
|
||||||
void close() @nogc
|
void close() @nogc
|
||||||
{
|
{
|
||||||
closing = true;
|
closing = true;
|
||||||
loop.reify(this, EventMask(Event.read, Event.write), EventMask(Event.write));
|
loop.reify(this,
|
||||||
}
|
EventMask(Event.read, Event.write),
|
||||||
|
EventMask(Event.write));
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Invokes the watcher callback.
|
* Invokes the watcher callback.
|
||||||
*/
|
*/
|
||||||
override void invoke() @nogc
|
override void invoke() @nogc
|
||||||
{
|
{
|
||||||
if (output.length)
|
if (output.length)
|
||||||
{
|
{
|
||||||
protocol.received(output[0 .. $]);
|
protocol.received(output[0 .. $]);
|
||||||
output.clear();
|
output.clear();
|
||||||
if (isClosing() && input.length == 0)
|
if (isClosing() && input.length == 0)
|
||||||
{
|
{
|
||||||
loop.kill(this);
|
loop.kill(this);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
protocol.disconnected(exception);
|
protocol.disconnected(exception);
|
||||||
MmapPool.instance.dispose(protocol_);
|
MmapPool.instance.dispose(protocol_);
|
||||||
defaultAllocator.dispose(exception);
|
defaultAllocator.dispose(exception);
|
||||||
active = false;
|
active = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Write some data to the transport.
|
* Write some data to the transport.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* data = Data to send.
|
* data = Data to send.
|
||||||
*/
|
*/
|
||||||
void write(ubyte[] data) @nogc
|
void write(ubyte[] data) @nogc
|
||||||
{
|
{
|
||||||
if (!data.length)
|
if (!data.length)
|
||||||
{
|
{
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
// Try to write if the socket is write ready.
|
// Try to write if the socket is write ready.
|
||||||
if (writeReady)
|
if (writeReady)
|
||||||
{
|
{
|
||||||
ptrdiff_t sent;
|
ptrdiff_t sent;
|
||||||
SocketException exception;
|
SocketException exception;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
sent = socket.send(data);
|
sent = socket.send(data);
|
||||||
if (sent == 0)
|
if (sent == 0)
|
||||||
{
|
{
|
||||||
writeReady = false;
|
writeReady = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
writeReady = false;
|
writeReady = false;
|
||||||
exception = e;
|
exception = e;
|
||||||
}
|
}
|
||||||
if (sent < data.length)
|
if (sent < data.length)
|
||||||
{
|
{
|
||||||
input ~= data[sent..$];
|
input ~= data[sent..$];
|
||||||
loop.feed(this, exception);
|
loop.feed(this, exception);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
input ~= data;
|
input ~= data;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
abstract class SelectorLoop : Loop
|
abstract class SelectorLoop : Loop
|
||||||
{
|
{
|
||||||
/// Pending connections.
|
/// Pending connections.
|
||||||
protected Vector!SocketWatcher connections;
|
protected Array!SocketWatcher connections;
|
||||||
|
|
||||||
this() @nogc
|
this() @nogc
|
||||||
{
|
{
|
||||||
super();
|
super();
|
||||||
connections = Vector!SocketWatcher(maxEvents, MmapPool.instance);
|
connections = Array!SocketWatcher(maxEvents, MmapPool.instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
foreach (ref connection; connections)
|
foreach (ref connection; connections)
|
||||||
{
|
{
|
||||||
// We want to free only the transports. ConnectionWatcher are created by the
|
// We want to free only the transports. ConnectionWatcher are
|
||||||
// user and should be freed by himself.
|
// created by the user and should be freed by himself.
|
||||||
if (cast(StreamTransport) connection !is null)
|
if (cast(StreamTransport) connection !is null)
|
||||||
{
|
{
|
||||||
MmapPool.instance.dispose(connection);
|
MmapPool.instance.dispose(connection);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Should be called if the backend configuration changes.
|
* Should be called if the backend configuration changes.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* watcher = Watcher.
|
* watcher = Watcher.
|
||||||
* oldEvents = The events were already set.
|
* oldEvents = The events were already set.
|
||||||
* events = The events should be set.
|
* events = The events should be set.
|
||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if the operation was successful.
|
* Returns: $(D_KEYWORD true) if the operation was successful.
|
||||||
*/
|
*/
|
||||||
override abstract protected bool reify(SocketWatcher watcher,
|
override abstract protected bool reify(SocketWatcher watcher,
|
||||||
EventMask oldEvents,
|
EventMask oldEvents,
|
||||||
EventMask events) @nogc;
|
EventMask events) @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Kills the watcher and closes the connection.
|
* Kills the watcher and closes the connection.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* transport = Transport.
|
* transport = Transport.
|
||||||
* exception = Occurred exception.
|
* exception = Occurred exception.
|
||||||
*/
|
*/
|
||||||
protected void kill(StreamTransport transport,
|
protected void kill(StreamTransport transport,
|
||||||
SocketException exception = null) @nogc
|
SocketException exception = null) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
transport.socket.shutdown();
|
transport.socket.shutdown();
|
||||||
defaultAllocator.dispose(transport.socket);
|
defaultAllocator.dispose(transport.socket);
|
||||||
transport.exception = exception;
|
transport.exception = exception;
|
||||||
pendings.enqueue(transport);
|
pendings.enqueue(transport);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* If the transport couldn't send the data, the further sending should
|
* If the transport couldn't send the data, the further sending should
|
||||||
* be handled by the event loop.
|
* be handled by the event loop.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* transport = Transport.
|
* transport = Transport.
|
||||||
* exception = Exception thrown on sending.
|
* exception = Exception thrown on sending.
|
||||||
*
|
*
|
||||||
* 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 will be destroyed then).
|
* transport will be destroyed then).
|
||||||
*/
|
*/
|
||||||
protected bool feed(StreamTransport transport,
|
protected bool feed(StreamTransport transport,
|
||||||
SocketException exception = null) @nogc
|
SocketException exception = null) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(transport !is null);
|
assert(transport !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
while (transport.input.length && transport.writeReady)
|
while (transport.input.length && transport.writeReady)
|
||||||
{
|
{
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
ptrdiff_t sent = transport.socket.send(transport.input[]);
|
ptrdiff_t sent = transport.socket.send(transport.input[]);
|
||||||
if (sent == 0)
|
if (sent == 0)
|
||||||
{
|
{
|
||||||
transport.writeReady = false;
|
transport.writeReady = false;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
transport.input += sent;
|
transport.input += sent;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
exception = e;
|
exception = e;
|
||||||
transport.writeReady = false;
|
transport.writeReady = false;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (exception !is null)
|
if (exception !is null)
|
||||||
{
|
{
|
||||||
kill(transport, exception);
|
kill(transport, exception);
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
if (transport.input.length == 0 && transport.isClosing())
|
if (transport.input.length == 0 && transport.isClosing())
|
||||||
{
|
{
|
||||||
kill(transport);
|
kill(transport);
|
||||||
}
|
}
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Start watching.
|
* Start watching.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* watcher = Watcher.
|
* watcher = Watcher.
|
||||||
*/
|
*/
|
||||||
override void start(ConnectionWatcher watcher) @nogc
|
override void start(ConnectionWatcher watcher) @nogc
|
||||||
{
|
{
|
||||||
if (watcher.active)
|
if (watcher.active)
|
||||||
{
|
{
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (connections.length <= watcher.socket)
|
if (connections.length <= watcher.socket)
|
||||||
{
|
{
|
||||||
connections.length = watcher.socket.handle + maxEvents / 2;
|
connections.length = watcher.socket.handle + maxEvents / 2;
|
||||||
}
|
}
|
||||||
connections[watcher.socket.handle] = watcher;
|
connections[watcher.socket.handle] = watcher;
|
||||||
|
|
||||||
super.start(watcher);
|
super.start(watcher);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Accept incoming connections.
|
* Accept incoming connections.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* connection = Connection watcher ready to accept.
|
* connection = Connection watcher ready to accept.
|
||||||
*/
|
*/
|
||||||
package void acceptConnections(ConnectionWatcher connection) @nogc
|
package void acceptConnections(ConnectionWatcher connection) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(connection !is null);
|
assert(connection !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
while (true)
|
while (true)
|
||||||
{
|
{
|
||||||
ConnectedSocket client;
|
ConnectedSocket client;
|
||||||
try
|
try
|
||||||
{
|
{
|
||||||
client = (cast(StreamSocket) connection.socket).accept();
|
client = (cast(StreamSocket) connection.socket).accept();
|
||||||
}
|
}
|
||||||
catch (SocketException e)
|
catch (SocketException e)
|
||||||
{
|
{
|
||||||
defaultAllocator.dispose(e);
|
defaultAllocator.dispose(e);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
if (client is null)
|
if (client is null)
|
||||||
{
|
{
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
StreamTransport transport;
|
StreamTransport transport;
|
||||||
|
|
||||||
if (connections.length > client.handle)
|
if (connections.length > client.handle)
|
||||||
{
|
{
|
||||||
transport = cast(StreamTransport) connections[client.handle];
|
transport = cast(StreamTransport) connections[client.handle];
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
connections.length = client.handle + maxEvents / 2;
|
connections.length = client.handle + maxEvents / 2;
|
||||||
}
|
}
|
||||||
if (transport is null)
|
if (transport is null)
|
||||||
{
|
{
|
||||||
transport = MmapPool.instance.make!StreamTransport(this, client);
|
transport = MmapPool.instance.make!StreamTransport(this, client);
|
||||||
connections[client.handle] = transport;
|
connections[client.handle] = transport;
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
transport.socket = client;
|
transport.socket = client;
|
||||||
}
|
}
|
||||||
|
|
||||||
reify(transport, EventMask(Event.none), EventMask(Event.read, Event.write));
|
reify(transport, EventMask(Event.none), EventMask(Event.read, Event.write));
|
||||||
connection.incoming.enqueue(transport);
|
connection.incoming.enqueue(transport);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!connection.incoming.empty)
|
if (!connection.incoming.empty)
|
||||||
{
|
{
|
||||||
pendings.enqueue(connection);
|
pendings.enqueue(connection);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -3,6 +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/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* This module provides API for Windows I/O Completion Ports.
|
||||||
|
*
|
||||||
|
* Note: Available only on Windows.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -10,7 +14,26 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.async.iocp;
|
module tanya.async.iocp;
|
||||||
|
|
||||||
version (Windows):
|
version (Windows)
|
||||||
|
{
|
||||||
|
version = WindowsDoc;
|
||||||
|
}
|
||||||
|
else version (D_Ddoc)
|
||||||
|
{
|
||||||
|
version = WindowsDoc;
|
||||||
|
version (Windows)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
private struct OVERLAPPED
|
||||||
|
{
|
||||||
|
}
|
||||||
|
private alias HANDLE = void*;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
version (WindowsDoc):
|
||||||
|
|
||||||
import core.sys.windows.winbase;
|
import core.sys.windows.winbase;
|
||||||
import core.sys.windows.windef;
|
import core.sys.windows.windef;
|
||||||
@ -21,12 +44,12 @@ import core.sys.windows.windef;
|
|||||||
*/
|
*/
|
||||||
class State
|
class State
|
||||||
{
|
{
|
||||||
/// For internal use by Windows API.
|
/// For internal use by Windows API.
|
||||||
align(1) OVERLAPPED overlapped;
|
align(1) OVERLAPPED overlapped;
|
||||||
|
|
||||||
/// File/socket handle.
|
/// File/socket handle.
|
||||||
HANDLE handle;
|
HANDLE handle;
|
||||||
|
|
||||||
/// For keeping events or event masks.
|
/// For keeping events or event masks.
|
||||||
int event;
|
int event;
|
||||||
}
|
}
|
||||||
|
@ -3,10 +3,8 @@
|
|||||||
* 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-2017.
|
* Interface for the event loop implementations and the default event loop
|
||||||
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
* chooser.
|
||||||
* Mozilla Public License, v. 2.0).
|
|
||||||
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
|
||||||
*
|
*
|
||||||
* ---
|
* ---
|
||||||
* import tanya.async;
|
* import tanya.async;
|
||||||
@ -15,52 +13,57 @@
|
|||||||
*
|
*
|
||||||
* class EchoProtocol : TransmissionControlProtocol
|
* class EchoProtocol : TransmissionControlProtocol
|
||||||
* {
|
* {
|
||||||
* private DuplexTransport transport;
|
* private DuplexTransport transport;
|
||||||
*
|
*
|
||||||
* void received(in ubyte[] data) @nogc
|
* void received(in ubyte[] data) @nogc
|
||||||
* {
|
* {
|
||||||
* transport.write(data);
|
* transport.write(data);
|
||||||
* }
|
* }
|
||||||
*
|
*
|
||||||
* void connected(DuplexTransport transport) @nogc
|
* void connected(DuplexTransport transport) @nogc
|
||||||
* {
|
* {
|
||||||
* this.transport = transport;
|
* this.transport = transport;
|
||||||
* }
|
* }
|
||||||
*
|
*
|
||||||
* void disconnected(SocketException e) @nogc
|
* void disconnected(SocketException e) @nogc
|
||||||
* {
|
* {
|
||||||
* }
|
* }
|
||||||
* }
|
* }
|
||||||
*
|
*
|
||||||
* void main()
|
* void main()
|
||||||
* {
|
* {
|
||||||
* auto address = defaultAllocator.make!InternetAddress("127.0.0.1", cast(ushort) 8192);
|
* auto address = defaultAllocator.make!InternetAddress("127.0.0.1", cast(ushort) 8192);
|
||||||
*
|
*
|
||||||
* version (Windows)
|
* version (Windows)
|
||||||
* {
|
* {
|
||||||
* auto sock = defaultAllocator.make!OverlappedStreamSocket(AddressFamily.INET);
|
* auto sock = defaultAllocator.make!OverlappedStreamSocket(AddressFamily.inet);
|
||||||
* }
|
* }
|
||||||
* else
|
* else
|
||||||
* {
|
* {
|
||||||
* auto sock = defaultAllocator.make!StreamSocket(AddressFamily.INET);
|
* auto sock = defaultAllocator.make!StreamSocket(AddressFamily.inet);
|
||||||
* sock.blocking = false;
|
* sock.blocking = false;
|
||||||
* }
|
* }
|
||||||
*
|
*
|
||||||
* sock.bind(address);
|
* sock.bind(address);
|
||||||
* sock.listen(5);
|
* sock.listen(5);
|
||||||
*
|
*
|
||||||
* auto io = defaultAllocator.make!ConnectionWatcher(sock);
|
* auto io = defaultAllocator.make!ConnectionWatcher(sock);
|
||||||
* io.setProtocol!EchoProtocol;
|
* io.setProtocol!EchoProtocol;
|
||||||
*
|
*
|
||||||
* defaultLoop.start(io);
|
* defaultLoop.start(io);
|
||||||
* defaultLoop.run();
|
* defaultLoop.run();
|
||||||
*
|
*
|
||||||
* sock.shutdown();
|
* sock.shutdown();
|
||||||
* defaultAllocator.dispose(io);
|
* defaultAllocator.dispose(io);
|
||||||
* defaultAllocator.dispose(sock);
|
* defaultAllocator.dispose(sock);
|
||||||
* defaultAllocator.dispose(address);
|
* defaultAllocator.dispose(address);
|
||||||
* }
|
* }
|
||||||
* ---
|
* ---
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
|
* 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.async.loop;
|
module tanya.async.loop;
|
||||||
|
|
||||||
@ -79,35 +82,62 @@ import tanya.network.socket;
|
|||||||
version (DisableBackends)
|
version (DisableBackends)
|
||||||
{
|
{
|
||||||
}
|
}
|
||||||
|
else version (D_Ddoc)
|
||||||
|
{
|
||||||
|
}
|
||||||
else version (linux)
|
else version (linux)
|
||||||
{
|
{
|
||||||
import tanya.async.event.epoll;
|
import tanya.async.event.epoll;
|
||||||
version = Epoll;
|
version = Epoll;
|
||||||
}
|
}
|
||||||
else version (Windows)
|
else version (Windows)
|
||||||
{
|
{
|
||||||
import tanya.async.event.iocp;
|
import tanya.async.event.iocp;
|
||||||
version = IOCP;
|
version = IOCP;
|
||||||
}
|
}
|
||||||
else version (OSX)
|
else version (OSX)
|
||||||
{
|
{
|
||||||
version = Kqueue;
|
version = Kqueue;
|
||||||
}
|
}
|
||||||
else version (iOS)
|
else version (iOS)
|
||||||
{
|
{
|
||||||
version = Kqueue;
|
version = Kqueue;
|
||||||
}
|
}
|
||||||
else version (FreeBSD)
|
else version (FreeBSD)
|
||||||
{
|
{
|
||||||
version = Kqueue;
|
version = Kqueue;
|
||||||
}
|
}
|
||||||
else version (OpenBSD)
|
else version (OpenBSD)
|
||||||
{
|
{
|
||||||
version = Kqueue;
|
version = Kqueue;
|
||||||
}
|
}
|
||||||
else version (DragonFlyBSD)
|
else version (DragonFlyBSD)
|
||||||
{
|
{
|
||||||
version = Kqueue;
|
version = Kqueue;
|
||||||
|
}
|
||||||
|
version (unittest)
|
||||||
|
{
|
||||||
|
final class TestLoop : Loop
|
||||||
|
{
|
||||||
|
override protected bool reify(SocketWatcher watcher,
|
||||||
|
EventMask oldEvents,
|
||||||
|
EventMask events) @nogc
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
override protected void poll() @nogc
|
||||||
|
{
|
||||||
|
assert(!this.done);
|
||||||
|
unloop();
|
||||||
|
}
|
||||||
|
|
||||||
|
override protected @property uint maxEvents()
|
||||||
|
const pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return 64U;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -115,11 +145,11 @@ else version (DragonFlyBSD)
|
|||||||
*/
|
*/
|
||||||
enum Event : uint
|
enum Event : uint
|
||||||
{
|
{
|
||||||
none = 0x00, /// No events.
|
none = 0x00, /// No events.
|
||||||
read = 0x01, /// Non-blocking read call.
|
read = 0x01, /// Non-blocking read call.
|
||||||
write = 0x02, /// Non-blocking write call.
|
write = 0x02, /// Non-blocking write call.
|
||||||
accept = 0x04, /// Connection made.
|
accept = 0x04, /// Connection made.
|
||||||
error = 0x80000000, /// Sent when an error occurs.
|
error = 0x80000000, /// Sent when an error occurs.
|
||||||
}
|
}
|
||||||
|
|
||||||
alias EventMask = BitFlags!Event;
|
alias EventMask = BitFlags!Event;
|
||||||
@ -129,150 +159,194 @@ alias EventMask = BitFlags!Event;
|
|||||||
*/
|
*/
|
||||||
abstract class Loop
|
abstract class Loop
|
||||||
{
|
{
|
||||||
private bool done;
|
private bool done = true;
|
||||||
|
|
||||||
/// Pending watchers.
|
/// Pending watchers.
|
||||||
protected Queue!Watcher pendings;
|
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 uint maxEvents()
|
protected @property uint maxEvents()
|
||||||
const pure nothrow @safe @nogc
|
const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return 128U;
|
return 128U;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
private unittest
|
||||||
* Initializes the loop.
|
{
|
||||||
*/
|
auto loop = defaultAllocator.make!TestLoop;
|
||||||
this() @nogc
|
assert(loop.maxEvents == 64);
|
||||||
{
|
|
||||||
pendings = Queue!Watcher(MmapPool.instance);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
defaultAllocator.dispose(loop);
|
||||||
* Frees loop internals.
|
}
|
||||||
*/
|
|
||||||
~this() @nogc
|
|
||||||
{
|
|
||||||
foreach (w; pendings)
|
|
||||||
{
|
|
||||||
MmapPool.instance.dispose(w);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Starts the loop.
|
* Initializes the loop.
|
||||||
*/
|
*/
|
||||||
void run() @nogc
|
this() @nogc
|
||||||
{
|
{
|
||||||
done = false;
|
pendings = Queue!Watcher(MmapPool.instance);
|
||||||
do
|
}
|
||||||
{
|
|
||||||
poll();
|
|
||||||
|
|
||||||
// Invoke pendings
|
/**
|
||||||
foreach (ref w; pendings)
|
* Frees loop internals.
|
||||||
{
|
*/
|
||||||
w.invoke();
|
~this() @nogc
|
||||||
}
|
{
|
||||||
}
|
foreach (w; pendings)
|
||||||
while (!done);
|
{
|
||||||
}
|
MmapPool.instance.dispose(w);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Break out of the loop.
|
* Starts the loop.
|
||||||
*/
|
*/
|
||||||
void unloop() @safe pure nothrow @nogc
|
void run() @nogc
|
||||||
{
|
{
|
||||||
done = true;
|
this.done = false;
|
||||||
}
|
do
|
||||||
|
{
|
||||||
|
poll();
|
||||||
|
|
||||||
/**
|
// Invoke pendings
|
||||||
* Start watching.
|
foreach (ref w; this.pendings)
|
||||||
*
|
{
|
||||||
* Params:
|
w.invoke();
|
||||||
* watcher = Watcher.
|
}
|
||||||
*/
|
}
|
||||||
void start(ConnectionWatcher watcher) @nogc
|
while (!this.done);
|
||||||
{
|
}
|
||||||
if (watcher.active)
|
|
||||||
{
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
watcher.active = true;
|
|
||||||
|
|
||||||
reify(watcher, EventMask(Event.none), EventMask(Event.accept));
|
/**
|
||||||
}
|
* Break out of the loop.
|
||||||
|
*/
|
||||||
|
void unloop() @safe pure nothrow @nogc
|
||||||
|
{
|
||||||
|
this.done = true;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
private unittest
|
||||||
* Stop watching.
|
{
|
||||||
*
|
auto loop = defaultAllocator.make!TestLoop;
|
||||||
* Params:
|
assert(loop.done);
|
||||||
* watcher = Watcher.
|
|
||||||
*/
|
|
||||||
void stop(ConnectionWatcher watcher) @nogc
|
|
||||||
{
|
|
||||||
if (!watcher.active)
|
|
||||||
{
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
watcher.active = false;
|
|
||||||
|
|
||||||
reify(watcher, EventMask(Event.accept), EventMask(Event.none));
|
loop.run();
|
||||||
}
|
assert(loop.done);
|
||||||
|
|
||||||
/**
|
defaultAllocator.dispose(loop);
|
||||||
* 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.
|
|
||||||
*/
|
|
||||||
abstract protected bool reify(SocketWatcher watcher,
|
|
||||||
EventMask oldEvents,
|
|
||||||
EventMask events) @nogc;
|
|
||||||
|
|
||||||
/**
|
private unittest
|
||||||
* Returns: The blocking time.
|
{
|
||||||
*/
|
auto loop = defaultAllocator.make!TestLoop;
|
||||||
protected @property inout(Duration) blockTime()
|
auto watcher = defaultAllocator.make!DummyWatcher;
|
||||||
inout @safe pure nothrow @nogc
|
loop.pendings.enqueue(watcher);
|
||||||
{
|
|
||||||
// Don't block if we have to do.
|
|
||||||
return pendings.empty ? blockTime_ : Duration.zero;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
assert(!watcher.invoked);
|
||||||
* Sets the blocking time for IO watchers.
|
loop.run();
|
||||||
*
|
assert(watcher.invoked);
|
||||||
* Params:
|
|
||||||
* blockTime = The blocking time. Cannot be larger than
|
|
||||||
* $(D_PSYMBOL maxBlockTime).
|
|
||||||
*/
|
|
||||||
protected @property void blockTime(in Duration blockTime) @safe pure nothrow @nogc
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(blockTime <= 1.dur!"hours", "Too long to wait.");
|
|
||||||
assert(!blockTime.isNegative);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
blockTime_ = blockTime;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
defaultAllocator.dispose(loop);
|
||||||
* Does the actual polling.
|
defaultAllocator.dispose(watcher);
|
||||||
*/
|
}
|
||||||
abstract protected void poll() @nogc;
|
|
||||||
|
|
||||||
/// Maximal block time.
|
/**
|
||||||
protected Duration blockTime_ = 1.dur!"minutes";
|
* Start watching.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* watcher = Watcher.
|
||||||
|
*/
|
||||||
|
void start(ConnectionWatcher watcher) @nogc
|
||||||
|
{
|
||||||
|
if (watcher.active)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
watcher.active = true;
|
||||||
|
|
||||||
|
reify(watcher, EventMask(Event.none), EventMask(Event.accept));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Stop watching.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* watcher = Watcher.
|
||||||
|
*/
|
||||||
|
void stop(ConnectionWatcher watcher) @nogc
|
||||||
|
{
|
||||||
|
if (!watcher.active)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
watcher.active = false;
|
||||||
|
|
||||||
|
reify(watcher, EventMask(Event.accept), EventMask(Event.none));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 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.
|
||||||
|
*/
|
||||||
|
abstract protected bool reify(SocketWatcher watcher,
|
||||||
|
EventMask oldEvents,
|
||||||
|
EventMask events) @nogc;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: The blocking time.
|
||||||
|
*/
|
||||||
|
protected @property inout(Duration) blockTime()
|
||||||
|
inout @safe pure nothrow @nogc
|
||||||
|
{
|
||||||
|
// Don't block if we have to do.
|
||||||
|
return pendings.empty ? blockTime_ : Duration.zero;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sets the blocking time for IO watchers.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* blockTime = The blocking time. Cannot be larger than
|
||||||
|
* $(D_PSYMBOL maxBlockTime).
|
||||||
|
*/
|
||||||
|
protected @property void blockTime(in Duration blockTime) @safe pure nothrow @nogc
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(blockTime <= 1.dur!"hours", "Too long to wait.");
|
||||||
|
assert(!blockTime.isNegative);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
blockTime_ = blockTime;
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto loop = defaultAllocator.make!TestLoop;
|
||||||
|
assert(loop.blockTime == 1.dur!"minutes");
|
||||||
|
|
||||||
|
loop.blockTime = 2.dur!"minutes";
|
||||||
|
assert(loop.blockTime == 2.dur!"minutes");
|
||||||
|
|
||||||
|
defaultAllocator.dispose(loop);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Does the actual polling.
|
||||||
|
*/
|
||||||
|
abstract protected void poll() @nogc;
|
||||||
|
|
||||||
|
/// Maximal block time.
|
||||||
|
protected Duration blockTime_ = 1.dur!"minutes";
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -280,17 +354,17 @@ abstract class Loop
|
|||||||
*/
|
*/
|
||||||
class BadLoopException : Exception
|
class BadLoopException : Exception
|
||||||
{
|
{
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
* file = The file where the exception occurred.
|
* file = The file where the exception occurred.
|
||||||
* line = The line number where the exception occurred.
|
* line = The line number where the exception occurred.
|
||||||
* next = The previous exception in the chain of exceptions, if any.
|
* next = The previous exception in the chain of exceptions, if any.
|
||||||
*/
|
*/
|
||||||
this(string file = __FILE__, size_t line = __LINE__, Throwable next = null)
|
this(string file = __FILE__, size_t line = __LINE__, Throwable next = null)
|
||||||
pure nothrow const @safe @nogc
|
pure nothrow const @safe @nogc
|
||||||
{
|
{
|
||||||
super("Event loop cannot be initialized.", file, line, next);
|
super("Event loop cannot be initialized.", file, line, next);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -302,24 +376,24 @@ class BadLoopException : Exception
|
|||||||
*/
|
*/
|
||||||
@property Loop defaultLoop() @nogc
|
@property Loop defaultLoop() @nogc
|
||||||
{
|
{
|
||||||
if (defaultLoop_ !is null)
|
if (defaultLoop_ !is null)
|
||||||
{
|
{
|
||||||
return defaultLoop_;
|
return defaultLoop_;
|
||||||
}
|
}
|
||||||
version (Epoll)
|
version (Epoll)
|
||||||
{
|
{
|
||||||
defaultLoop_ = MmapPool.instance.make!EpollLoop;
|
defaultLoop_ = MmapPool.instance.make!EpollLoop;
|
||||||
}
|
}
|
||||||
else version (IOCP)
|
else version (IOCP)
|
||||||
{
|
{
|
||||||
defaultLoop_ = MmapPool.instance.make!IOCPLoop;
|
defaultLoop_ = MmapPool.instance.make!IOCPLoop;
|
||||||
}
|
}
|
||||||
else version (Kqueue)
|
else version (Kqueue)
|
||||||
{
|
{
|
||||||
import tanya.async.event.kqueue;
|
import tanya.async.event.kqueue;
|
||||||
defaultLoop_ = MmapPool.instance.make!KqueueLoop;
|
defaultLoop_ = MmapPool.instance.make!KqueueLoop;
|
||||||
}
|
}
|
||||||
return defaultLoop_;
|
return defaultLoop_;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -331,16 +405,29 @@ class BadLoopException : Exception
|
|||||||
* your implementation to this property.
|
* your implementation to this property.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* loop = The event loop.
|
* loop = The event loop.
|
||||||
*/
|
*/
|
||||||
@property void defaultLoop(Loop loop) @nogc
|
@property void defaultLoop(Loop loop) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(loop !is null);
|
assert(loop !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
defaultLoop_ = loop;
|
defaultLoop_ = loop;
|
||||||
}
|
}
|
||||||
|
|
||||||
private Loop defaultLoop_;
|
private Loop defaultLoop_;
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto oldLoop = defaultLoop_;
|
||||||
|
auto loop = defaultAllocator.make!TestLoop;
|
||||||
|
|
||||||
|
defaultLoop = loop;
|
||||||
|
assert(defaultLoop_ is loop);
|
||||||
|
assert(defaultLoop is loop);
|
||||||
|
|
||||||
|
defaultLoop_ = oldLoop;
|
||||||
|
defaultAllocator.dispose(loop);
|
||||||
|
}
|
||||||
|
@ -3,6 +3,8 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* This package provides asynchronous capabilities.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
|
@ -3,6 +3,12 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* This module contains protocol which handle data in asynchronous
|
||||||
|
* applications.
|
||||||
|
*
|
||||||
|
* When an event from the network arrives, a protocol method gets
|
||||||
|
* called and can respond to the event.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -18,28 +24,28 @@ import tanya.async.transport;
|
|||||||
*/
|
*/
|
||||||
interface Protocol
|
interface Protocol
|
||||||
{
|
{
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
* data = Read data.
|
* data = Read data.
|
||||||
*/
|
*/
|
||||||
void received(in ubyte[] data) @nogc;
|
void received(in ubyte[] data) @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Called when a connection is made.
|
* Called when a connection is made.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* transport = Protocol transport.
|
* transport = Protocol transport.
|
||||||
*/
|
*/
|
||||||
void connected(DuplexTransport transport) @nogc;
|
void connected(DuplexTransport transport) @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Called when a connection is lost.
|
* Called when a connection is lost.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* 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) @nogc;
|
void disconnected(SocketException exception) @nogc;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -3,6 +3,9 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* This module contains transports which are responsible for data dilvery
|
||||||
|
* between two parties of an asynchronous communication.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -32,13 +35,13 @@ interface ReadTransport : Transport
|
|||||||
*/
|
*/
|
||||||
interface WriteTransport : Transport
|
interface WriteTransport : Transport
|
||||||
{
|
{
|
||||||
/**
|
/**
|
||||||
* Write some data to the transport.
|
* Write some data to the transport.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* data = Data to send.
|
* data = Data to send.
|
||||||
*/
|
*/
|
||||||
void write(ubyte[] data) @nogc;
|
void write(ubyte[] data) @nogc;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -46,46 +49,46 @@ interface WriteTransport : Transport
|
|||||||
*/
|
*/
|
||||||
interface DuplexTransport : ReadTransport, WriteTransport
|
interface DuplexTransport : ReadTransport, WriteTransport
|
||||||
{
|
{
|
||||||
/**
|
/**
|
||||||
* Returns: Application protocol.
|
* Returns: Application protocol.
|
||||||
*
|
*
|
||||||
* Postcondition: $(D_INLINECODE protocol !is null)
|
* Postcondition: $(D_INLINECODE protocol !is null)
|
||||||
*/
|
*/
|
||||||
@property Protocol protocol() pure nothrow @safe @nogc
|
@property Protocol protocol() pure nothrow @safe @nogc
|
||||||
out (protocol)
|
out (protocol)
|
||||||
{
|
{
|
||||||
assert(protocol !is null);
|
assert(protocol !is null);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Switches the protocol.
|
* Switches the protocol.
|
||||||
*
|
*
|
||||||
* The protocol is deallocated by the event loop, it should currently be
|
* The protocol is deallocated by the event loop, it should currently be
|
||||||
* allocated with $(D_PSYMBOL MmapPool).
|
* allocated with $(D_PSYMBOL MmapPool).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* protocol = Application protocol.
|
* protocol = Application protocol.
|
||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE protocol !is null)
|
* Precondition: $(D_INLINECODE protocol !is null)
|
||||||
*/
|
*/
|
||||||
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
@property void protocol(Protocol protocol) pure nothrow @safe @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(protocol !is null);
|
assert(protocol !is null);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns $(D_PARAM true) if the transport is closing or closed.
|
* Returns $(D_PARAM true) if the transport is closing or closed.
|
||||||
*/
|
*/
|
||||||
bool isClosing() const pure nothrow @safe @nogc;
|
bool isClosing() const pure nothrow @safe @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Close the transport.
|
* Close the transport.
|
||||||
*
|
*
|
||||||
* Buffered data will be flushed. No more data will be received.
|
* Buffered data will be flushed. No more data will be received.
|
||||||
*/
|
*/
|
||||||
void close() @nogc;
|
void close() @nogc;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -93,8 +96,8 @@ interface DuplexTransport : ReadTransport, WriteTransport
|
|||||||
*/
|
*/
|
||||||
interface SocketTransport : Transport
|
interface SocketTransport : Transport
|
||||||
{
|
{
|
||||||
/**
|
/**
|
||||||
* Returns: Socket.
|
* Returns: Socket.
|
||||||
*/
|
*/
|
||||||
@property Socket socket() pure nothrow @safe @nogc;
|
@property Socket socket() pure nothrow @safe @nogc;
|
||||||
}
|
}
|
||||||
|
@ -3,6 +3,8 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* Watchers register user's interest in some event.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -27,13 +29,26 @@ import tanya.network.socket;
|
|||||||
*/
|
*/
|
||||||
abstract class Watcher
|
abstract class Watcher
|
||||||
{
|
{
|
||||||
/// Whether the watcher is active.
|
/// Whether the watcher is active.
|
||||||
bool active;
|
bool active;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Invoke some action on event.
|
* Invoke some action on event.
|
||||||
*/
|
*/
|
||||||
void invoke() @nogc;
|
void invoke() @nogc;
|
||||||
|
}
|
||||||
|
|
||||||
|
version (unittest)
|
||||||
|
{
|
||||||
|
final class DummyWatcher : Watcher
|
||||||
|
{
|
||||||
|
bool invoked;
|
||||||
|
|
||||||
|
override void invoke() @nogc
|
||||||
|
{
|
||||||
|
this.invoked = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -41,32 +56,32 @@ abstract class Watcher
|
|||||||
*/
|
*/
|
||||||
abstract class SocketWatcher : Watcher
|
abstract class SocketWatcher : Watcher
|
||||||
{
|
{
|
||||||
/// Watched socket.
|
/// Watched socket.
|
||||||
protected Socket socket_;
|
protected Socket socket_;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
* socket = Socket.
|
* socket = Socket.
|
||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE socket !is null)
|
* Precondition: $(D_INLINECODE socket !is null)
|
||||||
*/
|
*/
|
||||||
this(Socket socket) pure nothrow @safe @nogc
|
this(Socket socket) pure nothrow @safe @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(socket !is null);
|
assert(socket !is null);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
socket_ = socket;
|
socket_ = socket;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Socket.
|
* Returns: Socket.
|
||||||
*/
|
*/
|
||||||
@property Socket socket() pure nothrow @safe @nogc
|
@property Socket socket() pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return socket_;
|
return socket_;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -74,44 +89,44 @@ abstract class SocketWatcher : Watcher
|
|||||||
*/
|
*/
|
||||||
class ConnectionWatcher : SocketWatcher
|
class ConnectionWatcher : SocketWatcher
|
||||||
{
|
{
|
||||||
/// Incoming connection queue.
|
/// Incoming connection queue.
|
||||||
Queue!DuplexTransport incoming;
|
Queue!DuplexTransport incoming;
|
||||||
|
|
||||||
private Protocol delegate() @nogc protocolFactory;
|
private Protocol delegate() @nogc protocolFactory;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
* socket = Socket.
|
* socket = Socket.
|
||||||
*/
|
*/
|
||||||
this(Socket socket) @nogc
|
this(Socket socket) @nogc
|
||||||
{
|
{
|
||||||
super(socket);
|
super(socket);
|
||||||
incoming = Queue!DuplexTransport(MmapPool.instance);
|
incoming = Queue!DuplexTransport(MmapPool.instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
* P = Protocol should be used.
|
* P = Protocol should be used.
|
||||||
*/
|
*/
|
||||||
void setProtocol(P : Protocol)() @nogc
|
void setProtocol(P : Protocol)() @nogc
|
||||||
{
|
{
|
||||||
this.protocolFactory = () @nogc => cast(Protocol) MmapPool.instance.make!P;
|
this.protocolFactory = () @nogc => cast(Protocol) MmapPool.instance.make!P;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Invokes new connection callback.
|
* Invokes new connection callback.
|
||||||
*/
|
*/
|
||||||
override void invoke() @nogc
|
override void invoke() @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(protocolFactory !is null, "Protocol isn't set.");
|
assert(protocolFactory !is null, "Protocol isn't set.");
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
foreach (transport; incoming)
|
foreach (transport; incoming)
|
||||||
{
|
{
|
||||||
transport.protocol = protocolFactory();
|
transport.protocol = protocolFactory();
|
||||||
transport.protocol.connected(transport);
|
transport.protocol.connected(transport);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -3,6 +3,8 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* This module contains buffers designed for C-style input/output APIs.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -333,9 +335,9 @@ struct WriteBuffer(T = ubyte)
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
* 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)
|
* Precondition: $(D_INLINECODE size > 0 && allocator !is null)
|
||||||
*/
|
*/
|
||||||
@ -428,7 +430,7 @@ struct WriteBuffer(T = ubyte)
|
|||||||
* Appends data to the buffer.
|
* Appends data to the buffer.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* buffer = Buffer chunk got with $(D_PSYMBOL opIndex).
|
* buffer = Buffer chunk got with $(D_PSYMBOL opIndex).
|
||||||
*/
|
*/
|
||||||
ref WriteBuffer opOpAssign(string op)(in T[] buffer)
|
ref WriteBuffer opOpAssign(string op)(in T[] buffer)
|
||||||
if (op == "~")
|
if (op == "~")
|
||||||
@ -535,7 +537,7 @@ struct WriteBuffer(T = ubyte)
|
|||||||
* appropriately. Always call it after $(D_PSYMBOL opIndex).
|
* appropriately. Always call it after $(D_PSYMBOL opIndex).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* length = Length of the written data.
|
* length = Length of the written data.
|
||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD this).
|
* Returns: $(D_KEYWORD this).
|
||||||
*/
|
*/
|
||||||
|
@ -12,11 +12,100 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.container.entry;
|
module tanya.container.entry;
|
||||||
|
|
||||||
|
import std.traits;
|
||||||
|
import tanya.typecons;
|
||||||
|
|
||||||
package struct SEntry(T)
|
package struct SEntry(T)
|
||||||
{
|
{
|
||||||
/// Item content.
|
// Item content.
|
||||||
T content;
|
T content;
|
||||||
|
|
||||||
/// Next item.
|
// Next item.
|
||||||
SEntry* next;
|
SEntry* next;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
package struct DEntry(T)
|
||||||
|
{
|
||||||
|
// Item content.
|
||||||
|
T content;
|
||||||
|
|
||||||
|
// Previous and next item.
|
||||||
|
DEntry* next, prev;
|
||||||
|
}
|
||||||
|
|
||||||
|
package struct HashEntry(K, V)
|
||||||
|
{
|
||||||
|
this(ref K key, ref V value)
|
||||||
|
{
|
||||||
|
this.pair = Pair!(K, V)(key, value);
|
||||||
|
}
|
||||||
|
|
||||||
|
Pair!(K, V) pair;
|
||||||
|
HashEntry* next;
|
||||||
|
}
|
||||||
|
|
||||||
|
package enum BucketStatus : byte
|
||||||
|
{
|
||||||
|
deleted = -1,
|
||||||
|
empty = 0,
|
||||||
|
used = 1,
|
||||||
|
}
|
||||||
|
|
||||||
|
package struct Bucket(T)
|
||||||
|
{
|
||||||
|
this(ref T content)
|
||||||
|
{
|
||||||
|
this.content = content;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property void content(ref T content)
|
||||||
|
{
|
||||||
|
this.content_ = content;
|
||||||
|
this.status = BucketStatus.used;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property ref inout(T) content() inout
|
||||||
|
{
|
||||||
|
return this.content_;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool opEquals(ref T content)
|
||||||
|
{
|
||||||
|
if (this.status == BucketStatus.used && this.content == content)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool opEquals(ref T content) const
|
||||||
|
{
|
||||||
|
if (this.status == BucketStatus.used && this.content == content)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool opEquals(ref typeof(this) that)
|
||||||
|
{
|
||||||
|
return this.content == that.content && this.status == that.status;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool opEquals(ref typeof(this) that) const
|
||||||
|
{
|
||||||
|
return this.content == that.content && this.status == that.status;
|
||||||
|
}
|
||||||
|
|
||||||
|
void remove()
|
||||||
|
{
|
||||||
|
static if (hasElaborateDestructor!T)
|
||||||
|
{
|
||||||
|
destroy(this.content);
|
||||||
|
}
|
||||||
|
this.status = BucketStatus.deleted;
|
||||||
|
}
|
||||||
|
|
||||||
|
T content_;
|
||||||
|
BucketStatus status = BucketStatus.empty;
|
||||||
|
}
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -12,7 +12,31 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.container;
|
module tanya.container;
|
||||||
|
|
||||||
|
public import tanya.container.array;
|
||||||
public import tanya.container.buffer;
|
public import tanya.container.buffer;
|
||||||
|
public import tanya.container.set;
|
||||||
public import tanya.container.list;
|
public import tanya.container.list;
|
||||||
public import tanya.container.vector;
|
public import tanya.container.string;
|
||||||
public import tanya.container.queue;
|
public import tanya.container.queue;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Thrown if $(D_PSYMBOL Set) cannot insert a new element because the container
|
||||||
|
* is full.
|
||||||
|
*/
|
||||||
|
class HashContainerFullException : Exception
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Params:
|
||||||
|
* msg = The message for the exception.
|
||||||
|
* file = The file where the exception occurred.
|
||||||
|
* line = The line number where the exception occurred.
|
||||||
|
* next = The previous exception in the chain of exceptions, if any.
|
||||||
|
*/
|
||||||
|
this(string msg,
|
||||||
|
string file = __FILE__,
|
||||||
|
size_t line = __LINE__,
|
||||||
|
Throwable next = null) @nogc @safe pure nothrow
|
||||||
|
{
|
||||||
|
super(msg, file, line, next);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
710
source/tanya/container/set.d
Normal file
710
source/tanya/container/set.d
Normal file
@ -0,0 +1,710 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This module implements a $(D_PSYMBOL Set) container that stores unique
|
||||||
|
* values without any particular order.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.container.set;
|
||||||
|
|
||||||
|
import std.algorithm.mutation;
|
||||||
|
import std.traits;
|
||||||
|
import tanya.container;
|
||||||
|
import tanya.container.entry;
|
||||||
|
import tanya.memory;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Bidirectional range that iterates over the $(D_PSYMBOL Set)'s values.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* E = Element type.
|
||||||
|
*/
|
||||||
|
struct Range(E)
|
||||||
|
{
|
||||||
|
static if (isMutable!E)
|
||||||
|
{
|
||||||
|
private alias DataRange = Array!(Bucket!(Unqual!E)).Range;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
private alias DataRange = Array!(Bucket!(Unqual!E)).ConstRange;
|
||||||
|
}
|
||||||
|
private DataRange dataRange;
|
||||||
|
|
||||||
|
@disable this();
|
||||||
|
|
||||||
|
private this(DataRange dataRange)
|
||||||
|
{
|
||||||
|
while (!dataRange.empty && dataRange.front.status != BucketStatus.used)
|
||||||
|
{
|
||||||
|
dataRange.popFront();
|
||||||
|
}
|
||||||
|
while (!dataRange.empty && dataRange.back.status != BucketStatus.used)
|
||||||
|
{
|
||||||
|
dataRange.popBack();
|
||||||
|
}
|
||||||
|
this.dataRange = dataRange;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property Range save()
|
||||||
|
{
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property bool empty() const
|
||||||
|
{
|
||||||
|
return this.dataRange.empty();
|
||||||
|
}
|
||||||
|
|
||||||
|
@property void popFront()
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(!this.dataRange.empty);
|
||||||
|
assert(this.dataRange.front.status == BucketStatus.used);
|
||||||
|
}
|
||||||
|
out
|
||||||
|
{
|
||||||
|
assert(this.dataRange.empty
|
||||||
|
|| this.dataRange.back.status == BucketStatus.used);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
do
|
||||||
|
{
|
||||||
|
dataRange.popFront();
|
||||||
|
}
|
||||||
|
while (!dataRange.empty && dataRange.front.status != BucketStatus.used);
|
||||||
|
}
|
||||||
|
|
||||||
|
@property void popBack()
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(!this.dataRange.empty);
|
||||||
|
assert(this.dataRange.back.status == BucketStatus.used);
|
||||||
|
}
|
||||||
|
out
|
||||||
|
{
|
||||||
|
assert(this.dataRange.empty
|
||||||
|
|| this.dataRange.back.status == BucketStatus.used);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
do
|
||||||
|
{
|
||||||
|
dataRange.popBack();
|
||||||
|
}
|
||||||
|
while (!dataRange.empty && dataRange.back.status != BucketStatus.used);
|
||||||
|
}
|
||||||
|
|
||||||
|
@property ref inout(E) front() inout
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(!this.dataRange.empty);
|
||||||
|
assert(this.dataRange.front.status == BucketStatus.used);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return dataRange.front.content;
|
||||||
|
}
|
||||||
|
|
||||||
|
@property ref inout(E) back() inout
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(!this.dataRange.empty);
|
||||||
|
assert(this.dataRange.back.status == BucketStatus.used);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return dataRange.back.content;
|
||||||
|
}
|
||||||
|
|
||||||
|
Range opIndex()
|
||||||
|
{
|
||||||
|
return typeof(return)(this.dataRange[]);
|
||||||
|
}
|
||||||
|
|
||||||
|
Range!(const E) opIndex() const
|
||||||
|
{
|
||||||
|
return typeof(return)(this.dataRange[]);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Set is a data structure that stores unique values without any particular
|
||||||
|
* order.
|
||||||
|
*
|
||||||
|
* This $(D_PSYMBOL Set) is implemented using closed hashing. Hash collisions
|
||||||
|
* are resolved with linear probing.
|
||||||
|
*
|
||||||
|
* Currently works only with integral types.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Element type.
|
||||||
|
*/
|
||||||
|
struct Set(T)
|
||||||
|
if (isIntegral!T || is(Unqual!T == bool))
|
||||||
|
{
|
||||||
|
/// The range types for $(D_PSYMBOL Set).
|
||||||
|
alias Range = .Range!T;
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
alias ConstRange = .Range!(const T);
|
||||||
|
|
||||||
|
invariant
|
||||||
|
{
|
||||||
|
assert(this.lengthIndex < primes.length);
|
||||||
|
assert(this.data.length == 0
|
||||||
|
|| this.data.length == primes[this.lengthIndex]);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructor.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* n = Minimum number of buckets.
|
||||||
|
* allocator = Allocator.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null).
|
||||||
|
*/
|
||||||
|
this(const size_t n, shared Allocator allocator = defaultAllocator)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this(allocator);
|
||||||
|
rehash(n);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
this(shared Allocator allocator)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.data = typeof(this.data)(allocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
{
|
||||||
|
auto set = Set!int(defaultAllocator);
|
||||||
|
assert(set.capacity == 0);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
auto set = Set!int(8);
|
||||||
|
assert(set.capacity == 13);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Initializes this $(D_PARAM Set) from another one.
|
||||||
|
*
|
||||||
|
* If $(D_PARAM init) is passed by reference, it will be copied.
|
||||||
|
* If $(D_PARAM init) is passed by value, it will be moved.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* S = Source set type.
|
||||||
|
* init = Source set.
|
||||||
|
* allocator = Allocator.
|
||||||
|
*/
|
||||||
|
this(S)(ref S init, shared Allocator allocator = defaultAllocator)
|
||||||
|
if (is(Unqual!S == Set))
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.data = typeof(this.data)(init.data, allocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
this(S)(S init, shared Allocator allocator = defaultAllocator)
|
||||||
|
if (is(S == Set))
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.data = typeof(this.data)(move(init.data), allocator);
|
||||||
|
this.lengthIndex = init.lengthIndex;
|
||||||
|
init.lengthIndex = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Assigns another set.
|
||||||
|
*
|
||||||
|
* If $(D_PARAM that) is passed by reference, it will be copied.
|
||||||
|
* If $(D_PARAM that) is passed by value, it will be moved.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* S = Content type.
|
||||||
|
* that = The value should be assigned.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD this).
|
||||||
|
*/
|
||||||
|
ref typeof(this) opAssign(S)(ref S that)
|
||||||
|
if (is(Unqual!S == Set))
|
||||||
|
{
|
||||||
|
this.data = that.data;
|
||||||
|
this.lengthIndex = that.lengthIndex;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ref typeof(this) opAssign(S)(S that) @trusted
|
||||||
|
if (is(S == Set))
|
||||||
|
{
|
||||||
|
swap(this.data, that.data);
|
||||||
|
swap(this.lengthIndex, that.lengthIndex);
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Used allocator.
|
||||||
|
*
|
||||||
|
* Postcondition: $(D_INLINECODE allocator !is null)
|
||||||
|
*/
|
||||||
|
@property shared(Allocator) allocator() const
|
||||||
|
out (allocator)
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return cast(shared Allocator) this.data.allocator;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Maximum amount of elements this $(D_PSYMBOL Set) can hold without
|
||||||
|
* resizing and rehashing. Note that it doesn't mean that the
|
||||||
|
* $(D_PSYMBOL Set) will hold $(I exactly) $(D_PSYMBOL capacity) elements.
|
||||||
|
* $(D_PSYMBOL capacity) tells the size of the container under a best-case
|
||||||
|
* distribution of elements.
|
||||||
|
*
|
||||||
|
* Returns: $(D_PSYMBOL Set) capacity.
|
||||||
|
*/
|
||||||
|
@property size_t capacity() const
|
||||||
|
{
|
||||||
|
return this.data.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
Set!int set;
|
||||||
|
assert(set.capacity == 0);
|
||||||
|
|
||||||
|
set.insert(8);
|
||||||
|
assert(set.capacity == 3);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Iterates over the $(D_PSYMBOL Set) and counts the elements.
|
||||||
|
*
|
||||||
|
* Returns: Count of elements within the $(D_PSYMBOL Set).
|
||||||
|
*/
|
||||||
|
@property size_t length() const
|
||||||
|
{
|
||||||
|
size_t count;
|
||||||
|
foreach (ref e; this.data[])
|
||||||
|
{
|
||||||
|
if (e.status == BucketStatus.used)
|
||||||
|
{
|
||||||
|
++count;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
Set!int set;
|
||||||
|
assert(set.length == 0);
|
||||||
|
|
||||||
|
set.insert(8);
|
||||||
|
assert(set.length == 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
private static const size_t[41] primes = [
|
||||||
|
3, 7, 13, 23, 29, 37, 53, 71, 97, 131, 163, 193, 239, 293, 389, 521,
|
||||||
|
769, 919, 1103, 1327, 1543, 2333, 3079, 4861, 6151, 12289, 24593,
|
||||||
|
49157, 98317, 196613, 393241, 786433, 1572869, 3145739, 6291469,
|
||||||
|
12582917, 25165843, 139022417, 282312799, 573292817, 1164186217,
|
||||||
|
];
|
||||||
|
|
||||||
|
/// The maximum number of buckets the container can have.
|
||||||
|
enum size_t maxBucketCount = primes[$ - 1];
|
||||||
|
|
||||||
|
static private size_t calculateHash(ref T value)
|
||||||
|
{
|
||||||
|
static if (isIntegral!T || isSomeChar!T || is(T == bool))
|
||||||
|
{
|
||||||
|
return (cast(size_t) value);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
static assert(false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static private size_t locateBucket(ref const DataType buckets, size_t hash)
|
||||||
|
{
|
||||||
|
return hash % buckets.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
private enum InsertStatus : byte
|
||||||
|
{
|
||||||
|
found = -1,
|
||||||
|
failed = 0,
|
||||||
|
added = 1,
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Inserts the value in an empty or deleted bucket. If the value is
|
||||||
|
* already in there, does nothing and returns InsertStatus.found. If the
|
||||||
|
* hash array is full returns InsertStatus.failed.
|
||||||
|
*/
|
||||||
|
private InsertStatus insertInUnusedBucket(ref T value)
|
||||||
|
{
|
||||||
|
auto bucketPosition = locateBucket(this.data, calculateHash(value));
|
||||||
|
|
||||||
|
foreach (ref e; this.data[bucketPosition .. $])
|
||||||
|
{
|
||||||
|
if (e == value) // Already in the set.
|
||||||
|
{
|
||||||
|
return InsertStatus.found;
|
||||||
|
}
|
||||||
|
else if (e.status != BucketStatus.used) // Insert the value.
|
||||||
|
{
|
||||||
|
e.content = value;
|
||||||
|
return InsertStatus.added;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return InsertStatus.failed;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Inserts a new element.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* value = Element value.
|
||||||
|
*
|
||||||
|
* Returns: Amount of new elements inserted.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL HashContainerFullException) if the insertion failed.
|
||||||
|
*/
|
||||||
|
size_t insert(T value)
|
||||||
|
{
|
||||||
|
if (this.data.length == 0)
|
||||||
|
{
|
||||||
|
this.data = DataType(primes[0], allocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
InsertStatus status = insertInUnusedBucket(value);
|
||||||
|
for (; !status; status = insertInUnusedBucket(value))
|
||||||
|
{
|
||||||
|
if ((this.primes.length - 1) == this.lengthIndex)
|
||||||
|
{
|
||||||
|
throw make!HashContainerFullException(defaultAllocator,
|
||||||
|
"Set is full");
|
||||||
|
}
|
||||||
|
rehashToSize(this.lengthIndex + 1);
|
||||||
|
}
|
||||||
|
return status == InsertStatus.added;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
Set!int set;
|
||||||
|
assert(8 !in set);
|
||||||
|
|
||||||
|
assert(set.insert(8) == 1);
|
||||||
|
assert(set.length == 1);
|
||||||
|
assert(8 in set);
|
||||||
|
|
||||||
|
assert(set.insert(8) == 0);
|
||||||
|
assert(set.length == 1);
|
||||||
|
assert(8 in set);
|
||||||
|
|
||||||
|
assert(set.remove(8));
|
||||||
|
assert(set.insert(8) == 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Removes an element.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* value = Element value.
|
||||||
|
*
|
||||||
|
* Returns: Number of elements removed, which is in the container with
|
||||||
|
* unique values `1` if an element existed, and `0` otherwise.
|
||||||
|
*/
|
||||||
|
size_t remove(T value)
|
||||||
|
{
|
||||||
|
if (this.data.length == 0)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto bucketPosition = locateBucket(this.data, calculateHash(value));
|
||||||
|
foreach (ref e; this.data[bucketPosition .. $])
|
||||||
|
{
|
||||||
|
if (e == value) // Found.
|
||||||
|
{
|
||||||
|
e.remove();
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
else if (e.status == BucketStatus.empty)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
Set!int set;
|
||||||
|
assert(8 !in set);
|
||||||
|
|
||||||
|
set.insert(8);
|
||||||
|
assert(8 in set);
|
||||||
|
|
||||||
|
assert(set.remove(8) == 1);
|
||||||
|
assert(set.remove(8) == 0);
|
||||||
|
assert(8 !in set);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* $(D_KEYWORD in) operator.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* value = Element to be searched for.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD true) if the given element exists in the container,
|
||||||
|
* $(D_KEYWORD false) otherwise.
|
||||||
|
*/
|
||||||
|
bool opBinaryRight(string op : "in")(auto ref T value)
|
||||||
|
{
|
||||||
|
if (this.data.length == 0)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto bucketPosition = locateBucket(this.data, calculateHash(value));
|
||||||
|
foreach (ref e; this.data[bucketPosition .. $])
|
||||||
|
{
|
||||||
|
if (e == value) // Found.
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
else if (e.status == BucketStatus.empty)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
bool opBinaryRight(string op : "in")(auto ref const T value) const
|
||||||
|
{
|
||||||
|
if (this.data.length == 0)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
auto bucketPosition = locateBucket(this.data, calculateHash(value));
|
||||||
|
foreach (ref e; this.data[bucketPosition .. $])
|
||||||
|
{
|
||||||
|
if (e.status == BucketStatus.used && e.content == value) // Found.
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
else if (e.status == BucketStatus.empty)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
Set!int set;
|
||||||
|
|
||||||
|
assert(5 !in set);
|
||||||
|
set.insert(5);
|
||||||
|
assert(5 in set);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sets the number of buckets in the container to at least $(D_PARAM n)
|
||||||
|
* and rearranges all the elements according to their hash values.
|
||||||
|
*
|
||||||
|
* If $(D_PARAM n) is greater than the current $(D_PSYMBOL capacity)
|
||||||
|
* and lower than or equal to $(D_PSYMBOL maxBucketCount), a rehash is
|
||||||
|
* forced.
|
||||||
|
*
|
||||||
|
* If $(D_PARAM n) is greater than $(D_PSYMBOL maxBucketCount),
|
||||||
|
* $(D_PSYMBOL maxBucketCount) is used instead as a new number of buckets.
|
||||||
|
*
|
||||||
|
* If $(D_PARAM n) is equal to the current $(D_PSYMBOL capacity), rehashing
|
||||||
|
* is forced without resizing the container.
|
||||||
|
*
|
||||||
|
* If $(D_PARAM n) is lower than the current $(D_PSYMBOL capacity), the
|
||||||
|
* function may have no effect.
|
||||||
|
*
|
||||||
|
* Rehashing is automatically performed whenever the container needs space
|
||||||
|
* to insert new elements.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* n = Minimum number of buckets.
|
||||||
|
*/
|
||||||
|
void rehash(const size_t n)
|
||||||
|
{
|
||||||
|
size_t lengthIndex;
|
||||||
|
for (; lengthIndex < primes.length; ++lengthIndex)
|
||||||
|
{
|
||||||
|
if (primes[lengthIndex] >= n)
|
||||||
|
{
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
rehashToSize(lengthIndex);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Takes an index in the primes array.
|
||||||
|
private void rehashToSize(const size_t n)
|
||||||
|
{
|
||||||
|
auto storage = DataType(primes[n], allocator);
|
||||||
|
DataLoop: foreach (ref e1; this.data[])
|
||||||
|
{
|
||||||
|
if (e1.status == BucketStatus.used)
|
||||||
|
{
|
||||||
|
auto bucketPosition = locateBucket(storage,
|
||||||
|
calculateHash(e1.content));
|
||||||
|
|
||||||
|
foreach (ref e2; storage[bucketPosition .. $])
|
||||||
|
{
|
||||||
|
if (e2.status != BucketStatus.used) // Insert the value.
|
||||||
|
{
|
||||||
|
e2.content = e1.content;
|
||||||
|
continue DataLoop;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return; // Rehashing failed.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
move(storage, this.data);
|
||||||
|
this.lengthIndex = n;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: A bidirectional range that iterates over the $(D_PSYMBOL Set)'s
|
||||||
|
* elements.
|
||||||
|
*/
|
||||||
|
Range opIndex()
|
||||||
|
{
|
||||||
|
return typeof(return)(this.data[]);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ConstRange opIndex() const
|
||||||
|
{
|
||||||
|
return typeof(return)(this.data[]);
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
Set!int set;
|
||||||
|
assert(set[].empty);
|
||||||
|
|
||||||
|
set.insert(8);
|
||||||
|
assert(!set[].empty);
|
||||||
|
assert(set[].front == 8);
|
||||||
|
assert(set[].back == 8);
|
||||||
|
|
||||||
|
set.remove(8);
|
||||||
|
assert(set[].empty);
|
||||||
|
}
|
||||||
|
|
||||||
|
private alias DataType = Array!(Bucket!T);
|
||||||
|
private DataType data;
|
||||||
|
private size_t lengthIndex;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Basic insertion logic.
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
Set!int set;
|
||||||
|
|
||||||
|
assert(set.insert(5) == 1);
|
||||||
|
assert(set.data[0].status == BucketStatus.empty);
|
||||||
|
assert(set.data[1].status == BucketStatus.empty);
|
||||||
|
assert(set.data[2].content == 5 && set.data[2].status == BucketStatus.used);
|
||||||
|
assert(set.data.length == 3);
|
||||||
|
|
||||||
|
assert(set.insert(5) == 0);
|
||||||
|
assert(set.data[0].status == BucketStatus.empty);
|
||||||
|
assert(set.data[1].status == BucketStatus.empty);
|
||||||
|
assert(set.data[2].content == 5 && set.data[2].status == BucketStatus.used);
|
||||||
|
assert(set.data.length == 3);
|
||||||
|
|
||||||
|
assert(set.insert(9) == 1);
|
||||||
|
assert(set.data[0].content == 9 && set.data[0].status == BucketStatus.used);
|
||||||
|
assert(set.data[1].status == BucketStatus.empty);
|
||||||
|
assert(set.data[2].content == 5 && set.data[2].status == BucketStatus.used);
|
||||||
|
assert(set.data.length == 3);
|
||||||
|
|
||||||
|
assert(set.insert(7) == 1);
|
||||||
|
assert(set.insert(8) == 1);
|
||||||
|
assert(set.data[0].content == 7);
|
||||||
|
assert(set.data[1].content == 8);
|
||||||
|
assert(set.data[2].content == 9);
|
||||||
|
assert(set.data[3].status == BucketStatus.empty);
|
||||||
|
assert(set.data[5].content == 5);
|
||||||
|
assert(set.data.length == 7);
|
||||||
|
|
||||||
|
assert(set.insert(16) == 1);
|
||||||
|
assert(set.data[2].content == 9);
|
||||||
|
assert(set.data[3].content == 16);
|
||||||
|
assert(set.data[4].status == BucketStatus.empty);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Static checks.
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
import std.range.primitives;
|
||||||
|
|
||||||
|
static assert(isBidirectionalRange!(Set!int.ConstRange));
|
||||||
|
static assert(isBidirectionalRange!(Set!int.Range));
|
||||||
|
|
||||||
|
static assert(!isInfinite!(Set!int.Range));
|
||||||
|
static assert(!hasLength!(Set!int.Range));
|
||||||
|
|
||||||
|
static assert(is(Set!uint));
|
||||||
|
static assert(is(Set!long));
|
||||||
|
static assert(is(Set!ulong));
|
||||||
|
static assert(is(Set!short));
|
||||||
|
static assert(is(Set!ushort));
|
||||||
|
static assert(is(Set!bool));
|
||||||
|
}
|
1541
source/tanya/container/string.d
Normal file
1541
source/tanya/container/string.d
Normal file
File diff suppressed because it is too large
Load Diff
537
source/tanya/format/conv.d
Normal file
537
source/tanya/format/conv.d
Normal file
@ -0,0 +1,537 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This module provides functions for converting between different types.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.format.conv;
|
||||||
|
|
||||||
|
import std.traits;
|
||||||
|
import tanya.memory;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Thrown if a type conversion fails.
|
||||||
|
*/
|
||||||
|
final class ConvException : Exception
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Params:
|
||||||
|
* msg = The message for the exception.
|
||||||
|
* file = The file where the exception occurred.
|
||||||
|
* line = The line number where the exception occurred.
|
||||||
|
* next = The previous exception in the chain of exceptions, if any.
|
||||||
|
*/
|
||||||
|
this(string msg,
|
||||||
|
string file = __FILE__,
|
||||||
|
size_t line = __LINE__,
|
||||||
|
Throwable next = null) @nogc @safe pure nothrow
|
||||||
|
{
|
||||||
|
super(msg, file, line, next);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* If the source type $(D_PARAM From) and the target type $(D_PARAM To) are
|
||||||
|
* equal, does nothing. If $(D_PARAM From) can be implicitly converted to
|
||||||
|
* $(D_PARAM To), just returns $(D_PARAM from).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* To = Target type.
|
||||||
|
*
|
||||||
|
* Returns: $(D_PARAM from).
|
||||||
|
*/
|
||||||
|
template to(To)
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Params:
|
||||||
|
* From = Source type.
|
||||||
|
* from = Source value.
|
||||||
|
*/
|
||||||
|
ref To to(From)(ref From from)
|
||||||
|
if (is(To == From))
|
||||||
|
{
|
||||||
|
return from;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
To to(From)(From from)
|
||||||
|
if (is(Unqual!To == Unqual!From) || isNumeric!From && isFloatingPoint!To)
|
||||||
|
{
|
||||||
|
return from;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
auto val = 5.to!int();
|
||||||
|
assert(val == 5);
|
||||||
|
static assert(is(typeof(val) == int));
|
||||||
|
}
|
||||||
|
|
||||||
|
private pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
int val = 5;
|
||||||
|
assert(val.to!int() == 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Performs checked conversion from an integral type $(D_PARAM From) to an
|
||||||
|
* integral type $(D_PARAM To).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* From = Source type.
|
||||||
|
* To = Target type.
|
||||||
|
* from = Source value.
|
||||||
|
*
|
||||||
|
* Returns: $(D_PARAM from) converted to $(D_PARAM To).
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL ConvException) if $(D_PARAM from) is too small or too
|
||||||
|
* large to be represented by $(D_PARAM To).
|
||||||
|
*/
|
||||||
|
To to(To, From)(From from)
|
||||||
|
if (isIntegral!From
|
||||||
|
&& isIntegral!To
|
||||||
|
&& !is(Unqual!To == Unqual!From)
|
||||||
|
&& !is(To == enum))
|
||||||
|
{
|
||||||
|
static if ((isUnsigned!From && isSigned!To && From.sizeof == To.sizeof)
|
||||||
|
|| From.sizeof > To.sizeof)
|
||||||
|
{
|
||||||
|
if (from > To.max)
|
||||||
|
{
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Positive integer overflow");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
static if (isSigned!From)
|
||||||
|
{
|
||||||
|
static if (isUnsigned!To)
|
||||||
|
{
|
||||||
|
if (from < 0)
|
||||||
|
{
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Negative integer overflow");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else static if (From.sizeof > To.sizeof)
|
||||||
|
{
|
||||||
|
if (from < To.min)
|
||||||
|
{
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Negative integer overflow");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
static if (From.sizeof <= To.sizeof)
|
||||||
|
{
|
||||||
|
return from;
|
||||||
|
}
|
||||||
|
else static if (isSigned!To)
|
||||||
|
{
|
||||||
|
return cast(To) from;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return from & To.max;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
// ubyte -> ushort
|
||||||
|
assert((cast(ubyte) 0).to!ushort == 0);
|
||||||
|
assert((cast(ubyte) 1).to!ushort == 1);
|
||||||
|
assert((cast(ubyte) (ubyte.max - 1)).to!ushort == ubyte.max - 1);
|
||||||
|
assert((cast(ubyte) ubyte.max).to!ushort == ubyte.max);
|
||||||
|
|
||||||
|
// ubyte -> short
|
||||||
|
assert((cast(ubyte) 0).to!short == 0);
|
||||||
|
assert((cast(ubyte) 1).to!short == 1);
|
||||||
|
assert((cast(ubyte) (ubyte.max - 1)).to!short == ubyte.max - 1);
|
||||||
|
assert((cast(ubyte) ubyte.max).to!short == ubyte.max);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
// ubyte <- ushort
|
||||||
|
assert((cast(ushort) 0).to!ubyte == 0);
|
||||||
|
assert((cast(ushort) 1).to!ubyte == 1);
|
||||||
|
assert((cast(ushort) (ubyte.max - 1)).to!ubyte == ubyte.max - 1);
|
||||||
|
assert((cast(ushort) ubyte.max).to!ubyte == ubyte.max);
|
||||||
|
|
||||||
|
// ubyte <- short
|
||||||
|
assert((cast(short) 0).to!ubyte == 0);
|
||||||
|
assert((cast(short) 1).to!ubyte == 1);
|
||||||
|
assert((cast(short) (ubyte.max - 1)).to!ubyte == ubyte.max - 1);
|
||||||
|
assert((cast(short) ubyte.max).to!ubyte == ubyte.max);
|
||||||
|
|
||||||
|
// short <-> int
|
||||||
|
assert(short.min.to!int == short.min);
|
||||||
|
assert((short.min + 1).to!int == short.min + 1);
|
||||||
|
assert((cast(short) -1).to!int == -1);
|
||||||
|
assert((cast(short) 0).to!int == 0);
|
||||||
|
assert((cast(short) 1).to!int == 1);
|
||||||
|
assert((short.max - 1).to!int == short.max - 1);
|
||||||
|
assert(short.max.to!int == short.max);
|
||||||
|
|
||||||
|
assert((cast(int) short.min).to!short == short.min);
|
||||||
|
assert((cast(int) short.min + 1).to!short == short.min + 1);
|
||||||
|
assert((cast(int) -1).to!short == -1);
|
||||||
|
assert((cast(int) 0).to!short == 0);
|
||||||
|
assert((cast(int) 1).to!short == 1);
|
||||||
|
assert((cast(int) short.max - 1).to!short == short.max - 1);
|
||||||
|
assert((cast(int) short.max).to!short == short.max);
|
||||||
|
|
||||||
|
// uint <-> int
|
||||||
|
assert((cast(uint) 0).to!int == 0);
|
||||||
|
assert((cast(uint) 1).to!int == 1);
|
||||||
|
assert((cast(uint) (int.max - 1)).to!int == int.max - 1);
|
||||||
|
assert((cast(uint) int.max).to!int == int.max);
|
||||||
|
|
||||||
|
assert((cast(int) 0).to!uint == 0);
|
||||||
|
assert((cast(int) 1).to!uint == 1);
|
||||||
|
assert((cast(int) (int.max - 1)).to!uint == int.max - 1);
|
||||||
|
assert((cast(int) int.max).to!uint == int.max);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert(int.min.to!short == int.min);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert(int.max.to!short == int.max);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert(uint.max.to!ushort == ushort.max);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert((-1).to!uint == -1);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
enum Test : int
|
||||||
|
{
|
||||||
|
one,
|
||||||
|
two,
|
||||||
|
}
|
||||||
|
assert(Test.one.to!int == 0);
|
||||||
|
assert(Test.two.to!int == 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Converts a number to a boolean.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* From = Source type.
|
||||||
|
* To = Target type.
|
||||||
|
* from = Source value.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD true) if $(D_INLINECODE from > 0 && from <= 1),
|
||||||
|
* otherwise $(D_KEYWORD false).
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL ConvException) if $(D_PARAM from) is greater than `1` or
|
||||||
|
* less than `0`.
|
||||||
|
*/
|
||||||
|
To to(To, From)(From from)
|
||||||
|
if (isNumeric!From && is(Unqual!To == bool) && !is(Unqual!To == Unqual!From))
|
||||||
|
{
|
||||||
|
if (from == 0)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
else if (from < 0)
|
||||||
|
{
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Negative number overflow");
|
||||||
|
}
|
||||||
|
else if (from <= 1)
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Positive number overflow");
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
assert(0.0.to!bool == false);
|
||||||
|
assert(0.2.to!bool == true);
|
||||||
|
assert(0.5.to!bool == true);
|
||||||
|
assert(1.0.to!bool == true);
|
||||||
|
|
||||||
|
assert(0.to!bool == false);
|
||||||
|
assert(1.to!bool == true);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert((-1).to!bool == true);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert(2.to!bool == true);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Converts a boolean to a number. $(D_KEYWORD true) is `1`, $(D_KEYWORD false)
|
||||||
|
* is `0`.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* From = Source type.
|
||||||
|
* To = Target type.
|
||||||
|
* from = Source value.
|
||||||
|
*
|
||||||
|
* Returns: `1` if $(D_PARAM from) is $(D_KEYWORD true), otherwise `0`.
|
||||||
|
*/
|
||||||
|
To to(To, From)(From from)
|
||||||
|
if (is(Unqual!From == bool) && isNumeric!To && !is(Unqual!To == Unqual!From))
|
||||||
|
{
|
||||||
|
return from;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
assert(true.to!float == 1.0);
|
||||||
|
assert(true.to!double == 1.0);
|
||||||
|
assert(true.to!ubyte == 1);
|
||||||
|
assert(true.to!byte == 1);
|
||||||
|
assert(true.to!ushort == 1);
|
||||||
|
assert(true.to!short == 1);
|
||||||
|
assert(true.to!uint == 1);
|
||||||
|
assert(true.to!int == 1);
|
||||||
|
|
||||||
|
assert(false.to!float == 0);
|
||||||
|
assert(false.to!double == 0);
|
||||||
|
assert(false.to!ubyte == 0);
|
||||||
|
assert(false.to!byte == 0);
|
||||||
|
assert(false.to!ushort == 0);
|
||||||
|
assert(false.to!short == 0);
|
||||||
|
assert(false.to!uint == 0);
|
||||||
|
assert(false.to!int == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Converts a floating point number to an integral type.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* From = Source type.
|
||||||
|
* To = Target type.
|
||||||
|
* from = Source value.
|
||||||
|
*
|
||||||
|
* Returns: Truncated $(D_PARAM from) (everything after the decimal point is
|
||||||
|
* dropped).
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL ConvException) if
|
||||||
|
* $(D_INLINECODE from < To.min || from > To.max).
|
||||||
|
*/
|
||||||
|
To to(To, From)(From from)
|
||||||
|
if (isFloatingPoint!From
|
||||||
|
&& isIntegral!To
|
||||||
|
&& !is(Unqual!To == Unqual!From)
|
||||||
|
&& !is(To == enum))
|
||||||
|
{
|
||||||
|
if (from > To.max)
|
||||||
|
{
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Positive number overflow");
|
||||||
|
}
|
||||||
|
else if (from < To.min)
|
||||||
|
{
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Negative number overflow");
|
||||||
|
}
|
||||||
|
return cast(To) from;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
assert(1.5.to!int == 1);
|
||||||
|
assert(2147483646.5.to!int == 2147483646);
|
||||||
|
assert((-2147483647.5).to!int == -2147483647);
|
||||||
|
assert(2147483646.5.to!uint == 2147483646);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert(2147483647.5.to!int == 2147483647);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert((-2147483648.5).to!int == -2147483648);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert((-21474.5).to!uint == -21474);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Performs checked conversion from an integral type $(D_PARAM From) to an
|
||||||
|
* $(D_KEYWORD enum).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* From = Source type.
|
||||||
|
* To = Target type.
|
||||||
|
* from = Source value.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD enum) value.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL ConvException) if $(D_PARAM from) is not a member of
|
||||||
|
* $(D_PSYMBOL To).
|
||||||
|
|
||||||
|
*/
|
||||||
|
To to(To, From)(From from)
|
||||||
|
if (isIntegral!From && is(To == enum))
|
||||||
|
{
|
||||||
|
foreach (m; EnumMembers!To)
|
||||||
|
{
|
||||||
|
if (from == m)
|
||||||
|
{
|
||||||
|
return m;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
throw make!ConvException(defaultAllocator,
|
||||||
|
"Value not found in enum '" ~ To.stringof ~ "'");
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
enum Test : int
|
||||||
|
{
|
||||||
|
one,
|
||||||
|
two,
|
||||||
|
}
|
||||||
|
static assert(is(typeof(1.to!Test) == Test));
|
||||||
|
assert(0.to!Test == Test.one);
|
||||||
|
assert(1.to!Test == Test.two);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
enum Test : uint
|
||||||
|
{
|
||||||
|
one,
|
||||||
|
two,
|
||||||
|
}
|
||||||
|
|
||||||
|
ConvException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
assert(5.to!Test == Test.one);
|
||||||
|
}
|
||||||
|
catch (ConvException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
15
source/tanya/format/package.d
Normal file
15
source/tanya/format/package.d
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This package contains formatting and conversion functions.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.format;
|
||||||
|
|
||||||
|
public import tanya.format.conv;
|
@ -16,7 +16,7 @@ import std.algorithm;
|
|||||||
import std.ascii;
|
import std.ascii;
|
||||||
import std.range;
|
import std.range;
|
||||||
import std.traits;
|
import std.traits;
|
||||||
import tanya.container.vector;
|
import tanya.container.array;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -160,6 +160,45 @@ struct Integer
|
|||||||
assert(integer == 7383520307673030126);
|
assert(integer == 7383520307673030126);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructs the integer from a two's complement representation.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* R = Range type.
|
||||||
|
* value = Range.
|
||||||
|
* allocator = Allocator.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null)
|
||||||
|
*/
|
||||||
|
this(R)(R value,
|
||||||
|
shared Allocator allocator = defaultAllocator)
|
||||||
|
if (isBidirectionalRange!R && hasLength!R
|
||||||
|
&& is(Unqual!(ElementType!R) == ubyte))
|
||||||
|
{
|
||||||
|
this(Sign.positive, value, allocator);
|
||||||
|
|
||||||
|
if (!value.empty && ((value.front & 0x80) != 0))
|
||||||
|
{
|
||||||
|
// Negative number.
|
||||||
|
opOpAssign!"-"(exp2(countBits()));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
{
|
||||||
|
ubyte[8] range = [ 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xdd, 0xee ];
|
||||||
|
auto integer = Integer(range[]);
|
||||||
|
assert(integer == 7383520307673030126);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
ubyte[8] range = [ 0xe6, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xdd, 0xee ];
|
||||||
|
auto integer = Integer(range[]);
|
||||||
|
assert(integer == -1839851729181745682);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Copies the integer.
|
* Copies the integer.
|
||||||
*/
|
*/
|
||||||
@ -178,12 +217,64 @@ struct Integer
|
|||||||
allocator.resize(this.rep, 0);
|
allocator.resize(this.rep, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static private const short[16] bitCounts = [
|
||||||
|
4, 0, 1, 0, 2, 0, 1, 0, 3, 0, 1, 0, 2, 0, 1, 0
|
||||||
|
];
|
||||||
|
|
||||||
|
// Counts the number of LSBs before the first non-zero bit.
|
||||||
|
private ptrdiff_t countLSBs() const pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
if (this.size == 0)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
ptrdiff_t bits;
|
||||||
|
for (bits = 0; (bits < this.size) && (this.rep[bits] == 0); ++bits)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
digit nonZero = this.rep[bits];
|
||||||
|
bits *= digitBitCount;
|
||||||
|
|
||||||
|
/* now scan this digit until a 1 is found */
|
||||||
|
if ((nonZero & 0x01) == 0)
|
||||||
|
{
|
||||||
|
digit bitCountsPos;
|
||||||
|
do
|
||||||
|
{
|
||||||
|
bitCountsPos = nonZero & 0x0f;
|
||||||
|
bits += bitCounts[bitCountsPos];
|
||||||
|
nonZero >>= 4;
|
||||||
|
}
|
||||||
|
while (bitCountsPos == 0);
|
||||||
|
}
|
||||||
|
return bits;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Integer byte length.
|
* Returns: Number of bytes in the two's complement representation.
|
||||||
*/
|
*/
|
||||||
@property size_t length() const pure nothrow @safe @nogc
|
@property size_t length() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return (countBits() + 7) / 8; // Round up.
|
if (this.sign)
|
||||||
|
{
|
||||||
|
const bc = countBits();
|
||||||
|
auto length = bc + (8 - (bc & 0x07));
|
||||||
|
|
||||||
|
if (((countLSBs() + 1) == bc) && ((bc & 0x07) == 0))
|
||||||
|
{
|
||||||
|
--length;
|
||||||
|
}
|
||||||
|
return length / 8;
|
||||||
|
}
|
||||||
|
else if (this.size == 0)
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return (countBits() / 8) + 1;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -245,16 +336,10 @@ struct Integer
|
|||||||
ref Integer opAssign(T)(T value) nothrow @safe @nogc
|
ref Integer opAssign(T)(T value) nothrow @safe @nogc
|
||||||
if (is(T == Integer))
|
if (is(T == Integer))
|
||||||
{
|
{
|
||||||
if (this.allocator is value.allocator)
|
swap(this.rep, value.rep);
|
||||||
{
|
swap(this.sign, value.sign);
|
||||||
swap(this.rep, value.rep);
|
swap(this.size, value.size);
|
||||||
swap(this.sign, value.sign);
|
swap(this.allocator_, value.allocator_);
|
||||||
swap(this.size, value.size);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
this = value;
|
|
||||||
}
|
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1343,79 +1428,72 @@ struct Integer
|
|||||||
return result;
|
return result;
|
||||||
}
|
}
|
||||||
|
|
||||||
Vector!ubyte toVector() const nothrow @safe @nogc
|
// Returns 2^^n.
|
||||||
|
private Integer exp2(size_t n) const nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
Vector!ubyte vector;
|
auto ret = Integer(allocator);
|
||||||
bool firstBit;
|
const bytes = n / digitBitCount;
|
||||||
ubyte carry;
|
|
||||||
|
ret.grow(bytes + 1);
|
||||||
|
ret.size = bytes + 1;
|
||||||
|
ret.rep[bytes] = (cast(digit) 1) << (n % digitBitCount);
|
||||||
|
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Two's complement representation of the integer.
|
||||||
|
*/
|
||||||
|
Array!ubyte toArray() const nothrow @safe @nogc
|
||||||
|
out (array)
|
||||||
|
{
|
||||||
|
assert(array.length == length);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
Array!ubyte array;
|
||||||
|
|
||||||
if (this.size == 0)
|
if (this.size == 0)
|
||||||
{
|
{
|
||||||
return vector;
|
return array;
|
||||||
}
|
}
|
||||||
vector.reserve(this.size * digit.sizeof);
|
const bc = countBits();
|
||||||
|
const remainingBits = bc & 0x07;
|
||||||
|
|
||||||
// The first digit needs extra handling since it can have leading
|
array.reserve(bc / 8);
|
||||||
// non significant zeros.
|
if (remainingBits == 0)
|
||||||
int digitCount = digitBitCount - 8;
|
|
||||||
const first = this.rep[this.size - 1];
|
|
||||||
const prevBitCount = ((this.size - 1) * digitBitCount);
|
|
||||||
const fullBytesBitCount = ((prevBitCount - 1) / 8 + 1) * 8;
|
|
||||||
|
|
||||||
// Find out the right alignment of the first byte.
|
|
||||||
if ((fullBytesBitCount - prevBitCount) == 0)
|
|
||||||
{
|
{
|
||||||
digitCount -= digit.sizeof * 8 - digitBitCount;
|
array.insertBack(ubyte.init);
|
||||||
|
|
||||||
}
|
}
|
||||||
for (; digitCount >= 0; digitCount -= 8)
|
|
||||||
|
Integer tmp;
|
||||||
|
if (this.sign)
|
||||||
{
|
{
|
||||||
if (firstBit || ((first >> digitCount) != 0))
|
auto length = bc + (8 - remainingBits);
|
||||||
|
|
||||||
|
if (((countLSBs() + 1) == bc) && (remainingBits == 0))
|
||||||
{
|
{
|
||||||
firstBit = true;
|
length -= 8;
|
||||||
vector.insertBack(cast(ubyte) (first >> digitCount));
|
|
||||||
}
|
}
|
||||||
}
|
|
||||||
if (digitCount >= -8)
|
tmp = exp2(length) + this;
|
||||||
{
|
|
||||||
carry = (first << -digitCount) & 0xff;
|
|
||||||
digitCount += digitBitCount;
|
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
carry = 0;
|
tmp = this;
|
||||||
digitCount = digitBitCount - 8;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
foreach_reverse (d; this.rep[0 .. this.size - 1])
|
do
|
||||||
{
|
{
|
||||||
if (carry != 0) // Check the carry from the previous digit.
|
array.insertBack(cast(ubyte) (tmp.rep[0] & 0xff));
|
||||||
{
|
tmp >>= 8;
|
||||||
vector.insertBack(cast(ubyte) (carry | (d >> digitCount)));
|
|
||||||
digitCount -= 8;
|
|
||||||
}
|
|
||||||
// Write the digit by bytes.
|
|
||||||
for (; digitCount >= 0; digitCount -= 8)
|
|
||||||
{
|
|
||||||
vector.insertBack(cast(ubyte) (d >> digitCount));
|
|
||||||
}
|
|
||||||
// Check for an incomplete byte.
|
|
||||||
if (digitCount >= -8)
|
|
||||||
{
|
|
||||||
carry = (d << -digitCount) & 0xff;
|
|
||||||
digitCount += digitBitCount;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
carry = 0;
|
|
||||||
digitCount = digitBitCount - 8;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (carry != 0)
|
|
||||||
{
|
|
||||||
vector.insertBack(cast(ubyte) (carry >> (digitBitCount - digitCount)));
|
|
||||||
}
|
}
|
||||||
|
while (tmp != 0);
|
||||||
|
|
||||||
return vector;
|
array[].reverse();
|
||||||
|
|
||||||
|
return array;
|
||||||
}
|
}
|
||||||
|
|
||||||
///
|
///
|
||||||
@ -1425,15 +1503,15 @@ struct Integer
|
|||||||
auto integer = Integer(0x66778899aabbddee);
|
auto integer = Integer(0x66778899aabbddee);
|
||||||
ubyte[8] expected = [ 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xdd, 0xee ];
|
ubyte[8] expected = [ 0x66, 0x77, 0x88, 0x99, 0xaa, 0xbb, 0xdd, 0xee ];
|
||||||
|
|
||||||
auto vector = integer.toVector();
|
auto array = integer.toArray();
|
||||||
assert(equal(vector[], expected[]));
|
assert(equal(array[], expected[]));
|
||||||
}
|
}
|
||||||
{
|
{
|
||||||
auto integer = Integer(0x03);
|
auto integer = Integer(0x03);
|
||||||
ubyte[1] expected = [ 0x03 ];
|
ubyte[1] expected = [ 0x03 ];
|
||||||
|
|
||||||
auto vector = integer.toVector();
|
auto array = integer.toArray();
|
||||||
assert(equal(vector[], expected[]));
|
assert(equal(array[], expected[]));
|
||||||
}
|
}
|
||||||
{
|
{
|
||||||
ubyte[63] expected = [
|
ubyte[63] expected = [
|
||||||
@ -1448,8 +1526,8 @@ struct Integer
|
|||||||
];
|
];
|
||||||
auto integer = Integer(Sign.positive, expected[]);
|
auto integer = Integer(Sign.positive, expected[]);
|
||||||
|
|
||||||
auto vector = integer.toVector();
|
auto array = integer.toArray();
|
||||||
assert(equal(vector[], expected[]));
|
assert(equal(array[], expected[]));
|
||||||
}
|
}
|
||||||
{
|
{
|
||||||
ubyte[14] expected = [
|
ubyte[14] expected = [
|
||||||
@ -1458,8 +1536,8 @@ struct Integer
|
|||||||
];
|
];
|
||||||
auto integer = Integer(Sign.positive, expected[]);
|
auto integer = Integer(Sign.positive, expected[]);
|
||||||
|
|
||||||
auto vector = integer.toVector();
|
auto array = integer.toArray();
|
||||||
assert(equal(vector[], expected[]));
|
assert(equal(array[], expected[]));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3,6 +3,8 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* This package provides mathematical functions.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
|
@ -54,17 +54,17 @@ abstract class EntropySource
|
|||||||
/**
|
/**
|
||||||
* Returns: Minimum bytes required from the entropy source.
|
* Returns: Minimum bytes required from the entropy source.
|
||||||
*/
|
*/
|
||||||
@property immutable(ubyte) threshold() const @safe pure nothrow;
|
@property ubyte threshold() const pure nothrow @safe @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Whether this entropy source is strong.
|
* Returns: Whether this entropy source is strong.
|
||||||
*/
|
*/
|
||||||
@property immutable(bool) strong() const @safe pure nothrow;
|
@property bool strong() const pure nothrow @safe @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Amount of already generated entropy.
|
* Returns: Amount of already generated entropy.
|
||||||
*/
|
*/
|
||||||
@property ushort size() const @safe pure nothrow
|
@property ushort size() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return size_;
|
return size_;
|
||||||
}
|
}
|
||||||
@ -74,7 +74,7 @@ abstract class EntropySource
|
|||||||
* size = Amount of already generated entropy. Cannot be smaller than the
|
* size = Amount of already generated entropy. Cannot be smaller than the
|
||||||
* already set value.
|
* already set value.
|
||||||
*/
|
*/
|
||||||
@property void size(ushort size) @safe pure nothrow
|
@property void size(ushort size) pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
size_ = size;
|
size_ = size;
|
||||||
}
|
}
|
||||||
@ -89,12 +89,12 @@ abstract class EntropySource
|
|||||||
* Returns: Number of bytes that were copied to the $(D_PARAM output)
|
* Returns: Number of bytes that were copied to the $(D_PARAM output)
|
||||||
* or $(D_PSYMBOL Nullable!ubyte.init) on error.
|
* or $(D_PSYMBOL Nullable!ubyte.init) on error.
|
||||||
*/
|
*/
|
||||||
Nullable!ubyte poll(out ubyte[maxGather] output);
|
Nullable!ubyte poll(out ubyte[maxGather] output) @nogc;
|
||||||
}
|
}
|
||||||
|
|
||||||
version (linux)
|
version (linux)
|
||||||
{
|
{
|
||||||
extern (C) long syscall(long number, ...) nothrow;
|
extern (C) long syscall(long number, ...) nothrow @system @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Uses getrandom system call.
|
* Uses getrandom system call.
|
||||||
@ -104,7 +104,7 @@ version (linux)
|
|||||||
/**
|
/**
|
||||||
* Returns: Minimum bytes required from the entropy source.
|
* Returns: Minimum bytes required from the entropy source.
|
||||||
*/
|
*/
|
||||||
override @property immutable(ubyte) threshold() const @safe pure nothrow
|
override @property ubyte threshold() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return 32;
|
return 32;
|
||||||
}
|
}
|
||||||
@ -112,7 +112,7 @@ version (linux)
|
|||||||
/**
|
/**
|
||||||
* Returns: Whether this entropy source is strong.
|
* Returns: Whether this entropy source is strong.
|
||||||
*/
|
*/
|
||||||
override @property immutable(bool) strong() const @safe pure nothrow
|
override @property bool strong() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -127,7 +127,7 @@ version (linux)
|
|||||||
* Returns: Number of bytes that were copied to the $(D_PARAM output)
|
* Returns: Number of bytes that were copied to the $(D_PARAM output)
|
||||||
* or $(D_PSYMBOL Nullable!ubyte.init) on error.
|
* or $(D_PSYMBOL Nullable!ubyte.init) on error.
|
||||||
*/
|
*/
|
||||||
override Nullable!ubyte poll(out ubyte[maxGather] output) nothrow
|
override Nullable!ubyte poll(out ubyte[maxGather] output) nothrow @nogc
|
||||||
out (length)
|
out (length)
|
||||||
{
|
{
|
||||||
assert(length <= maxGather);
|
assert(length <= maxGather);
|
||||||
@ -145,6 +145,18 @@ version (linux)
|
|||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
version (X86_64)
|
||||||
|
{
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto entropy = defaultAllocator.make!Entropy();
|
||||||
|
ubyte[blockSize] output;
|
||||||
|
output = entropy.random;
|
||||||
|
|
||||||
|
defaultAllocator.dispose(entropy);
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -177,7 +189,8 @@ class Entropy
|
|||||||
* allocator = Allocator to allocate entropy sources available on the
|
* allocator = Allocator to allocate entropy sources available on the
|
||||||
* system.
|
* system.
|
||||||
*/
|
*/
|
||||||
this(size_t maxSources = 20, shared Allocator allocator = defaultAllocator)
|
this(const size_t maxSources = 20,
|
||||||
|
shared Allocator allocator = defaultAllocator) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(maxSources > 0 && maxSources <= ubyte.max);
|
assert(maxSources > 0 && maxSources <= ubyte.max);
|
||||||
@ -196,7 +209,7 @@ class Entropy
|
|||||||
/**
|
/**
|
||||||
* Returns: Amount of the registered entropy sources.
|
* Returns: Amount of the registered entropy sources.
|
||||||
*/
|
*/
|
||||||
@property ubyte sourceCount() const @safe pure nothrow
|
@property ubyte sourceCount() const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return sourceCount_;
|
return sourceCount_;
|
||||||
}
|
}
|
||||||
@ -212,7 +225,7 @@ class Entropy
|
|||||||
* See_Also:
|
* See_Also:
|
||||||
* $(D_PSYMBOL EntropySource)
|
* $(D_PSYMBOL EntropySource)
|
||||||
*/
|
*/
|
||||||
Entropy opOpAssign(string Op)(EntropySource source) @safe pure nothrow
|
Entropy opOpAssign(string Op)(EntropySource source) pure nothrow @safe @nogc
|
||||||
if (Op == "~")
|
if (Op == "~")
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
@ -230,7 +243,7 @@ class Entropy
|
|||||||
* Throws: $(D_PSYMBOL EntropyException) if no strong entropy source was
|
* Throws: $(D_PSYMBOL EntropyException) if no strong entropy source was
|
||||||
* registered or it failed.
|
* registered or it failed.
|
||||||
*/
|
*/
|
||||||
@property ubyte[blockSize] random()
|
@property ubyte[blockSize] random() @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(sourceCount_ > 0, "No entropy sources defined.");
|
assert(sourceCount_ > 0, "No entropy sources defined.");
|
||||||
@ -301,13 +314,13 @@ class Entropy
|
|||||||
*/
|
*/
|
||||||
protected void update(in ubyte sourceId,
|
protected void update(in ubyte sourceId,
|
||||||
ref ubyte[maxGather] data,
|
ref ubyte[maxGather] data,
|
||||||
ubyte length) @safe pure nothrow
|
ubyte length) pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
ubyte[2] header;
|
ubyte[2] header;
|
||||||
|
|
||||||
if (length > blockSize)
|
if (length > blockSize)
|
||||||
{
|
{
|
||||||
data[0..64] = sha512Of(data);
|
data[0 .. 64] = sha512Of(data);
|
||||||
length = blockSize;
|
length = blockSize;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -315,6 +328,6 @@ class Entropy
|
|||||||
header[1] = length;
|
header[1] = length;
|
||||||
|
|
||||||
accumulator.put(header);
|
accumulator.put(header);
|
||||||
accumulator.put(data[0..length]);
|
accumulator.put(data[0 .. length]);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -3,6 +3,11 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* This module contains the interface for implementing custom allocators.
|
||||||
|
*
|
||||||
|
* Allocators are classes encapsulating memory allocation strategy. This allows
|
||||||
|
* to decouple memory management from the algorithms and the data.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -28,7 +33,7 @@ interface Allocator
|
|||||||
*
|
*
|
||||||
* Returns: Pointer to the new allocated memory.
|
* Returns: Pointer to the new allocated memory.
|
||||||
*/
|
*/
|
||||||
void[] allocate(in size_t size) shared nothrow @nogc;
|
void[] allocate(const size_t size) shared nothrow @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Deallocates a memory block.
|
* Deallocates a memory block.
|
||||||
@ -49,7 +54,7 @@ interface Allocator
|
|||||||
*
|
*
|
||||||
* Returns: Pointer to the allocated memory.
|
* Returns: Pointer to the allocated memory.
|
||||||
*/
|
*/
|
||||||
bool reallocate(ref void[] p, in size_t size) shared nothrow @nogc;
|
bool reallocate(ref void[] p, const size_t size) shared nothrow @nogc;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Reallocates a memory block in place if possible or returns
|
* Reallocates a memory block in place if possible or returns
|
||||||
@ -63,5 +68,6 @@ interface Allocator
|
|||||||
*
|
*
|
||||||
* Returns: $(D_KEYWORD true) if successful, $(D_KEYWORD false) otherwise.
|
* Returns: $(D_KEYWORD true) if successful, $(D_KEYWORD false) otherwise.
|
||||||
*/
|
*/
|
||||||
bool reallocateInPlace(ref void[] p, in size_t size) shared nothrow @nogc;
|
bool reallocateInPlace(ref void[] p, const size_t size)
|
||||||
|
shared nothrow @nogc;
|
||||||
}
|
}
|
||||||
|
@ -3,6 +3,8 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* Allocator based on $(D_PSYMBOL malloc), $(D_PSYMBOL realloc) and $(D_PSYMBOL free).
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2017.
|
* Copyright: Eugene Wissner 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).
|
||||||
@ -11,11 +13,11 @@
|
|||||||
module tanya.memory.mallocator;
|
module tanya.memory.mallocator;
|
||||||
|
|
||||||
import core.stdc.stdlib;
|
import core.stdc.stdlib;
|
||||||
import std.algorithm.comparison;
|
|
||||||
import tanya.memory.allocator;
|
import tanya.memory.allocator;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Wrapper for malloc/realloc/free from the C standard library.
|
* Wrapper for $(D_PSYMBOL malloc)/$(D_PSYMBOL realloc)/$(D_PSYMBOL free) from
|
||||||
|
* the C standard library.
|
||||||
*/
|
*/
|
||||||
final class Mallocator : Allocator
|
final class Mallocator : Allocator
|
||||||
{
|
{
|
||||||
@ -27,9 +29,9 @@ final class Mallocator : Allocator
|
|||||||
*
|
*
|
||||||
* Returns: The pointer to the new allocated memory.
|
* Returns: The pointer to the new allocated memory.
|
||||||
*/
|
*/
|
||||||
void[] allocate(in size_t size) shared nothrow @nogc
|
void[] allocate(const size_t size) shared nothrow @nogc
|
||||||
{
|
{
|
||||||
if (!size)
|
if (size == 0)
|
||||||
{
|
{
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
@ -42,10 +44,11 @@ final class Mallocator : Allocator
|
|||||||
@nogc nothrow unittest
|
@nogc nothrow unittest
|
||||||
{
|
{
|
||||||
auto p = Mallocator.instance.allocate(20);
|
auto p = Mallocator.instance.allocate(20);
|
||||||
|
|
||||||
assert(p.length == 20);
|
assert(p.length == 20);
|
||||||
|
|
||||||
Mallocator.instance.deallocate(p);
|
Mallocator.instance.deallocate(p);
|
||||||
|
|
||||||
|
p = Mallocator.instance.allocate(0);
|
||||||
|
assert(p.length == 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -89,6 +92,13 @@ final class Mallocator : Allocator
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc nothrow unittest
|
||||||
|
{
|
||||||
|
void[] p;
|
||||||
|
assert(!Mallocator.instance.reallocateInPlace(p, 8));
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Increases or decreases the size of a memory block.
|
* Increases or decreases the size of a memory block.
|
||||||
*
|
*
|
||||||
@ -144,12 +154,27 @@ final class Mallocator : Allocator
|
|||||||
assert(p is null);
|
assert(p is null);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Fails with false.
|
||||||
|
private @nogc nothrow unittest
|
||||||
|
{
|
||||||
|
void[] p = Mallocator.instance.allocate(20);
|
||||||
|
void[] oldP = p;
|
||||||
|
assert(!Mallocator.instance.reallocate(p, size_t.max - Mallocator.psize * 2));
|
||||||
|
assert(oldP is p);
|
||||||
|
Mallocator.instance.deallocate(p);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: The alignment offered.
|
* Returns: The alignment offered.
|
||||||
*/
|
*/
|
||||||
@property uint alignment() shared const pure nothrow @safe @nogc
|
@property uint alignment() shared const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return cast(uint) max(double.alignof, real.alignof);
|
return (void*).alignof;
|
||||||
|
}
|
||||||
|
|
||||||
|
private nothrow @nogc unittest
|
||||||
|
{
|
||||||
|
assert(Mallocator.instance.alignment == (void*).alignof);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -161,7 +186,7 @@ final class Mallocator : Allocator
|
|||||||
{
|
{
|
||||||
if (instance_ is null)
|
if (instance_ is null)
|
||||||
{
|
{
|
||||||
immutable size = __traits(classInstanceSize, Mallocator) + psize;
|
const size = __traits(classInstanceSize, Mallocator) + psize;
|
||||||
void* p = malloc(size);
|
void* p = malloc(size);
|
||||||
|
|
||||||
if (p !is null)
|
if (p !is null)
|
||||||
@ -179,7 +204,7 @@ final class Mallocator : Allocator
|
|||||||
assert(instance is instance);
|
assert(instance is instance);
|
||||||
}
|
}
|
||||||
|
|
||||||
private enum psize = 8;
|
private enum ushort psize = 8;
|
||||||
|
|
||||||
private shared static Mallocator instance_;
|
private shared static Mallocator instance_;
|
||||||
}
|
}
|
||||||
|
File diff suppressed because it is too large
Load Diff
@ -3,6 +3,8 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* Dynamic memory management.
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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).
|
||||||
@ -15,6 +17,7 @@ import std.algorithm.iteration;
|
|||||||
public import std.experimental.allocator : make;
|
public import std.experimental.allocator : make;
|
||||||
import std.traits;
|
import std.traits;
|
||||||
public import tanya.memory.allocator;
|
public import tanya.memory.allocator;
|
||||||
|
import tanya.memory.mmappool;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* The mixin generates common methods for classes and structs using
|
* The mixin generates common methods for classes and structs using
|
||||||
@ -88,7 +91,6 @@ shared Allocator allocator;
|
|||||||
|
|
||||||
shared static this() nothrow @trusted @nogc
|
shared static this() nothrow @trusted @nogc
|
||||||
{
|
{
|
||||||
import tanya.memory.mmappool;
|
|
||||||
allocator = MmapPool.instance;
|
allocator = MmapPool.instance;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -112,6 +114,17 @@ body
|
|||||||
.allocator = allocator;
|
.allocator = allocator;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
private nothrow @nogc unittest
|
||||||
|
{
|
||||||
|
import tanya.memory.mallocator;
|
||||||
|
|
||||||
|
auto oldAllocator = defaultAllocator;
|
||||||
|
defaultAllocator = Mallocator.instance;
|
||||||
|
assert(defaultAllocator is Mallocator.instance);
|
||||||
|
|
||||||
|
defaultAllocator = oldAllocator;
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns the size in bytes of the state that needs to be allocated to hold an
|
* Returns the size in bytes of the state that needs to be allocated to hold an
|
||||||
* object of type $(D_PARAM T).
|
* object of type $(D_PARAM T).
|
||||||
@ -307,3 +320,19 @@ private unittest
|
|||||||
|
|
||||||
defaultAllocator.dispose(p);
|
defaultAllocator.dispose(p);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Works with interfaces.
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
interface I
|
||||||
|
{
|
||||||
|
}
|
||||||
|
class C : I
|
||||||
|
{
|
||||||
|
}
|
||||||
|
auto c = defaultAllocator.make!C();
|
||||||
|
I i = c;
|
||||||
|
|
||||||
|
defaultAllocator.dispose(i);
|
||||||
|
defaultAllocator.dispose(i);
|
||||||
|
}
|
||||||
|
924
source/tanya/memory/smartref.d
Normal file
924
source/tanya/memory/smartref.d
Normal file
@ -0,0 +1,924 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Smart pointers.
|
||||||
|
*
|
||||||
|
* A smart pointer is an object that wraps a raw pointer or a reference
|
||||||
|
* (class, array) to manage its lifetime.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
|
* 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.memory.smartref;
|
||||||
|
|
||||||
|
import core.exception;
|
||||||
|
import std.algorithm.comparison;
|
||||||
|
import std.algorithm.mutation;
|
||||||
|
import std.conv;
|
||||||
|
import std.range;
|
||||||
|
import std.traits;
|
||||||
|
import tanya.memory;
|
||||||
|
|
||||||
|
package template Payload(T)
|
||||||
|
{
|
||||||
|
static if (is(T == class) || is(T == interface) || isArray!T)
|
||||||
|
{
|
||||||
|
alias Payload = T;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
alias Payload = T*;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
package final class RefCountedStore(T)
|
||||||
|
{
|
||||||
|
T payload;
|
||||||
|
size_t counter = 1;
|
||||||
|
|
||||||
|
size_t opUnary(string op)()
|
||||||
|
if (op == "--" || op == "++")
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(this.counter > 0);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
mixin("return " ~ op ~ "counter;");
|
||||||
|
}
|
||||||
|
|
||||||
|
int opCmp(const size_t counter)
|
||||||
|
{
|
||||||
|
if (this.counter > counter)
|
||||||
|
{
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
else if (this.counter < counter)
|
||||||
|
{
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
package void separateDeleter(T)(RefCountedStore!T storage,
|
||||||
|
shared Allocator allocator)
|
||||||
|
{
|
||||||
|
allocator.dispose(storage.payload);
|
||||||
|
allocator.dispose(storage);
|
||||||
|
}
|
||||||
|
|
||||||
|
package void unifiedDeleter(T)(RefCountedStore!T storage,
|
||||||
|
shared Allocator allocator)
|
||||||
|
{
|
||||||
|
auto ptr1 = finalize(storage);
|
||||||
|
auto ptr2 = finalize(storage.payload);
|
||||||
|
allocator.deallocate(ptr1.ptr[0 .. ptr1.length + ptr2.length]);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Reference-counted object containing a $(D_PARAM T) value as payload.
|
||||||
|
* $(D_PSYMBOL RefCounted) keeps track of all references of an object, and
|
||||||
|
* when the reference count goes down to zero, frees the underlying store.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Type of the reference-counted value.
|
||||||
|
*/
|
||||||
|
struct RefCounted(T)
|
||||||
|
{
|
||||||
|
private alias Storage = RefCountedStore!(Payload!T);
|
||||||
|
|
||||||
|
private Storage storage;
|
||||||
|
private void function(Storage storage,
|
||||||
|
shared Allocator allocator) @nogc deleter;
|
||||||
|
|
||||||
|
invariant
|
||||||
|
{
|
||||||
|
assert(this.storage is null || this.allocator_ !is null);
|
||||||
|
assert(this.storage is null || this.deleter !is null);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Takes ownership over $(D_PARAM value), setting the counter to 1.
|
||||||
|
* $(D_PARAM value) may be a pointer, an object or a dynamic array.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* value = Value whose ownership is taken over.
|
||||||
|
* allocator = Allocator used to destroy the $(D_PARAM value) and to
|
||||||
|
* allocate/deallocate internal storage.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null)
|
||||||
|
*/
|
||||||
|
this(Payload!T value, shared Allocator allocator = defaultAllocator)
|
||||||
|
{
|
||||||
|
this(allocator);
|
||||||
|
this.storage = allocator.make!Storage();
|
||||||
|
this.deleter = &separateDeleter!(Payload!T);
|
||||||
|
|
||||||
|
this.storage.payload = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
this(shared Allocator allocator)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.allocator_ = allocator;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Increases the reference counter by one.
|
||||||
|
*/
|
||||||
|
this(this)
|
||||||
|
{
|
||||||
|
if (count != 0)
|
||||||
|
{
|
||||||
|
++this.storage;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Decreases the reference counter by one.
|
||||||
|
*
|
||||||
|
* If the counter reaches 0, destroys the owned object.
|
||||||
|
*/
|
||||||
|
~this()
|
||||||
|
{
|
||||||
|
if (this.storage !is null && !(this.storage > 0 && --this.storage))
|
||||||
|
{
|
||||||
|
deleter(this.storage, allocator);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Takes ownership over $(D_PARAM rhs). Initializes this
|
||||||
|
* $(D_PSYMBOL RefCounted) if needed.
|
||||||
|
*
|
||||||
|
* If it is the last reference of the previously owned object,
|
||||||
|
* it will be destroyed.
|
||||||
|
*
|
||||||
|
* To reset $(D_PSYMBOL RefCounted) assign $(D_KEYWORD null).
|
||||||
|
*
|
||||||
|
* If the allocator wasn't set before, $(D_PSYMBOL defaultAllocator) will
|
||||||
|
* be used. If you need a different allocator, create a new
|
||||||
|
* $(D_PSYMBOL RefCounted) and assign it.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* rhs = New object.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD this).
|
||||||
|
*/
|
||||||
|
ref typeof(this) opAssign(Payload!T rhs)
|
||||||
|
{
|
||||||
|
if (this.storage is null)
|
||||||
|
{
|
||||||
|
this.storage = allocator.make!Storage();
|
||||||
|
this.deleter = &separateDeleter!(Payload!T);
|
||||||
|
}
|
||||||
|
else if (this.storage > 1)
|
||||||
|
{
|
||||||
|
--this.storage;
|
||||||
|
this.storage = allocator.make!Storage();
|
||||||
|
this.deleter = &separateDeleter!(Payload!T);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
finalize(this.storage.payload);
|
||||||
|
this.storage.payload = Payload!T.init;
|
||||||
|
}
|
||||||
|
this.storage.payload = rhs;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!int(5);
|
||||||
|
rc = defaultAllocator.make!int(7);
|
||||||
|
assert(*rc == 7);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ref typeof(this) opAssign(typeof(null))
|
||||||
|
{
|
||||||
|
if (this.storage is null)
|
||||||
|
{
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
else if (this.storage > 1)
|
||||||
|
{
|
||||||
|
--this.storage;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
deleter(this.storage, allocator);
|
||||||
|
}
|
||||||
|
this.storage = null;
|
||||||
|
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
RefCounted!int rc;
|
||||||
|
assert(!rc.isInitialized);
|
||||||
|
rc = null;
|
||||||
|
assert(!rc.isInitialized);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ref typeof(this) opAssign(typeof(this) rhs)
|
||||||
|
{
|
||||||
|
swap(this.allocator_, rhs.allocator_);
|
||||||
|
swap(this.storage, rhs.storage);
|
||||||
|
swap(this.deleter, rhs.deleter);
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Reference to the owned object.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE cound > 0).
|
||||||
|
*/
|
||||||
|
inout(Payload!T) get() inout
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(count > 0, "Attempted to access an uninitialized reference");
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return this.storage.payload;
|
||||||
|
}
|
||||||
|
|
||||||
|
version (D_Ddoc)
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Dereferences the pointer. It is defined only for pointers, not for
|
||||||
|
* reference types like classes, that can be accessed directly.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* op = Operation.
|
||||||
|
*
|
||||||
|
* Returns: Reference to the pointed value.
|
||||||
|
*/
|
||||||
|
ref inout(T) opUnary(string op)() inout
|
||||||
|
if (op == "*");
|
||||||
|
}
|
||||||
|
else static if (isPointer!(Payload!T))
|
||||||
|
{
|
||||||
|
ref inout(T) opUnary(string op)() inout
|
||||||
|
if (op == "*")
|
||||||
|
{
|
||||||
|
return *this.storage.payload;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Whether this $(D_PSYMBOL RefCounted) already has an internal
|
||||||
|
* storage.
|
||||||
|
*/
|
||||||
|
@property bool isInitialized() const
|
||||||
|
{
|
||||||
|
return this.storage !is null;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: The number of $(D_PSYMBOL RefCounted) instances that share
|
||||||
|
* ownership over the same pointer (including $(D_KEYWORD this)).
|
||||||
|
* If this $(D_PSYMBOL RefCounted) isn't initialized, returns `0`.
|
||||||
|
*/
|
||||||
|
@property size_t count() const
|
||||||
|
{
|
||||||
|
return this.storage is null ? 0 : this.storage.counter;
|
||||||
|
}
|
||||||
|
|
||||||
|
mixin DefaultAllocator;
|
||||||
|
alias get this;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
auto rc = RefCounted!int(defaultAllocator.make!int(5), defaultAllocator);
|
||||||
|
auto val = rc.get();
|
||||||
|
|
||||||
|
*val = 8;
|
||||||
|
assert(*rc.storage.payload == 8);
|
||||||
|
|
||||||
|
val = null;
|
||||||
|
assert(rc.storage.payload !is null);
|
||||||
|
assert(*rc.storage.payload == 8);
|
||||||
|
|
||||||
|
*rc = 9;
|
||||||
|
assert(*rc.storage.payload == 9);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!int(5);
|
||||||
|
|
||||||
|
void func(RefCounted!int param) @nogc
|
||||||
|
{
|
||||||
|
assert(param.count == 2);
|
||||||
|
param = defaultAllocator.make!int(7);
|
||||||
|
assert(param.count == 1);
|
||||||
|
assert(*param == 7);
|
||||||
|
}
|
||||||
|
func(rc);
|
||||||
|
assert(rc.count == 1);
|
||||||
|
assert(*rc == 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
RefCounted!int rc;
|
||||||
|
|
||||||
|
void func(RefCounted!int param) @nogc
|
||||||
|
{
|
||||||
|
assert(param.count == 0);
|
||||||
|
param = defaultAllocator.make!int(7);
|
||||||
|
assert(param.count == 1);
|
||||||
|
assert(*param == 7);
|
||||||
|
}
|
||||||
|
func(rc);
|
||||||
|
assert(rc.count == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
RefCounted!int rc1, rc2;
|
||||||
|
static assert(is(typeof(rc1 = rc2)));
|
||||||
|
}
|
||||||
|
|
||||||
|
version (unittest)
|
||||||
|
{
|
||||||
|
private class A
|
||||||
|
{
|
||||||
|
uint *destroyed;
|
||||||
|
|
||||||
|
this(ref uint destroyed) @nogc
|
||||||
|
{
|
||||||
|
this.destroyed = &destroyed;
|
||||||
|
}
|
||||||
|
|
||||||
|
~this() @nogc
|
||||||
|
{
|
||||||
|
++(*destroyed);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private struct B
|
||||||
|
{
|
||||||
|
int prop;
|
||||||
|
@disable this();
|
||||||
|
this(int param1) @nogc
|
||||||
|
{
|
||||||
|
prop = param1;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
uint destroyed;
|
||||||
|
auto a = defaultAllocator.make!A(destroyed);
|
||||||
|
|
||||||
|
assert(destroyed == 0);
|
||||||
|
{
|
||||||
|
auto rc = RefCounted!A(a, defaultAllocator);
|
||||||
|
assert(rc.count == 1);
|
||||||
|
|
||||||
|
void func(RefCounted!A rc) @nogc
|
||||||
|
{
|
||||||
|
assert(rc.count == 2);
|
||||||
|
}
|
||||||
|
func(rc);
|
||||||
|
|
||||||
|
assert(rc.count == 1);
|
||||||
|
}
|
||||||
|
assert(destroyed == 1);
|
||||||
|
|
||||||
|
RefCounted!int rc;
|
||||||
|
assert(rc.count == 0);
|
||||||
|
rc = defaultAllocator.make!int(8);
|
||||||
|
assert(rc.count == 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto rc = RefCounted!int(defaultAllocator);
|
||||||
|
assert(!rc.isInitialized);
|
||||||
|
assert(rc.allocator is defaultAllocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!int(5);
|
||||||
|
assert(rc.count == 1);
|
||||||
|
|
||||||
|
void func(RefCounted!int rc) @nogc
|
||||||
|
{
|
||||||
|
assert(rc.count == 2);
|
||||||
|
rc = null;
|
||||||
|
assert(!rc.isInitialized);
|
||||||
|
assert(rc.count == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
assert(rc.count == 1);
|
||||||
|
func(rc);
|
||||||
|
assert(rc.count == 1);
|
||||||
|
|
||||||
|
rc = null;
|
||||||
|
assert(!rc.isInitialized);
|
||||||
|
assert(rc.count == 0);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!int(5);
|
||||||
|
assert(*rc == 5);
|
||||||
|
|
||||||
|
void func(RefCounted!int rc) @nogc
|
||||||
|
{
|
||||||
|
assert(rc.count == 2);
|
||||||
|
rc = defaultAllocator.refCounted!int(4);
|
||||||
|
assert(*rc == 4);
|
||||||
|
assert(rc.count == 1);
|
||||||
|
}
|
||||||
|
func(rc);
|
||||||
|
assert(*rc == 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
static assert(is(typeof(RefCounted!int.storage.payload) == int*));
|
||||||
|
static assert(is(typeof(RefCounted!A.storage.payload) == A));
|
||||||
|
|
||||||
|
static assert(is(RefCounted!B));
|
||||||
|
static assert(is(RefCounted!A));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructs a new object of type $(D_PARAM T) and wraps it in a
|
||||||
|
* $(D_PSYMBOL RefCounted) using $(D_PARAM args) as the parameter list for
|
||||||
|
* the constructor of $(D_PARAM T).
|
||||||
|
*
|
||||||
|
* This function is more efficient than the using of $(D_PSYMBOL RefCounted)
|
||||||
|
* directly, since it allocates only ones (the internal storage and the
|
||||||
|
* object).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Type of the constructed object.
|
||||||
|
* A = Types of the arguments to the constructor of $(D_PARAM T).
|
||||||
|
* allocator = Allocator.
|
||||||
|
* args = Constructor arguments of $(D_PARAM T).
|
||||||
|
*
|
||||||
|
* Returns: Newly created $(D_PSYMBOL RefCounted!T).
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null)
|
||||||
|
*/
|
||||||
|
RefCounted!T refCounted(T, A...)(shared Allocator allocator, auto ref A args)
|
||||||
|
if (!is(T == interface) && !isAbstractClass!T
|
||||||
|
&& !isAssociativeArray!T && !isArray!T)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
auto rc = typeof(return)(allocator);
|
||||||
|
|
||||||
|
const storageSize = alignedSize(stateSize!(RefCounted!T.Storage));
|
||||||
|
const size = alignedSize(stateSize!T + storageSize);
|
||||||
|
|
||||||
|
auto mem = (() @trusted => allocator.allocate(size))();
|
||||||
|
if (mem is null)
|
||||||
|
{
|
||||||
|
onOutOfMemoryError();
|
||||||
|
}
|
||||||
|
scope (failure)
|
||||||
|
{
|
||||||
|
() @trusted { allocator.deallocate(mem); }();
|
||||||
|
}
|
||||||
|
rc.storage = emplace!((RefCounted!T.Storage))(mem[0 .. storageSize]);
|
||||||
|
|
||||||
|
static if (is(T == class))
|
||||||
|
{
|
||||||
|
rc.storage.payload = emplace!T(mem[storageSize .. $], args);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
auto ptr = (() @trusted => (cast(T*) mem[storageSize .. $].ptr))();
|
||||||
|
rc.storage.payload = emplace!T(ptr, args);
|
||||||
|
}
|
||||||
|
rc.deleter = &unifiedDeleter!(Payload!T);
|
||||||
|
return rc;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructs a new array with $(D_PARAM size) elements and wraps it in a
|
||||||
|
* $(D_PSYMBOL RefCounted).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Array type.
|
||||||
|
* size = Array size.
|
||||||
|
* allocator = Allocator.
|
||||||
|
*
|
||||||
|
* Returns: Newly created $(D_PSYMBOL RefCounted!T).
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null
|
||||||
|
* && size <= size_t.max / ElementType!T.sizeof)
|
||||||
|
*/
|
||||||
|
RefCounted!T refCounted(T)(shared Allocator allocator, const size_t size)
|
||||||
|
@trusted
|
||||||
|
if (isArray!T)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
assert(size <= size_t.max / ElementType!T.sizeof);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
auto payload = allocator.resize!(ElementType!T)(null, size);
|
||||||
|
return RefCounted!T(payload, allocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!int(5);
|
||||||
|
assert(rc.count == 1);
|
||||||
|
|
||||||
|
void func(RefCounted!int param) @nogc
|
||||||
|
{
|
||||||
|
if (param.count == 2)
|
||||||
|
{
|
||||||
|
func(param);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
assert(param.count == 3);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
func(rc);
|
||||||
|
|
||||||
|
assert(rc.count == 1);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
struct E
|
||||||
|
{
|
||||||
|
}
|
||||||
|
auto b = defaultAllocator.refCounted!B(15);
|
||||||
|
static assert(is(typeof(b.storage.payload) == B*));
|
||||||
|
static assert(is(typeof(b.prop) == int));
|
||||||
|
static assert(!is(typeof(defaultAllocator.refCounted!B())));
|
||||||
|
|
||||||
|
static assert(is(typeof(defaultAllocator.refCounted!E())));
|
||||||
|
static assert(!is(typeof(defaultAllocator.refCounted!E(5))));
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!B(3);
|
||||||
|
assert(rc.get().prop == 3);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!E();
|
||||||
|
assert(rc.count);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!(int[])(5);
|
||||||
|
assert(rc.length == 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto p1 = defaultAllocator.make!int(5);
|
||||||
|
auto p2 = p1;
|
||||||
|
auto rc = RefCounted!int(p1, defaultAllocator);
|
||||||
|
assert(rc.get() is p2);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
static bool destroyed = false;
|
||||||
|
|
||||||
|
struct F
|
||||||
|
{
|
||||||
|
~this() @nogc
|
||||||
|
{
|
||||||
|
destroyed = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.refCounted!F();
|
||||||
|
}
|
||||||
|
assert(destroyed);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* $(D_PSYMBOL Unique) stores an object that gets destroyed at the end of its scope.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Value type.
|
||||||
|
*/
|
||||||
|
struct Unique(T)
|
||||||
|
{
|
||||||
|
private Payload!T payload;
|
||||||
|
|
||||||
|
invariant
|
||||||
|
{
|
||||||
|
assert(payload is null || allocator_ !is null);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Takes ownership over $(D_PARAM value), setting the counter to 1.
|
||||||
|
* $(D_PARAM value) may be a pointer, an object or a dynamic array.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* value = Value whose ownership is taken over.
|
||||||
|
* allocator = Allocator used to destroy the $(D_PARAM value) and to
|
||||||
|
* allocate/deallocate internal storage.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null)
|
||||||
|
*/
|
||||||
|
this(Payload!T value, shared Allocator allocator = defaultAllocator)
|
||||||
|
{
|
||||||
|
this(allocator);
|
||||||
|
this.payload = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
this(shared Allocator allocator)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.allocator_ = allocator;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* $(D_PSYMBOL Unique) is noncopyable.
|
||||||
|
*/
|
||||||
|
@disable this(this);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Destroys the owned object.
|
||||||
|
*/
|
||||||
|
~this()
|
||||||
|
{
|
||||||
|
allocator.dispose(this.payload);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Initialized this $(D_PARAM Unique) and takes ownership over
|
||||||
|
* $(D_PARAM rhs).
|
||||||
|
*
|
||||||
|
* To reset $(D_PSYMBOL Unique) assign $(D_KEYWORD null).
|
||||||
|
*
|
||||||
|
* If the allocator wasn't set before, $(D_PSYMBOL defaultAllocator) will
|
||||||
|
* be used. If you need a different allocator, create a new
|
||||||
|
* $(D_PSYMBOL Unique) and assign it.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* rhs = New object.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD this).
|
||||||
|
*/
|
||||||
|
ref typeof(this) opAssign(Payload!T rhs)
|
||||||
|
{
|
||||||
|
allocator.dispose(this.payload);
|
||||||
|
this.payload = rhs;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ref typeof(this) opAssign(typeof(null))
|
||||||
|
{
|
||||||
|
allocator.dispose(this.payload);
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ref typeof(this) opAssign(typeof(this) rhs)
|
||||||
|
{
|
||||||
|
swap(this.allocator_, rhs.allocator_);
|
||||||
|
swap(this.payload, rhs.payload);
|
||||||
|
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
auto rc = defaultAllocator.unique!int(5);
|
||||||
|
rc = defaultAllocator.make!int(7);
|
||||||
|
assert(*rc == 7);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Reference to the owned object.
|
||||||
|
*/
|
||||||
|
inout(Payload!T) get() inout
|
||||||
|
{
|
||||||
|
return this.payload;
|
||||||
|
}
|
||||||
|
|
||||||
|
version (D_Ddoc)
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Dereferences the pointer. It is defined only for pointers, not for
|
||||||
|
* reference types like classes, that can be accessed directly.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* op = Operation.
|
||||||
|
*
|
||||||
|
* Returns: Reference to the pointed value.
|
||||||
|
*/
|
||||||
|
ref inout(T) opUnary(string op)() inout
|
||||||
|
if (op == "*");
|
||||||
|
}
|
||||||
|
else static if (isPointer!(Payload!T))
|
||||||
|
{
|
||||||
|
ref inout(T) opUnary(string op)() inout
|
||||||
|
if (op == "*")
|
||||||
|
{
|
||||||
|
return *this.payload;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Whether this $(D_PSYMBOL Unique) holds some value.
|
||||||
|
*/
|
||||||
|
@property bool isInitialized() const
|
||||||
|
{
|
||||||
|
return this.payload !is null;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
Unique!int u;
|
||||||
|
assert(!u.isInitialized);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sets the internal pointer to $(D_KEYWORD). The allocator isn't changed.
|
||||||
|
*
|
||||||
|
* Returns: Reference to the owned object.
|
||||||
|
*/
|
||||||
|
Payload!T release()
|
||||||
|
{
|
||||||
|
auto payload = this.payload;
|
||||||
|
this.payload = null;
|
||||||
|
return payload;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
auto u = defaultAllocator.unique!int(5);
|
||||||
|
assert(u.isInitialized);
|
||||||
|
|
||||||
|
auto i = u.release();
|
||||||
|
assert(*i == 5);
|
||||||
|
assert(!u.isInitialized);
|
||||||
|
}
|
||||||
|
|
||||||
|
mixin DefaultAllocator;
|
||||||
|
alias get this;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
auto p = defaultAllocator.make!int(5);
|
||||||
|
auto s = Unique!int(p, defaultAllocator);
|
||||||
|
assert(*s == 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
static bool destroyed = false;
|
||||||
|
|
||||||
|
struct F
|
||||||
|
{
|
||||||
|
~this() @nogc
|
||||||
|
{
|
||||||
|
destroyed = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
{
|
||||||
|
auto s = Unique!F(defaultAllocator.make!F(), defaultAllocator);
|
||||||
|
}
|
||||||
|
assert(destroyed);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructs a new object of type $(D_PARAM T) and wraps it in a
|
||||||
|
* $(D_PSYMBOL Unique) using $(D_PARAM args) as the parameter list for
|
||||||
|
* the constructor of $(D_PARAM T).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Type of the constructed object.
|
||||||
|
* A = Types of the arguments to the constructor of $(D_PARAM T).
|
||||||
|
* allocator = Allocator.
|
||||||
|
* args = Constructor arguments of $(D_PARAM T).
|
||||||
|
*
|
||||||
|
* Returns: Newly created $(D_PSYMBOL Unique!T).
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null)
|
||||||
|
*/
|
||||||
|
Unique!T unique(T, A...)(shared Allocator allocator, auto ref A args)
|
||||||
|
if (!is(T == interface) && !isAbstractClass!T
|
||||||
|
&& !isAssociativeArray!T && !isArray!T)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
auto payload = allocator.make!(T, shared Allocator, A)(args);
|
||||||
|
return Unique!T(payload, allocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructs a new array with $(D_PARAM size) elements and wraps it in a
|
||||||
|
* $(D_PSYMBOL Unique).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Array type.
|
||||||
|
* size = Array size.
|
||||||
|
* allocator = Allocator.
|
||||||
|
*
|
||||||
|
* Returns: Newly created $(D_PSYMBOL Unique!T).
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE allocator !is null
|
||||||
|
* && size <= size_t.max / ElementType!T.sizeof)
|
||||||
|
*/
|
||||||
|
Unique!T unique(T)(shared Allocator allocator, const size_t size)
|
||||||
|
@trusted
|
||||||
|
if (isArray!T)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(allocator !is null);
|
||||||
|
assert(size <= size_t.max / ElementType!T.sizeof);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
auto payload = allocator.resize!(ElementType!T)(null, size);
|
||||||
|
return Unique!T(payload, allocator);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
static assert(is(typeof(defaultAllocator.unique!B(5))));
|
||||||
|
static assert(is(typeof(defaultAllocator.unique!(int[])(5))));
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto s = defaultAllocator.unique!int(5);
|
||||||
|
assert(*s == 5);
|
||||||
|
|
||||||
|
s = null;
|
||||||
|
assert(s is null);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto s = defaultAllocator.unique!int(5);
|
||||||
|
assert(*s == 5);
|
||||||
|
|
||||||
|
s = defaultAllocator.unique!int(4);
|
||||||
|
assert(*s == 4);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto p1 = defaultAllocator.make!int(5);
|
||||||
|
auto p2 = p1;
|
||||||
|
|
||||||
|
auto rc = Unique!int(p1, defaultAllocator);
|
||||||
|
assert(rc.get() is p2);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
auto rc = Unique!int(defaultAllocator);
|
||||||
|
assert(rc.allocator is defaultAllocator);
|
||||||
|
}
|
@ -3,11 +3,17 @@
|
|||||||
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
* Smart pointers.
|
||||||
|
*
|
||||||
|
* $(RED Deprecated. Use $(D_PSYMBOL tanya.memory.smartref) instead.
|
||||||
|
* This module will be removed in 0.8.0.)
|
||||||
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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)
|
||||||
*/
|
*/
|
||||||
|
deprecated("Use tanya.memory.smartref instead")
|
||||||
module tanya.memory.types;
|
module tanya.memory.types;
|
||||||
|
|
||||||
import core.exception;
|
import core.exception;
|
||||||
@ -17,88 +23,34 @@ import std.conv;
|
|||||||
import std.range;
|
import std.range;
|
||||||
import std.traits;
|
import std.traits;
|
||||||
import tanya.memory;
|
import tanya.memory;
|
||||||
|
public import tanya.memory.smartref : RefCounted, Payload;
|
||||||
|
|
||||||
package(tanya) final class RefCountedStore(T)
|
version (unittest)
|
||||||
{
|
{
|
||||||
T payload;
|
private struct B
|
||||||
size_t counter = 1;
|
|
||||||
|
|
||||||
size_t opUnary(string op)()
|
|
||||||
if (op == "--" || op == "++")
|
|
||||||
in
|
|
||||||
{
|
{
|
||||||
assert(counter > 0);
|
int prop;
|
||||||
}
|
@disable this();
|
||||||
body
|
this(int param1) @nogc
|
||||||
{
|
|
||||||
mixin("return " ~ op ~ "counter;");
|
|
||||||
}
|
|
||||||
|
|
||||||
int opCmp(size_t counter)
|
|
||||||
{
|
|
||||||
if (this.counter > counter)
|
|
||||||
{
|
{
|
||||||
return 1;
|
prop = param1;
|
||||||
}
|
|
||||||
else if (this.counter < counter)
|
|
||||||
{
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
int opEquals(size_t counter)
|
|
||||||
{
|
|
||||||
return this.counter == counter;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private void separateDeleter(T)(RefCountedStore!T storage,
|
|
||||||
shared Allocator allocator)
|
|
||||||
{
|
|
||||||
allocator.dispose(storage.payload);
|
|
||||||
allocator.dispose(storage);
|
|
||||||
}
|
|
||||||
|
|
||||||
private void unifiedDeleter(T)(RefCountedStore!T storage,
|
|
||||||
shared Allocator allocator)
|
|
||||||
{
|
|
||||||
auto ptr1 = finalize(storage);
|
|
||||||
auto ptr2 = finalize(storage.payload);
|
|
||||||
allocator.deallocate(ptr1.ptr[0 .. ptr1.length + ptr2.length]);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Reference-counted object containing a $(D_PARAM T) value as payload.
|
* $(D_PSYMBOL Scoped) stores an object that gets destroyed at the end of its scope.
|
||||||
* $(D_PSYMBOL RefCounted) keeps track of all references of an object, and
|
|
||||||
* when the reference count goes down to zero, frees the underlying store.
|
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* T = Type of the reference-counted value.
|
* T = Value type.
|
||||||
*/
|
*/
|
||||||
struct RefCounted(T)
|
struct Scoped(T)
|
||||||
{
|
{
|
||||||
static if (is(T == class) || is(T == interface) || isArray!T)
|
private Payload!T payload;
|
||||||
{
|
|
||||||
private alias Payload = T;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
private alias Payload = T*;
|
|
||||||
}
|
|
||||||
private alias Storage = RefCountedStore!Payload;
|
|
||||||
|
|
||||||
private Storage storage;
|
|
||||||
private void function(Storage storage,
|
|
||||||
shared Allocator allocator) @nogc deleter;
|
|
||||||
|
|
||||||
invariant
|
invariant
|
||||||
{
|
{
|
||||||
assert(storage is null || allocator_ !is null);
|
assert(payload is null || allocator_ !is null);
|
||||||
assert(storage is null || deleter !is null);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -112,13 +64,16 @@ struct RefCounted(T)
|
|||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE allocator !is null)
|
* Precondition: $(D_INLINECODE allocator !is null)
|
||||||
*/
|
*/
|
||||||
this()(auto ref Payload value,
|
this()(auto ref Payload!T value,
|
||||||
shared Allocator allocator = defaultAllocator)
|
shared Allocator allocator = defaultAllocator)
|
||||||
{
|
{
|
||||||
this(allocator);
|
this(allocator);
|
||||||
this.storage = allocator.make!Storage();
|
|
||||||
this.deleter = &separateDeleter!Payload;
|
move(value, this.payload);
|
||||||
move(value, this.storage.payload);
|
static if (__traits(isRef, value))
|
||||||
|
{
|
||||||
|
value = null;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
@ -133,88 +88,48 @@ struct RefCounted(T)
|
|||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Increases the reference counter by one.
|
* $(D_PSYMBOL Scoped) is noncopyable.
|
||||||
*/
|
*/
|
||||||
this(this)
|
@disable this(this);
|
||||||
{
|
|
||||||
if (count != 0)
|
|
||||||
{
|
|
||||||
++storage;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Decreases the reference counter by one.
|
* Destroys the owned object.
|
||||||
*
|
|
||||||
* If the counter reaches 0, destroys the owned object.
|
|
||||||
*/
|
*/
|
||||||
~this()
|
~this()
|
||||||
{
|
{
|
||||||
if (this.storage !is null && !(this.storage.counter && --this.storage))
|
if (this.payload !is null)
|
||||||
{
|
{
|
||||||
deleter(storage, allocator);
|
allocator.dispose(this.payload);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Takes ownership over $(D_PARAM rhs). Initializes this
|
* Initialized this $(D_PARAM Scoped) and takes ownership over
|
||||||
* $(D_PSYMBOL RefCounted) if needed.
|
* $(D_PARAM rhs).
|
||||||
*
|
*
|
||||||
* If it is the last reference of the previously owned object,
|
* To reset $(D_PSYMBOL Scoped) assign $(D_KEYWORD null).
|
||||||
* it will be destroyed.
|
|
||||||
*
|
|
||||||
* To reset the $(D_PSYMBOL RefCounted) assign $(D_KEYWORD null).
|
|
||||||
*
|
*
|
||||||
* If the allocator wasn't set before, $(D_PSYMBOL defaultAllocator) will
|
* If the allocator wasn't set before, $(D_PSYMBOL defaultAllocator) will
|
||||||
* be used. If you need a different allocator, create a new
|
* be used. If you need a different allocator, create a new
|
||||||
* $(D_PSYMBOL RefCounted) and assign it.
|
* $(D_PSYMBOL Scoped) and assign it.
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* rhs = $(D_KEYWORD this).
|
* rhs = New object.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD this).
|
||||||
*/
|
*/
|
||||||
ref typeof(this) opAssign()(auto ref Payload rhs)
|
ref typeof(this) opAssign()(auto ref Payload!T rhs)
|
||||||
{
|
{
|
||||||
if (this.storage is null)
|
allocator.dispose(this.payload);
|
||||||
{
|
move(rhs, this.payload);
|
||||||
this.storage = allocator.make!Storage();
|
|
||||||
this.deleter = &separateDeleter!Payload;
|
|
||||||
}
|
|
||||||
else if (this.storage > 1)
|
|
||||||
{
|
|
||||||
--this.storage;
|
|
||||||
this.storage = allocator.make!Storage();
|
|
||||||
this.deleter = &separateDeleter!Payload;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// Created with refCounted. Always destroyed togethter with the pointer.
|
|
||||||
assert(this.storage.counter != 0);
|
|
||||||
finalize(this.storage.payload);
|
|
||||||
this.storage.payload = Payload.init;
|
|
||||||
}
|
|
||||||
move(rhs, this.storage.payload);
|
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
ref typeof(this) opAssign(typeof(null))
|
ref typeof(this) opAssign(typeof(null))
|
||||||
{
|
{
|
||||||
if (this.storage is null)
|
allocator.dispose(this.payload);
|
||||||
{
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
else if (this.storage > 1)
|
|
||||||
{
|
|
||||||
--this.storage;
|
|
||||||
this.storage = null;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// Created with refCounted. Always destroyed togethter with the pointer.
|
|
||||||
assert(this.storage.counter != 0);
|
|
||||||
finalize(this.storage.payload);
|
|
||||||
this.storage.payload = Payload.init;
|
|
||||||
}
|
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -222,25 +137,20 @@ struct RefCounted(T)
|
|||||||
ref typeof(this) opAssign(typeof(this) rhs)
|
ref typeof(this) opAssign(typeof(this) rhs)
|
||||||
{
|
{
|
||||||
swap(this.allocator_, rhs.allocator_);
|
swap(this.allocator_, rhs.allocator_);
|
||||||
swap(this.storage, rhs.storage);
|
swap(this.payload, rhs.payload);
|
||||||
swap(this.deleter, rhs.deleter);
|
|
||||||
return this;
|
return this;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Reference to the owned object.
|
* Returns: Reference to the owned object.
|
||||||
*/
|
*/
|
||||||
inout(Payload) get() inout pure nothrow @safe @nogc
|
Payload!T get() pure nothrow @safe @nogc
|
||||||
in
|
|
||||||
{
|
{
|
||||||
assert(count > 0, "Attempted to access an uninitialized reference.");
|
return payload;
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
return storage.payload;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
static if (isPointer!Payload)
|
version (D_Ddoc)
|
||||||
{
|
{
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
@ -251,136 +161,65 @@ struct RefCounted(T)
|
|||||||
*
|
*
|
||||||
* Returns: Reference to the pointed value.
|
* Returns: Reference to the pointed value.
|
||||||
*/
|
*/
|
||||||
|
ref T opUnary(string op)()
|
||||||
|
if (op == "*");
|
||||||
|
}
|
||||||
|
else static if (isPointer!(Payload!T))
|
||||||
|
{
|
||||||
ref T opUnary(string op)()
|
ref T opUnary(string op)()
|
||||||
if (op == "*")
|
if (op == "*")
|
||||||
{
|
{
|
||||||
return *storage.payload;
|
return *payload;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Whether this $(D_PSYMBOL RefCounted) already has an internal
|
|
||||||
* storage.
|
|
||||||
*/
|
|
||||||
@property bool isInitialized() const
|
|
||||||
{
|
|
||||||
return storage !is null;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: The number of $(D_PSYMBOL RefCounted) instances that share
|
|
||||||
* ownership over the same pointer (including $(D_KEYWORD this)).
|
|
||||||
* If this $(D_PSYMBOL RefCounted) isn't initialized, returns `0`.
|
|
||||||
*/
|
|
||||||
@property size_t count() const
|
|
||||||
{
|
|
||||||
return storage is null ? 0 : storage.counter;
|
|
||||||
}
|
|
||||||
|
|
||||||
mixin DefaultAllocator;
|
mixin DefaultAllocator;
|
||||||
alias get this;
|
alias get this;
|
||||||
}
|
}
|
||||||
|
|
||||||
///
|
///
|
||||||
unittest
|
@nogc unittest
|
||||||
{
|
{
|
||||||
auto rc = RefCounted!int(defaultAllocator.make!int(5), defaultAllocator);
|
auto p = defaultAllocator.make!int(5);
|
||||||
auto val = rc.get;
|
auto s = Scoped!int(p, defaultAllocator);
|
||||||
|
assert(p is null);
|
||||||
*val = 8;
|
assert(*s == 5);
|
||||||
assert(*rc.storage.payload == 8);
|
|
||||||
|
|
||||||
val = null;
|
|
||||||
assert(rc.storage.payload !is null);
|
|
||||||
assert(*rc.storage.payload == 8);
|
|
||||||
|
|
||||||
*rc = 9;
|
|
||||||
assert(*rc.storage.payload == 9);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
version (unittest)
|
///
|
||||||
|
@nogc unittest
|
||||||
{
|
{
|
||||||
private class A
|
static bool destroyed = false;
|
||||||
|
|
||||||
|
struct F
|
||||||
{
|
{
|
||||||
uint *destroyed;
|
|
||||||
|
|
||||||
this(ref uint destroyed) @nogc
|
|
||||||
{
|
|
||||||
this.destroyed = &destroyed;
|
|
||||||
}
|
|
||||||
|
|
||||||
~this() @nogc
|
~this() @nogc
|
||||||
{
|
{
|
||||||
++(*destroyed);
|
destroyed = true;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private struct B
|
|
||||||
{
|
{
|
||||||
int prop;
|
auto s = Scoped!F(defaultAllocator.make!F(), defaultAllocator);
|
||||||
@disable this();
|
|
||||||
this(int param1) @nogc
|
|
||||||
{
|
|
||||||
prop = param1;
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
assert(destroyed);
|
||||||
|
|
||||||
private unittest
|
|
||||||
{
|
|
||||||
uint destroyed;
|
|
||||||
auto a = defaultAllocator.make!A(destroyed);
|
|
||||||
|
|
||||||
assert(destroyed == 0);
|
|
||||||
{
|
|
||||||
auto rc = RefCounted!A(a, defaultAllocator);
|
|
||||||
assert(rc.count == 1);
|
|
||||||
|
|
||||||
void func(RefCounted!A rc)
|
|
||||||
{
|
|
||||||
assert(rc.count == 2);
|
|
||||||
}
|
|
||||||
func(rc);
|
|
||||||
|
|
||||||
assert(rc.count == 1);
|
|
||||||
}
|
|
||||||
assert(destroyed == 1);
|
|
||||||
|
|
||||||
RefCounted!int rc;
|
|
||||||
assert(rc.count == 0);
|
|
||||||
rc = defaultAllocator.make!int(8);
|
|
||||||
assert(rc.count == 1);
|
|
||||||
}
|
|
||||||
|
|
||||||
private unittest
|
|
||||||
{
|
|
||||||
static assert(is(typeof(RefCounted!int.storage.payload) == int*));
|
|
||||||
static assert(is(typeof(RefCounted!A.storage.payload) == A));
|
|
||||||
|
|
||||||
static assert(is(RefCounted!B));
|
|
||||||
static assert(is(RefCounted!A));
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Constructs a new object of type $(D_PARAM T) and wraps it in a
|
* Constructs a new object of type $(D_PARAM T) and wraps it in a
|
||||||
* $(D_PSYMBOL RefCounted) using $(D_PARAM args) as the parameter list for
|
* $(D_PSYMBOL Scoped) using $(D_PARAM args) as the parameter list for
|
||||||
* the constructor of $(D_PARAM T).
|
* the constructor of $(D_PARAM T).
|
||||||
*
|
*
|
||||||
* This function is more efficient than the using of $(D_PSYMBOL RefCounted)
|
|
||||||
* directly, since it allocates only ones (the internal storage and the
|
|
||||||
* object).
|
|
||||||
*
|
|
||||||
* Params:
|
* Params:
|
||||||
* T = Type of the constructed object.
|
* T = Type of the constructed object.
|
||||||
* A = Types of the arguments to the constructor of $(D_PARAM T).
|
* A = Types of the arguments to the constructor of $(D_PARAM T).
|
||||||
* allocator = Allocator.
|
* allocator = Allocator.
|
||||||
* args = Constructor arguments of $(D_PARAM T).
|
* args = Constructor arguments of $(D_PARAM T).
|
||||||
*
|
*
|
||||||
* Returns: Newly created $(D_PSYMBOL RefCounted!T).
|
* Returns: Newly created $(D_PSYMBOL Scoped!T).
|
||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE allocator !is null)
|
* Precondition: $(D_INLINECODE allocator !is null)
|
||||||
*/
|
*/
|
||||||
RefCounted!T refCounted(T, A...)(shared Allocator allocator, auto ref A args)
|
Scoped!T scoped(T, A...)(shared Allocator allocator, auto ref A args)
|
||||||
if (!is(T == interface) && !isAbstractClass!T
|
if (!is(T == interface) && !isAbstractClass!T
|
||||||
&& !isAssociativeArray!T && !isArray!T)
|
&& !isAssociativeArray!T && !isArray!T)
|
||||||
in
|
in
|
||||||
@ -389,50 +228,26 @@ in
|
|||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
auto rc = typeof(return)(allocator);
|
auto payload = allocator.make!(T, shared Allocator, A)(args);
|
||||||
|
return Scoped!T(payload, allocator);
|
||||||
const storageSize = alignedSize(stateSize!(RefCounted!T.Storage));
|
|
||||||
const size = alignedSize(stateSize!T + storageSize);
|
|
||||||
|
|
||||||
auto mem = (() @trusted => allocator.allocate(size))();
|
|
||||||
if (mem is null)
|
|
||||||
{
|
|
||||||
onOutOfMemoryError();
|
|
||||||
}
|
|
||||||
scope (failure)
|
|
||||||
{
|
|
||||||
() @trusted { allocator.deallocate(mem); }();
|
|
||||||
}
|
|
||||||
rc.storage = emplace!((RefCounted!T.Storage))(mem[0 .. storageSize]);
|
|
||||||
|
|
||||||
static if (is(T == class))
|
|
||||||
{
|
|
||||||
rc.storage.payload = emplace!T(mem[storageSize .. $], args);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
auto ptr = (() @trusted => (cast(T*) mem[storageSize .. $].ptr))();
|
|
||||||
rc.storage.payload = emplace!T(ptr, args);
|
|
||||||
}
|
|
||||||
rc.deleter = &unifiedDeleter!(RefCounted!T.Payload);
|
|
||||||
return rc;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Constructs a new array with $(D_PARAM size) elements and wraps it in a
|
* Constructs a new array with $(D_PARAM size) elements and wraps it in a
|
||||||
* $(D_PSYMBOL RefCounted) using.
|
* $(D_PSYMBOL Scoped).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* T = Array type.
|
* T = Array type.
|
||||||
* size = Array size.
|
* size = Array size.
|
||||||
* allocator = Allocator.
|
* allocator = Allocator.
|
||||||
*
|
*
|
||||||
* Returns: Newly created $(D_PSYMBOL RefCounted!T).
|
* Returns: Newly created $(D_PSYMBOL Scoped!T).
|
||||||
*
|
*
|
||||||
* Precondition: $(D_INLINECODE allocator !is null
|
* Precondition: $(D_INLINECODE allocator !is null
|
||||||
* && size <= size_t.max / ElementType!T.sizeof)
|
* && size <= size_t.max / ElementType!T.sizeof)
|
||||||
*/
|
*/
|
||||||
RefCounted!T refCounted(T)(shared Allocator allocator, const size_t size)
|
Scoped!T scoped(T)(shared Allocator allocator, const size_t size)
|
||||||
|
@trusted
|
||||||
if (isArray!T)
|
if (isArray!T)
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
@ -441,57 +256,46 @@ in
|
|||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
auto payload = cast(T) allocator.allocate(ElementType!T.sizeof * size);
|
auto payload = allocator.resize!(ElementType!T)(null, size);
|
||||||
initializeAll(payload);
|
return Scoped!T(payload, allocator);
|
||||||
return RefCounted!T(payload, allocator);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
///
|
private unittest
|
||||||
unittest
|
|
||||||
{
|
{
|
||||||
auto rc = defaultAllocator.refCounted!int(5);
|
static assert(is(typeof(defaultAllocator.scoped!B(5))));
|
||||||
assert(rc.count == 1);
|
static assert(is(typeof(defaultAllocator.scoped!(int[])(5))));
|
||||||
|
}
|
||||||
|
|
||||||
void func(RefCounted!int param)
|
private unittest
|
||||||
{
|
{
|
||||||
if (param.count == 2)
|
auto s = defaultAllocator.scoped!int(5);
|
||||||
{
|
assert(*s == 5);
|
||||||
func(param);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
assert(param.count == 3);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
func(rc);
|
|
||||||
|
|
||||||
assert(rc.count == 1);
|
s = null;
|
||||||
|
assert(s is null);
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto s = defaultAllocator.scoped!int(5);
|
||||||
|
assert(*s == 5);
|
||||||
|
|
||||||
|
s = defaultAllocator.scoped!int(4);
|
||||||
|
assert(*s == 4);
|
||||||
}
|
}
|
||||||
|
|
||||||
private @nogc unittest
|
private @nogc unittest
|
||||||
{
|
{
|
||||||
struct E
|
auto p1 = defaultAllocator.make!int(5);
|
||||||
{
|
auto p2 = p1;
|
||||||
}
|
auto rc = Scoped!int(p1, defaultAllocator);
|
||||||
auto b = defaultAllocator.refCounted!B(15);
|
|
||||||
static assert(is(typeof(b.storage.payload) == B*));
|
|
||||||
static assert(is(typeof(b.prop) == int));
|
|
||||||
static assert(!is(typeof(defaultAllocator.refCounted!B())));
|
|
||||||
|
|
||||||
static assert(is(typeof(defaultAllocator.refCounted!E())));
|
assert(p1 is null);
|
||||||
static assert(!is(typeof(defaultAllocator.refCounted!E(5))));
|
assert(rc.get() is p2);
|
||||||
{
|
|
||||||
auto rc = defaultAllocator.refCounted!B(3);
|
|
||||||
assert(rc.get.prop == 3);
|
|
||||||
}
|
|
||||||
{
|
|
||||||
auto rc = defaultAllocator.refCounted!E();
|
|
||||||
assert(rc.count);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private @nogc unittest
|
private @nogc unittest
|
||||||
{
|
{
|
||||||
auto rc = defaultAllocator.refCounted!(int[])(5);
|
auto rc = Scoped!int(defaultAllocator);
|
||||||
assert(rc.length == 5);
|
assert(rc.allocator is defaultAllocator);
|
||||||
}
|
}
|
||||||
|
356
source/tanya/net/inet.d
Normal file
356
source/tanya/net/inet.d
Normal file
@ -0,0 +1,356 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Internet utilities.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2016-2017.
|
||||||
|
* 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.net.inet;
|
||||||
|
|
||||||
|
import std.math;
|
||||||
|
import std.range.primitives;
|
||||||
|
import std.traits;
|
||||||
|
|
||||||
|
version (unittest)
|
||||||
|
{
|
||||||
|
version (Windows)
|
||||||
|
{
|
||||||
|
import core.sys.windows.winsock2;
|
||||||
|
version = PlattformUnittest;
|
||||||
|
}
|
||||||
|
else version (Posix)
|
||||||
|
{
|
||||||
|
import core.sys.posix.arpa.inet;
|
||||||
|
version = PlattformUnittest;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Represents an unsigned integer as an $(D_KEYWORD ubyte) range.
|
||||||
|
*
|
||||||
|
* The range is bidirectional. The byte order is always big-endian.
|
||||||
|
*
|
||||||
|
* It can accept any unsigned integral type but the value should fit
|
||||||
|
* in $(D_PARAM L) bytes.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* L = Desired range length.
|
||||||
|
*/
|
||||||
|
struct NetworkOrder(uint L)
|
||||||
|
if (L > ubyte.sizeof && L <= ulong.sizeof)
|
||||||
|
{
|
||||||
|
static if (L > uint.sizeof)
|
||||||
|
{
|
||||||
|
private alias StorageType = ulong;
|
||||||
|
}
|
||||||
|
else static if (L > ushort.sizeof)
|
||||||
|
{
|
||||||
|
private alias StorageType = uint;
|
||||||
|
}
|
||||||
|
else static if (L > ubyte.sizeof)
|
||||||
|
{
|
||||||
|
private alias StorageType = ushort;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
private alias StorageType = ubyte;
|
||||||
|
}
|
||||||
|
|
||||||
|
private StorageType value;
|
||||||
|
private size_t size = L;
|
||||||
|
|
||||||
|
const pure nothrow @safe @nogc invariant
|
||||||
|
{
|
||||||
|
assert(this.size <= L);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructs a new range.
|
||||||
|
*
|
||||||
|
* $(D_PARAM T) can be any unsigned type but $(D_PARAM value) cannot be
|
||||||
|
* larger than the maximum can be stored in $(D_PARAM L) bytes. Otherwise
|
||||||
|
* an assertion failure will be caused.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Value type.
|
||||||
|
* value = The value should be represented by this range.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE value <= 2 ^^ (length * 8) - 1).
|
||||||
|
*/
|
||||||
|
this(T)(const T value)
|
||||||
|
if (isUnsigned!T)
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(value <= pow(2, L * 8) - 1);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.value = value & StorageType.max;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: LSB.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE length > 0).
|
||||||
|
*/
|
||||||
|
@property ubyte back() const
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(this.length > 0);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return this.value & 0xff;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: MSB.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE length > 0).
|
||||||
|
*/
|
||||||
|
@property ubyte front() const
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(this.length > 0);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
return (this.value >> ((this.length - 1) * 8)) & 0xff;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Eliminates the LSB.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE length > 0).
|
||||||
|
*/
|
||||||
|
void popBack()
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(this.length > 0);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.value >>= 8;
|
||||||
|
--this.size;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Eliminates the MSB.
|
||||||
|
*
|
||||||
|
* Precondition: $(D_INLINECODE length > 0).
|
||||||
|
*/
|
||||||
|
void popFront()
|
||||||
|
in
|
||||||
|
{
|
||||||
|
assert(this.length > 0);
|
||||||
|
}
|
||||||
|
body
|
||||||
|
{
|
||||||
|
this.value &= StorageType.max >> ((StorageType.sizeof - this.length) * 8);
|
||||||
|
--this.size;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Copy of this range.
|
||||||
|
*/
|
||||||
|
typeof(this) save() const
|
||||||
|
{
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Whether the range is empty.
|
||||||
|
*/
|
||||||
|
@property bool empty() const
|
||||||
|
{
|
||||||
|
return this.length == 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Byte length.
|
||||||
|
*/
|
||||||
|
@property size_t length() const
|
||||||
|
{
|
||||||
|
return this.size;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
auto networkOrder = NetworkOrder!3(0xae34e2u);
|
||||||
|
assert(!networkOrder.empty);
|
||||||
|
assert(networkOrder.front == 0xae);
|
||||||
|
|
||||||
|
networkOrder.popFront();
|
||||||
|
assert(networkOrder.length == 2);
|
||||||
|
assert(networkOrder.front == 0x34);
|
||||||
|
assert(networkOrder.back == 0xe2);
|
||||||
|
|
||||||
|
networkOrder.popBack();
|
||||||
|
assert(networkOrder.length == 1);
|
||||||
|
assert(networkOrder.front == 0x34);
|
||||||
|
assert(networkOrder.front == 0x34);
|
||||||
|
|
||||||
|
networkOrder.popFront();
|
||||||
|
assert(networkOrder.empty);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Static.
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
static assert(isBidirectionalRange!(NetworkOrder!4));
|
||||||
|
static assert(isBidirectionalRange!(NetworkOrder!8));
|
||||||
|
static assert(!is(NetworkOrder!9));
|
||||||
|
static assert(!is(NetworkOrder!1));
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tests against the system's htonl, htons.
|
||||||
|
version (PlattformUnittest)
|
||||||
|
{
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
for (uint counter; counter <= 8 * uint.sizeof; ++counter)
|
||||||
|
{
|
||||||
|
const value = pow(2, counter) - 1;
|
||||||
|
const inNetworkOrder = htonl(value);
|
||||||
|
const p = cast(ubyte*) &inNetworkOrder;
|
||||||
|
auto networkOrder = NetworkOrder!4(value);
|
||||||
|
|
||||||
|
assert(networkOrder.length == 4);
|
||||||
|
assert(!networkOrder.empty);
|
||||||
|
assert(networkOrder.front == *p);
|
||||||
|
assert(networkOrder.back == *(p + 3));
|
||||||
|
|
||||||
|
networkOrder.popBack();
|
||||||
|
assert(networkOrder.length == 3);
|
||||||
|
assert(networkOrder.front == *p);
|
||||||
|
assert(networkOrder.back == *(p + 2));
|
||||||
|
|
||||||
|
networkOrder.popFront();
|
||||||
|
assert(networkOrder.length == 2);
|
||||||
|
assert(networkOrder.front == *(p + 1));
|
||||||
|
assert(networkOrder.back == *(p + 2));
|
||||||
|
|
||||||
|
networkOrder.popFront();
|
||||||
|
assert(networkOrder.length == 1);
|
||||||
|
assert(networkOrder.front == *(p + 2));
|
||||||
|
assert(networkOrder.back == *(p + 2));
|
||||||
|
|
||||||
|
networkOrder.popBack();
|
||||||
|
assert(networkOrder.length == 0);
|
||||||
|
assert(networkOrder.empty);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (ushort counter; counter <= 8 * ushort.sizeof; ++counter)
|
||||||
|
{
|
||||||
|
const value = cast(ushort) (pow(2, counter) - 1);
|
||||||
|
const inNetworkOrder = htons(value);
|
||||||
|
const p = cast(ubyte*) &inNetworkOrder;
|
||||||
|
|
||||||
|
auto networkOrder = NetworkOrder!2(value);
|
||||||
|
|
||||||
|
assert(networkOrder.length == 2);
|
||||||
|
assert(!networkOrder.empty);
|
||||||
|
assert(networkOrder.front == *p);
|
||||||
|
assert(networkOrder.back == *(p + 1));
|
||||||
|
|
||||||
|
networkOrder.popBack();
|
||||||
|
assert(networkOrder.length == 1);
|
||||||
|
assert(networkOrder.front == *p);
|
||||||
|
assert(networkOrder.back == *p);
|
||||||
|
|
||||||
|
networkOrder.popBack();
|
||||||
|
assert(networkOrder.length == 0);
|
||||||
|
assert(networkOrder.empty);
|
||||||
|
|
||||||
|
networkOrder = NetworkOrder!2(value);
|
||||||
|
|
||||||
|
networkOrder.popFront();
|
||||||
|
assert(networkOrder.length == 1);
|
||||||
|
assert(networkOrder.front == *(p + 1));
|
||||||
|
assert(networkOrder.back == *(p + 1));
|
||||||
|
|
||||||
|
networkOrder.popFront();
|
||||||
|
assert(networkOrder.length == 0);
|
||||||
|
assert(networkOrder.empty);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Converts the $(D_KEYWORD ubyte) input range $(D_PARAM range) to
|
||||||
|
* $(D_PARAM T).
|
||||||
|
*
|
||||||
|
* The byte order of $(D_PARAM r) is assumed to be big-endian. The length
|
||||||
|
* cannot be larger than $(D_INLINECODE T.sizeof). Otherwise an assertion
|
||||||
|
* failure will be caused.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = Desired return type.
|
||||||
|
* R = Range type.
|
||||||
|
* range = Input range.
|
||||||
|
*
|
||||||
|
* Returns: Integral representation of $(D_PARAM range) with the host byte
|
||||||
|
* order.
|
||||||
|
*/
|
||||||
|
T toHostOrder(T = size_t, R)(R range)
|
||||||
|
if (isInputRange!R
|
||||||
|
&& !isInfinite!R
|
||||||
|
&& is(Unqual!(ElementType!R) == ubyte)
|
||||||
|
&& isUnsigned!T)
|
||||||
|
{
|
||||||
|
T ret;
|
||||||
|
ushort pos = T.sizeof * 8;
|
||||||
|
|
||||||
|
for (; !range.empty && range.front == 0; pos -= 8, range.popFront())
|
||||||
|
{
|
||||||
|
}
|
||||||
|
for (; !range.empty; range.popFront())
|
||||||
|
{
|
||||||
|
assert(pos != 0);
|
||||||
|
pos -= 8;
|
||||||
|
ret |= (cast(T) range.front) << pos;
|
||||||
|
}
|
||||||
|
|
||||||
|
return ret >> pos;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
const value = 0xae34e2u;
|
||||||
|
auto networkOrder = NetworkOrder!4(value);
|
||||||
|
assert(networkOrder.toHostOrder() == value);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Tests against the system's htonl, htons.
|
||||||
|
version (PlattformUnittest)
|
||||||
|
{
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
for (uint counter; counter <= 8 * uint.sizeof; ++counter)
|
||||||
|
{
|
||||||
|
const value = pow(2, counter) - 1;
|
||||||
|
const inNetworkOrder = htonl(value);
|
||||||
|
const p = cast(ubyte*) &inNetworkOrder;
|
||||||
|
auto networkOrder = NetworkOrder!4(value);
|
||||||
|
|
||||||
|
assert(p[0 .. uint.sizeof].toHostOrder() == value);
|
||||||
|
}
|
||||||
|
for (ushort counter; counter <= 8 * ushort.sizeof; ++counter)
|
||||||
|
{
|
||||||
|
const value = cast(ushort) (pow(2, counter) - 1);
|
||||||
|
const inNetworkOrder = htons(value);
|
||||||
|
const p = cast(ubyte*) &inNetworkOrder;
|
||||||
|
auto networkOrder = NetworkOrder!2(value);
|
||||||
|
|
||||||
|
assert(p[0 .. ushort.sizeof].toHostOrder() == value);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
16
source/tanya/net/package.d
Normal file
16
source/tanya/net/package.d
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Network programming.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.net;
|
||||||
|
|
||||||
|
public import tanya.net.inet;
|
||||||
|
public import tanya.net.uri;
|
580
source/tanya/net/uri.d
Normal file
580
source/tanya/net/uri.d
Normal file
@ -0,0 +1,580 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* URL parser.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.net.uri;
|
||||||
|
|
||||||
|
import std.ascii : isAlphaNum, isDigit;
|
||||||
|
import std.traits : isSomeString;
|
||||||
|
import std.uni : isAlpha, isNumber;
|
||||||
|
import tanya.memory;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Thrown if an invalid URI was specified.
|
||||||
|
*/
|
||||||
|
final class URIException : Exception
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Params:
|
||||||
|
* msg = The message for the exception.
|
||||||
|
* file = The file where the exception occurred.
|
||||||
|
* line = The line number where the exception occurred.
|
||||||
|
* next = The previous exception in the chain of exceptions, if any.
|
||||||
|
*/
|
||||||
|
this(string msg,
|
||||||
|
string file = __FILE__,
|
||||||
|
size_t line = __LINE__,
|
||||||
|
Throwable next = null) @nogc @safe pure nothrow
|
||||||
|
{
|
||||||
|
super(msg, file, line, next);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* A Unique Resource Locator.
|
||||||
|
*/
|
||||||
|
struct URL
|
||||||
|
{
|
||||||
|
/// The URL scheme.
|
||||||
|
const(char)[] scheme;
|
||||||
|
|
||||||
|
/// The username.
|
||||||
|
const(char)[] user;
|
||||||
|
|
||||||
|
/// The password.
|
||||||
|
const(char)[] pass;
|
||||||
|
|
||||||
|
/// The hostname.
|
||||||
|
const(char)[] host;
|
||||||
|
|
||||||
|
/// The port number.
|
||||||
|
ushort port;
|
||||||
|
|
||||||
|
/// The path.
|
||||||
|
const(char)[] path;
|
||||||
|
|
||||||
|
/// The query string.
|
||||||
|
const(char)[] query;
|
||||||
|
|
||||||
|
/// The anchor.
|
||||||
|
const(char)[] fragment;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Attempts to parse an URL from a string.
|
||||||
|
* Output string data (scheme, user, etc.) are just slices of input string
|
||||||
|
* (e.g., no memory allocation and copying).
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* source = The string containing the URL.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL URIException) if the URL is malformed.
|
||||||
|
*/
|
||||||
|
this(const char[] source) @nogc
|
||||||
|
{
|
||||||
|
auto value = source;
|
||||||
|
ptrdiff_t pos = -1, endPos = value.length, start;
|
||||||
|
|
||||||
|
foreach (i, ref c; source)
|
||||||
|
{
|
||||||
|
if (pos == -1 && c == ':')
|
||||||
|
{
|
||||||
|
pos = i;
|
||||||
|
}
|
||||||
|
if (endPos == value.length && (c == '?' || c == '#'))
|
||||||
|
{
|
||||||
|
endPos = i;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if the colon is a part of the scheme or the port and parse
|
||||||
|
// the appropriate part
|
||||||
|
if (value.length > 1 && value[0] == '/' && value[1] == '/')
|
||||||
|
{
|
||||||
|
// Relative scheme
|
||||||
|
start = 2;
|
||||||
|
}
|
||||||
|
else if (pos > 0)
|
||||||
|
{
|
||||||
|
// Validate scheme
|
||||||
|
// [ toLower(alpha) | digit | "+" | "-" | "." ]
|
||||||
|
foreach (ref c; value[0 .. pos])
|
||||||
|
{
|
||||||
|
if (!c.isAlphaNum && c != '+' && c != '-' && c != '.')
|
||||||
|
{
|
||||||
|
if (endPos > pos)
|
||||||
|
{
|
||||||
|
if (!parsePort(value[pos .. $]))
|
||||||
|
{
|
||||||
|
throw make!URIException(defaultAllocator,
|
||||||
|
"Failed to parse port");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
goto ParsePath;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (value.length == pos + 1) // only scheme is available
|
||||||
|
{
|
||||||
|
this.scheme = value[0 .. $ - 1];
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if (value.length > pos + 1 && value[pos + 1] == '/')
|
||||||
|
{
|
||||||
|
this.scheme = value[0 .. pos];
|
||||||
|
|
||||||
|
if (value.length > pos + 2 && value[pos + 2] == '/')
|
||||||
|
{
|
||||||
|
start = pos + 3;
|
||||||
|
if (this.scheme == "file"
|
||||||
|
&& value.length > start
|
||||||
|
&& value[start] == '/')
|
||||||
|
{
|
||||||
|
// Windows drive letters
|
||||||
|
if (value.length - start > 2 && value[start + 2] == ':')
|
||||||
|
{
|
||||||
|
++start;
|
||||||
|
}
|
||||||
|
goto ParsePath;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
start = pos + 1;
|
||||||
|
goto ParsePath;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else // certain schemas like mailto: and zlib: may not have any / after them
|
||||||
|
{
|
||||||
|
|
||||||
|
if (!parsePort(value[pos .. $]))
|
||||||
|
{
|
||||||
|
this.scheme = value[0 .. pos];
|
||||||
|
start = pos + 1;
|
||||||
|
goto ParsePath;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (pos == 0 && parsePort(value[pos .. $]))
|
||||||
|
{
|
||||||
|
// An URL shouldn't begin with a port number
|
||||||
|
throw defaultAllocator.make!URIException("URL begins with port");
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
goto ParsePath;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parse host
|
||||||
|
pos = -1;
|
||||||
|
for (ptrdiff_t i = start; i < value.length; ++i)
|
||||||
|
{
|
||||||
|
if (value[i] == '@')
|
||||||
|
{
|
||||||
|
pos = i;
|
||||||
|
}
|
||||||
|
else if (value[i] == '/')
|
||||||
|
{
|
||||||
|
endPos = i;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check for login and password
|
||||||
|
if (pos != -1)
|
||||||
|
{
|
||||||
|
// *( unreserved / pct-encoded / sub-delims / ":" )
|
||||||
|
foreach (i, c; value[start .. pos])
|
||||||
|
{
|
||||||
|
if (c == ':')
|
||||||
|
{
|
||||||
|
if (this.user is null)
|
||||||
|
{
|
||||||
|
this.user = value[start .. start + i];
|
||||||
|
this.pass = value[start + i + 1 .. pos];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (!c.isAlpha &&
|
||||||
|
!c.isNumber &&
|
||||||
|
c != '!' &&
|
||||||
|
c != ';' &&
|
||||||
|
c != '=' &&
|
||||||
|
c != '_' &&
|
||||||
|
c != '~' &&
|
||||||
|
!(c >= '$' && c <= '.'))
|
||||||
|
{
|
||||||
|
this.scheme = this.user = this.pass = null;
|
||||||
|
throw make!URIException(defaultAllocator,
|
||||||
|
"Restricted characters in user information");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (this.user is null)
|
||||||
|
{
|
||||||
|
this.user = value[start .. pos];
|
||||||
|
}
|
||||||
|
|
||||||
|
start = ++pos;
|
||||||
|
}
|
||||||
|
|
||||||
|
pos = endPos;
|
||||||
|
if (endPos <= 1 || value[start] != '[' || value[endPos - 1] != ']')
|
||||||
|
{
|
||||||
|
// Short circuit portscan
|
||||||
|
// IPv6 embedded address
|
||||||
|
for (ptrdiff_t i = endPos - 1; i >= start; --i)
|
||||||
|
{
|
||||||
|
if (value[i] == ':')
|
||||||
|
{
|
||||||
|
pos = i;
|
||||||
|
if (this.port == 0 && !parsePort(value[i .. endPos]))
|
||||||
|
{
|
||||||
|
this.scheme = this.user = this.pass = null;
|
||||||
|
throw defaultAllocator.make!URIException("Invalid port");
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Check if we have a valid host, if we don't reject the string as url
|
||||||
|
if (pos <= start)
|
||||||
|
{
|
||||||
|
this.scheme = this.user = this.pass = null;
|
||||||
|
throw defaultAllocator.make!URIException("Invalid host");
|
||||||
|
}
|
||||||
|
|
||||||
|
this.host = value[start .. pos];
|
||||||
|
|
||||||
|
if (endPos == value.length)
|
||||||
|
{
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
start = endPos;
|
||||||
|
|
||||||
|
ParsePath:
|
||||||
|
endPos = value.length;
|
||||||
|
pos = -1;
|
||||||
|
foreach (i, ref c; value[start .. $])
|
||||||
|
{
|
||||||
|
if (c == '?' && pos == -1)
|
||||||
|
{
|
||||||
|
pos = start + i;
|
||||||
|
}
|
||||||
|
else if (c == '#')
|
||||||
|
{
|
||||||
|
endPos = start + i;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (pos == -1)
|
||||||
|
{
|
||||||
|
pos = endPos;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pos > start)
|
||||||
|
{
|
||||||
|
this.path = value[start .. pos];
|
||||||
|
}
|
||||||
|
if (endPos >= ++pos)
|
||||||
|
{
|
||||||
|
this.query = value[pos .. endPos];
|
||||||
|
}
|
||||||
|
if (++endPos <= value.length)
|
||||||
|
{
|
||||||
|
this.fragment = value[endPos .. $];
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/*
|
||||||
|
* Attempts to parse and set the port.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* port = String beginning with a colon followed by the port number and
|
||||||
|
* an optional path (query string and/or fragment), like:
|
||||||
|
* `:12345/some_path` or `:12345`.
|
||||||
|
*
|
||||||
|
* Returns: Whether the port could be found.
|
||||||
|
*/
|
||||||
|
private bool parsePort(const char[] port) pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
ptrdiff_t i = 1;
|
||||||
|
float lPort = 0;
|
||||||
|
|
||||||
|
for (; i < port.length && port[i].isDigit() && i <= 6; ++i)
|
||||||
|
{
|
||||||
|
lPort += (port[i] - '0') / cast(float)(10 ^^ (i - 1));
|
||||||
|
}
|
||||||
|
if (i == 1 && (i == port.length || port[i] == '/'))
|
||||||
|
{
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
else if (i == port.length || port[i] == '/')
|
||||||
|
{
|
||||||
|
lPort *= 10 ^^ (i - 2);
|
||||||
|
if (lPort > ushort.max)
|
||||||
|
{
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
this.port = cast(ushort) lPort;
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
auto u = URL("example.org");
|
||||||
|
assert(u.path == "example.org");
|
||||||
|
|
||||||
|
u = URL("relative/path");
|
||||||
|
assert(u.path == "relative/path");
|
||||||
|
|
||||||
|
// Host and scheme
|
||||||
|
u = URL("https://example.org");
|
||||||
|
assert(u.scheme == "https");
|
||||||
|
assert(u.host == "example.org");
|
||||||
|
assert(u.path is null);
|
||||||
|
assert(u.port == 0);
|
||||||
|
assert(u.fragment is null);
|
||||||
|
|
||||||
|
// With user and port and path
|
||||||
|
u = URL("https://hilary:putnam@example.org:443/foo/bar");
|
||||||
|
assert(u.scheme == "https");
|
||||||
|
assert(u.host == "example.org");
|
||||||
|
assert(u.path == "/foo/bar");
|
||||||
|
assert(u.port == 443);
|
||||||
|
assert(u.user == "hilary");
|
||||||
|
assert(u.pass == "putnam");
|
||||||
|
assert(u.fragment is null);
|
||||||
|
|
||||||
|
// With query string
|
||||||
|
u = URL("https://example.org/?login=true");
|
||||||
|
assert(u.scheme == "https");
|
||||||
|
assert(u.host == "example.org");
|
||||||
|
assert(u.path == "/");
|
||||||
|
assert(u.query == "login=true");
|
||||||
|
assert(u.fragment is null);
|
||||||
|
|
||||||
|
// With query string and fragment
|
||||||
|
u = URL("https://example.org/?login=false#label");
|
||||||
|
assert(u.scheme == "https");
|
||||||
|
assert(u.host == "example.org");
|
||||||
|
assert(u.path == "/");
|
||||||
|
assert(u.query == "login=false");
|
||||||
|
assert(u.fragment == "label");
|
||||||
|
|
||||||
|
u = URL("redis://root:password@localhost:2201/path?query=value#fragment");
|
||||||
|
assert(u.scheme == "redis");
|
||||||
|
assert(u.user == "root");
|
||||||
|
assert(u.pass == "password");
|
||||||
|
assert(u.host == "localhost");
|
||||||
|
assert(u.port == 2201);
|
||||||
|
assert(u.path == "/path");
|
||||||
|
assert(u.query == "query=value");
|
||||||
|
assert(u.fragment == "fragment");
|
||||||
|
}
|
||||||
|
|
||||||
|
private unittest
|
||||||
|
{
|
||||||
|
auto u = URL("127.0.0.1");
|
||||||
|
assert(u.path == "127.0.0.1");
|
||||||
|
|
||||||
|
u = URL("http://127.0.0.1");
|
||||||
|
assert(u.scheme == "http");
|
||||||
|
assert(u.host == "127.0.0.1");
|
||||||
|
|
||||||
|
u = URL("http://127.0.0.1:9000");
|
||||||
|
assert(u.scheme == "http");
|
||||||
|
assert(u.host == "127.0.0.1");
|
||||||
|
assert(u.port == 9000);
|
||||||
|
|
||||||
|
u = URL("//example.net");
|
||||||
|
assert(u.host == "example.net");
|
||||||
|
assert(u.scheme is null);
|
||||||
|
|
||||||
|
u = URL("//example.net?q=before:after");
|
||||||
|
assert(u.host == "example.net");
|
||||||
|
assert(u.query == "q=before:after");
|
||||||
|
|
||||||
|
u = URL("localhost:8080");
|
||||||
|
assert(u.host == "localhost");
|
||||||
|
assert(u.port == 8080);
|
||||||
|
|
||||||
|
u = URL("ftp:");
|
||||||
|
assert(u.scheme == "ftp");
|
||||||
|
|
||||||
|
u = URL("file:///C:\\Users");
|
||||||
|
assert(u.scheme == "file");
|
||||||
|
assert(u.path == "C:\\Users");
|
||||||
|
|
||||||
|
u = URL("localhost:66000");
|
||||||
|
assert(u.scheme == "localhost");
|
||||||
|
assert(u.path == "66000");
|
||||||
|
|
||||||
|
u = URL("file:///home/");
|
||||||
|
assert(u.scheme == "file");
|
||||||
|
assert(u.path == "/home/");
|
||||||
|
|
||||||
|
u = URL("file:///home/?q=asdf");
|
||||||
|
assert(u.scheme == "file");
|
||||||
|
assert(u.path == "/home/");
|
||||||
|
assert(u.query == "q=asdf");
|
||||||
|
|
||||||
|
u = URL("http://secret@example.org");
|
||||||
|
assert(u.scheme == "http");
|
||||||
|
assert(u.host == "example.org");
|
||||||
|
assert(u.user == "secret");
|
||||||
|
|
||||||
|
u = URL("h_tp://:80");
|
||||||
|
assert(u.path == "h_tp://:80");
|
||||||
|
assert(u.port == 0);
|
||||||
|
|
||||||
|
u = URL("zlib:/home/user/file.gz");
|
||||||
|
assert(u.scheme == "zlib");
|
||||||
|
assert(u.path == "/home/user/file.gz");
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
URIException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
auto u = URL("h_tp:asdf");
|
||||||
|
}
|
||||||
|
catch (URIException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
URIException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
auto u = URL("http://:80");
|
||||||
|
}
|
||||||
|
catch (URIException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
URIException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
auto u = URL(":80");
|
||||||
|
}
|
||||||
|
catch (URIException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
URIException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
auto u = URL("http://user1:pass1@user2:pass2@example.org");
|
||||||
|
}
|
||||||
|
catch (URIException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
URIException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
auto u = URL("http://blah.com:port");
|
||||||
|
}
|
||||||
|
catch (URIException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
private @nogc unittest
|
||||||
|
{
|
||||||
|
URIException exception;
|
||||||
|
try
|
||||||
|
{
|
||||||
|
auto u = URL(":/");
|
||||||
|
}
|
||||||
|
catch (URIException e)
|
||||||
|
{
|
||||||
|
exception = e;
|
||||||
|
}
|
||||||
|
assert(exception !is null);
|
||||||
|
defaultAllocator.dispose(exception);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Attempts to parse an URL from a string and returns the specified component
|
||||||
|
* of the URL or $(D_PSYMBOL URL) if no component is specified.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* T = "scheme", "host", "port", "user", "pass", "path", "query",
|
||||||
|
* "fragment" or $(D_KEYWORD null) for a struct with all components.
|
||||||
|
* source = The string containing the URL.
|
||||||
|
*
|
||||||
|
* Returns: Requested URL component.
|
||||||
|
*/
|
||||||
|
URL parseURL(const char[] source) @nogc
|
||||||
|
{
|
||||||
|
return URL(source);
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
auto parseURL(string T)(const char[] source)
|
||||||
|
if (T == "scheme"
|
||||||
|
|| T == "host"
|
||||||
|
|| T == "user"
|
||||||
|
|| T == "pass"
|
||||||
|
|| T == "path"
|
||||||
|
|| T == "query"
|
||||||
|
|| T == "fragment"
|
||||||
|
|| T == "port")
|
||||||
|
{
|
||||||
|
auto ret = URL(source);
|
||||||
|
return mixin("ret." ~ T);
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
@nogc unittest
|
||||||
|
{
|
||||||
|
auto u = parseURL("http://example.org:5326");
|
||||||
|
assert(u.scheme == parseURL!"scheme"("http://example.org:5326"));
|
||||||
|
assert(u.host == parseURL!"host"("http://example.org:5326"));
|
||||||
|
assert(u.user == parseURL!"user"("http://example.org:5326"));
|
||||||
|
assert(u.pass == parseURL!"pass"("http://example.org:5326"));
|
||||||
|
assert(u.path == parseURL!"path"("http://example.org:5326"));
|
||||||
|
assert(u.query == parseURL!"query"("http://example.org:5326"));
|
||||||
|
assert(u.fragment == parseURL!"fragment"("http://example.org:5326"));
|
||||||
|
assert(u.port == parseURL!"port"("http://example.org:5326"));
|
||||||
|
}
|
@ -5,352 +5,15 @@
|
|||||||
/**
|
/**
|
||||||
* Internet utilities.
|
* Internet utilities.
|
||||||
*
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* $(RED Deprecated. Use $(D_PSYMBOL tanya.net.inet) instead.
|
||||||
|
* This module will be removed in 0.8.0.)
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 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)
|
||||||
*/
|
*/
|
||||||
|
deprecated("Use tanya.net.inet instead")
|
||||||
module tanya.network.inet;
|
module tanya.network.inet;
|
||||||
|
|
||||||
import std.math;
|
public import tanya.net.inet;
|
||||||
import std.range.primitives;
|
|
||||||
import std.traits;
|
|
||||||
|
|
||||||
version (unittest)
|
|
||||||
{
|
|
||||||
version (Windows)
|
|
||||||
{
|
|
||||||
import core.sys.windows.winsock2;
|
|
||||||
version = PlattformUnittest;
|
|
||||||
}
|
|
||||||
else version (Posix)
|
|
||||||
{
|
|
||||||
import core.sys.posix.arpa.inet;
|
|
||||||
version = PlattformUnittest;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Represents an unsigned integer as an $(D_KEYWORD ubyte) range.
|
|
||||||
*
|
|
||||||
* The range is bidirectional. The byte order is always big-endian.
|
|
||||||
*
|
|
||||||
* It can accept any unsigned integral type but the value should fit
|
|
||||||
* in $(D_PARAM L) bytes.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* L = Desired range length.
|
|
||||||
*/
|
|
||||||
struct NetworkOrder(uint L)
|
|
||||||
if (L > ubyte.sizeof && L <= ulong.sizeof)
|
|
||||||
{
|
|
||||||
static if (L > uint.sizeof)
|
|
||||||
{
|
|
||||||
private alias StorageType = ulong;
|
|
||||||
}
|
|
||||||
else static if (L > ushort.sizeof)
|
|
||||||
{
|
|
||||||
private alias StorageType = uint;
|
|
||||||
}
|
|
||||||
else static if (L > ubyte.sizeof)
|
|
||||||
{
|
|
||||||
private alias StorageType = ushort;
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
private alias StorageType = ubyte;
|
|
||||||
}
|
|
||||||
|
|
||||||
private StorageType value;
|
|
||||||
private size_t size = L;
|
|
||||||
|
|
||||||
const pure nothrow @safe @nogc invariant
|
|
||||||
{
|
|
||||||
assert(this.size <= L);
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Constructs a new range.
|
|
||||||
*
|
|
||||||
* $(D_PARAM T) can be any unsigned type but $(D_PARAM value) cannot be
|
|
||||||
* larger than the maximum can be stored in $(D_PARAM L) bytes. Otherwise
|
|
||||||
* an assertion failure will be caused.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* T = Value type.
|
|
||||||
* value = The value should be represented by this range.
|
|
||||||
*
|
|
||||||
* Precondition: $(D_INLINECODE value <= 2 ^^ (length * 8) - 1).
|
|
||||||
*/
|
|
||||||
this(T)(const T value)
|
|
||||||
if (isUnsigned!T)
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(value <= pow(2, L * 8) - 1);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
this.value = value & StorageType.max;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: LSB.
|
|
||||||
*
|
|
||||||
* Precondition: $(D_INLINECODE length > 0).
|
|
||||||
*/
|
|
||||||
@property ubyte back() const
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(this.length > 0);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
return this.value & 0xff;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: MSB.
|
|
||||||
*
|
|
||||||
* Precondition: $(D_INLINECODE length > 0).
|
|
||||||
*/
|
|
||||||
@property ubyte front() const
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(this.length > 0);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
return (this.value >> ((this.length - 1) * 8)) & 0xff;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Eliminates the LSB.
|
|
||||||
*
|
|
||||||
* Precondition: $(D_INLINECODE length > 0).
|
|
||||||
*/
|
|
||||||
void popBack()
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(this.length > 0);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
this.value >>= 8;
|
|
||||||
--this.size;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Eliminates the MSB.
|
|
||||||
*
|
|
||||||
* Precondition: $(D_INLINECODE length > 0).
|
|
||||||
*/
|
|
||||||
void popFront()
|
|
||||||
in
|
|
||||||
{
|
|
||||||
assert(this.length > 0);
|
|
||||||
}
|
|
||||||
body
|
|
||||||
{
|
|
||||||
this.value &= StorageType.max >> ((StorageType.sizeof - this.length) * 8);
|
|
||||||
--this.size;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Copy of this range.
|
|
||||||
*/
|
|
||||||
typeof(this) save() const
|
|
||||||
{
|
|
||||||
return this;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Whether the range is empty.
|
|
||||||
*/
|
|
||||||
@property bool empty() const
|
|
||||||
{
|
|
||||||
return this.length == 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Returns: Byte length.
|
|
||||||
*/
|
|
||||||
@property size_t length() const
|
|
||||||
{
|
|
||||||
return this.size;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
pure nothrow @safe @nogc unittest
|
|
||||||
{
|
|
||||||
auto networkOrder = NetworkOrder!3(0xae34e2u);
|
|
||||||
assert(!networkOrder.empty);
|
|
||||||
assert(networkOrder.front == 0xae);
|
|
||||||
|
|
||||||
networkOrder.popFront();
|
|
||||||
assert(networkOrder.length == 2);
|
|
||||||
assert(networkOrder.front == 0x34);
|
|
||||||
assert(networkOrder.back == 0xe2);
|
|
||||||
|
|
||||||
networkOrder.popBack();
|
|
||||||
assert(networkOrder.length == 1);
|
|
||||||
assert(networkOrder.front == 0x34);
|
|
||||||
assert(networkOrder.front == 0x34);
|
|
||||||
|
|
||||||
networkOrder.popFront();
|
|
||||||
assert(networkOrder.empty);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Static.
|
|
||||||
private unittest
|
|
||||||
{
|
|
||||||
static assert(isBidirectionalRange!(NetworkOrder!4));
|
|
||||||
static assert(isBidirectionalRange!(NetworkOrder!8));
|
|
||||||
static assert(!is(NetworkOrder!9));
|
|
||||||
static assert(!is(NetworkOrder!1));
|
|
||||||
}
|
|
||||||
|
|
||||||
// Tests against the system's htonl, htons.
|
|
||||||
version (PlattformUnittest)
|
|
||||||
{
|
|
||||||
private unittest
|
|
||||||
{
|
|
||||||
for (uint counter; counter <= 8 * uint.sizeof; ++counter)
|
|
||||||
{
|
|
||||||
const value = pow(2, counter) - 1;
|
|
||||||
const inNetworkOrder = htonl(value);
|
|
||||||
const p = cast(ubyte*) &inNetworkOrder;
|
|
||||||
auto networkOrder = NetworkOrder!4(value);
|
|
||||||
|
|
||||||
assert(networkOrder.length == 4);
|
|
||||||
assert(!networkOrder.empty);
|
|
||||||
assert(networkOrder.front == *p);
|
|
||||||
assert(networkOrder.back == *(p + 3));
|
|
||||||
|
|
||||||
networkOrder.popBack();
|
|
||||||
assert(networkOrder.length == 3);
|
|
||||||
assert(networkOrder.front == *p);
|
|
||||||
assert(networkOrder.back == *(p + 2));
|
|
||||||
|
|
||||||
networkOrder.popFront();
|
|
||||||
assert(networkOrder.length == 2);
|
|
||||||
assert(networkOrder.front == *(p + 1));
|
|
||||||
assert(networkOrder.back == *(p + 2));
|
|
||||||
|
|
||||||
networkOrder.popFront();
|
|
||||||
assert(networkOrder.length == 1);
|
|
||||||
assert(networkOrder.front == *(p + 2));
|
|
||||||
assert(networkOrder.back == *(p + 2));
|
|
||||||
|
|
||||||
networkOrder.popBack();
|
|
||||||
assert(networkOrder.length == 0);
|
|
||||||
assert(networkOrder.empty);
|
|
||||||
}
|
|
||||||
|
|
||||||
for (ushort counter; counter <= 8 * ushort.sizeof; ++counter)
|
|
||||||
{
|
|
||||||
const value = cast(ushort) (pow(2, counter) - 1);
|
|
||||||
const inNetworkOrder = htons(value);
|
|
||||||
const p = cast(ubyte*) &inNetworkOrder;
|
|
||||||
|
|
||||||
auto networkOrder = NetworkOrder!2(value);
|
|
||||||
|
|
||||||
assert(networkOrder.length == 2);
|
|
||||||
assert(!networkOrder.empty);
|
|
||||||
assert(networkOrder.front == *p);
|
|
||||||
assert(networkOrder.back == *(p + 1));
|
|
||||||
|
|
||||||
networkOrder.popBack();
|
|
||||||
assert(networkOrder.length == 1);
|
|
||||||
assert(networkOrder.front == *p);
|
|
||||||
assert(networkOrder.back == *p);
|
|
||||||
|
|
||||||
networkOrder.popBack();
|
|
||||||
assert(networkOrder.length == 0);
|
|
||||||
assert(networkOrder.empty);
|
|
||||||
|
|
||||||
networkOrder = NetworkOrder!2(value);
|
|
||||||
|
|
||||||
networkOrder.popFront();
|
|
||||||
assert(networkOrder.length == 1);
|
|
||||||
assert(networkOrder.front == *(p + 1));
|
|
||||||
assert(networkOrder.back == *(p + 1));
|
|
||||||
|
|
||||||
networkOrder.popFront();
|
|
||||||
assert(networkOrder.length == 0);
|
|
||||||
assert(networkOrder.empty);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Converts the $(D_KEYWORD ubyte) input range $(D_PARAM range) to
|
|
||||||
* $(D_PARAM T).
|
|
||||||
*
|
|
||||||
* The byte order of $(D_PARAM r) is assumed to be big-endian. The length
|
|
||||||
* cannot be larger than $(D_INLINECODE T.sizeof). Otherwise an assertion
|
|
||||||
* failure will be caused.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* T = Desired return type.
|
|
||||||
* R = Range type.
|
|
||||||
* range = Input range.
|
|
||||||
*
|
|
||||||
* Returns: Integral representation of $(D_PARAM range) with the host byte
|
|
||||||
* order.
|
|
||||||
*/
|
|
||||||
T toHostOrder(T = size_t, R)(R range)
|
|
||||||
if (isInputRange!R
|
|
||||||
&& !isInfinite!R
|
|
||||||
&& is(Unqual!(ElementType!R) == ubyte)
|
|
||||||
&& isUnsigned!T)
|
|
||||||
{
|
|
||||||
T ret;
|
|
||||||
ushort pos = T.sizeof * 8;
|
|
||||||
|
|
||||||
for (; !range.empty && range.front == 0; pos -= 8, range.popFront())
|
|
||||||
{
|
|
||||||
}
|
|
||||||
for (; !range.empty; range.popFront())
|
|
||||||
{
|
|
||||||
assert(pos != 0);
|
|
||||||
pos -= 8;
|
|
||||||
ret |= (cast(T) range.front) << pos;
|
|
||||||
}
|
|
||||||
|
|
||||||
return ret >> pos;
|
|
||||||
}
|
|
||||||
|
|
||||||
///
|
|
||||||
pure nothrow @safe @nogc unittest
|
|
||||||
{
|
|
||||||
const value = 0xae34e2u;
|
|
||||||
auto networkOrder = NetworkOrder!4(value);
|
|
||||||
assert(networkOrder.toHostOrder() == value);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Tests against the system's htonl, htons.
|
|
||||||
version (PlattformUnittest)
|
|
||||||
{
|
|
||||||
private unittest
|
|
||||||
{
|
|
||||||
for (uint counter; counter <= 8 * uint.sizeof; ++counter)
|
|
||||||
{
|
|
||||||
const value = pow(2, counter) - 1;
|
|
||||||
const inNetworkOrder = htonl(value);
|
|
||||||
const p = cast(ubyte*) &inNetworkOrder;
|
|
||||||
auto networkOrder = NetworkOrder!4(value);
|
|
||||||
|
|
||||||
assert(p[0 .. uint.sizeof].toHostOrder() == value);
|
|
||||||
}
|
|
||||||
for (ushort counter; counter <= 8 * ushort.sizeof; ++counter)
|
|
||||||
{
|
|
||||||
const value = cast(ushort) (pow(2, counter) - 1);
|
|
||||||
const inNetworkOrder = htons(value);
|
|
||||||
const p = cast(ubyte*) &inNetworkOrder;
|
|
||||||
auto networkOrder = NetworkOrder!2(value);
|
|
||||||
|
|
||||||
assert(p[0 .. ushort.sizeof].toHostOrder() == value);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
@ -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/. */
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Socket programming.
|
* Low-level socket programming.
|
||||||
*
|
*
|
||||||
* Copyright: Eugene Wissner 2016-2017.
|
* 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/,
|
||||||
@ -12,15 +12,16 @@
|
|||||||
*/
|
*/
|
||||||
module tanya.network.socket;
|
module tanya.network.socket;
|
||||||
|
|
||||||
import tanya.memory;
|
|
||||||
import core.stdc.errno;
|
import core.stdc.errno;
|
||||||
import core.time;
|
import core.time;
|
||||||
import std.algorithm.comparison;
|
import std.algorithm.comparison;
|
||||||
import std.algorithm.searching;
|
public import std.socket : SocketOptionLevel, SocketOption;
|
||||||
public import std.socket : socket_t, Linger, SocketOptionLevel, SocketOption,
|
|
||||||
SocketType, AddressFamily, AddressInfo;
|
|
||||||
import std.traits;
|
import std.traits;
|
||||||
import std.typecons;
|
import std.typecons;
|
||||||
|
import tanya.memory;
|
||||||
|
|
||||||
|
/// Value returned by socket operations on error.
|
||||||
|
enum int socketError = -1;
|
||||||
|
|
||||||
version (Posix)
|
version (Posix)
|
||||||
{
|
{
|
||||||
@ -32,7 +33,12 @@ version (Posix)
|
|||||||
import core.sys.posix.sys.time;
|
import core.sys.posix.sys.time;
|
||||||
import core.sys.posix.unistd;
|
import core.sys.posix.unistd;
|
||||||
|
|
||||||
private enum SOCKET_ERROR = -1;
|
enum SocketType : int
|
||||||
|
{
|
||||||
|
init = -1,
|
||||||
|
}
|
||||||
|
|
||||||
|
private alias LingerField = int;
|
||||||
}
|
}
|
||||||
else version (Windows)
|
else version (Windows)
|
||||||
{
|
{
|
||||||
@ -43,16 +49,23 @@ else version (Windows)
|
|||||||
import core.sys.windows.windef;
|
import core.sys.windows.windef;
|
||||||
import core.sys.windows.winsock2;
|
import core.sys.windows.winsock2;
|
||||||
|
|
||||||
|
enum SocketType : size_t
|
||||||
|
{
|
||||||
|
init = ~0,
|
||||||
|
}
|
||||||
|
|
||||||
|
private alias LingerField = ushort;
|
||||||
|
|
||||||
enum : uint
|
enum : uint
|
||||||
{
|
{
|
||||||
IOC_UNIX = 0x00000000,
|
IOC_UNIX = 0x00000000,
|
||||||
IOC_WS2 = 0x08000000,
|
IOC_WS2 = 0x08000000,
|
||||||
IOC_PROTOCOL = 0x10000000,
|
IOC_PROTOCOL = 0x10000000,
|
||||||
IOC_VOID = 0x20000000, /// No parameters.
|
IOC_VOID = 0x20000000, // No parameters.
|
||||||
IOC_OUT = 0x40000000, /// Copy parameters back.
|
IOC_OUT = 0x40000000, // Copy parameters back.
|
||||||
IOC_IN = 0x80000000, /// Copy parameters into.
|
IOC_IN = 0x80000000, // Copy parameters into.
|
||||||
IOC_VENDOR = 0x18000000,
|
IOC_VENDOR = 0x18000000,
|
||||||
IOC_INOUT = (IOC_IN | IOC_OUT), /// Copy parameter into and get back.
|
IOC_INOUT = (IOC_IN | IOC_OUT), // Copy parameter into and get back.
|
||||||
}
|
}
|
||||||
|
|
||||||
template _WSAIO(int x, int y)
|
template _WSAIO(int x, int y)
|
||||||
@ -183,19 +196,26 @@ else version (Windows)
|
|||||||
LPINT,
|
LPINT,
|
||||||
SOCKADDR**,
|
SOCKADDR**,
|
||||||
LPINT);
|
LPINT);
|
||||||
const GUID WSAID_GETACCEPTEXSOCKADDRS = {0xb5367df2,0xcbac,0x11cf,[0x95,0xca,0x00,0x80,0x5f,0x48,0xa1,0x92]};
|
const GUID WSAID_GETACCEPTEXSOCKADDRS = {
|
||||||
|
0xb5367df2, 0xcbac, 0x11cf,
|
||||||
|
[ 0x95, 0xca, 0x00, 0x80, 0x5f, 0x48, 0xa1, 0x92 ],
|
||||||
|
};
|
||||||
|
|
||||||
struct WSABUF {
|
struct WSABUF
|
||||||
|
{
|
||||||
ULONG len;
|
ULONG len;
|
||||||
CHAR* buf;
|
CHAR* buf;
|
||||||
}
|
}
|
||||||
alias WSABUF* LPWSABUF;
|
alias WSABUF* LPWSABUF;
|
||||||
|
|
||||||
struct WSAOVERLAPPED {
|
struct WSAOVERLAPPED
|
||||||
|
{
|
||||||
ULONG_PTR Internal;
|
ULONG_PTR Internal;
|
||||||
ULONG_PTR InternalHigh;
|
ULONG_PTR InternalHigh;
|
||||||
union {
|
union
|
||||||
struct {
|
{
|
||||||
|
struct
|
||||||
|
{
|
||||||
DWORD Offset;
|
DWORD Offset;
|
||||||
DWORD OffsetHigh;
|
DWORD OffsetHigh;
|
||||||
}
|
}
|
||||||
@ -219,36 +239,13 @@ else version (Windows)
|
|||||||
private WSABUF buffer;
|
private WSABUF buffer;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Socket returned if a connection has been established.
|
|
||||||
*/
|
|
||||||
class OverlappedConnectedSocket : ConnectedSocket
|
class OverlappedConnectedSocket : ConnectedSocket
|
||||||
{
|
{
|
||||||
/**
|
this(SocketType handle, AddressFamily af) @nogc
|
||||||
* Create a socket.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* handle = Socket handle.
|
|
||||||
* af = Address family.
|
|
||||||
*/
|
|
||||||
this(socket_t handle, AddressFamily af) @nogc
|
|
||||||
{
|
{
|
||||||
super(handle, af);
|
super(handle, af);
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Begins to asynchronously receive data from a connected socket.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* buffer = Storage location for the received data.
|
|
||||||
* flags = Flags.
|
|
||||||
* overlapped = Unique operation identifier.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD true) if the operation could be finished synchronously.
|
|
||||||
* $(D_KEYWORD false) otherwise.
|
|
||||||
*
|
|
||||||
* Throws: $(D_PSYMBOL SocketException) if unable to receive.
|
|
||||||
*/
|
|
||||||
bool beginReceive(ubyte[] buffer,
|
bool beginReceive(ubyte[] buffer,
|
||||||
SocketState overlapped,
|
SocketState overlapped,
|
||||||
Flags flags = Flags(Flag.none)) @nogc @trusted
|
Flags flags = Flags(Flag.none)) @nogc @trusted
|
||||||
@ -268,23 +265,13 @@ else version (Windows)
|
|||||||
&overlapped.overlapped,
|
&overlapped.overlapped,
|
||||||
NULL);
|
NULL);
|
||||||
|
|
||||||
if (result == SOCKET_ERROR && !wouldHaveBlocked)
|
if (result == socketError && !wouldHaveBlocked)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Unable to receive");
|
throw defaultAllocator.make!SocketException("Unable to receive");
|
||||||
}
|
}
|
||||||
return result == 0;
|
return result == 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Ends a pending asynchronous read.
|
|
||||||
*
|
|
||||||
* Params
|
|
||||||
* overlapped = Unique operation identifier.
|
|
||||||
*
|
|
||||||
* Returns: Number of bytes received.
|
|
||||||
*
|
|
||||||
* Throws: $(D_PSYMBOL SocketException) if unable to receive.
|
|
||||||
*/
|
|
||||||
int endReceive(SocketState overlapped) @nogc @trusted
|
int endReceive(SocketState overlapped) @nogc @trusted
|
||||||
out (count)
|
out (count)
|
||||||
{
|
{
|
||||||
@ -309,19 +296,6 @@ else version (Windows)
|
|||||||
return lpNumber;
|
return lpNumber;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Sends data asynchronously to a connected socket.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* buffer = Data to be sent.
|
|
||||||
* flags = Flags.
|
|
||||||
* overlapped = Unique operation identifier.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD true) if the operation could be finished synchronously.
|
|
||||||
* $(D_KEYWORD false) otherwise.
|
|
||||||
*
|
|
||||||
* Throws: $(D_PSYMBOL SocketException) if unable to send.
|
|
||||||
*/
|
|
||||||
bool beginSend(ubyte[] buffer,
|
bool beginSend(ubyte[] buffer,
|
||||||
SocketState overlapped,
|
SocketState overlapped,
|
||||||
Flags flags = Flags(Flag.none)) @nogc @trusted
|
Flags flags = Flags(Flag.none)) @nogc @trusted
|
||||||
@ -339,7 +313,7 @@ else version (Windows)
|
|||||||
&overlapped.overlapped,
|
&overlapped.overlapped,
|
||||||
NULL);
|
NULL);
|
||||||
|
|
||||||
if (result == SOCKET_ERROR && !wouldHaveBlocked)
|
if (result == socketError && !wouldHaveBlocked)
|
||||||
{
|
{
|
||||||
disconnected_ = true;
|
disconnected_ = true;
|
||||||
throw defaultAllocator.make!SocketException("Unable to send");
|
throw defaultAllocator.make!SocketException("Unable to send");
|
||||||
@ -347,16 +321,6 @@ else version (Windows)
|
|||||||
return result == 0;
|
return result == 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Ends a pending asynchronous send.
|
|
||||||
*
|
|
||||||
* Params
|
|
||||||
* overlapped = Unique operation identifier.
|
|
||||||
*
|
|
||||||
* Returns: Number of bytes sent.
|
|
||||||
*
|
|
||||||
* Throws: $(D_PSYMBOL SocketException) if unable to receive.
|
|
||||||
*/
|
|
||||||
int endSend(SocketState overlapped) @nogc @trusted
|
int endSend(SocketState overlapped) @nogc @trusted
|
||||||
out (count)
|
out (count)
|
||||||
{
|
{
|
||||||
@ -380,17 +344,9 @@ else version (Windows)
|
|||||||
|
|
||||||
class OverlappedStreamSocket : StreamSocket
|
class OverlappedStreamSocket : StreamSocket
|
||||||
{
|
{
|
||||||
/// Accept extension function pointer.
|
// Accept extension function pointer.
|
||||||
package LPFN_ACCEPTEX acceptExtension;
|
package LPFN_ACCEPTEX acceptExtension;
|
||||||
|
|
||||||
/**
|
|
||||||
* Create a socket.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* af = Address family.
|
|
||||||
*
|
|
||||||
* Throws: $(D_PSYMBOL SocketException) on errors.
|
|
||||||
*/
|
|
||||||
this(AddressFamily af) @nogc @trusted
|
this(AddressFamily af) @nogc @trusted
|
||||||
{
|
{
|
||||||
super(af);
|
super(af);
|
||||||
@ -412,28 +368,17 @@ else version (Windows)
|
|||||||
&dwBytes,
|
&dwBytes,
|
||||||
NULL,
|
NULL,
|
||||||
NULL);
|
NULL);
|
||||||
if (!result == SOCKET_ERROR)
|
if (!result == socketError)
|
||||||
{
|
{
|
||||||
throw make!SocketException(defaultAllocator,
|
throw make!SocketException(defaultAllocator,
|
||||||
"Unable to retrieve an accept extension function pointer");
|
"Unable to retrieve an accept extension function pointer");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
|
||||||
* Begins an asynchronous operation to accept an incoming connection attempt.
|
|
||||||
*
|
|
||||||
* Params:
|
|
||||||
* overlapped = Unique operation identifier.
|
|
||||||
*
|
|
||||||
* Returns: $(D_KEYWORD true) if the operation could be finished synchronously.
|
|
||||||
* $(D_KEYWORD false) otherwise.
|
|
||||||
*
|
|
||||||
* Throws: $(D_PSYMBOL SocketException) on accept errors.
|
|
||||||
*/
|
|
||||||
bool beginAccept(SocketState overlapped) @nogc @trusted
|
bool beginAccept(SocketState overlapped) @nogc @trusted
|
||||||
{
|
{
|
||||||
auto socket = cast(socket_t) socket(addressFamily, SOCK_STREAM, 0);
|
auto socket = cast(SocketType) socket(addressFamily, 1, 0);
|
||||||
if (socket == socket_t.init)
|
if (socket == SocketType.init)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Unable to create socket");
|
throw defaultAllocator.make!SocketException("Unable to create socket");
|
||||||
}
|
}
|
||||||
@ -445,7 +390,7 @@ else version (Windows)
|
|||||||
overlapped.handle = cast(HANDLE) socket;
|
overlapped.handle = cast(HANDLE) socket;
|
||||||
overlapped.event = OverlappedSocketEvent.accept;
|
overlapped.event = OverlappedSocketEvent.accept;
|
||||||
|
|
||||||
immutable len = (sockaddr_in.sizeof + 16) * 2;
|
const len = (sockaddr_in.sizeof + 16) * 2;
|
||||||
overlapped.buffer.len = len;
|
overlapped.buffer.len = len;
|
||||||
overlapped.buffer.buf = cast(char*) defaultAllocator.allocate(len).ptr;
|
overlapped.buffer.buf = cast(char*) defaultAllocator.allocate(len).ptr;
|
||||||
|
|
||||||
@ -465,6 +410,139 @@ else version (Windows)
|
|||||||
return result == TRUE;
|
return result == TRUE;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
OverlappedConnectedSocket endAccept(SocketState overlapped)
|
||||||
|
@nogc @trusted
|
||||||
|
{
|
||||||
|
scope (exit)
|
||||||
|
{
|
||||||
|
defaultAllocator.dispose(overlapped.buffer.buf[0 .. overlapped.buffer.len]);
|
||||||
|
}
|
||||||
|
auto socket = make!OverlappedConnectedSocket(defaultAllocator,
|
||||||
|
cast(SocketType) overlapped.handle,
|
||||||
|
addressFamily);
|
||||||
|
scope (failure)
|
||||||
|
{
|
||||||
|
defaultAllocator.dispose(socket);
|
||||||
|
}
|
||||||
|
socket.setOption(SocketOptionLevel.SOCKET,
|
||||||
|
cast(SocketOption) SO_UPDATE_ACCEPT_CONTEXT,
|
||||||
|
cast(size_t) handle);
|
||||||
|
return socket;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else version (D_Ddoc)
|
||||||
|
{
|
||||||
|
/// Native socket representation type.
|
||||||
|
enum SocketType;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Socket returned if a connection has been established.
|
||||||
|
*
|
||||||
|
* Note: Available only on Windows.
|
||||||
|
*/
|
||||||
|
class OverlappedConnectedSocket : ConnectedSocket
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Create a socket.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* handle = Socket handle.
|
||||||
|
* af = Address family.
|
||||||
|
*/
|
||||||
|
this(SocketType handle, AddressFamily af) @nogc;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Begins to asynchronously receive data from a connected socket.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* buffer = Storage location for the received data.
|
||||||
|
* flags = Flags.
|
||||||
|
* overlapped = Unique operation identifier.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD true) if the operation could be finished synchronously.
|
||||||
|
* $(D_KEYWORD false) otherwise.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL SocketException) if unable to receive.
|
||||||
|
*/
|
||||||
|
bool beginReceive(ubyte[] buffer,
|
||||||
|
SocketState overlapped,
|
||||||
|
Flags flags = Flags(Flag.none)) @nogc @trusted;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Ends a pending asynchronous read.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* overlapped = Unique operation identifier.
|
||||||
|
*
|
||||||
|
* Returns: Number of bytes received.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL SocketException) if unable to receive.
|
||||||
|
*
|
||||||
|
* Postcondition: $(D_INLINECODE result >= 0).
|
||||||
|
*/
|
||||||
|
int endReceive(SocketState overlapped) @nogc @trusted;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sends data asynchronously to a connected socket.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* buffer = Data to be sent.
|
||||||
|
* flags = Flags.
|
||||||
|
* overlapped = Unique operation identifier.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD true) if the operation could be finished synchronously.
|
||||||
|
* $(D_KEYWORD false) otherwise.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL SocketException) if unable to send.
|
||||||
|
*/
|
||||||
|
bool beginSend(ubyte[] buffer,
|
||||||
|
SocketState overlapped,
|
||||||
|
Flags flags = Flags(Flag.none)) @nogc @trusted;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Ends a pending asynchronous send.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* overlapped = Unique operation identifier.
|
||||||
|
*
|
||||||
|
* Returns: Number of bytes sent.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL SocketException) if unable to receive.
|
||||||
|
*
|
||||||
|
* Postcondition: $(D_INLINECODE result >= 0).
|
||||||
|
*/
|
||||||
|
int endSend(SocketState overlapped) @nogc @trusted;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Windows stream socket overlapped I/O.
|
||||||
|
*/
|
||||||
|
class OverlappedStreamSocket : StreamSocket
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Create a socket.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* af = Address family.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL SocketException) on errors.
|
||||||
|
*/
|
||||||
|
this(AddressFamily af) @nogc @trusted;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Begins an asynchronous operation to accept an incoming connection attempt.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* overlapped = Unique operation identifier.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD true) if the operation could be finished synchronously.
|
||||||
|
* $(D_KEYWORD false) otherwise.
|
||||||
|
*
|
||||||
|
* Throws: $(D_PSYMBOL SocketException) on accept errors.
|
||||||
|
*/
|
||||||
|
bool beginAccept(SocketState overlapped) @nogc @trusted;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Asynchronously accepts an incoming connection attempt and creates a
|
* Asynchronously accepts an incoming connection attempt and creates a
|
||||||
* new socket to handle remote host communication.
|
* new socket to handle remote host communication.
|
||||||
@ -476,24 +554,116 @@ else version (Windows)
|
|||||||
*
|
*
|
||||||
* Throws: $(D_PSYMBOL SocketException) if unable to accept.
|
* Throws: $(D_PSYMBOL SocketException) if unable to accept.
|
||||||
*/
|
*/
|
||||||
OverlappedConnectedSocket endAccept(SocketState overlapped) @nogc @trusted
|
OverlappedConnectedSocket endAccept(SocketState overlapped)
|
||||||
|
@nogc @trusted;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Socket option that specifies what should happen when the socket that
|
||||||
|
* promises reliable delivery still has untransmitted messages when
|
||||||
|
* it is closed.
|
||||||
|
*/
|
||||||
|
struct Linger
|
||||||
|
{
|
||||||
|
/// If nonzero, $(D_PSYMBOL close) and $(D_PSYMBOL shutdown) block until
|
||||||
|
/// the data are transmitted or the timeout period has expired.
|
||||||
|
LingerField l_onoff;
|
||||||
|
|
||||||
|
/// Time, in seconds to wait before any buffered data to be sent is
|
||||||
|
/// discarded.
|
||||||
|
LingerField l_linger;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* If $(D_PARAM timeout) is `0`, linger is disabled, otherwise enables the
|
||||||
|
* linger and sets the timeout.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* timeout = Timeout, in seconds.
|
||||||
|
*/
|
||||||
|
this(const ushort timeout)
|
||||||
|
{
|
||||||
|
time = timeout;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
{
|
{
|
||||||
scope (exit)
|
auto linger = Linger(5);
|
||||||
{
|
assert(linger.enabled);
|
||||||
defaultAllocator.dispose(overlapped.buffer.buf[0 .. overlapped.buffer.len]);
|
assert(linger.time == 5);
|
||||||
}
|
|
||||||
auto socket = make!OverlappedConnectedSocket(defaultAllocator,
|
|
||||||
cast(socket_t) overlapped.handle,
|
|
||||||
addressFamily);
|
|
||||||
scope (failure)
|
|
||||||
{
|
|
||||||
defaultAllocator.dispose(socket);
|
|
||||||
}
|
|
||||||
socket.setOption(SocketOptionLevel.SOCKET,
|
|
||||||
cast(SocketOption) SO_UPDATE_ACCEPT_CONTEXT,
|
|
||||||
cast(size_t) handle);
|
|
||||||
return socket;
|
|
||||||
}
|
}
|
||||||
|
{
|
||||||
|
auto linger = Linger(0);
|
||||||
|
assert(!linger.enabled);
|
||||||
|
}
|
||||||
|
{ // Default constructor.
|
||||||
|
Linger linger;
|
||||||
|
assert(!linger.enabled);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* System dependent constructor.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* l_onoff = $(D_PSYMBOL l_onoff) value.
|
||||||
|
* l_linger = $(D_PSYMBOL l_linger) value.
|
||||||
|
*/
|
||||||
|
this(LingerField l_onoff, LingerField l_linger)
|
||||||
|
{
|
||||||
|
this.l_onoff = l_onoff;
|
||||||
|
this.l_linger = l_linger;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
auto linger = Linger(1, 5);
|
||||||
|
assert(linger.l_onoff == 1);
|
||||||
|
assert(linger.l_linger == 5);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Params:
|
||||||
|
* value = Whether to linger after the socket is closed.
|
||||||
|
*
|
||||||
|
* See_Also: $(D_PSYMBOL time).
|
||||||
|
*/
|
||||||
|
@property enabled(const bool value) pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
this.l_onoff = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Whether to linger after the socket is closed.
|
||||||
|
*/
|
||||||
|
@property bool enabled() const pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return this.l_onoff != 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Timeout period, in seconds, to wait before closing the socket
|
||||||
|
* if the $(D_PSYMBOL Linger) is $(D_PSYMBOL enabled).
|
||||||
|
*/
|
||||||
|
@property ushort time() const pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return this.l_linger & ushort.max;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sets timeout period, to wait before closing the socket if the
|
||||||
|
* $(D_PSYMBOL Linger) is $(D_PSYMBOL enabled), ignored otherwise.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* timeout = Timeout period, in seconds.
|
||||||
|
*/
|
||||||
|
@property void time(const ushort timeout) pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
this.l_onoff = timeout > 0;
|
||||||
|
this.l_linger = timeout;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -525,7 +695,7 @@ else version (DragonFlyBSD)
|
|||||||
|
|
||||||
version (MacBSD)
|
version (MacBSD)
|
||||||
{
|
{
|
||||||
enum ESOCKTNOSUPPORT = 44; /// Socket type not suppoted.
|
enum ESOCKTNOSUPPORT = 44; // Socket type not suppoted.
|
||||||
}
|
}
|
||||||
|
|
||||||
private immutable
|
private immutable
|
||||||
@ -552,12 +722,32 @@ shared static this()
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* $(D_PSYMBOL AddressFamily) specifies a communication domain; this selects
|
||||||
|
* the protocol family which will be used for communication.
|
||||||
|
*/
|
||||||
|
enum AddressFamily : int
|
||||||
|
{
|
||||||
|
unspec = 0, /// Unspecified.
|
||||||
|
local = 1, /// Local to host (pipes and file-domain).
|
||||||
|
unix = local, /// POSIX name for PF_LOCAL.
|
||||||
|
inet = 2, /// IP protocol family.
|
||||||
|
ax25 = 3, /// Amateur Radio AX.25.
|
||||||
|
ipx = 4, /// Novell Internet Protocol.
|
||||||
|
appletalk = 5, /// Appletalk DDP.
|
||||||
|
netrom = 6, /// Amateur radio NetROM.
|
||||||
|
bridge = 7, /// Multiprotocol bridge.
|
||||||
|
atmpvc = 8, /// ATM PVCs.
|
||||||
|
x25 = 9, /// Reserved for X.25 project.
|
||||||
|
inet6 = 10, /// IP version 6.
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Error codes for $(D_PSYMBOL Socket).
|
* Error codes for $(D_PSYMBOL Socket).
|
||||||
*/
|
*/
|
||||||
enum SocketError : int
|
enum SocketError : int
|
||||||
{
|
{
|
||||||
/// Unknown error
|
/// Unknown error.
|
||||||
unknown = 0,
|
unknown = 0,
|
||||||
/// Firewall rules forbid connection.
|
/// Firewall rules forbid connection.
|
||||||
accessDenied = EPERM,
|
accessDenied = EPERM,
|
||||||
@ -587,12 +777,12 @@ enum SocketError : int
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* $(D_PSYMBOL SocketException) should be thrown only if one of the socket functions
|
* $(D_PSYMBOL SocketException) should be thrown only if one of the socket functions
|
||||||
* returns -1 or $(D_PSYMBOL SOCKET_ERROR) and sets $(D_PSYMBOL errno),
|
* $(D_PSYMBOL socketError) and sets $(D_PSYMBOL errno), because
|
||||||
* because $(D_PSYMBOL SocketException) relies on the $(D_PSYMBOL errno) value.
|
* $(D_PSYMBOL SocketException) relies on the $(D_PSYMBOL errno) value.
|
||||||
*/
|
*/
|
||||||
class SocketException : Exception
|
class SocketException : Exception
|
||||||
{
|
{
|
||||||
immutable SocketError error = SocketError.unknown;
|
const SocketError error = SocketError.unknown;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Params:
|
* Params:
|
||||||
@ -680,16 +870,16 @@ abstract class Socket
|
|||||||
}
|
}
|
||||||
|
|
||||||
/// Socket handle.
|
/// Socket handle.
|
||||||
protected socket_t handle_;
|
protected SocketType handle_;
|
||||||
|
|
||||||
/// Address family.
|
/// Address family.
|
||||||
protected AddressFamily family;
|
protected AddressFamily family;
|
||||||
|
|
||||||
private @property void handle(socket_t handle) @nogc
|
private @property void handle(SocketType handle) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(handle != socket_t.init);
|
assert(handle != SocketType.init);
|
||||||
assert(handle_ == socket_t.init, "Socket handle cannot be changed");
|
assert(handle_ == SocketType.init, "Socket handle cannot be changed");
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
@ -703,7 +893,7 @@ abstract class Socket
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@property inout(socket_t) handle() inout const pure nothrow @safe @nogc
|
@property inout(SocketType) handle() inout const pure nothrow @safe @nogc
|
||||||
{
|
{
|
||||||
return handle_;
|
return handle_;
|
||||||
}
|
}
|
||||||
@ -715,10 +905,10 @@ abstract class Socket
|
|||||||
* handle = Socket.
|
* handle = Socket.
|
||||||
* af = Address family.
|
* af = Address family.
|
||||||
*/
|
*/
|
||||||
this(socket_t handle, AddressFamily af) @nogc
|
this(SocketType handle, AddressFamily af) @nogc
|
||||||
in
|
in
|
||||||
{
|
{
|
||||||
assert(handle != socket_t.init);
|
assert(handle != SocketType.init);
|
||||||
}
|
}
|
||||||
body
|
body
|
||||||
{
|
{
|
||||||
@ -759,7 +949,7 @@ abstract class Socket
|
|||||||
cast(int) level,
|
cast(int) level,
|
||||||
cast(int) option,
|
cast(int) option,
|
||||||
result.ptr,
|
result.ptr,
|
||||||
&length) == SOCKET_ERROR)
|
&length) == socketError)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Unable to get socket option");
|
throw defaultAllocator.make!SocketException("Unable to get socket option");
|
||||||
}
|
}
|
||||||
@ -771,7 +961,7 @@ abstract class Socket
|
|||||||
SocketOption option,
|
SocketOption option,
|
||||||
out size_t result) const @trusted @nogc
|
out size_t result) const @trusted @nogc
|
||||||
{
|
{
|
||||||
return getOption(level, option, (&result)[0..1]);
|
return getOption(level, option, (&result)[0 .. 1]);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
@ -779,7 +969,7 @@ abstract class Socket
|
|||||||
SocketOption option,
|
SocketOption option,
|
||||||
out Linger result) const @trusted @nogc
|
out Linger result) const @trusted @nogc
|
||||||
{
|
{
|
||||||
return getOption(level, option, (&result.clinger)[0..1]);
|
return getOption(level, option, (&result)[0 .. 1]);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
@ -792,7 +982,7 @@ abstract class Socket
|
|||||||
version (Posix)
|
version (Posix)
|
||||||
{
|
{
|
||||||
timeval tv;
|
timeval tv;
|
||||||
auto ret = getOption(level, option, (&tv)[0..1]);
|
auto ret = getOption(level, option, (&tv)[0 .. 1]);
|
||||||
result = dur!"seconds"(tv.tv_sec) + dur!"usecs"(tv.tv_usec);
|
result = dur!"seconds"(tv.tv_sec) + dur!"usecs"(tv.tv_usec);
|
||||||
}
|
}
|
||||||
else version (Windows)
|
else version (Windows)
|
||||||
@ -826,7 +1016,7 @@ abstract class Socket
|
|||||||
cast(int)level,
|
cast(int)level,
|
||||||
cast(int)option,
|
cast(int)option,
|
||||||
value.ptr,
|
value.ptr,
|
||||||
cast(uint) value.length) == SOCKET_ERROR)
|
cast(uint) value.length) == socketError)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Unable to set socket option");
|
throw defaultAllocator.make!SocketException("Unable to set socket option");
|
||||||
}
|
}
|
||||||
@ -836,14 +1026,14 @@ abstract class Socket
|
|||||||
void setOption(SocketOptionLevel level, SocketOption option, size_t value)
|
void setOption(SocketOptionLevel level, SocketOption option, size_t value)
|
||||||
const @trusted @nogc
|
const @trusted @nogc
|
||||||
{
|
{
|
||||||
setOption(level, option, (&value)[0..1]);
|
setOption(level, option, (&value)[0 .. 1]);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
void setOption(SocketOptionLevel level, SocketOption option, Linger value)
|
void setOption(SocketOptionLevel level, SocketOption option, Linger value)
|
||||||
const @trusted @nogc
|
const @trusted @nogc
|
||||||
{
|
{
|
||||||
setOption(level, option, (&value.clinger)[0..1]);
|
setOption(level, option, (&value)[0 .. 1]);
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Ditto.
|
/// Ditto.
|
||||||
@ -854,7 +1044,7 @@ abstract class Socket
|
|||||||
{
|
{
|
||||||
timeval tv;
|
timeval tv;
|
||||||
value.split!("seconds", "usecs")(tv.tv_sec, tv.tv_usec);
|
value.split!("seconds", "usecs")(tv.tv_sec, tv.tv_usec);
|
||||||
setOption(level, option, (&tv)[0..1]);
|
setOption(level, option, (&tv)[0 .. 1]);
|
||||||
}
|
}
|
||||||
else version (Windows)
|
else version (Windows)
|
||||||
{
|
{
|
||||||
@ -878,7 +1068,7 @@ abstract class Socket
|
|||||||
}
|
}
|
||||||
else version (Windows)
|
else version (Windows)
|
||||||
{
|
{
|
||||||
return blocking_;
|
return this.blocking_;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -892,12 +1082,12 @@ abstract class Socket
|
|||||||
{
|
{
|
||||||
int fl = fcntl(handle_, F_GETFL, 0);
|
int fl = fcntl(handle_, F_GETFL, 0);
|
||||||
|
|
||||||
if (fl != SOCKET_ERROR)
|
if (fl != socketError)
|
||||||
{
|
{
|
||||||
fl = yes ? fl & ~O_NONBLOCK : fl | O_NONBLOCK;
|
fl = yes ? fl & ~O_NONBLOCK : fl | O_NONBLOCK;
|
||||||
fl = fcntl(handle_, F_SETFL, fl);
|
fl = fcntl(handle_, F_SETFL, fl);
|
||||||
}
|
}
|
||||||
if (fl == SOCKET_ERROR)
|
if (fl == socketError)
|
||||||
{
|
{
|
||||||
throw make!SocketException(defaultAllocator,
|
throw make!SocketException(defaultAllocator,
|
||||||
"Unable to set socket blocking");
|
"Unable to set socket blocking");
|
||||||
@ -906,12 +1096,12 @@ abstract class Socket
|
|||||||
else version (Windows)
|
else version (Windows)
|
||||||
{
|
{
|
||||||
uint num = !yes;
|
uint num = !yes;
|
||||||
if (ioctlsocket(handle_, FIONBIO, &num) == SOCKET_ERROR)
|
if (ioctlsocket(handle_, FIONBIO, &num) == socketError)
|
||||||
{
|
{
|
||||||
throw make!SocketException(defaultAllocator,
|
throw make!SocketException(defaultAllocator,
|
||||||
"Unable to set socket blocking");
|
"Unable to set socket blocking");
|
||||||
}
|
}
|
||||||
blocking_ = yes;
|
this.blocking_ = yes;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -963,7 +1153,7 @@ abstract class Socket
|
|||||||
{
|
{
|
||||||
.close(handle_);
|
.close(handle_);
|
||||||
}
|
}
|
||||||
handle_ = socket_t.init;
|
handle_ = SocketType.init;
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
@ -971,12 +1161,12 @@ abstract class Socket
|
|||||||
* can $(D_PSYMBOL listen).
|
* can $(D_PSYMBOL listen).
|
||||||
*
|
*
|
||||||
* Params:
|
* Params:
|
||||||
* backlog = Request of how many pending incoming connections are
|
* backlog = Request of how many pending incoming connections are
|
||||||
* queued until $(D_PSYMBOL accept)ed.
|
* queued until $(D_PSYMBOL accept)ed.
|
||||||
*/
|
*/
|
||||||
void listen(int backlog) const @trusted @nogc
|
void listen(int backlog) const @trusted @nogc
|
||||||
{
|
{
|
||||||
if (.listen(handle_, backlog) == SOCKET_ERROR)
|
if (.listen(handle_, backlog) == socketError)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Unable to listen on socket");
|
throw defaultAllocator.make!SocketException("Unable to listen on socket");
|
||||||
}
|
}
|
||||||
@ -1029,8 +1219,8 @@ class StreamSocket : Socket, ConnectionOrientedSocket
|
|||||||
*/
|
*/
|
||||||
this(AddressFamily af) @trusted @nogc
|
this(AddressFamily af) @trusted @nogc
|
||||||
{
|
{
|
||||||
auto handle = cast(socket_t) socket(af, SOCK_STREAM, 0);
|
auto handle = cast(SocketType) socket(af, 1, 0);
|
||||||
if (handle == socket_t.init)
|
if (handle == SocketType.init)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Unable to create socket");
|
throw defaultAllocator.make!SocketException("Unable to create socket");
|
||||||
}
|
}
|
||||||
@ -1047,7 +1237,7 @@ class StreamSocket : Socket, ConnectionOrientedSocket
|
|||||||
*/
|
*/
|
||||||
void bind(Address address) const @trusted @nogc
|
void bind(Address address) const @trusted @nogc
|
||||||
{
|
{
|
||||||
if (.bind(handle_, address.name, address.length) == SOCKET_ERROR)
|
if (.bind(handle_, address.name, address.length) == socketError)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Unable to bind socket");
|
throw defaultAllocator.make!SocketException("Unable to bind socket");
|
||||||
}
|
}
|
||||||
@ -1066,7 +1256,7 @@ class StreamSocket : Socket, ConnectionOrientedSocket
|
|||||||
*/
|
*/
|
||||||
ConnectedSocket accept() @trusted @nogc
|
ConnectedSocket accept() @trusted @nogc
|
||||||
{
|
{
|
||||||
socket_t sock;
|
SocketType sock;
|
||||||
|
|
||||||
version (linux)
|
version (linux)
|
||||||
{
|
{
|
||||||
@ -1075,14 +1265,14 @@ class StreamSocket : Socket, ConnectionOrientedSocket
|
|||||||
{
|
{
|
||||||
flags |= SOCK_NONBLOCK;
|
flags |= SOCK_NONBLOCK;
|
||||||
}
|
}
|
||||||
sock = cast(socket_t).accept4(handle_, null, null, flags);
|
sock = cast(SocketType).accept4(handle_, null, null, flags);
|
||||||
}
|
}
|
||||||
else
|
else
|
||||||
{
|
{
|
||||||
sock = cast(socket_t).accept(handle_, null, null);
|
sock = cast(SocketType).accept(handle_, null, null);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (sock == socket_t.init)
|
if (sock == SocketType.init)
|
||||||
{
|
{
|
||||||
if (wouldHaveBlocked())
|
if (wouldHaveBlocked())
|
||||||
{
|
{
|
||||||
@ -1147,7 +1337,7 @@ class ConnectedSocket : Socket, ConnectionOrientedSocket
|
|||||||
* handle = Socket.
|
* handle = Socket.
|
||||||
* af = Address family.
|
* af = Address family.
|
||||||
*/
|
*/
|
||||||
this(socket_t handle, AddressFamily af) @nogc
|
this(SocketType handle, AddressFamily af) @nogc
|
||||||
{
|
{
|
||||||
super(handle, af);
|
super(handle, af);
|
||||||
}
|
}
|
||||||
@ -1196,7 +1386,7 @@ class ConnectedSocket : Socket, ConnectionOrientedSocket
|
|||||||
{
|
{
|
||||||
disconnected_ = true;
|
disconnected_ = true;
|
||||||
}
|
}
|
||||||
else if (ret == SOCKET_ERROR)
|
else if (ret == socketError)
|
||||||
{
|
{
|
||||||
if (wouldHaveBlocked())
|
if (wouldHaveBlocked())
|
||||||
{
|
{
|
||||||
@ -1233,7 +1423,7 @@ class ConnectedSocket : Socket, ConnectionOrientedSocket
|
|||||||
}
|
}
|
||||||
|
|
||||||
sent = .send(handle_, buf.ptr, capToMaxBuffer(buf.length), sendFlags);
|
sent = .send(handle_, buf.ptr, capToMaxBuffer(buf.length), sendFlags);
|
||||||
if (sent != SOCKET_ERROR)
|
if (sent != socketError)
|
||||||
{
|
{
|
||||||
return sent;
|
return sent;
|
||||||
}
|
}
|
||||||
@ -1273,14 +1463,11 @@ class InternetAddress : Address
|
|||||||
/// Internal internet address representation.
|
/// Internal internet address representation.
|
||||||
protected sockaddr_storage storage;
|
protected sockaddr_storage storage;
|
||||||
}
|
}
|
||||||
immutable ushort port_;
|
const ushort port_;
|
||||||
|
|
||||||
enum
|
enum ushort anyPort = 0;
|
||||||
{
|
|
||||||
anyPort = 0,
|
|
||||||
}
|
|
||||||
|
|
||||||
this(in string host, ushort port = anyPort) @nogc
|
this(string host, const ushort port = anyPort) @nogc
|
||||||
{
|
{
|
||||||
if (getaddrinfoPointer is null || freeaddrinfoPointer is null)
|
if (getaddrinfoPointer is null || freeaddrinfoPointer is null)
|
||||||
{
|
{
|
||||||
@ -1288,11 +1475,11 @@ class InternetAddress : Address
|
|||||||
"Address info lookup is not available on this system");
|
"Address info lookup is not available on this system");
|
||||||
}
|
}
|
||||||
addrinfo* ai_res;
|
addrinfo* ai_res;
|
||||||
port_ = port;
|
this.port_ = port;
|
||||||
|
|
||||||
// Make C-string from host.
|
// Make C-string from host.
|
||||||
auto node = cast(char[]) allocator.allocate(host.length + 1);
|
auto node = cast(char[]) allocator.allocate(host.length + 1);
|
||||||
node[0.. $ - 1] = host;
|
node[0 .. $ - 1] = host;
|
||||||
node[$ - 1] = '\0';
|
node[$ - 1] = '\0';
|
||||||
scope (exit)
|
scope (exit)
|
||||||
{
|
{
|
||||||
@ -1304,18 +1491,19 @@ class InternetAddress : Address
|
|||||||
const(char)* servicePointer;
|
const(char)* servicePointer;
|
||||||
if (port)
|
if (port)
|
||||||
{
|
{
|
||||||
|
ushort originalPort = port;
|
||||||
ushort start;
|
ushort start;
|
||||||
for (ushort j = 10, i = 4; i > 0; j *= 10, --i)
|
for (ushort j = 10, i = 4; i > 0; j *= 10, --i)
|
||||||
{
|
{
|
||||||
ushort rest = port % 10;
|
ushort rest = originalPort % 10;
|
||||||
if (rest != 0)
|
if (rest != 0)
|
||||||
{
|
{
|
||||||
service[i] = cast(char) (rest + '0');
|
service[i] = cast(char) (rest + '0');
|
||||||
start = i;
|
start = i;
|
||||||
}
|
}
|
||||||
port /= 10;
|
originalPort /= 10;
|
||||||
}
|
}
|
||||||
servicePointer = service[start..$].ptr;
|
servicePointer = service[start .. $].ptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto ret = getaddrinfoPointer(node.ptr, servicePointer, null, &ai_res);
|
auto ret = getaddrinfoPointer(node.ptr, servicePointer, null, &ai_res);
|
||||||
@ -1327,18 +1515,29 @@ class InternetAddress : Address
|
|||||||
{
|
{
|
||||||
freeaddrinfoPointer(ai_res);
|
freeaddrinfoPointer(ai_res);
|
||||||
}
|
}
|
||||||
|
|
||||||
ubyte* dp = cast(ubyte*) &storage, sp = cast(ubyte*) ai_res.ai_addr;
|
ubyte* dp = cast(ubyte*) &storage, sp = cast(ubyte*) ai_res.ai_addr;
|
||||||
for (auto i = ai_res.ai_addrlen; i > 0; --i, *dp++, *sp++)
|
for (auto i = ai_res.ai_addrlen; i > 0; --i, *dp++, *sp++)
|
||||||
{
|
{
|
||||||
*dp = *sp;
|
*dp = *sp;
|
||||||
}
|
}
|
||||||
if (ai_res.ai_family != AddressFamily.INET && ai_res.ai_family != AddressFamily.INET6)
|
if (ai_res.ai_family != AddressFamily.inet && ai_res.ai_family != AddressFamily.inet6)
|
||||||
{
|
{
|
||||||
throw defaultAllocator.make!SocketException("Wrong address family");
|
throw defaultAllocator.make!SocketException("Wrong address family");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
auto address = defaultAllocator.make!InternetAddress("127.0.0.1");
|
||||||
|
assert(address.port == InternetAddress.anyPort);
|
||||||
|
assert(address.name !is null);
|
||||||
|
assert(address.family == AddressFamily.inet);
|
||||||
|
|
||||||
|
defaultAllocator.dispose(address);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Returns: Pointer to underlying $(D_PSYMBOL sockaddr) structure.
|
* Returns: Pointer to underlying $(D_PSYMBOL sockaddr) structure.
|
||||||
*/
|
*/
|
||||||
@ -1355,9 +1554,9 @@ class InternetAddress : Address
|
|||||||
// FreeBSD wants to know the exact length of the address on bind.
|
// FreeBSD wants to know the exact length of the address on bind.
|
||||||
switch (family)
|
switch (family)
|
||||||
{
|
{
|
||||||
case AddressFamily.INET:
|
case AddressFamily.inet:
|
||||||
return sockaddr_in.sizeof;
|
return sockaddr_in.sizeof;
|
||||||
case AddressFamily.INET6:
|
case AddressFamily.inet6:
|
||||||
return sockaddr_in6.sizeof;
|
return sockaddr_in6.sizeof;
|
||||||
default:
|
default:
|
||||||
assert(false);
|
assert(false);
|
||||||
@ -1376,6 +1575,15 @@ class InternetAddress : Address
|
|||||||
{
|
{
|
||||||
return port_;
|
return port_;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
auto address = defaultAllocator.make!InternetAddress("127.0.0.1",
|
||||||
|
cast(ushort) 1234);
|
||||||
|
assert(address.port == 1234);
|
||||||
|
defaultAllocator.dispose(address);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -10,6 +10,7 @@
|
|||||||
* 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)
|
||||||
*/
|
*/
|
||||||
|
deprecated("Use tanya.net.uri instead")
|
||||||
module tanya.network.url;
|
module tanya.network.url;
|
||||||
|
|
||||||
import std.ascii : isAlphaNum, isDigit;
|
import std.ascii : isAlphaNum, isDigit;
|
||||||
|
337
source/tanya/os/error.d
Normal file
337
source/tanya/os/error.d
Normal file
@ -0,0 +1,337 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This module provides a portable way of using operating system error codes.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.os.error;
|
||||||
|
|
||||||
|
// Socket API error.
|
||||||
|
private template SAError(int posix, int wsa = posix)
|
||||||
|
{
|
||||||
|
version (Windows)
|
||||||
|
{
|
||||||
|
enum SAError = 10000 + wsa;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
alias SAError = posix;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Error for Windows and Posix separately.
|
||||||
|
private template NativeError(int posix, int win)
|
||||||
|
{
|
||||||
|
version (Windows)
|
||||||
|
{
|
||||||
|
alias NativeError = win;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
alias NativeError = posix;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
version (Windows)
|
||||||
|
{
|
||||||
|
private enum eProtocolError = -71;
|
||||||
|
}
|
||||||
|
else version (OpenBSD)
|
||||||
|
{
|
||||||
|
private enum eProtocolError = -71;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
private enum eProtocolError = 71;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* System error code.
|
||||||
|
*/
|
||||||
|
struct ErrorCode
|
||||||
|
{
|
||||||
|
/**
|
||||||
|
* Error code numbers.
|
||||||
|
*/
|
||||||
|
enum ErrorNo : int
|
||||||
|
{
|
||||||
|
/// The operation completed successfully.
|
||||||
|
success = 0,
|
||||||
|
|
||||||
|
/// Operation not permitted.
|
||||||
|
noPermission = NativeError!(1, 5),
|
||||||
|
|
||||||
|
/// Interrupted system call.
|
||||||
|
interrupted = SAError!4,
|
||||||
|
|
||||||
|
/// Bad file descriptor.
|
||||||
|
badDescriptor = SAError!9,
|
||||||
|
|
||||||
|
/// An operation on a non-blocking socket would block.
|
||||||
|
wouldBlock = SAError!(11, 35),
|
||||||
|
|
||||||
|
/// Out of memory.
|
||||||
|
noMemory = NativeError!(12, 14),
|
||||||
|
|
||||||
|
/// Access denied.
|
||||||
|
accessDenied = SAError!13,
|
||||||
|
|
||||||
|
/// An invalid pointer address detected.
|
||||||
|
fault = SAError!14,
|
||||||
|
|
||||||
|
/// No such device.
|
||||||
|
noSuchDevice = NativeError!(19, 20),
|
||||||
|
|
||||||
|
/// An invalid argument was supplied.
|
||||||
|
invalidArgument = SAError!22,
|
||||||
|
|
||||||
|
/// The limit on the number of open file descriptors.
|
||||||
|
tooManyDescriptors = NativeError!(23, 331),
|
||||||
|
|
||||||
|
/// The limit on the number of open file descriptors.
|
||||||
|
noDescriptors = SAError!24,
|
||||||
|
|
||||||
|
/// Broken pipe.
|
||||||
|
brokenPipe = NativeError!(32, 109),
|
||||||
|
|
||||||
|
/// The name was too long.
|
||||||
|
nameTooLong = SAError!(36, 63),
|
||||||
|
|
||||||
|
/// A socket operation was attempted on a non-socket.
|
||||||
|
notSocket = SAError!(88, 38),
|
||||||
|
|
||||||
|
/// Protocol error.
|
||||||
|
protocolError = eProtocolError,
|
||||||
|
|
||||||
|
/// Message too long.
|
||||||
|
messageTooLong = SAError!(90, 40),
|
||||||
|
|
||||||
|
/// Wrong protocol type for socket.
|
||||||
|
wrongProtocolType = SAError!(91, 41),
|
||||||
|
|
||||||
|
/// Protocol not available.
|
||||||
|
noProtocolOption = SAError!(92, 42),
|
||||||
|
|
||||||
|
/// The protocol is not implemented orR has not been configured.
|
||||||
|
protocolNotSupported = SAError!(93, 43),
|
||||||
|
|
||||||
|
/// The support for the specified socket type does not exist in this
|
||||||
|
/// address family.
|
||||||
|
socketNotSupported = SAError!(94, 44),
|
||||||
|
|
||||||
|
/// The address family is no supported by the protocol family.
|
||||||
|
operationNotSupported = SAError!(95, 45),
|
||||||
|
|
||||||
|
/// Address family specified is not supported.
|
||||||
|
addressFamilyNotSupported = SAError!(97, 47),
|
||||||
|
|
||||||
|
/// Address already in use.
|
||||||
|
addressInUse = SAError!(98, 48),
|
||||||
|
|
||||||
|
/// The network is not available.
|
||||||
|
networkDown = SAError!(100, 50),
|
||||||
|
|
||||||
|
/// No route to host.
|
||||||
|
networkUnreachable = SAError!(101, 51),
|
||||||
|
|
||||||
|
/// Network dropped connection because of reset.
|
||||||
|
networkReset = SAError!(102, 52),
|
||||||
|
|
||||||
|
/// The connection has been aborted.
|
||||||
|
connectionAborted = SAError!(103, 53),
|
||||||
|
|
||||||
|
/// Connection reset by peer.
|
||||||
|
connectionReset = SAError!(104, 54),
|
||||||
|
|
||||||
|
/// No free buffer space is available for a socket operation.
|
||||||
|
noBufferSpace = SAError!(105, 55),
|
||||||
|
|
||||||
|
/// Transport endpoint is already connected.
|
||||||
|
alreadyConnected = SAError!(106, 56),
|
||||||
|
|
||||||
|
/// Transport endpoint is not connected.
|
||||||
|
notConnected = SAError!(107, 57),
|
||||||
|
|
||||||
|
/// Cannot send after transport endpoint shutdown.
|
||||||
|
shutdown = SAError!(108, 58),
|
||||||
|
|
||||||
|
/// The connection attempt timed out, or the connected host has failed
|
||||||
|
/// to respond.
|
||||||
|
timedOut = SAError!(110, 60),
|
||||||
|
|
||||||
|
/// Connection refused.
|
||||||
|
connectionRefused = SAError!(111, 61),
|
||||||
|
|
||||||
|
/// Host is down.
|
||||||
|
hostDown = SAError!(112, 64),
|
||||||
|
|
||||||
|
/// No route to host.
|
||||||
|
hostUnreachable = SAError!(113, 65),
|
||||||
|
|
||||||
|
/// Operation already in progress.
|
||||||
|
alreadyStarted = SAError!(114, 37),
|
||||||
|
|
||||||
|
/// Operation now in progress.
|
||||||
|
inProgress = SAError!(115, 36),
|
||||||
|
|
||||||
|
/// Operation cancelled.
|
||||||
|
cancelled = SAError!(125, 103),
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Constructor.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* value = Numeric error code.
|
||||||
|
*/
|
||||||
|
this(const ErrorNo value) pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
this.value_ = value;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
ErrorCode ec;
|
||||||
|
assert(ec == ErrorCode.success);
|
||||||
|
|
||||||
|
ec = ErrorCode.fault;
|
||||||
|
assert(ec == ErrorCode.fault);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Resets this $(D_PSYMBOL ErrorCode) to default
|
||||||
|
* ($(D_PSYMBOL ErrorCode.success)).
|
||||||
|
*/
|
||||||
|
void reset() pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
this.value_ = ErrorNo.success;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
auto ec = ErrorCode(ErrorCode.fault);
|
||||||
|
assert(ec == ErrorCode.fault);
|
||||||
|
|
||||||
|
ec.reset();
|
||||||
|
assert(ec == ErrorCode.success);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns: Numeric error code.
|
||||||
|
*/
|
||||||
|
ErrorNo opCast(T : ErrorNo)() const
|
||||||
|
{
|
||||||
|
return this.value_;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ErrorNo opCast(T : int)() const
|
||||||
|
{
|
||||||
|
return this.value_;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
ErrorCode ec = ErrorCode.fault;
|
||||||
|
auto errorNo = cast(ErrorCode.ErrorNo) ec;
|
||||||
|
|
||||||
|
assert(errorNo == ErrorCode.fault);
|
||||||
|
static assert(is(typeof(cast(int) ec)));
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Assigns another error code or error code number.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* that = Numeric error code.
|
||||||
|
*
|
||||||
|
* Returns: $(D_KEYWORD this).
|
||||||
|
*/
|
||||||
|
ref ErrorCode opAssign(const ErrorNo that) pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
this.value_ = that;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
ref ErrorCode opAssign()(auto ref const ErrorCode that)
|
||||||
|
pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
this.value_ = that.value_;
|
||||||
|
return this;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
{
|
||||||
|
ErrorCode ec;
|
||||||
|
assert(ec == ErrorCode.success);
|
||||||
|
|
||||||
|
ec = ErrorCode.fault;
|
||||||
|
assert(ec == ErrorCode.fault);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
auto ec1 = ErrorCode(ErrorCode.fault);
|
||||||
|
ErrorCode ec2;
|
||||||
|
assert(ec2 == ErrorCode.success);
|
||||||
|
|
||||||
|
ec2 = ec1;
|
||||||
|
assert(ec1 == ec2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Equality with another error code or error code number.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* that = Numeric error code.
|
||||||
|
*
|
||||||
|
* Returns: Whether $(D_KEYWORD this) and $(D_PARAM that) are equal.
|
||||||
|
*/
|
||||||
|
bool opEquals(const ErrorNo that) const pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return this.value_ == that;
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Ditto.
|
||||||
|
bool opEquals()(auto ref const ErrorCode that)
|
||||||
|
const pure nothrow @safe @nogc
|
||||||
|
{
|
||||||
|
return this.value_ == that.value_;
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
pure nothrow @safe @nogc unittest
|
||||||
|
{
|
||||||
|
{
|
||||||
|
ErrorCode ec1 = ErrorCode.fault;
|
||||||
|
ErrorCode ec2 = ErrorCode.accessDenied;
|
||||||
|
|
||||||
|
assert(ec1 != ec2);
|
||||||
|
assert(ec1 != ErrorCode.accessDenied);
|
||||||
|
assert(ErrorCode.fault != ec2);
|
||||||
|
}
|
||||||
|
{
|
||||||
|
ErrorCode ec1 = ErrorCode.fault;
|
||||||
|
ErrorCode ec2 = ErrorCode.fault;
|
||||||
|
|
||||||
|
assert(ec1 == ec2);
|
||||||
|
assert(ec1 == ErrorCode.fault);
|
||||||
|
assert(ErrorCode.fault == ec2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private ErrorNo value_ = ErrorNo.success;
|
||||||
|
|
||||||
|
alias ErrorNo this;
|
||||||
|
}
|
16
source/tanya/os/package.d
Normal file
16
source/tanya/os/package.d
Normal file
@ -0,0 +1,16 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* This package provides platform-independent interfaces to operating system
|
||||||
|
* functionality.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.os;
|
||||||
|
|
||||||
|
public import tanya.os.error;
|
107
source/tanya/typecons.d
Normal file
107
source/tanya/typecons.d
Normal file
@ -0,0 +1,107 @@
|
|||||||
|
/* 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/. */
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Type constructors.
|
||||||
|
*
|
||||||
|
* This module contains templates that allow to build new types from the
|
||||||
|
* available ones.
|
||||||
|
*
|
||||||
|
* Copyright: Eugene Wissner 2017.
|
||||||
|
* 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.typecons;
|
||||||
|
|
||||||
|
import std.meta;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* $(D_PSYMBOL Pair) can store two heterogeneous objects.
|
||||||
|
*
|
||||||
|
* The objects can by accessed by index as $(D_INLINECODE obj[0]) and
|
||||||
|
* $(D_INLINECODE obj[1]) or by optional names (e.g.
|
||||||
|
* $(D_INLINECODE obj.first)).
|
||||||
|
*
|
||||||
|
* $(D_PARAM Specs) contains a list of object types and names. First
|
||||||
|
* comes the object type, then an optional string containing the name.
|
||||||
|
* If you want the object be accessible only by its index (`0` or `1`),
|
||||||
|
* just skip the name.
|
||||||
|
*
|
||||||
|
* Params:
|
||||||
|
* Specs = Field types and names.
|
||||||
|
*/
|
||||||
|
template Pair(Specs...)
|
||||||
|
{
|
||||||
|
template parseSpecs(int fieldCount, Specs...)
|
||||||
|
{
|
||||||
|
static if (Specs.length == 0)
|
||||||
|
{
|
||||||
|
alias parseSpecs = AliasSeq!();
|
||||||
|
}
|
||||||
|
else static if (is(Specs[0]) && fieldCount < 2)
|
||||||
|
{
|
||||||
|
static if (is(typeof(Specs[1]) == string))
|
||||||
|
{
|
||||||
|
alias parseSpecs
|
||||||
|
= AliasSeq!(Specs[0],
|
||||||
|
parseSpecs!(fieldCount + 1, Specs[2 .. $]));
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
alias parseSpecs
|
||||||
|
= AliasSeq!(Specs[0],
|
||||||
|
parseSpecs!(fieldCount + 1, Specs[1 .. $]));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
static assert(false, "Invalid argument: " ~ Specs[0].stringof);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
struct Pair
|
||||||
|
{
|
||||||
|
/// Field types.
|
||||||
|
alias Types = parseSpecs!(0, Specs);
|
||||||
|
|
||||||
|
static assert(Types.length == 2, "Invalid argument count.");
|
||||||
|
|
||||||
|
// Create field aliases.
|
||||||
|
static if (is(typeof(Specs[1]) == string))
|
||||||
|
{
|
||||||
|
mixin("alias " ~ Specs[1] ~ " = expand[0];");
|
||||||
|
}
|
||||||
|
static if (is(typeof(Specs[2]) == string))
|
||||||
|
{
|
||||||
|
mixin("alias " ~ Specs[2] ~ " = expand[1];");
|
||||||
|
}
|
||||||
|
else static if (is(typeof(Specs[3]) == string))
|
||||||
|
{
|
||||||
|
mixin("alias " ~ Specs[3] ~ " = expand[1];");
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Represents the values of the $(D_PSYMBOL Pair) as a list of values.
|
||||||
|
Types expand;
|
||||||
|
|
||||||
|
alias expand this;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
///
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
static assert(is(Pair!(int, int)));
|
||||||
|
static assert(!is(Pair!(int, 5)));
|
||||||
|
|
||||||
|
static assert(is(Pair!(int, "first", int)));
|
||||||
|
static assert(is(Pair!(int, "first", int, "second")));
|
||||||
|
static assert(is(Pair!(int, "first", int)));
|
||||||
|
|
||||||
|
static assert(is(Pair!(int, int, "second")));
|
||||||
|
static assert(!is(Pair!("first", int, "second", int)));
|
||||||
|
static assert(!is(Pair!(int, int, int)));
|
||||||
|
|
||||||
|
static assert(!is(Pair!(int, "first")));
|
||||||
|
}
|
Reference in New Issue
Block a user