MegurineLuka/chuniio-yubideck
1
0
Fork 0
forked from beerpsi/chuniio-yubideck
Code Pull Requests Activity

add aimeio

Browse Source
This commit is contained in:
beerpsi 2024-01-02 11:27:10 +07:00
parent de021005fb
commit 4a28826edb
18 changed files with 1468 additions and 238 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

539
Cargo.lock generated
View File

@ -3,10 +3,44 @@
version = 3
[[package]]
name = "anyhow"
version = "1.0.76"
name = "aho-corasick"
version = "1.1.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "59d2a3357dde987206219e78ecfbbb6e8dad06cbb65292758d3270e6254f7355"
checksum = "b2969dcb958b36655471fc61f7e416fa76033bdd4bfed0678d8fee1e2d07a1f0"
dependencies = [
"memchr",
]
[[package]]
name = "aimeio-yubideck"
version = "0.1.0"
dependencies = [
"cfg_aliases",
"faster-hex",
"lazy_static",
"log",
"rand",
"shared_memory",
"winapi",
"yubideck-common",
]
[[package]]
name = "anyhow"
version = "1.0.78"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ca87830a3e3fb156dc96cfbd31cb620265dd053be734723f22b760d6cc3c3051"
[[package]]
name = "atty"
version = "0.2.14"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d9b39be18770d11421cdb1b9947a45dd3f37e93092cbf377614828a319d5fee8"
dependencies = [
"hermit-abi 0.1.19",
"libc",
"winapi",
]
[[package]]
name = "autocfg"
@ -20,6 +54,12 @@ version = "1.3.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
[[package]]
name = "bitflags"
version = "2.4.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "327762f6e5a765692301e5bb513e0d9fef63be86bbc14528052b1cd3e6f03e07"
[[package]]
name = "cc"
version = "1.0.83"
@ -29,6 +69,12 @@ dependencies = [
"libc",
]
[[package]]
name = "cfg-if"
version = "0.1.10"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4785bdd1c96b2a846b2bd7cc02e86b6b3dbf14e7e53446c4f54c92a361040822"
[[package]]
name = "cfg-if"
version = "1.0.0"
@ -46,13 +92,53 @@ name = "chuniio-yubideck"
version = "0.1.0"
dependencies = [
"anyhow",
"cfg-if 1.0.0",
"cfg_aliases",
"env_logger",
"lazy_static",
"log",
"rusb",
"shared_memory",
"winapi",
"yubideck-common",
]
[[package]]
name = "clap"
version = "3.2.25"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4ea181bf566f71cb9a5d17a59e1871af638180a18fb0035c92ae62b705207123"
dependencies = [
"atty",
"bitflags 1.3.2",
"clap_derive",
"clap_lex",
"indexmap",
"once_cell",
"strsim",
"termcolor",
"textwrap",
]
[[package]]
name = "clap_derive"
version = "3.2.25"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ae6371b8bdc8b7d3959e9cf7b22d4435ef3e79e138688421ec654acf8c81b008"
dependencies = [
"heck",
"proc-macro-error",
"proc-macro2",
"quote",
"syn 1.0.109",
]
[[package]]
name = "clap_lex"
version = "0.2.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2850f2f5a82cbf437dd5af4d49848fbdfc27c157c3d010345776f952765261c5"
dependencies = [
"os_str_bytes",
]
[[package]]
@ -61,7 +147,95 @@ version = "0.10.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "95b3f3e67048839cb0d0781f445682a35113da7121f7c949db0e2be96a4fbece"
dependencies = [
"humantime",
"is-terminal",
"log",
"regex",
"termcolor",
]
[[package]]
name = "errno"
version = "0.3.8"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a258e46cdc063eb8519c00b9fc845fc47bcfca4130e2f08e88665ceda8474245"
dependencies = [
"libc",
"windows-sys",
]
[[package]]
name = "faster-hex"
version = "0.9.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a2a2b11eda1d40935b26cf18f6833c526845ae8c41e58d09af6adeb6f0269183"
dependencies = [
"serde",
]
[[package]]
name = "getrandom"
version = "0.2.11"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "fe9006bed769170c11f845cf00c7c1e9092aeb3f268e007c3e760ac68008070f"
dependencies = [
"cfg-if 1.0.0",
"libc",
"wasi",
]
[[package]]
name = "hashbrown"
version = "0.12.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8a9ee70c43aaf417c914396645a0fa852624801b24ebb7ae78fe8272889ac888"
[[package]]
name = "heck"
version = "0.4.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "95505c38b4572b2d910cecb0281560f54b440a19336cbbcb27bf6ce6adc6f5a8"
[[package]]
name = "hermit-abi"
version = "0.1.19"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "62b467343b94ba476dcb2500d242dadbb39557df889310ac77c5d99100aaac33"
dependencies = [
"libc",
]
[[package]]
name = "hermit-abi"
version = "0.3.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d77f7ec81a6d05a3abb01ab6eb7590f6083d08449fe5a1c8b1e620283546ccb7"
[[package]]
name = "humantime"
version = "2.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9a3a5bfb195931eeb336b2a7b4d761daec841b97f947d34394601737a7bba5e4"
[[package]]
name = "indexmap"
version = "1.9.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bd070e393353796e801d209ad339e89596eb4c8d430d18ede6a1cced8fafbd99"
dependencies = [
"autocfg",
"hashbrown",
]
[[package]]
name = "is-terminal"
version = "0.4.10"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0bad00257d07be169d870ab665980b06cdb366d792ad690bf2e76876dc503455"
dependencies = [
"hermit-abi 0.3.3",
"rustix",
"windows-sys",
]
[[package]]
@ -88,12 +262,24 @@ dependencies = [
"vcpkg",
]
[[package]]
name = "linux-raw-sys"
version = "0.4.12"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c4cd1a83af159aa67994778be9070f0ae1bd732942279cabb14f86f986a21456"
[[package]]
name = "log"
version = "0.4.20"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b5e6163cb8c49088c2c36f57875e58ccd8c87c7427f7fbd50ea6710b2f3f2e8f"
[[package]]
name = "memchr"
version = "2.7.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "523dc4f511e55ab87b694dc30d0f820d60906ef06413f93d4d7a1385599cc149"
[[package]]
name = "memoffset"
version = "0.6.5"
@ -109,19 +295,162 @@ version = "0.23.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8f3790c00a0150112de0f4cd161e3d7fc4b2d8a5542ffc35f099a2562aecb35c"
dependencies = [
"bitflags",
"bitflags 1.3.2",
"cc",
"cfg-if",
"cfg-if 1.0.0",
"libc",
"memoffset",
]
[[package]]
name = "nix"
version = "0.27.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2eb04e9c688eff1c89d72b407f168cf79bb9e867a9d3323ed6c01519eb9cc053"
dependencies = [
"bitflags 2.4.1",
"cfg-if 1.0.0",
"libc",
]
[[package]]
name = "once_cell"
version = "1.19.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3fdb12b2476b595f9358c5161aa467c2438859caa136dec86c26fdd2efe17b92"
[[package]]
name = "os_str_bytes"
version = "6.6.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e2355d85b9a3786f481747ced0e0ff2ba35213a1f9bd406ed906554d7af805a1"
[[package]]
name = "pkg-config"
version = "0.3.28"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "69d3587f8a9e599cc7ec2c00e331f71c4e69a5f9a4b8a6efd5b07466b9736f9a"
[[package]]
name = "ppv-lite86"
version = "0.2.17"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5b40af805b3121feab8a3c29f04d8ad262fa8e0561883e7653e024ae4479e6de"
[[package]]
name = "proc-macro-error"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "da25490ff9892aab3fcf7c36f08cfb902dd3e71ca0f9f9517bea02a73a5ce38c"
dependencies = [
"proc-macro-error-attr",
"proc-macro2",
"quote",
"syn 1.0.109",
"version_check",
]
[[package]]
name = "proc-macro-error-attr"
version = "1.0.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a1be40180e52ecc98ad80b184934baf3d0d29f979574e439af5a55274b35f869"
dependencies = [
"proc-macro2",
"quote",
"version_check",
]
[[package]]
name = "proc-macro2"
version = "1.0.73"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2dd5e8a1f1029c43224ad5898e50140c2aebb1705f19e67c918ebf5b9e797fe1"
dependencies = [
"unicode-ident",
]
[[package]]
name = "quote"
version = "1.0.34"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "22a37c9326af5ed140c86a46655b5278de879853be5573c01df185b6f49a580a"
dependencies = [
"proc-macro2",
]
[[package]]
name = "rand"
version = "0.8.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "34af8d1a0e25924bc5b7c43c079c942339d8f0a8b57c39049bef581b46327404"
dependencies = [
"libc",
"rand_chacha",
"rand_core",
]
[[package]]
name = "rand_chacha"
version = "0.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e6c10a63a0fa32252be49d21e7709d4d4baf8d231c2dbce1eaa8141b9b127d88"
dependencies = [
"ppv-lite86",
"rand_core",
]
[[package]]
name = "rand_core"
version = "0.6.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ec0be4795e2f6a28069bec0b5ff3e2ac9bafc99e6a9a7dc3547996c5c816922c"
dependencies = [
"getrandom",
]
[[package]]
name = "raw_sync"
version = "0.1.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2a34bde3561f980a51c70495164200569a11662644fe5af017f0b5d7015688cc"
dependencies = [
"cfg-if 0.1.10",
"libc",
"nix 0.27.1",
"rand",
"winapi",
]
[[package]]
name = "regex"
version = "1.10.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "380b951a9c5e80ddfd6136919eef32310721aa4aacd4889a8d39124b026ab343"
dependencies = [
"aho-corasick",
"memchr",
"regex-automata",
"regex-syntax",
]
[[package]]
name = "regex-automata"
version = "0.4.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5f804c7828047e88b2d32e2d7fe5a105da8ee3264f01902f796c8e067dc2483f"
dependencies = [
"aho-corasick",
"memchr",
"regex-syntax",
]
[[package]]
name = "regex-syntax"
version = "0.8.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "c08c74e62047bb2de4ff487b251e4a92e24f48745648451635cec7d591162d9f"
[[package]]
name = "rusb"
version = "0.9.3"
@ -132,22 +461,120 @@ dependencies = [
"libusb1-sys",
]
[[package]]
name = "rustix"
version = "0.38.28"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "72e572a5e8ca657d7366229cdde4bd14c4eb5499a9573d4d366fe1b599daa316"
dependencies = [
"bitflags 2.4.1",
"errno",
"libc",
"linux-raw-sys",
"windows-sys",
]
[[package]]
name = "serde"
version = "1.0.193"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "25dd9975e68d0cb5aa1120c288333fc98731bd1dd12f561e468ea4728c042b89"
dependencies = [
"serde_derive",
]
[[package]]
name = "serde_derive"
version = "1.0.193"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "43576ca501357b9b071ac53cdc7da8ef0cbd9493d8df094cd821777ea6e894d3"
dependencies = [
"proc-macro2",
"quote",
"syn 2.0.45",
]
[[package]]
name = "shared_memory"
version = "0.12.4-chuniio"
dependencies = [
"cfg-if",
"cfg-if 1.0.0",
"clap",
"env_logger",
"libc",
"nix",
"log",
"nix 0.23.2",
"raw_sync",
"win-sys",
]
[[package]]
name = "strsim"
version = "0.10.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "73473c0e59e6d5812c5dfe2a064a6444949f089e20eec9a2e5506596494e4623"
[[package]]
name = "syn"
version = "1.0.109"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
dependencies = [
"proc-macro2",
"quote",
"unicode-ident",
]
[[package]]
name = "syn"
version = "2.0.45"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0eae3c679c56dc214320b67a1bc04ef3dfbd6411f6443974b5e4893231298e66"
dependencies = [
"proc-macro2",
"quote",
"unicode-ident",
]
[[package]]
name = "termcolor"
version = "1.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ff1bc3d3f05aff0403e8ac0d92ced918ec05b666a43f83297ccef5bea8a3d449"
dependencies = [
"winapi-util",
]
[[package]]
name = "textwrap"
version = "0.16.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "222a222a5bfe1bba4a77b45ec488a741b3cb8872e5e499451fd7d0129c9c7c3d"
[[package]]
name = "unicode-ident"
version = "1.0.12"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3354b9ac3fae1ff6755cb6db53683adb661634f67557942dea4facebec0fee4b"
[[package]]
name = "vcpkg"
version = "0.2.15"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "accd4ea62f7bb7a82fe23066fb0957d48ef677f6eeb8215f372f52e48bb32426"
[[package]]
name = "version_check"
version = "0.9.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "49874b5167b65d7193b8aba1567f5c7d93d001cafc34600cee003eda787e483f"
[[package]]
name = "wasi"
version = "0.11.0+wasi-snapshot-preview1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
[[package]]
name = "win-sys"
version = "0.3.1"
@ -173,6 +600,15 @@ version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
[[package]]
name = "winapi-util"
version = "0.1.6"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f29e6f9198ba0d26b4c9f07dbe6f9ed633e1f3d5b8b414090084349e46a52596"
dependencies = [
"winapi",
]
[[package]]
name = "winapi-x86_64-pc-windows-gnu"
version = "0.4.0"
@ -185,39 +621,116 @@ version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "45296b64204227616fdbf2614cefa4c236b98ee64dfaaaa435207ed99fe7829f"
dependencies = [
"windows_aarch64_msvc",
"windows_i686_gnu",
"windows_i686_msvc",
"windows_x86_64_gnu",
"windows_x86_64_msvc",
"windows_aarch64_msvc 0.34.0",
"windows_i686_gnu 0.34.0",
"windows_i686_msvc 0.34.0",
"windows_x86_64_gnu 0.34.0",
"windows_x86_64_msvc 0.34.0",
]
[[package]]
name = "windows-sys"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "282be5f36a8ce781fad8c8ae18fa3f9beff57ec1b52cb3de0789201425d9a33d"
dependencies = [
"windows-targets",
]
[[package]]
name = "windows-targets"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8a18201040b24831fbb9e4eb208f8892e1f50a37feb53cc7ff887feb8f50e7cd"
dependencies = [
"windows_aarch64_gnullvm",
"windows_aarch64_msvc 0.52.0",
"windows_i686_gnu 0.52.0",
"windows_i686_msvc 0.52.0",
"windows_x86_64_gnu 0.52.0",
"windows_x86_64_gnullvm",
"windows_x86_64_msvc 0.52.0",
]
[[package]]
name = "windows_aarch64_gnullvm"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "cb7764e35d4db8a7921e09562a0304bf2f93e0a51bfccee0bd0bb0b666b015ea"
[[package]]
name = "windows_aarch64_msvc"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "17cffbe740121affb56fad0fc0e421804adf0ae00891205213b5cecd30db881d"
[[package]]
name = "windows_aarch64_msvc"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bbaa0368d4f1d2aaefc55b6fcfee13f41544ddf36801e793edbbfd7d7df075ef"
[[package]]
name = "windows_i686_gnu"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2564fde759adb79129d9b4f54be42b32c89970c18ebf93124ca8870a498688ed"
[[package]]
name = "windows_i686_gnu"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "a28637cb1fa3560a16915793afb20081aba2c92ee8af57b4d5f28e4b3e7df313"
[[package]]
name = "windows_i686_msvc"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9cd9d32ba70453522332c14d38814bceeb747d80b3958676007acadd7e166956"
[[package]]
name = "windows_i686_msvc"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ffe5e8e31046ce6230cc7215707b816e339ff4d4d67c65dffa206fd0f7aa7b9a"
[[package]]
name = "windows_x86_64_gnu"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "cfce6deae227ee8d356d19effc141a509cc503dfd1f850622ec4b0f84428e1f4"
[[package]]
name = "windows_x86_64_gnu"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "3d6fa32db2bc4a2f5abeacf2b69f7992cd09dca97498da74a151a3132c26befd"
[[package]]
name = "windows_x86_64_gnullvm"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "1a657e1e9d3f514745a572a6846d3c7aa7dbe1658c056ed9c3344c4109a6949e"
[[package]]
name = "windows_x86_64_msvc"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d19538ccc21819d01deaf88d6a17eae6596a12e9aafdbb97916fb49896d89de9"
[[package]]
name = "windows_x86_64_msvc"
version = "0.52.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "dff9641d1cd4be8d1a070daf9e3773c5f67e78b4d9d42263020c057706765c04"
[[package]]
name = "yubideck-common"
version = "0.1.0"
dependencies = [
"anyhow",
"env_logger",
"log",
"shared_memory",
"winapi",
]

23
Cargo.toml
View File

@ -1,10 +1,6 @@
[package]
name = "chuniio-yubideck"
version = "0.1.0"
edition = "2021"
[lib]
crate-type = ["cdylib"]
[workspace]
members = ["aimeio", "chuniio", "common"]
resolver = "2"
[profile.release]
strip = true # Automatically strip symbols from the binary.
@ -13,18 +9,9 @@ lto = true
codegen-units = 1
panic = "abort"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[features]
chusan = []
[dependencies]
[workspace.dependencies]
anyhow = "1.0.76"
env_logger = { version = "0.10.1", default-features = false }
lazy_static = "1.4.0"
log = "0.4.20"
rusb = "0.9.3"
shared_memory = { path = "vendor/shared_memory" }
winapi = { version = "0.3.9", features = ["minwindef", "winnt", "winerror", "debugapi", "winbase", "winuser"] }
[build-dependencies]
cfg_aliases = "0.2.0"
winapi = "0.3.9"

23
aimeio/Cargo.toml Normal file
View File

@ -0,0 +1,23 @@
[package]
name = "aimeio-yubideck"
version = "0.1.0"
edition = "2021"
[lib]
crate-type = ["cdylib"]
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[features]
chusan = []
[dependencies]
faster-hex = "0.9.0"
lazy_static = { workspace = true }
log = { workspace = true }
rand = "0.8.5"
shared_memory = { workspace = true }
winapi = { workspace = true, features = ["winbase", "winerror", "winnt", "winuser"] }
yubideck-common = { path = "../common" }
[build-dependencies]
cfg_aliases = "0.2.0"

10
aimeio/src/configuration.rs Normal file
View File

@ -0,0 +1,10 @@
use std::path::PathBuf;
#[derive(Debug, Clone)]
pub struct Configuration {
pub aime_path: PathBuf,
pub felica_path: PathBuf,
pub aime_gen: bool,
pub felica_gen: bool,
pub vk_scan: u32,
}

406
aimeio/src/lib.rs Normal file
View File

@ -0,0 +1,406 @@
#![allow(clippy::missing_safety_doc)]
mod configuration;
use std::{
ffi::{c_char, CStr, CString},
path::PathBuf,
rc::Rc,
sync::{
atomic::{AtomicBool, Ordering},
Mutex,
},
};
use configuration::Configuration;
use log::error;
use shared_memory::Shmem;
use winapi::{
shared::{
minwindef::{BOOL, DWORD, HINSTANCE, LPVOID, MAX_PATH, TRUE},
winerror::{E_FAIL, S_FALSE, S_OK},
},
um::{
winbase::{GetPrivateProfileIntA, GetPrivateProfileStringA},
winnt::{DLL_PROCESS_ATTACH, HRESULT},
winuser::{GetAsyncKeyState, VK_RETURN},
},
};
use yubideck_common::{create_shared_memory, init_logger, INPUT_SHMEM_SIZE, OUTPUT_SHMEM_SIZE};
static mut INPUT_SHMEM: Option<Rc<Shmem>> = None;
static mut OUTPUT_SHMEM: Option<Rc<Mutex<Shmem>>> = None;
static AIME_ID_PRESENT: AtomicBool = AtomicBool::new(false);
static FELICA_ID_PRESENT: AtomicBool = AtomicBool::new(false);
static mut REMOTE_CARD_ID: [u8; 10] = [0u8; 10];
lazy_static::lazy_static! {
static ref CONFIGURATION: Configuration = {
let aime = CString::new("aime").unwrap();
let aime_path = CString::new("aimePath").unwrap();
let aime_path_default = CString::new("DEVICE\\aime.txt").unwrap();
let felica_path = CString::new("felicaPath").unwrap();
let felica_path_default = CString::new("DEVICE\\felica.txt").unwrap();
let felica_gen = CString::new("felicaGen").unwrap();
let aime_gen = CString::new("aimeGen").unwrap();
let scan = CString::new("scan").unwrap();
let cfg_file = CString::new(".\\segatools.ini").unwrap();
let mut aime_path_buf: [c_char; MAX_PATH] = [0; MAX_PATH];
let mut felica_path_buf: [c_char; MAX_PATH] = [0; MAX_PATH];
unsafe {
GetPrivateProfileStringA(
aime.as_ptr(),
aime_path.as_ptr(),
aime_path_default.as_ptr(),
aime_path_buf.as_mut_ptr(),
MAX_PATH as u32,
cfg_file.as_ptr()
);
GetPrivateProfileStringA(
aime.as_ptr(),
felica_path.as_ptr(),
felica_path_default.as_ptr(),
felica_path_buf.as_mut_ptr(),
MAX_PATH as u32,
cfg_file.as_ptr()
);
Configuration {
aime_path: CStr::from_ptr(aime_path_buf.as_ptr()).to_str().unwrap().into(),
felica_path: CStr::from_ptr(felica_path_buf.as_ptr()).to_str().unwrap().into(),
aime_gen: GetPrivateProfileIntA(
aime.as_ptr(),
aime_gen.as_ptr(),
1,
cfg_file.as_ptr()
) == 1,
felica_gen: GetPrivateProfileIntA(
aime.as_ptr(),
felica_gen.as_ptr(),
1,
cfg_file.as_ptr()
) == 1,
vk_scan: GetPrivateProfileIntA(
aime.as_ptr(),
scan.as_ptr(),
VK_RETURN,
cfg_file.as_ptr()
),
}
}
};
}
#[no_mangle]
extern "system" fn DllMain(_dll_module: HINSTANCE, call_reason: DWORD, _reserved: LPVOID) -> BOOL {
if call_reason == DLL_PROCESS_ATTACH {
init_logger()
}
TRUE
}
/// Get the version of the Aime IO API that this DLL supports. This function
/// should return a positive 16-bit integer, where the high byte is the major
/// version and the low byte is the minor version (as defined by the Semantic
/// Versioning standard).
/// The latest API version as of this writing is 0x0100.
#[no_mangle]
pub extern "C" fn aime_io_get_api_version() -> u16 {
0x0100
}
/// Initialize Aime IO provider DLL. Only called once, before any other
/// functions exported from this DLL are called (except for
/// aime_io_get_api_version).
///
/// Minimum API version: 0x0100
#[no_mangle]
pub extern "C" fn aime_io_init() -> HRESULT {
match create_shared_memory("YubideckInput", INPUT_SHMEM_SIZE, false) {
Ok(s) => unsafe { INPUT_SHMEM = Some(Rc::new(s)) },
Err(e) => {
error!("Could not acquire shared input memory: {e:#?}");
return E_FAIL;
}
}
match create_shared_memory("YubideckOutput", OUTPUT_SHMEM_SIZE, false) {
Ok(s) => {
unsafe { OUTPUT_SHMEM = Some(Rc::new(Mutex::new(s))) };
S_OK
}
Err(e) => {
error!("Could not obtain shared output memory: {e:#?}");
E_FAIL
}
}
}
fn aime_io_read_id_file<const N: usize>(path: &PathBuf) -> Option<[u8; N]> {
let min_length = N * 2;
let id_string = match std::fs::read_to_string(path) {
Ok(s) => s,
Err(_) => return None,
};
if id_string.len() < min_length {
return None;
}
let mut out = [0u8; N];
if faster_hex::hex_decode(id_string[0..min_length].as_bytes(), &mut out).is_err() {
return None;
}
Some(out)
}
/// # IMPORTANT
/// This is not how Aime access codes are created! This is literally
/// just 10 random bytes! Do not use this for anything serious!
fn aime_io_generate_aime(path: &PathBuf) -> Option<[u8; 10]> {
let mut access_code = [0u8; 10];
loop {
for nib in access_code.iter_mut() {
*nib = ((rand::random::<u8>() % 10) << 4) | (rand::random::<u8>() % 10);
}
if access_code[0] >> 4 != 3 {
break;
}
}
let mut encoded_ac = [0u8; 20];
if let Err(e) = faster_hex::hex_encode(&access_code, &mut encoded_ac) {
error!("Encoding access code failed: {e:#?}");
return None;
}
if let Err(e) = std::fs::write(path, encoded_ac) {
error!("Could not write generated access code to file: {e:#?}");
return None;
}
Some(access_code)
}
/// # IMPORTANT
/// This is not how FeliCa IDms are created! This is literally
/// just 8 random bytes! Do not use this for anything serious!
fn aime_io_generate_felica(path: &PathBuf) -> Option<[u8; 8]> {
let mut idm = [0u8; 8];
for nib in idm.iter_mut() {
*nib = rand::random::<u8>();
}
// FeliCa IDm values should have a 0 in their high nibble. I think.
idm[0] &= 0x0F;
let mut encoded_idm = [0u8; 16];
if let Err(e) = faster_hex::hex_encode(&idm, &mut encoded_idm) {
error!("Encoding Felica IDm failed: {e:#?}");
return None;
}
if let Err(e) = std::fs::write(path, encoded_idm) {
error!("Could not write generated FeliCa IDm to file: {e:#?}");
return None;
}
Some(idm)
}
/// Poll for IC cards in the vicinity.
///
/// - unit_no: Always 0 as of the current API version
///
/// Minimum API version: 0x0100
#[no_mangle]
pub extern "C" fn aime_io_nfc_poll(unit_no: u8) -> HRESULT {
if unit_no != 0 {
return S_OK;
}
let Some(input_shmem) = (unsafe { &INPUT_SHMEM }) else {
return S_OK;
};
let input = unsafe { input_shmem.as_slice() };
AIME_ID_PRESENT.store(false, Ordering::Relaxed);
FELICA_ID_PRESENT.store(false, Ordering::Relaxed);
if input[34] != 0 {
match input[34] {
1 => AIME_ID_PRESENT.store(true, Ordering::Relaxed),
2 => FELICA_ID_PRESENT.store(true, Ordering::Relaxed),
_ => {}
}
unsafe {
REMOTE_CARD_ID.copy_from_slice(&input[35..45]);
}
return S_OK;
}
if unsafe { GetAsyncKeyState(CONFIGURATION.vk_scan as i32) } == 0 {
return S_OK;
}
if let Some(aime_id) = aime_io_read_id_file::<10>(&CONFIGURATION.aime_path) {
AIME_ID_PRESENT.store(true, Ordering::Relaxed);
unsafe { REMOTE_CARD_ID.copy_from_slice(&aime_id) }
return S_OK;
}
if CONFIGURATION.aime_gen {
if let Some(aime_id) = aime_io_generate_aime(&CONFIGURATION.aime_path) {
AIME_ID_PRESENT.store(true, Ordering::Relaxed);
unsafe { REMOTE_CARD_ID.copy_from_slice(&aime_id) }
return S_OK;
}
return E_FAIL;
}
if let Some(felica_id) = aime_io_read_id_file::<8>(&CONFIGURATION.felica_path) {
FELICA_ID_PRESENT.store(true, Ordering::Relaxed);
unsafe {
REMOTE_CARD_ID[..8].copy_from_slice(&felica_id);
}
return S_OK;
}
if CONFIGURATION.felica_gen {
if let Some(felica_id) = aime_io_generate_felica(&CONFIGURATION.felica_path) {
FELICA_ID_PRESENT.store(true, Ordering::Relaxed);
unsafe { REMOTE_CARD_ID[..8].copy_from_slice(&felica_id) }
return S_OK;
}
return E_FAIL;
}
S_OK
}
/// Attempt to read out a classic Aime card ID
///
/// - unit_no: Always 0 as of the current API version
/// - luid: Pointer to a ten-byte buffer that will receive the ID
/// - luid_size: Size of the buffer at *luid. Always 10.
///
/// Returns:
///
/// - S_OK if a classic Aime is present and was read successfully
/// - S_FALSE if no classic Aime card is present (*luid will be ignored)
/// - Any HRESULT error if an error occured.
///
/// Minimum API version: 0x0100
#[no_mangle]
pub unsafe extern "C" fn aime_io_nfc_get_aime_id(
unit_no: u8,
access_code: *mut u8,
access_code_size: usize,
) -> HRESULT {
if access_code.is_null() || access_code_size != 10 {
return S_FALSE;
}
if unit_no != 0 || !AIME_ID_PRESENT.load(Ordering::Relaxed) {
return S_FALSE;
}
let ac_slice = std::slice::from_raw_parts_mut(access_code, access_code_size);
ac_slice.copy_from_slice(&REMOTE_CARD_ID);
S_OK
}
/// Attempt to read out a FeliCa card ID ("IDm"). The following are examples
/// of FeliCa cards:
///
/// - Amuse IC (which includes new-style Aime-branded cards, among others)
/// - Smartphones with FeliCa NFC capability (uncommon outside Japan)
/// - Various Japanese e-cash cards and train passes
///
/// Parameters:
///
/// - unit_no: Always 0 as of the current API version
/// - IDm: Output parameter that will receive the card ID
///
/// Returns:
///
/// - S_OK if a FeliCa device is present and was read successfully
/// - S_FALSE if no FeliCa device is present (*IDm will be ignored)
/// - Any HRESULT error if an error occured.
///
/// Minimum API version: 0x0100
#[no_mangle]
pub unsafe extern "C" fn aime_io_nfc_get_felica_id(unit_no: u8, idm: *mut u64) -> HRESULT {
if idm.is_null() {
return S_FALSE;
}
if unit_no != 0 || !FELICA_ID_PRESENT.load(Ordering::Relaxed) {
return S_FALSE;
}
*idm = 0;
for nib in REMOTE_CARD_ID.iter().take(8) {
*idm = (*idm << 8) | (*nib as u64);
}
S_OK
}
/// Change the color and brightness of the card reader's RGB lighting
///
/// - unit_no: Always 0 as of the current API version
/// - r, g, b: Primary color intensity, from 0 to 255 inclusive.
///
/// Minimum API version: 0x0100
#[no_mangle]
pub extern "C" fn aime_io_led_set_color(unit_no: u8, r: u8, g: u8, b: u8) {
if unit_no != 0 {
return;
}
let Some(output_shmem) = (unsafe { &OUTPUT_SHMEM }) else {
return;
};
let Ok(mut output_shmem) = output_shmem.lock() else {
error!("Could not acquire mutex of output shared memory");
return;
};
let buf = unsafe { output_shmem.as_slice_mut() };
buf[101] = r;
buf[102] = g;
buf[103] = b;
buf[122] = 1;
}

224
chuniio/Cargo.lock generated Normal file
View File

@ -0,0 +1,224 @@
# This file is automatically @generated by Cargo.
# It is not intended for manual editing.
version = 3
[[package]]
name = "anyhow"
version = "1.0.76"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "59d2a3357dde987206219e78ecfbbb6e8dad06cbb65292758d3270e6254f7355"
[[package]]
name = "autocfg"
version = "1.1.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d468802bab17cbc0cc575e9b053f41e72aa36bfa6b7f55e3529ffa43161b97fa"
[[package]]
name = "bitflags"
version = "1.3.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
[[package]]
name = "cc"
version = "1.0.83"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f1174fb0b6ec23863f8b971027804a42614e347eafb0a95bf0b12cdae21fc4d0"
dependencies = [
"libc",
]
[[package]]
name = "cfg-if"
version = "1.0.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "baf1de4339761588bc0619e3cbc0120ee582ebb74b53b4efbf79117bd2da40fd"
[[package]]
name = "cfg_aliases"
version = "0.2.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "77e53693616d3075149f4ead59bdeecd204ac6b8192d8969757601b74bddf00f"
[[package]]
name = "chuniio-yubideck"
version = "0.1.0"
dependencies = [
"anyhow",
"cfg-if",
"cfg_aliases",
"env_logger",
"lazy_static",
"log",
"rusb",
"shared_memory",
"winapi",
]
[[package]]
name = "env_logger"
version = "0.10.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "95b3f3e67048839cb0d0781f445682a35113da7121f7c949db0e2be96a4fbece"
dependencies = [
"log",
]
[[package]]
name = "lazy_static"
version = "1.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "e2abad23fbc42b3700f2f279844dc832adb2b2eb069b2df918f455c4e18cc646"
[[package]]
name = "libc"
version = "0.2.151"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "302d7ab3130588088d277783b1e2d2e10c9e9e4a16dd9050e6ec93fb3e7048f4"
[[package]]
name = "libusb1-sys"
version = "0.6.4"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "f9d0e2afce4245f2c9a418511e5af8718bcaf2fa408aefb259504d1a9cb25f27"
dependencies = [
"cc",
"libc",
"pkg-config",
"vcpkg",
]
[[package]]
name = "log"
version = "0.4.20"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "b5e6163cb8c49088c2c36f57875e58ccd8c87c7427f7fbd50ea6710b2f3f2e8f"
[[package]]
name = "memoffset"
version = "0.6.5"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5aa361d4faea93603064a027415f07bd8e1d5c88c9fbf68bf56a285428fd79ce"
dependencies = [
"autocfg",
]
[[package]]
name = "nix"
version = "0.23.2"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8f3790c00a0150112de0f4cd161e3d7fc4b2d8a5542ffc35f099a2562aecb35c"
dependencies = [
"bitflags",
"cc",
"cfg-if",
"libc",
"memoffset",
]
[[package]]
name = "pkg-config"
version = "0.3.28"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "69d3587f8a9e599cc7ec2c00e331f71c4e69a5f9a4b8a6efd5b07466b9736f9a"
[[package]]
name = "rusb"
version = "0.9.3"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "45fff149b6033f25e825cbb7b2c625a11ee8e6dac09264d49beb125e39aa97bf"
dependencies = [
"libc",
"libusb1-sys",
]
[[package]]
name = "shared_memory"
version = "0.12.4-chuniio"
dependencies = [
"cfg-if",
"libc",
"nix",
"win-sys",
]
[[package]]
name = "vcpkg"
version = "0.2.15"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "accd4ea62f7bb7a82fe23066fb0957d48ef677f6eeb8215f372f52e48bb32426"
[[package]]
name = "win-sys"
version = "0.3.1"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5b7b128a98c1cfa201b09eb49ba285887deb3cbe7466a98850eb1adabb452be5"
dependencies = [
"windows",
]
[[package]]
name = "winapi"
version = "0.3.9"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
dependencies = [
"winapi-i686-pc-windows-gnu",
"winapi-x86_64-pc-windows-gnu",
]
[[package]]
name = "winapi-i686-pc-windows-gnu"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
[[package]]
name = "winapi-x86_64-pc-windows-gnu"
version = "0.4.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
[[package]]
name = "windows"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "45296b64204227616fdbf2614cefa4c236b98ee64dfaaaa435207ed99fe7829f"
dependencies = [
"windows_aarch64_msvc",
"windows_i686_gnu",
"windows_i686_msvc",
"windows_x86_64_gnu",
"windows_x86_64_msvc",
]
[[package]]
name = "windows_aarch64_msvc"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "17cffbe740121affb56fad0fc0e421804adf0ae00891205213b5cecd30db881d"
[[package]]
name = "windows_i686_gnu"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "2564fde759adb79129d9b4f54be42b32c89970c18ebf93124ca8870a498688ed"
[[package]]
name = "windows_i686_msvc"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "9cd9d32ba70453522332c14d38814bceeb747d80b3958676007acadd7e166956"
[[package]]
name = "windows_x86_64_gnu"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "cfce6deae227ee8d356d19effc141a509cc503dfd1f850622ec4b0f84428e1f4"
[[package]]
name = "windows_x86_64_msvc"
version = "0.34.0"
source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "d19538ccc21819d01deaf88d6a17eae6596a12e9aafdbb97916fb49896d89de9"

24
chuniio/Cargo.toml Normal file
View File

@ -0,0 +1,24 @@
[package]
name = "chuniio-yubideck"
version = "0.1.0"
edition = "2021"
[lib]
crate-type = ["cdylib"]
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[features]
chusan = []
[dependencies]
anyhow = { workspace = true }
cfg-if = "1.0.0"
lazy_static = { workspace = true }
log = { workspace = true }
rusb = "0.9.3"
shared_memory = { workspace = true }
yubideck-common = { path = "../common" }
winapi = { workspace = true, features = ["minwindef", "winnt", "winerror", "winbase", "winuser"] }
[build-dependencies]
cfg_aliases = "0.2.0"

2
build.rs → chuniio/build.rs
View File

@ -6,4 +6,4 @@ fn main() {
amdaemon: { all(feature = "chusan", target_pointer_width = "64") },
chusanapp: { all(feature = "chusan", target_pointer_width = "32") },
}
}
}

12
src/configuration.rs → chuniio/src/configuration.rs
View File

@ -1,6 +1,6 @@
#[derive(Debug, Clone, Copy)]
pub struct Configuration {
pub test_key: u32,
pub service_key: u32,
pub coin_key: u32,
}
#[derive(Debug, Clone, Copy)]
pub struct Configuration {
pub test_key: u32,
pub service_key: u32,
pub coin_key: u32,
}

320
src/lib.rs → chuniio/src/lib.rs
View File

@ -1,109 +1,103 @@
mod configuration;
mod log;
#![allow(clippy::missing_safety_doc)]
mod configuration;
#[cfg(any(chuni, amdaemon))]
use std::{ffi::c_int, sync::atomic::AtomicU16};
use std::{
ffi::{c_void, CString},
fmt::Display,
ffi::c_void,
rc::Rc,
sync::{
atomic::{AtomicBool, Ordering},
Arc,
Mutex,
},
};
#[cfg(any(chuni, chusanapp))]
use std::{
sync::{Mutex, OnceLock},
thread::{self, JoinHandle},
time::Duration,
};
use ::log::error;
use anyhow::{anyhow, Result};
use lazy_static::lazy_static;
use shared_memory::{Shmem, ShmemConf, ShmemError};
#[cfg(any(chuni, amdaemon))]
use winapi::um::winuser::GetAsyncKeyState;
use shared_memory::Shmem;
use winapi::{
shared::{
minwindef::{BOOL, DWORD, HINSTANCE, LPVOID, TRUE},
winerror::{E_FAIL, S_OK},
},
um::{
winbase::GetPrivateProfileIntA,
winnt::{DLL_PROCESS_ATTACH, HRESULT},
},
};
#[cfg(any(chuni, chusanapp))]
use {
::log::info,
rusb::{DeviceHandle, GlobalContext},
um::winnt::{DLL_PROCESS_ATTACH, HRESULT},
};
use yubideck_common::{create_shared_memory, init_logger, INPUT_SHMEM_SIZE, OUTPUT_SHMEM_SIZE};
use crate::{configuration::Configuration, log::init_logger};
static mut INPUT_SHMEM: Option<Rc<Shmem>> = None;
static mut OUTPUT_SHMEM: Option<Rc<Mutex<Shmem>>> = None;
static mut INPUT_SHMEM: Option<Arc<Shmem>> = None;
cfg_if::cfg_if! {
if #[cfg(any(chuni, chusanapp))] {
use std::{
sync::OnceLock,
thread::{self, JoinHandle},
time::Duration,
};
#[cfg(any(chuni, chusanapp))]
type SliderCallbackFn = unsafe extern "C" fn(data: *const u8);
use ::log::info;
use rusb::{DeviceHandle, GlobalContext};
#[cfg(any(chuni, chusanapp))]
static DEVICE: OnceLock<DeviceHandle<GlobalContext>> = OnceLock::new();
type SliderCallbackFn = unsafe extern "C" fn(data: *const u8);
#[cfg(any(chuni, chusanapp))]
static mut SLIDER_THREAD: OnceLock<JoinHandle<()>> = OnceLock::new();
const TIMEOUT: Duration = Duration::from_millis(20);
#[cfg(any(chuni, chusanapp))]
static SLIDER_ACTIVE: AtomicBool = AtomicBool::new(false);
static DEVICE: OnceLock<DeviceHandle<GlobalContext>> = OnceLock::new();
static mut SLIDER_THREAD: OnceLock<JoinHandle<()>> = OnceLock::new();
static SLIDER_ACTIVE: AtomicBool = AtomicBool::new(false);
}
}
#[cfg(any(chuni, amdaemon))]
static COIN_COUNT: AtomicU16 = AtomicU16::new(0);
cfg_if::cfg_if! {
if #[cfg(any(chuni, amdaemon))] {
use std::{ffi::{c_int, CString}, sync::atomic::AtomicU16};
#[cfg(any(chuni, amdaemon))]
static COIN_PRESSED: AtomicBool = AtomicBool::new(false);
use lazy_static::lazy_static;
use winapi::um::{winuser::GetAsyncKeyState, winbase::GetPrivateProfileIntA};
#[cfg(any(chuni, chusanapp))]
static mut OUTPUT_SHMEM: Option<Arc<Mutex<Shmem>>> = None;
use crate::configuration::Configuration;
#[cfg(any(chuni, amdaemon))]
lazy_static! {
static ref CONFIGURATION: Configuration = {
let io3 = CString::new("io3").unwrap();
let test = CString::new("test").unwrap();
let service = CString::new("service").unwrap();
let coin = CString::new("coin").unwrap();
let cfg_file = CString::new(".\\segatools.ini").unwrap();
static COIN_COUNT: AtomicU16 = AtomicU16::new(0);
static COIN_PRESSED: AtomicBool = AtomicBool::new(false);
unsafe {
Configuration {
test_key: GetPrivateProfileIntA(
io3.as_ptr(),
test.as_ptr(),
0x31,
cfg_file.as_ptr(),
),
service_key: GetPrivateProfileIntA(
io3.as_ptr(),
service.as_ptr(),
0x32,
cfg_file.as_ptr(),
),
coin_key: GetPrivateProfileIntA(
io3.as_ptr(),
coin.as_ptr(),
0x33,
cfg_file.as_ptr(),
),
}
lazy_static! {
static ref CONFIGURATION: Configuration = {
let io3 = CString::new("io3").unwrap();
let test = CString::new("test").unwrap();
let service = CString::new("service").unwrap();
let coin = CString::new("coin").unwrap();
let cfg_file = CString::new(".\\segatools.ini").unwrap();
unsafe {
Configuration {
test_key: GetPrivateProfileIntA(
io3.as_ptr(),
test.as_ptr(),
0x31,
cfg_file.as_ptr(),
),
service_key: GetPrivateProfileIntA(
io3.as_ptr(),
service.as_ptr(),
0x32,
cfg_file.as_ptr(),
),
coin_key: GetPrivateProfileIntA(
io3.as_ptr(),
coin.as_ptr(),
0x33,
cfg_file.as_ptr(),
),
}
}
};
}
};
}
}
#[no_mangle]
extern "system" fn DllMain(_dll_module: HINSTANCE, call_reason: DWORD, _reserved: LPVOID) -> BOOL {
match call_reason {
DLL_PROCESS_ATTACH => init_logger(),
_ => {}
if call_reason == DLL_PROCESS_ATTACH {
init_logger()
}
TRUE
@ -115,9 +109,24 @@ pub extern "C" fn chuni_io_get_api_version() -> u16 {
}
#[no_mangle]
#[cfg(any(chuni, amdaemon))]
pub extern "C" fn chuni_io_jvs_init() -> HRESULT {
#[cfg(not(feature = "chusan"))]
match create_shared_memory("YubideckInput", INPUT_SHMEM_SIZE, true) {
Ok(s) => unsafe { INPUT_SHMEM = Some(Rc::new(s)) },
Err(e) => {
error!("Could not acquire shared memory for YubiDeck input: {e:#?}");
return E_FAIL;
}
}
match create_shared_memory("YubideckOutput", OUTPUT_SHMEM_SIZE, true) {
Ok(s) => unsafe { OUTPUT_SHMEM = Some(Rc::new(Mutex::new(s))) },
Err(e) => {
error!("Could not obtain shared memory for YubiDeck output: {e:#?}");
return E_FAIL;
}
}
#[cfg(not(amdaemon))]
{
if let Err(e) = yubideck_init() {
error!("Could not initialize YubiDeck: {e:#?}");
@ -125,18 +134,12 @@ pub extern "C" fn chuni_io_jvs_init() -> HRESULT {
}
}
return create_input_shared_memory();
}
#[no_mangle]
#[cfg(chusanapp)]
pub extern "C" fn chuni_io_jvs_init() -> HRESULT {
S_OK
}
#[no_mangle]
#[cfg(any(chuni, amdaemon))]
pub extern "C" fn chuni_io_jvs_poll(opbtn: *mut u8, beams: *mut u8) {
pub unsafe extern "C" fn chuni_io_jvs_poll(opbtn: *mut u8, beams: *mut u8) {
if opbtn.is_null() || beams.is_null() {
return;
}
@ -148,18 +151,16 @@ pub extern "C" fn chuni_io_jvs_poll(opbtn: *mut u8, beams: *mut u8) {
let ir_value = input[0];
let mut buttons = input[1] & 3; // Buttons are in order: coin, service, test. We take the last 2 bits
unsafe {
if GetAsyncKeyState(CONFIGURATION.test_key as c_int) != 0 {
buttons |= 1;
}
if GetAsyncKeyState(CONFIGURATION.service_key as c_int) != 0 {
buttons |= 2;
}
*opbtn = buttons;
*beams = ir_value;
if GetAsyncKeyState(CONFIGURATION.test_key as c_int) != 0 {
buttons |= 1;
}
if GetAsyncKeyState(CONFIGURATION.service_key as c_int) != 0 {
buttons |= 2;
}
*opbtn = buttons;
*beams = ir_value;
}
#[no_mangle]
@ -168,7 +169,7 @@ pub extern "C" fn chuni_io_jvs_poll(_opbtn: *mut u8, _beams: *mut u8) {}
#[no_mangle]
#[cfg(any(chuni, amdaemon))]
pub extern "C" fn chuni_io_jvs_read_coin_counter(total: *mut u16) {
pub unsafe extern "C" fn chuni_io_jvs_read_coin_counter(total: *mut u16) {
if total.is_null() {
return;
}
@ -188,9 +189,7 @@ pub extern "C" fn chuni_io_jvs_read_coin_counter(total: *mut u16) {
COIN_PRESSED.store(false, Ordering::Relaxed);
}
unsafe {
*total = COIN_COUNT.load(Ordering::Relaxed);
}
*total = COIN_COUNT.load(Ordering::Relaxed);
}
#[no_mangle]
@ -198,34 +197,25 @@ pub extern "C" fn chuni_io_jvs_read_coin_counter(total: *mut u16) {
pub extern "C" fn chuni_io_jvs_read_coin_counter(_total: *mut u16) {}
#[no_mangle]
#[allow(unreachable_code)]
#[cfg(any(chuni, chusanapp))]
pub unsafe extern "C" fn chuni_io_slider_init() -> HRESULT {
match create_shared_memory("Local\\YubideckOutput", 122, true) {
Ok(s) => OUTPUT_SHMEM = Some(Arc::new(Mutex::new(s))),
Err(e) => {
error!("Could not obtain shared memory: {e:#?}");
return E_FAIL;
}
}
#[cfg(not(feature = "chusan"))]
pub extern "C" fn chuni_io_slider_init() -> HRESULT {
#[cfg(any(chuni, chusanapp))]
{
// Already initialized in chuni_io_jvs_init()
return S_OK;
let Some(out_shmem) = (unsafe { &OUTPUT_SHMEM }) else {
error!("OUTPUT_SHMEM is unset.");
return E_FAIL;
};
let Ok(mut out_shmem) = out_shmem.lock() else {
error!("Could not acquire mutex of output shared memory");
return E_FAIL;
};
let buf = unsafe { out_shmem.as_slice_mut() };
buf[0] = 0;
buf[61] = 1;
}
if let Err(e) = yubideck_init() {
error!("Failed to initialize YubiDeck: {e:#?}");
return E_FAIL;
}
return create_input_shared_memory();
}
#[no_mangle]
#[cfg(amdaemon)]
pub extern "C" fn chuni_io_slider_init(_callback: *const c_void) -> HRESULT {
S_OK
}
@ -251,10 +241,8 @@ pub unsafe extern "C" fn chuni_io_slider_start(callback: *const c_void) {
let mut pressure = [0u8; 32];
while SLIDER_ACTIVE.load(Ordering::Relaxed) {
pressure.copy_from_slice(&usb_in[2..34]);
for i in 0..16 {
pressure.swap(i * 2, i * 2 + 1);
for (i, p) in usb_in.iter().skip(2).take(32).enumerate() {
pressure[if i % 2 == 0 { 30 - i } else { 32 - i }] = *p
}
callback(pressure.as_ptr());
@ -304,9 +292,6 @@ pub unsafe extern "C" fn chuni_io_slider_set_leds(rgb: *const u8) {
let buf = out_shmem.as_slice_mut();
let ground = std::slice::from_raw_parts(rgb, 93);
buf[0] = 0;
buf[61] = 1;
for (buf_chunk, state_chunk) in buf[1..61]
.chunks_mut(3)
.zip(ground.chunks(3).skip(11).take(20).rev())
@ -325,11 +310,11 @@ pub unsafe extern "C" fn chuni_io_slider_set_leds(rgb: *const u8) {
buf_chunk[2] = state_chunk[2];
}
if let Err(e) = device.write_interrupt(0x02, &buf[0..61], Duration::from_millis(20)) {
if let Err(e) = device.write_interrupt(0x02, &buf[0..61], TIMEOUT) {
error!("Error writing first batch of output data: {e:#?}");
}
if let Err(e) = device.write_interrupt(0x02, &buf[61..], Duration::from_millis(20)) {
if let Err(e) = device.write_interrupt(0x02, &buf[61..122], TIMEOUT) {
error!("Error writing second batch of output data: {e:#?}");
}
}
@ -358,8 +343,6 @@ pub unsafe extern "C" fn chuni_io_led_set_colors(board: u8, rgb: *const u8) {
let buf = out_shmem.as_slice_mut();
let data = std::slice::from_raw_parts(rgb, 183);
buf[61] = 1;
match board {
0 => {
// left air
@ -376,7 +359,7 @@ pub unsafe extern "C" fn chuni_io_led_set_colors(board: u8, rgb: *const u8) {
_ => {}
}
if let Err(e) = device.write_interrupt(0x02, &buf[61..], Duration::from_millis(20)) {
if let Err(e) = device.write_interrupt(0x02, &buf[61..122], TIMEOUT) {
error!("Error writing second batch of output data: {e:#?}");
}
}
@ -385,42 +368,6 @@ pub unsafe extern "C" fn chuni_io_led_set_colors(board: u8, rgb: *const u8) {
#[cfg(amdaemon)]
pub extern "C" fn chuni_io_led_set_colors(_rgb: *const u8) {}
fn create_input_shared_memory() -> HRESULT {
match create_shared_memory("Local\\YubideckInput", 45, false) {
Ok(s) => {
unsafe { INPUT_SHMEM = Some(Arc::new(s)) };
S_OK
}
Err(e) => {
error!("Could not acquire shared memory: {e:#?}");
E_FAIL
}
}
}
fn create_shared_memory<S>(os_id: S, size: usize, is_owner: bool) -> Result<Shmem>
where
S: AsRef<str> + Display + Copy,
{
let shmem_conf = ShmemConf::new().size(size).os_id(os_id);
let shmem_result = match shmem_conf.clone().create() {
Ok(s) => Ok(s),
Err(ShmemError::MappingIdExists) => shmem_conf.open(),
Err(e) => {
return Err(anyhow!(
"Failed to create/open shared memory {os_id}: {e:#}"
));
}
};
shmem_result
.map(|mut m| {
m.set_owner(is_owner);
m
})
.map_err(|e| anyhow!("Failed to create/open shared memory {os_id}: {e:#}"))
}
#[cfg(any(chuni, chusanapp))]
fn yubideck_init() -> Result<()> {
let Some(mut device) = rusb::open_device_with_vid_pid(0x1973, 0x2001) else {
@ -439,22 +386,41 @@ fn yubideck_init() -> Result<()> {
}
#[cfg(any(chuni, chusanapp))]
fn input_thread_proc() {
fn input_thread_proc() {
info!("Input thread started");
let mut shmem = match create_shared_memory("Local\\YubideckInput", 45, true) {
let device = DEVICE.get().unwrap();
let mut input_shmem = match create_shared_memory("YubideckInput", INPUT_SHMEM_SIZE, true) {
Ok(s) => s,
Err(e) => {
error!("Could not obtain shared memory for YubiDeck input: {e:#}");
return;
}
};
let usb_in = unsafe { shmem.as_slice_mut() };
let device = DEVICE.get().unwrap();
let usb_in = unsafe { input_shmem.as_slice_mut() };
let mut output_shmem = match create_shared_memory("YubideckOutput", OUTPUT_SHMEM_SIZE, true) {
Ok(s) => s,
Err(e) => {
error!("Could not obtain shared memory for YubiDeck output: {e:#?}");
return;
}
};
let usb_out = unsafe { output_shmem.as_slice_mut() };
loop {
if let Err(e) = device.read_interrupt(0x81, usb_in, Duration::from_millis(20)) {
if let Err(e) = device.read_interrupt(0x81, usb_in, TIMEOUT) {
error!("Failed to read data from YubiDeck: {e:#}");
}
// Update reader LED data
if usb_out[122] == 1 {
usb_out[122] = 0;
if let Err(e) = device.write_interrupt(0x02, &usb_out[61..122], TIMEOUT) {
error!("Error writing second batch of output data: {e:#?}");
}
}
}
}

11
common/Cargo.toml Normal file
View File

@ -0,0 +1,11 @@
[package]
name = "yubideck-common"
version = "0.1.0"
edition = "2021"
[dependencies]
anyhow = { workspace = true }
env_logger = { version = "0.10.1", default-features = false }
log = { workspace = true }
shared_memory = { workspace = true }
winapi = { workspace = true, features = ["debugapi"] }

61
common/src/lib.rs Normal file
View File

@ -0,0 +1,61 @@
pub mod log;
use std::fmt::Display;
use anyhow::{anyhow, Result};
pub use log::init_logger;
use shared_memory::{Shmem, ShmemConf, ShmemError};
// #[repr(C)]
// struct YubideckInput {
// pub ir_value: u8,
// pub buttons: u8,
// pub touch_data: [u8; 32],
// pub card_status: u8,
// pub card_id: [u8; 10],
// }
pub const INPUT_SHMEM_SIZE: usize = 45;
// #[repr(C)]
// struct YubideckOutputGaosan {
// // Always 0
// pub packet_type: u8,
// pub slider_led_data: [u8; 60],
// }
//
// #[repr(C)]
// struct YubideckOutputDasi {
// // Always 1
// pub packet_type: u8,
// pub slider_led_data: [u8; 33],
// pub left_air_led: [u8; 3],
// pub right_air_led: [u8; 3],
// pub card_reader_led: [u8; 3],
// pub padding: [u8; 18],
// }
//
// There's an extra byte (byte 122) for AimeIO to communicate to ChuniIO
// that reader LED has changed and ChuniIO should update it.
pub const OUTPUT_SHMEM_SIZE: usize = 123;
pub fn create_shared_memory<S>(os_id: S, size: usize, is_owner: bool) -> Result<Shmem>
where
S: AsRef<str> + Display + Copy,
{
let shmem_conf = ShmemConf::new().size(size).os_id(os_id);
let shmem_result = match shmem_conf.clone().create() {
Ok(s) => Ok(s),
Err(ShmemError::MappingIdExists) => shmem_conf.open(),
Err(e) => {
return Err(anyhow!(
"Failed to create/open shared memory {os_id}: {e:#}"
));
}
};
shmem_result
.map(|mut m| {
m.set_owner(is_owner);
m
})
.map_err(|e| anyhow!("Failed to create/open shared memory {os_id}: {e:#}"))
}

0
src/log.rs → common/src/log.rs
View File

6
vendor/shared_memory/examples/mutex.rs vendored
View File

@ -1,5 +1,7 @@
use std::sync::atomic::{AtomicU8, Ordering};
use std::thread;
use std::{
sync::atomic::{AtomicU8, Ordering},
thread,
};
use clap::Parser;
use raw_sync::locks::*;

10
vendor/shared_memory/src/lib.rs vendored
View File

@ -2,11 +2,11 @@
//!
//! For help on how to get started, take a look at the [examples](https://github.com/elast0ny/shared_memory-rs/tree/master/examples) !
use std::fs::{File, OpenOptions};
use std::io::{ErrorKind, Read, Write};
use std::fs::remove_file;
use std::path::{Path, PathBuf};
use std::{
fs::{remove_file, File, OpenOptions},
io::{ErrorKind, Read, Write},
path::{Path, PathBuf},
};
use cfg_if::cfg_if;

18
vendor/shared_memory/src/unix.rs vendored
View File

@ -1,13 +1,15 @@
use std::os::unix::io::RawFd;
use std::ptr::null_mut;
use std::{os::unix::io::RawFd, ptr::null_mut};
use crate::log::*;
use nix::fcntl::OFlag;
use nix::sys::mman::{mmap, munmap, shm_open, shm_unlink, MapFlags, ProtFlags};
use nix::sys::stat::{fstat, Mode};
use nix::unistd::{close, ftruncate};
use nix::{
fcntl::OFlag,
sys::{
mman::{mmap, munmap, shm_open, shm_unlink, MapFlags, ProtFlags},
stat::{fstat, Mode},
},
unistd::{close, ftruncate},
};
use crate::ShmemError;
use crate::{log::*, ShmemError};
#[derive(Clone, Default)]
pub struct ShmemConfExt;

13
vendor/shared_memory/src/windows.rs vendored
View File

@ -1,12 +1,13 @@
use std::fs::{File, OpenOptions};
use std::io::ErrorKind;
use std::os::windows::{fs::OpenOptionsExt, io::AsRawHandle};
use std::path::PathBuf;
use std::{
fs::{File, OpenOptions},
io::ErrorKind,
os::windows::{fs::OpenOptionsExt, io::AsRawHandle},
path::PathBuf,
};
use crate::{log::*, ShmemConf};
use win_sys::*;
use crate::ShmemError;
use crate::{log::*, ShmemConf, ShmemError};
#[derive(Clone, Default)]
pub struct ShmemConfExt {

4
vendor/shared_memory/tests/posix_semantics.rs vendored
View File

@ -1,6 +1,6 @@
use std::{sync::mpsc::channel, thread};
use shared_memory::ShmemConf;
use std::sync::mpsc::channel;
use std::thread;
#[test]
fn persistence() {