Compare commits
2 Commits
feature/ch
...
feature/lo
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
d0e925bf70 | ||
|
|
e519ffa1e6 |
19
Makefile
19
Makefile
@@ -1,4 +1,4 @@
|
|||||||
.PHONY: all build lint fmt fmt-check test check clean run debug docker hooks ensure-web-dist
|
.PHONY: all build lint fmt fmt-check test check clean run debug docker hooks
|
||||||
|
|
||||||
BINARY := neoircd
|
BINARY := neoircd
|
||||||
VERSION := $(shell git describe --tags --always --dirty 2>/dev/null || echo "dev")
|
VERSION := $(shell git describe --tags --always --dirty 2>/dev/null || echo "dev")
|
||||||
@@ -7,21 +7,10 @@ LDFLAGS := -X main.Version=$(VERSION) -X main.Buildarch=$(BUILDARCH)
|
|||||||
|
|
||||||
all: check build
|
all: check build
|
||||||
|
|
||||||
# ensure-web-dist creates placeholder files so //go:embed dist/* in
|
build:
|
||||||
# web/embed.go resolves without a full Node.js build. The real SPA is
|
|
||||||
# built by the web-builder Docker stage; these placeholders let
|
|
||||||
# "make test" and "make build" work outside Docker.
|
|
||||||
ensure-web-dist:
|
|
||||||
@if [ ! -d web/dist ]; then \
|
|
||||||
mkdir -p web/dist && \
|
|
||||||
touch web/dist/index.html web/dist/style.css web/dist/app.js && \
|
|
||||||
echo "==> Created placeholder web/dist/ for go:embed"; \
|
|
||||||
fi
|
|
||||||
|
|
||||||
build: ensure-web-dist
|
|
||||||
go build -ldflags "$(LDFLAGS)" -o bin/$(BINARY) ./cmd/neoircd
|
go build -ldflags "$(LDFLAGS)" -o bin/$(BINARY) ./cmd/neoircd
|
||||||
|
|
||||||
lint: ensure-web-dist
|
lint:
|
||||||
golangci-lint run --config .golangci.yml ./...
|
golangci-lint run --config .golangci.yml ./...
|
||||||
|
|
||||||
fmt:
|
fmt:
|
||||||
@@ -31,7 +20,7 @@ fmt:
|
|||||||
fmt-check:
|
fmt-check:
|
||||||
@test -z "$$(gofmt -l .)" || (echo "Files not formatted:" && gofmt -l . && exit 1)
|
@test -z "$$(gofmt -l .)" || (echo "Files not formatted:" && gofmt -l . && exit 1)
|
||||||
|
|
||||||
test: ensure-web-dist
|
test:
|
||||||
go test -timeout 30s -v -race -cover ./...
|
go test -timeout 30s -v -race -cover ./...
|
||||||
|
|
||||||
# check runs all validation without making changes
|
# check runs all validation without making changes
|
||||||
|
|||||||
100
README.md
100
README.md
@@ -523,7 +523,7 @@ the same JSON envelope:
|
|||||||
| `params` | array of strings | Sometimes | Sometimes | Additional IRC-style positional parameters. Used by commands like `MODE`, `KICK`, and numeric replies like `353` (NAMES). |
|
| `params` | array of strings | Sometimes | Sometimes | Additional IRC-style positional parameters. Used by commands like `MODE`, `KICK`, and numeric replies like `353` (NAMES). |
|
||||||
| `body` | array or object | Usually | Usually | Structured message body. For text messages: array of strings (one per line). For structured data (e.g., `PUBKEY`): JSON object. **Never a raw string.** |
|
| `body` | array or object | Usually | Usually | Structured message body. For text messages: array of strings (one per line). For structured data (e.g., `PUBKEY`): JSON object. **Never a raw string.** |
|
||||||
| `ts` | string (ISO 8601) | Ignored | Always | Server-assigned timestamp in RFC 3339 / ISO 8601 format with nanosecond precision. Example: `"2026-02-10T20:00:00.000000000Z"`. Always UTC. |
|
| `ts` | string (ISO 8601) | Ignored | Always | Server-assigned timestamp in RFC 3339 / ISO 8601 format with nanosecond precision. Example: `"2026-02-10T20:00:00.000000000Z"`. Always UTC. |
|
||||||
| `meta` | object | Optional | If present | Extensible metadata. Used for cryptographic signatures (`meta.sig`, `meta.alg`), hashcash proof-of-work (`meta.hashcash`), content hashes, or any client-defined key/value pairs. Server relays `meta` verbatim except for `hashcash` which is validated on channels with `+H` mode. |
|
| `meta` | object | Optional | If present | Extensible metadata. Used for cryptographic signatures (`meta.sig`, `meta.alg`), content hashes, or any client-defined key/value pairs. Server relays `meta` verbatim — it does not interpret or validate it. |
|
||||||
|
|
||||||
**Important invariants:**
|
**Important invariants:**
|
||||||
|
|
||||||
@@ -1013,13 +1013,12 @@ carries IRC-style parameters (e.g., channel name, target nick).
|
|||||||
Inspired by IRC, simplified:
|
Inspired by IRC, simplified:
|
||||||
|
|
||||||
| Mode | Name | Meaning |
|
| Mode | Name | Meaning |
|
||||||
|------|----------------|---------|
|
|------|--------------|---------|
|
||||||
| `+i` | Invite-only | Only invited users can join |
|
| `+i` | Invite-only | Only invited users can join |
|
||||||
| `+m` | Moderated | Only voiced (`+v`) users and operators (`+o`) can send |
|
| `+m` | Moderated | Only voiced (`+v`) users and operators (`+o`) can send |
|
||||||
| `+s` | Secret | Channel hidden from LIST response |
|
| `+s` | Secret | Channel hidden from LIST response |
|
||||||
| `+t` | Topic lock | Only operators can change the topic |
|
| `+t` | Topic lock | Only operators can change the topic |
|
||||||
| `+n` | No external | Only channel members can send messages to the channel |
|
| `+n` | No external | Only channel members can send messages to the channel |
|
||||||
| `+H` | Hashcash | Requires proof-of-work for PRIVMSG (parameter: bits, e.g. `+H 20`) |
|
|
||||||
|
|
||||||
**User channel modes (set per-user per-channel):**
|
**User channel modes (set per-user per-channel):**
|
||||||
|
|
||||||
@@ -1030,56 +1029,6 @@ Inspired by IRC, simplified:
|
|||||||
|
|
||||||
**Status:** Channel modes are defined but not yet enforced. The `modes` column
|
**Status:** Channel modes are defined but not yet enforced. The `modes` column
|
||||||
exists in the channels table but the server does not check modes on actions.
|
exists in the channels table but the server does not check modes on actions.
|
||||||
Exception: `+H` (hashcash) is fully enforced — see below.
|
|
||||||
|
|
||||||
### Per-Channel Hashcash (Anti-Spam)
|
|
||||||
|
|
||||||
Channels can require hashcash proof-of-work for every `PRIVMSG`. This is an
|
|
||||||
anti-spam mechanism: channel operators set a difficulty level, and clients must
|
|
||||||
compute a proof-of-work stamp bound to the specific channel and message before
|
|
||||||
sending.
|
|
||||||
|
|
||||||
**Setting the requirement:**
|
|
||||||
|
|
||||||
```
|
|
||||||
MODE #channel +H <bits> — require <bits> leading zero bits (1-40)
|
|
||||||
MODE #channel -H — disable hashcash requirement
|
|
||||||
```
|
|
||||||
|
|
||||||
**Stamp format:** `1:bits:YYMMDD:channel:bodyhash:counter`
|
|
||||||
|
|
||||||
- `bits` — difficulty (leading zero bits in SHA-256 hash of the stamp)
|
|
||||||
- `YYMMDD` — current date (prevents old token reuse)
|
|
||||||
- `channel` — channel name (prevents cross-channel reuse)
|
|
||||||
- `bodyhash` — hex-encoded SHA-256 of the message body (binds stamp to message)
|
|
||||||
- `counter` — hex nonce
|
|
||||||
|
|
||||||
**Sending a message to a hashcash-protected channel:**
|
|
||||||
|
|
||||||
Include the hashcash stamp in the `meta` field:
|
|
||||||
|
|
||||||
```json
|
|
||||||
{
|
|
||||||
"command": "PRIVMSG",
|
|
||||||
"to": "#general",
|
|
||||||
"body": ["hello world"],
|
|
||||||
"meta": {
|
|
||||||
"hashcash": "1:20:260317:#general:a1b2c3...bodyhash:1f4a"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
```
|
|
||||||
|
|
||||||
**Server validation:** The server checks that the stamp is well-formed, meets
|
|
||||||
the required difficulty, is bound to the correct channel and message body, has a
|
|
||||||
recent date, and has not been previously used. Spent stamps are cached for 1
|
|
||||||
year to prevent replay attacks.
|
|
||||||
|
|
||||||
**Error responses:** If the channel requires hashcash and the stamp is missing,
|
|
||||||
invalid, or replayed, the server returns `ERR_CANNOTSENDTOCHAN (404)` with a
|
|
||||||
descriptive reason.
|
|
||||||
|
|
||||||
**Client minting:** The CLI provides `MintChannelHashcash(bits, channel, body)`
|
|
||||||
to compute stamps. Higher bit counts take exponentially longer to compute.
|
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
@@ -2102,6 +2051,8 @@ directory is also loaded automatically via
|
|||||||
| `METRICS_USERNAME` | string | `""` | Basic auth username for `/metrics` endpoint. If empty, metrics endpoint is disabled. |
|
| `METRICS_USERNAME` | string | `""` | Basic auth username for `/metrics` endpoint. If empty, metrics endpoint is disabled. |
|
||||||
| `METRICS_PASSWORD` | string | `""` | Basic auth password for `/metrics` endpoint |
|
| `METRICS_PASSWORD` | string | `""` | Basic auth password for `/metrics` endpoint |
|
||||||
| `NEOIRC_HASHCASH_BITS` | int | `20` | Required hashcash proof-of-work difficulty (leading zero bits in SHA-256) for session creation. Set to `0` to disable. |
|
| `NEOIRC_HASHCASH_BITS` | int | `20` | Required hashcash proof-of-work difficulty (leading zero bits in SHA-256) for session creation. Set to `0` to disable. |
|
||||||
|
| `LOGIN_RATE_LIMIT` | float | `1` | Allowed login attempts per second per IP address. |
|
||||||
|
| `LOGIN_RATE_BURST` | int | `5` | Maximum burst of login attempts per IP before rate limiting kicks in. |
|
||||||
| `MAINTENANCE_MODE` | bool | `false` | Maintenance mode flag (reserved) |
|
| `MAINTENANCE_MODE` | bool | `false` | Maintenance mode flag (reserved) |
|
||||||
|
|
||||||
### Example `.env` file
|
### Example `.env` file
|
||||||
@@ -2506,6 +2457,49 @@ creating one session pays once and keeps their session.
|
|||||||
- **Language-agnostic**: SHA-256 is available in every programming language.
|
- **Language-agnostic**: SHA-256 is available in every programming language.
|
||||||
The proof computation is trivially implementable in any client.
|
The proof computation is trivially implementable in any client.
|
||||||
|
|
||||||
|
### Login Rate Limiting
|
||||||
|
|
||||||
|
The login endpoint (`POST /api/v1/login`) has per-IP rate limiting to prevent
|
||||||
|
brute-force password attacks. This uses a token-bucket algorithm
|
||||||
|
(`golang.org/x/time/rate`) with configurable rate and burst.
|
||||||
|
|
||||||
|
| Environment Variable | Default | Description |
|
||||||
|
|---------------------|---------|-------------|
|
||||||
|
| `LOGIN_RATE_LIMIT` | `1` | Allowed login attempts per second per IP |
|
||||||
|
| `LOGIN_RATE_BURST` | `5` | Maximum burst of login attempts per IP |
|
||||||
|
|
||||||
|
When the limit is exceeded, the server returns **429 Too Many Requests** with a
|
||||||
|
`Retry-After: 1` header. Stale per-IP entries are automatically cleaned up
|
||||||
|
every 10 minutes.
|
||||||
|
|
||||||
|
> **⚠️ Security: Reverse Proxy Required for Production Use**
|
||||||
|
>
|
||||||
|
> The rate limiter extracts the client IP by checking the `X-Forwarded-For`
|
||||||
|
> header first, then `X-Real-IP`, and finally falling back to the TCP
|
||||||
|
> `RemoteAddr`. Both `X-Forwarded-For` and `X-Real-IP` are **client-controlled
|
||||||
|
> request headers** — any client can set them to arbitrary values.
|
||||||
|
>
|
||||||
|
> Without a properly configured reverse proxy in front of this server:
|
||||||
|
>
|
||||||
|
> - An attacker can **bypass rate limiting entirely** by rotating
|
||||||
|
> `X-Forwarded-For` values on each request (each value is treated as a
|
||||||
|
> distinct IP).
|
||||||
|
> - An attacker can **deny service to a specific user** by spoofing that user's
|
||||||
|
> IP in the `X-Forwarded-For` header, exhausting their rate limit bucket.
|
||||||
|
>
|
||||||
|
> **Recommendation:** Always deploy behind a reverse proxy (e.g. nginx, Caddy,
|
||||||
|
> Traefik) that strips or overwrites incoming `X-Forwarded-For` and `X-Real-IP`
|
||||||
|
> headers with the actual client IP. If running without a reverse proxy, be
|
||||||
|
> aware that the rate limiting provides no meaningful protection against a
|
||||||
|
> targeted attack.
|
||||||
|
|
||||||
|
**Why rate limits here but not on session creation?** Session creation is
|
||||||
|
protected by hashcash proof-of-work (stateless, no IP tracking needed). Login
|
||||||
|
involves bcrypt password verification against a registered account — a
|
||||||
|
fundamentally different threat model where an attacker targets a specific
|
||||||
|
account. Per-IP rate limiting is appropriate here because the cost of a wrong
|
||||||
|
guess is borne by the server (bcrypt), not the client.
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
## Roadmap
|
## Roadmap
|
||||||
|
|||||||
1
go.mod
1
go.mod
@@ -16,6 +16,7 @@ require (
|
|||||||
github.com/spf13/viper v1.21.0
|
github.com/spf13/viper v1.21.0
|
||||||
go.uber.org/fx v1.24.0
|
go.uber.org/fx v1.24.0
|
||||||
golang.org/x/crypto v0.48.0
|
golang.org/x/crypto v0.48.0
|
||||||
|
golang.org/x/time v0.6.0
|
||||||
modernc.org/sqlite v1.45.0
|
modernc.org/sqlite v1.45.0
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
2
go.sum
2
go.sum
@@ -151,6 +151,8 @@ golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
|||||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||||
golang.org/x/text v0.34.0 h1:oL/Qq0Kdaqxa1KbNeMKwQq0reLCCaFtqu2eNuSeNHbk=
|
golang.org/x/text v0.34.0 h1:oL/Qq0Kdaqxa1KbNeMKwQq0reLCCaFtqu2eNuSeNHbk=
|
||||||
golang.org/x/text v0.34.0/go.mod h1:homfLqTYRFyVYemLBFl5GgL/DWEiH5wcsQ5gSh1yziA=
|
golang.org/x/text v0.34.0/go.mod h1:homfLqTYRFyVYemLBFl5GgL/DWEiH5wcsQ5gSh1yziA=
|
||||||
|
golang.org/x/time v0.6.0 h1:eTDhh4ZXt5Qf0augr54TN6suAUudPcawVZeIAPU7D4U=
|
||||||
|
golang.org/x/time v0.6.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM=
|
||||||
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
|
||||||
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
|
||||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||||
|
|||||||
@@ -7,8 +7,6 @@ import (
|
|||||||
"fmt"
|
"fmt"
|
||||||
"math/big"
|
"math/big"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"git.eeqj.de/sneak/neoirc/internal/hashcash"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -39,23 +37,6 @@ func MintHashcash(bits int, resource string) string {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MintChannelHashcash computes a hashcash stamp bound to
|
|
||||||
// a specific channel and message body. The stamp format
|
|
||||||
// is 1:bits:YYMMDD:channel:bodyhash:counter where
|
|
||||||
// bodyhash is the hex-encoded SHA-256 of the message
|
|
||||||
// body bytes. Delegates to the internal/hashcash package.
|
|
||||||
func MintChannelHashcash(
|
|
||||||
bits int,
|
|
||||||
channel string,
|
|
||||||
body []byte,
|
|
||||||
) string {
|
|
||||||
bodyHash := hashcash.BodyHash(body)
|
|
||||||
|
|
||||||
return hashcash.MintChannelStamp(
|
|
||||||
bits, channel, bodyHash,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// hasLeadingZeroBits checks if hash has at least numBits
|
// hasLeadingZeroBits checks if hash has at least numBits
|
||||||
// leading zero bits.
|
// leading zero bits.
|
||||||
func hasLeadingZeroBits(
|
func hasLeadingZeroBits(
|
||||||
|
|||||||
@@ -46,6 +46,8 @@ type Config struct {
|
|||||||
FederationKey string
|
FederationKey string
|
||||||
SessionIdleTimeout string
|
SessionIdleTimeout string
|
||||||
HashcashBits int
|
HashcashBits int
|
||||||
|
LoginRateLimit float64
|
||||||
|
LoginRateBurst int
|
||||||
params *Params
|
params *Params
|
||||||
log *slog.Logger
|
log *slog.Logger
|
||||||
}
|
}
|
||||||
@@ -78,6 +80,8 @@ func New(
|
|||||||
viper.SetDefault("FEDERATION_KEY", "")
|
viper.SetDefault("FEDERATION_KEY", "")
|
||||||
viper.SetDefault("SESSION_IDLE_TIMEOUT", "720h")
|
viper.SetDefault("SESSION_IDLE_TIMEOUT", "720h")
|
||||||
viper.SetDefault("NEOIRC_HASHCASH_BITS", "20")
|
viper.SetDefault("NEOIRC_HASHCASH_BITS", "20")
|
||||||
|
viper.SetDefault("LOGIN_RATE_LIMIT", "1")
|
||||||
|
viper.SetDefault("LOGIN_RATE_BURST", "5")
|
||||||
|
|
||||||
err := viper.ReadInConfig()
|
err := viper.ReadInConfig()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -104,6 +108,8 @@ func New(
|
|||||||
FederationKey: viper.GetString("FEDERATION_KEY"),
|
FederationKey: viper.GetString("FEDERATION_KEY"),
|
||||||
SessionIdleTimeout: viper.GetString("SESSION_IDLE_TIMEOUT"),
|
SessionIdleTimeout: viper.GetString("SESSION_IDLE_TIMEOUT"),
|
||||||
HashcashBits: viper.GetInt("NEOIRC_HASHCASH_BITS"),
|
HashcashBits: viper.GetInt("NEOIRC_HASHCASH_BITS"),
|
||||||
|
LoginRateLimit: viper.GetFloat64("LOGIN_RATE_LIMIT"),
|
||||||
|
LoginRateBurst: viper.GetInt("LOGIN_RATE_BURST"),
|
||||||
log: log,
|
log: log,
|
||||||
params: ¶ms,
|
params: ¶ms,
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1305,110 +1305,3 @@ func (database *Database) GetQueueEntryCount(
|
|||||||
|
|
||||||
return count, nil
|
return count, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetChannelHashcashBits returns the hashcash difficulty
|
|
||||||
// requirement for a channel. Returns 0 if not set.
|
|
||||||
func (database *Database) GetChannelHashcashBits(
|
|
||||||
ctx context.Context,
|
|
||||||
channelID int64,
|
|
||||||
) (int, error) {
|
|
||||||
var bits int
|
|
||||||
|
|
||||||
err := database.conn.QueryRowContext(
|
|
||||||
ctx,
|
|
||||||
"SELECT hashcash_bits FROM channels WHERE id = ?",
|
|
||||||
channelID,
|
|
||||||
).Scan(&bits)
|
|
||||||
if err != nil {
|
|
||||||
return 0, fmt.Errorf(
|
|
||||||
"get channel hashcash bits: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return bits, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// SetChannelHashcashBits sets the hashcash difficulty
|
|
||||||
// requirement for a channel. A value of 0 disables the
|
|
||||||
// requirement.
|
|
||||||
func (database *Database) SetChannelHashcashBits(
|
|
||||||
ctx context.Context,
|
|
||||||
channelID int64,
|
|
||||||
bits int,
|
|
||||||
) error {
|
|
||||||
_, err := database.conn.ExecContext(ctx,
|
|
||||||
`UPDATE channels
|
|
||||||
SET hashcash_bits = ?, updated_at = ?
|
|
||||||
WHERE id = ?`,
|
|
||||||
bits, time.Now(), channelID)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"set channel hashcash bits: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// RecordSpentHashcash stores a spent hashcash stamp hash
|
|
||||||
// for replay prevention.
|
|
||||||
func (database *Database) RecordSpentHashcash(
|
|
||||||
ctx context.Context,
|
|
||||||
stampHash string,
|
|
||||||
) error {
|
|
||||||
_, err := database.conn.ExecContext(ctx,
|
|
||||||
`INSERT OR IGNORE INTO spent_hashcash
|
|
||||||
(stamp_hash, created_at)
|
|
||||||
VALUES (?, ?)`,
|
|
||||||
stampHash, time.Now())
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"record spent hashcash: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// IsHashcashSpent checks whether a hashcash stamp hash
|
|
||||||
// has already been used.
|
|
||||||
func (database *Database) IsHashcashSpent(
|
|
||||||
ctx context.Context,
|
|
||||||
stampHash string,
|
|
||||||
) (bool, error) {
|
|
||||||
var count int
|
|
||||||
|
|
||||||
err := database.conn.QueryRowContext(ctx,
|
|
||||||
`SELECT COUNT(*) FROM spent_hashcash
|
|
||||||
WHERE stamp_hash = ?`,
|
|
||||||
stampHash,
|
|
||||||
).Scan(&count)
|
|
||||||
if err != nil {
|
|
||||||
return false, fmt.Errorf(
|
|
||||||
"check spent hashcash: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return count > 0, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// PruneSpentHashcash deletes spent hashcash tokens older
|
|
||||||
// than the cutoff and returns the number of rows removed.
|
|
||||||
func (database *Database) PruneSpentHashcash(
|
|
||||||
ctx context.Context,
|
|
||||||
cutoff time.Time,
|
|
||||||
) (int64, error) {
|
|
||||||
res, err := database.conn.ExecContext(ctx,
|
|
||||||
"DELETE FROM spent_hashcash WHERE created_at < ?",
|
|
||||||
cutoff,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
return 0, fmt.Errorf(
|
|
||||||
"prune spent hashcash: %w", err,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
deleted, _ := res.RowsAffected()
|
|
||||||
|
|
||||||
return deleted, nil
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -33,7 +33,6 @@ CREATE TABLE IF NOT EXISTS channels (
|
|||||||
topic TEXT NOT NULL DEFAULT '',
|
topic TEXT NOT NULL DEFAULT '',
|
||||||
topic_set_by TEXT NOT NULL DEFAULT '',
|
topic_set_by TEXT NOT NULL DEFAULT '',
|
||||||
topic_set_at DATETIME,
|
topic_set_at DATETIME,
|
||||||
hashcash_bits INTEGER NOT NULL DEFAULT 0,
|
|
||||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||||
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
||||||
);
|
);
|
||||||
@@ -62,14 +61,6 @@ CREATE TABLE IF NOT EXISTS messages (
|
|||||||
CREATE INDEX IF NOT EXISTS idx_messages_to_id ON messages(msg_to, id);
|
CREATE INDEX IF NOT EXISTS idx_messages_to_id ON messages(msg_to, id);
|
||||||
CREATE INDEX IF NOT EXISTS idx_messages_created ON messages(created_at);
|
CREATE INDEX IF NOT EXISTS idx_messages_created ON messages(created_at);
|
||||||
|
|
||||||
-- Spent hashcash tokens for replay prevention (1-year TTL)
|
|
||||||
CREATE TABLE IF NOT EXISTS spent_hashcash (
|
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
||||||
stamp_hash TEXT NOT NULL UNIQUE,
|
|
||||||
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
|
|
||||||
);
|
|
||||||
CREATE INDEX IF NOT EXISTS idx_spent_hashcash_created ON spent_hashcash(created_at);
|
|
||||||
|
|
||||||
-- Per-client message queues for fan-out delivery
|
-- Per-client message queues for fan-out delivery
|
||||||
CREATE TABLE IF NOT EXISTS client_queues (
|
CREATE TABLE IF NOT EXISTS client_queues (
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
|||||||
@@ -3,7 +3,6 @@ package handlers
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"errors"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"regexp"
|
"regexp"
|
||||||
@@ -12,16 +11,10 @@ import (
|
|||||||
"time"
|
"time"
|
||||||
|
|
||||||
"git.eeqj.de/sneak/neoirc/internal/db"
|
"git.eeqj.de/sneak/neoirc/internal/db"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/hashcash"
|
|
||||||
"git.eeqj.de/sneak/neoirc/pkg/irc"
|
"git.eeqj.de/sneak/neoirc/pkg/irc"
|
||||||
"github.com/go-chi/chi/v5"
|
"github.com/go-chi/chi/v5"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
|
||||||
errHashcashRequired = errors.New("hashcash required")
|
|
||||||
errHashcashReused = errors.New("hashcash reused")
|
|
||||||
)
|
|
||||||
|
|
||||||
var validNickRe = regexp.MustCompile(
|
var validNickRe = regexp.MustCompile(
|
||||||
`^[a-zA-Z_][a-zA-Z0-9_\-\[\]\\^{}|` + "`" + `]{0,31}$`,
|
`^[a-zA-Z_][a-zA-Z0-9_\-\[\]\\^{}|` + "`" + `]{0,31}$`,
|
||||||
)
|
)
|
||||||
@@ -95,11 +88,10 @@ func (hdlr *Handlers) fanOut(
|
|||||||
request *http.Request,
|
request *http.Request,
|
||||||
command, from, target string,
|
command, from, target string,
|
||||||
body json.RawMessage,
|
body json.RawMessage,
|
||||||
meta json.RawMessage,
|
|
||||||
sessionIDs []int64,
|
sessionIDs []int64,
|
||||||
) (string, error) {
|
) (string, error) {
|
||||||
dbID, msgUUID, err := hdlr.params.Database.InsertMessage(
|
dbID, msgUUID, err := hdlr.params.Database.InsertMessage(
|
||||||
request.Context(), command, from, target, nil, body, meta,
|
request.Context(), command, from, target, nil, body, nil,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return "", fmt.Errorf("insert message: %w", err)
|
return "", fmt.Errorf("insert message: %w", err)
|
||||||
@@ -125,11 +117,10 @@ func (hdlr *Handlers) fanOutSilent(
|
|||||||
request *http.Request,
|
request *http.Request,
|
||||||
command, from, target string,
|
command, from, target string,
|
||||||
body json.RawMessage,
|
body json.RawMessage,
|
||||||
meta json.RawMessage,
|
|
||||||
sessionIDs []int64,
|
sessionIDs []int64,
|
||||||
) error {
|
) error {
|
||||||
_, err := hdlr.fanOut(
|
_, err := hdlr.fanOut(
|
||||||
request, command, from, target, body, meta, sessionIDs,
|
request, command, from, target, body, sessionIDs,
|
||||||
)
|
)
|
||||||
|
|
||||||
return err
|
return err
|
||||||
@@ -303,7 +294,7 @@ func (hdlr *Handlers) deliverWelcome(
|
|||||||
[]string{
|
[]string{
|
||||||
"CHANTYPES=#",
|
"CHANTYPES=#",
|
||||||
"NICKLEN=32",
|
"NICKLEN=32",
|
||||||
"CHANMODES=,,H," + "imnst",
|
"CHANMODES=,,," + "imnst",
|
||||||
"NETWORK=neoirc",
|
"NETWORK=neoirc",
|
||||||
"CASEMAPPING=ascii",
|
"CASEMAPPING=ascii",
|
||||||
},
|
},
|
||||||
@@ -834,7 +825,7 @@ func (hdlr *Handlers) HandleSendCommand() http.HandlerFunc {
|
|||||||
writer, request,
|
writer, request,
|
||||||
sessionID, clientID, nick,
|
sessionID, clientID, nick,
|
||||||
payload.Command, payload.To,
|
payload.Command, payload.To,
|
||||||
payload.Body, payload.Meta, bodyLines,
|
payload.Body, bodyLines,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -845,7 +836,6 @@ func (hdlr *Handlers) dispatchCommand(
|
|||||||
sessionID, clientID int64,
|
sessionID, clientID int64,
|
||||||
nick, command, target string,
|
nick, command, target string,
|
||||||
body json.RawMessage,
|
body json.RawMessage,
|
||||||
meta json.RawMessage,
|
|
||||||
bodyLines func() []string,
|
bodyLines func() []string,
|
||||||
) {
|
) {
|
||||||
switch command {
|
switch command {
|
||||||
@@ -858,7 +848,7 @@ func (hdlr *Handlers) dispatchCommand(
|
|||||||
hdlr.handlePrivmsg(
|
hdlr.handlePrivmsg(
|
||||||
writer, request,
|
writer, request,
|
||||||
sessionID, clientID, nick,
|
sessionID, clientID, nick,
|
||||||
command, target, body, meta, bodyLines,
|
command, target, body, bodyLines,
|
||||||
)
|
)
|
||||||
case irc.CmdJoin:
|
case irc.CmdJoin:
|
||||||
hdlr.handleJoin(
|
hdlr.handleJoin(
|
||||||
@@ -959,7 +949,6 @@ func (hdlr *Handlers) handlePrivmsg(
|
|||||||
sessionID, clientID int64,
|
sessionID, clientID int64,
|
||||||
nick, command, target string,
|
nick, command, target string,
|
||||||
body json.RawMessage,
|
body json.RawMessage,
|
||||||
meta json.RawMessage,
|
|
||||||
bodyLines func() []string,
|
bodyLines func() []string,
|
||||||
) {
|
) {
|
||||||
if target == "" {
|
if target == "" {
|
||||||
@@ -997,7 +986,7 @@ func (hdlr *Handlers) handlePrivmsg(
|
|||||||
hdlr.handleChannelMsg(
|
hdlr.handleChannelMsg(
|
||||||
writer, request,
|
writer, request,
|
||||||
sessionID, clientID, nick,
|
sessionID, clientID, nick,
|
||||||
command, target, body, meta,
|
command, target, body,
|
||||||
)
|
)
|
||||||
|
|
||||||
return
|
return
|
||||||
@@ -1006,7 +995,7 @@ func (hdlr *Handlers) handlePrivmsg(
|
|||||||
hdlr.handleDirectMsg(
|
hdlr.handleDirectMsg(
|
||||||
writer, request,
|
writer, request,
|
||||||
sessionID, clientID, nick,
|
sessionID, clientID, nick,
|
||||||
command, target, body, meta,
|
command, target, body,
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1037,7 +1026,6 @@ func (hdlr *Handlers) handleChannelMsg(
|
|||||||
sessionID, clientID int64,
|
sessionID, clientID int64,
|
||||||
nick, command, target string,
|
nick, command, target string,
|
||||||
body json.RawMessage,
|
body json.RawMessage,
|
||||||
meta json.RawMessage,
|
|
||||||
) {
|
) {
|
||||||
chID, err := hdlr.params.Database.GetChannelByName(
|
chID, err := hdlr.params.Database.GetChannelByName(
|
||||||
request.Context(), target,
|
request.Context(), target,
|
||||||
@@ -1078,180 +1066,16 @@ func (hdlr *Handlers) handleChannelMsg(
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
hashcashErr := hdlr.validateChannelHashcash(
|
|
||||||
request, clientID, sessionID,
|
|
||||||
writer, nick, target, body, meta, chID,
|
|
||||||
)
|
|
||||||
if hashcashErr != nil {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
hdlr.sendChannelMsg(
|
hdlr.sendChannelMsg(
|
||||||
writer, request, command, nick, target,
|
writer, request, command, nick, target, body, chID,
|
||||||
body, meta, chID,
|
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
// validateChannelHashcash checks whether the channel
|
|
||||||
// requires hashcash proof-of-work for messages and
|
|
||||||
// validates the stamp from the message meta field.
|
|
||||||
// Returns nil on success or if the channel has no
|
|
||||||
// hashcash requirement. On failure, it sends the
|
|
||||||
// appropriate IRC error and returns a non-nil error.
|
|
||||||
func (hdlr *Handlers) validateChannelHashcash(
|
|
||||||
request *http.Request,
|
|
||||||
clientID, sessionID int64,
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
nick, target string,
|
|
||||||
body json.RawMessage,
|
|
||||||
meta json.RawMessage,
|
|
||||||
chID int64,
|
|
||||||
) error {
|
|
||||||
ctx := request.Context()
|
|
||||||
|
|
||||||
bits, bitsErr := hdlr.params.Database.GetChannelHashcashBits(
|
|
||||||
ctx, chID,
|
|
||||||
)
|
|
||||||
if bitsErr != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"get channel hashcash bits", "error", bitsErr,
|
|
||||||
)
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"internal error",
|
|
||||||
http.StatusInternalServerError,
|
|
||||||
)
|
|
||||||
|
|
||||||
return fmt.Errorf("channel hashcash bits: %w", bitsErr)
|
|
||||||
}
|
|
||||||
|
|
||||||
if bits <= 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
stamp := hdlr.extractHashcashFromMeta(meta)
|
|
||||||
if stamp == "" {
|
|
||||||
hdlr.respondIRCError(
|
|
||||||
writer, request, clientID, sessionID,
|
|
||||||
irc.ErrCannotSendToChan, nick, []string{target},
|
|
||||||
"Channel requires hashcash proof-of-work",
|
|
||||||
)
|
|
||||||
|
|
||||||
return errHashcashRequired
|
|
||||||
}
|
|
||||||
|
|
||||||
return hdlr.verifyChannelStamp(
|
|
||||||
request, writer,
|
|
||||||
clientID, sessionID,
|
|
||||||
nick, target, body, stamp, bits,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// verifyChannelStamp validates a channel hashcash stamp
|
|
||||||
// and checks for replay attacks.
|
|
||||||
func (hdlr *Handlers) verifyChannelStamp(
|
|
||||||
request *http.Request,
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
clientID, sessionID int64,
|
|
||||||
nick, target string,
|
|
||||||
body json.RawMessage,
|
|
||||||
stamp string,
|
|
||||||
bits int,
|
|
||||||
) error {
|
|
||||||
ctx := request.Context()
|
|
||||||
bodyHashStr := hashcash.BodyHash(body)
|
|
||||||
|
|
||||||
valErr := hdlr.channelHashcash.ValidateStamp(
|
|
||||||
stamp, bits, target, bodyHashStr,
|
|
||||||
)
|
|
||||||
if valErr != nil {
|
|
||||||
hdlr.respondIRCError(
|
|
||||||
writer, request, clientID, sessionID,
|
|
||||||
irc.ErrCannotSendToChan, nick, []string{target},
|
|
||||||
"Invalid hashcash: "+valErr.Error(),
|
|
||||||
)
|
|
||||||
|
|
||||||
return fmt.Errorf("channel hashcash: %w", valErr)
|
|
||||||
}
|
|
||||||
|
|
||||||
stampKey := hashcash.StampHash(stamp)
|
|
||||||
|
|
||||||
spent, spentErr := hdlr.params.Database.IsHashcashSpent(
|
|
||||||
ctx, stampKey,
|
|
||||||
)
|
|
||||||
if spentErr != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"check spent hashcash", "error", spentErr,
|
|
||||||
)
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"internal error",
|
|
||||||
http.StatusInternalServerError,
|
|
||||||
)
|
|
||||||
|
|
||||||
return fmt.Errorf("check spent hashcash: %w", spentErr)
|
|
||||||
}
|
|
||||||
|
|
||||||
if spent {
|
|
||||||
hdlr.respondIRCError(
|
|
||||||
writer, request, clientID, sessionID,
|
|
||||||
irc.ErrCannotSendToChan, nick, []string{target},
|
|
||||||
"Hashcash stamp already used",
|
|
||||||
)
|
|
||||||
|
|
||||||
return errHashcashReused
|
|
||||||
}
|
|
||||||
|
|
||||||
recordErr := hdlr.params.Database.RecordSpentHashcash(
|
|
||||||
ctx, stampKey,
|
|
||||||
)
|
|
||||||
if recordErr != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"record spent hashcash", "error", recordErr,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// extractHashcashFromMeta parses the meta JSON and
|
|
||||||
// returns the hashcash stamp string, or empty string
|
|
||||||
// if not present.
|
|
||||||
func (hdlr *Handlers) extractHashcashFromMeta(
|
|
||||||
meta json.RawMessage,
|
|
||||||
) string {
|
|
||||||
if len(meta) == 0 {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
var metaMap map[string]json.RawMessage
|
|
||||||
|
|
||||||
err := json.Unmarshal(meta, &metaMap)
|
|
||||||
if err != nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
raw, ok := metaMap["hashcash"]
|
|
||||||
if !ok {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
var stamp string
|
|
||||||
|
|
||||||
err = json.Unmarshal(raw, &stamp)
|
|
||||||
if err != nil {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
return stamp
|
|
||||||
}
|
|
||||||
|
|
||||||
func (hdlr *Handlers) sendChannelMsg(
|
func (hdlr *Handlers) sendChannelMsg(
|
||||||
writer http.ResponseWriter,
|
writer http.ResponseWriter,
|
||||||
request *http.Request,
|
request *http.Request,
|
||||||
command, nick, target string,
|
command, nick, target string,
|
||||||
body json.RawMessage,
|
body json.RawMessage,
|
||||||
meta json.RawMessage,
|
|
||||||
chID int64,
|
chID int64,
|
||||||
) {
|
) {
|
||||||
memberIDs, err := hdlr.params.Database.GetChannelMemberIDs(
|
memberIDs, err := hdlr.params.Database.GetChannelMemberIDs(
|
||||||
@@ -1271,7 +1095,7 @@ func (hdlr *Handlers) sendChannelMsg(
|
|||||||
}
|
}
|
||||||
|
|
||||||
msgUUID, err := hdlr.fanOut(
|
msgUUID, err := hdlr.fanOut(
|
||||||
request, command, nick, target, body, meta, memberIDs,
|
request, command, nick, target, body, memberIDs,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
hdlr.log.Error("send message failed", "error", err)
|
hdlr.log.Error("send message failed", "error", err)
|
||||||
@@ -1295,7 +1119,6 @@ func (hdlr *Handlers) handleDirectMsg(
|
|||||||
sessionID, clientID int64,
|
sessionID, clientID int64,
|
||||||
nick, command, target string,
|
nick, command, target string,
|
||||||
body json.RawMessage,
|
body json.RawMessage,
|
||||||
meta json.RawMessage,
|
|
||||||
) {
|
) {
|
||||||
targetSID, err := hdlr.params.Database.GetSessionByNick(
|
targetSID, err := hdlr.params.Database.GetSessionByNick(
|
||||||
request.Context(), target,
|
request.Context(), target,
|
||||||
@@ -1320,7 +1143,7 @@ func (hdlr *Handlers) handleDirectMsg(
|
|||||||
}
|
}
|
||||||
|
|
||||||
msgUUID, err := hdlr.fanOut(
|
msgUUID, err := hdlr.fanOut(
|
||||||
request, command, nick, target, body, meta, recipients,
|
request, command, nick, target, body, recipients,
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
hdlr.log.Error("send dm failed", "error", err)
|
hdlr.log.Error("send dm failed", "error", err)
|
||||||
@@ -1431,7 +1254,7 @@ func (hdlr *Handlers) executeJoin(
|
|||||||
)
|
)
|
||||||
|
|
||||||
_ = hdlr.fanOutSilent(
|
_ = hdlr.fanOutSilent(
|
||||||
request, irc.CmdJoin, nick, channel, nil, nil, memberIDs,
|
request, irc.CmdJoin, nick, channel, nil, memberIDs,
|
||||||
)
|
)
|
||||||
|
|
||||||
hdlr.deliverJoinNumerics(
|
hdlr.deliverJoinNumerics(
|
||||||
@@ -1601,7 +1424,7 @@ func (hdlr *Handlers) handlePart(
|
|||||||
)
|
)
|
||||||
|
|
||||||
_ = hdlr.fanOutSilent(
|
_ = hdlr.fanOutSilent(
|
||||||
request, irc.CmdPart, nick, channel, body, nil, memberIDs,
|
request, irc.CmdPart, nick, channel, body, memberIDs,
|
||||||
)
|
)
|
||||||
|
|
||||||
err = hdlr.params.Database.PartChannel(
|
err = hdlr.params.Database.PartChannel(
|
||||||
@@ -1881,7 +1704,7 @@ func (hdlr *Handlers) executeTopic(
|
|||||||
)
|
)
|
||||||
|
|
||||||
_ = hdlr.fanOutSilent(
|
_ = hdlr.fanOutSilent(
|
||||||
request, irc.CmdTopic, nick, channel, body, nil, memberIDs,
|
request, irc.CmdTopic, nick, channel, body, memberIDs,
|
||||||
)
|
)
|
||||||
|
|
||||||
hdlr.enqueueNumeric(
|
hdlr.enqueueNumeric(
|
||||||
@@ -2044,10 +1867,11 @@ func (hdlr *Handlers) handleMode(
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
_ = bodyLines
|
||||||
|
|
||||||
hdlr.handleChannelMode(
|
hdlr.handleChannelMode(
|
||||||
writer, request,
|
writer, request,
|
||||||
sessionID, clientID, nick, channel,
|
sessionID, clientID, nick, channel,
|
||||||
bodyLines,
|
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -2056,7 +1880,6 @@ func (hdlr *Handlers) handleChannelMode(
|
|||||||
request *http.Request,
|
request *http.Request,
|
||||||
sessionID, clientID int64,
|
sessionID, clientID int64,
|
||||||
nick, channel string,
|
nick, channel string,
|
||||||
bodyLines func() []string,
|
|
||||||
) {
|
) {
|
||||||
ctx := request.Context()
|
ctx := request.Context()
|
||||||
|
|
||||||
@@ -2073,47 +1896,10 @@ func (hdlr *Handlers) handleChannelMode(
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
lines := bodyLines()
|
|
||||||
if len(lines) > 0 {
|
|
||||||
hdlr.applyChannelMode(
|
|
||||||
writer, request,
|
|
||||||
sessionID, clientID, nick,
|
|
||||||
channel, chID, lines,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
hdlr.queryChannelMode(
|
|
||||||
writer, request,
|
|
||||||
sessionID, clientID, nick, channel, chID,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// queryChannelMode sends RPL_CHANNELMODEIS and
|
|
||||||
// RPL_CREATIONTIME for a channel. Includes +H if
|
|
||||||
// the channel has a hashcash requirement.
|
|
||||||
func (hdlr *Handlers) queryChannelMode(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
sessionID, clientID int64,
|
|
||||||
nick, channel string,
|
|
||||||
chID int64,
|
|
||||||
) {
|
|
||||||
ctx := request.Context()
|
|
||||||
|
|
||||||
modeStr := "+n"
|
|
||||||
|
|
||||||
bits, bitsErr := hdlr.params.Database.
|
|
||||||
GetChannelHashcashBits(ctx, chID)
|
|
||||||
if bitsErr == nil && bits > 0 {
|
|
||||||
modeStr = fmt.Sprintf("+nH %d", bits)
|
|
||||||
}
|
|
||||||
|
|
||||||
// 324 RPL_CHANNELMODEIS
|
// 324 RPL_CHANNELMODEIS
|
||||||
hdlr.enqueueNumeric(
|
hdlr.enqueueNumeric(
|
||||||
ctx, clientID, irc.RplChannelModeIs, nick,
|
ctx, clientID, irc.RplChannelModeIs, nick,
|
||||||
[]string{channel, modeStr}, "",
|
[]string{channel, "+n"}, "",
|
||||||
)
|
)
|
||||||
|
|
||||||
// 329 RPL_CREATIONTIME
|
// 329 RPL_CREATIONTIME
|
||||||
@@ -2138,156 +1924,6 @@ func (hdlr *Handlers) queryChannelMode(
|
|||||||
http.StatusOK)
|
http.StatusOK)
|
||||||
}
|
}
|
||||||
|
|
||||||
// applyChannelMode handles setting channel modes.
|
|
||||||
// Currently supports +H/-H for hashcash bits.
|
|
||||||
func (hdlr *Handlers) applyChannelMode(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
sessionID, clientID int64,
|
|
||||||
nick, channel string,
|
|
||||||
chID int64,
|
|
||||||
modeArgs []string,
|
|
||||||
) {
|
|
||||||
ctx := request.Context()
|
|
||||||
modeStr := modeArgs[0]
|
|
||||||
|
|
||||||
switch modeStr {
|
|
||||||
case "+H":
|
|
||||||
hdlr.setHashcashMode(
|
|
||||||
writer, request,
|
|
||||||
sessionID, clientID, nick,
|
|
||||||
channel, chID, modeArgs,
|
|
||||||
)
|
|
||||||
case "-H":
|
|
||||||
hdlr.clearHashcashMode(
|
|
||||||
writer, request,
|
|
||||||
sessionID, clientID, nick,
|
|
||||||
channel, chID,
|
|
||||||
)
|
|
||||||
default:
|
|
||||||
// Unknown or unsupported mode change.
|
|
||||||
hdlr.enqueueNumeric(
|
|
||||||
ctx, clientID, irc.ErrUnknownMode, nick,
|
|
||||||
[]string{modeStr},
|
|
||||||
"is unknown mode char to me",
|
|
||||||
)
|
|
||||||
hdlr.broker.Notify(sessionID)
|
|
||||||
hdlr.respondJSON(writer, request,
|
|
||||||
map[string]string{"status": "error"},
|
|
||||||
http.StatusOK)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const (
|
|
||||||
// minHashcashBits is the minimum allowed hashcash
|
|
||||||
// difficulty for channels.
|
|
||||||
minHashcashBits = 1
|
|
||||||
// maxHashcashBits is the maximum allowed hashcash
|
|
||||||
// difficulty for channels.
|
|
||||||
maxHashcashBits = 40
|
|
||||||
)
|
|
||||||
|
|
||||||
// setHashcashMode handles MODE #channel +H <bits>.
|
|
||||||
func (hdlr *Handlers) setHashcashMode(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
sessionID, clientID int64,
|
|
||||||
nick, channel string,
|
|
||||||
chID int64,
|
|
||||||
modeArgs []string,
|
|
||||||
) {
|
|
||||||
ctx := request.Context()
|
|
||||||
|
|
||||||
if len(modeArgs) < 2 { //nolint:mnd // +H requires a bits arg
|
|
||||||
hdlr.respondIRCError(
|
|
||||||
writer, request, clientID, sessionID,
|
|
||||||
irc.ErrNeedMoreParams, nick, []string{irc.CmdMode},
|
|
||||||
"Not enough parameters (+H requires bits)",
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
bits, err := strconv.Atoi(modeArgs[1])
|
|
||||||
if err != nil || bits < minHashcashBits ||
|
|
||||||
bits > maxHashcashBits {
|
|
||||||
hdlr.respondIRCError(
|
|
||||||
writer, request, clientID, sessionID,
|
|
||||||
irc.ErrUnknownMode, nick, []string{"+H"},
|
|
||||||
fmt.Sprintf(
|
|
||||||
"Invalid hashcash bits (must be %d-%d)",
|
|
||||||
minHashcashBits, maxHashcashBits,
|
|
||||||
),
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
err = hdlr.params.Database.SetChannelHashcashBits(
|
|
||||||
ctx, chID, bits,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"set channel hashcash bits", "error", err,
|
|
||||||
)
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"internal error",
|
|
||||||
http.StatusInternalServerError,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
hdlr.enqueueNumeric(
|
|
||||||
ctx, clientID, irc.RplChannelModeIs, nick,
|
|
||||||
[]string{
|
|
||||||
channel,
|
|
||||||
fmt.Sprintf("+H %d", bits),
|
|
||||||
}, "",
|
|
||||||
)
|
|
||||||
hdlr.broker.Notify(sessionID)
|
|
||||||
hdlr.respondJSON(writer, request,
|
|
||||||
map[string]string{"status": "ok"},
|
|
||||||
http.StatusOK)
|
|
||||||
}
|
|
||||||
|
|
||||||
// clearHashcashMode handles MODE #channel -H.
|
|
||||||
func (hdlr *Handlers) clearHashcashMode(
|
|
||||||
writer http.ResponseWriter,
|
|
||||||
request *http.Request,
|
|
||||||
sessionID, clientID int64,
|
|
||||||
nick, channel string,
|
|
||||||
chID int64,
|
|
||||||
) {
|
|
||||||
ctx := request.Context()
|
|
||||||
|
|
||||||
err := hdlr.params.Database.SetChannelHashcashBits(
|
|
||||||
ctx, chID, 0,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"clear channel hashcash bits", "error", err,
|
|
||||||
)
|
|
||||||
hdlr.respondError(
|
|
||||||
writer, request,
|
|
||||||
"internal error",
|
|
||||||
http.StatusInternalServerError,
|
|
||||||
)
|
|
||||||
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
hdlr.enqueueNumeric(
|
|
||||||
ctx, clientID, irc.RplChannelModeIs, nick,
|
|
||||||
[]string{channel, "+n"}, "",
|
|
||||||
)
|
|
||||||
hdlr.broker.Notify(sessionID)
|
|
||||||
hdlr.respondJSON(writer, request,
|
|
||||||
map[string]string{"status": "ok"},
|
|
||||||
http.StatusOK)
|
|
||||||
}
|
|
||||||
|
|
||||||
// handleNames sends NAMES reply for a channel.
|
// handleNames sends NAMES reply for a channel.
|
||||||
func (hdlr *Handlers) handleNames(
|
func (hdlr *Handlers) handleNames(
|
||||||
writer http.ResponseWriter,
|
writer http.ResponseWriter,
|
||||||
|
|||||||
@@ -22,7 +22,6 @@ import (
|
|||||||
"git.eeqj.de/sneak/neoirc/internal/db"
|
"git.eeqj.de/sneak/neoirc/internal/db"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/globals"
|
"git.eeqj.de/sneak/neoirc/internal/globals"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/handlers"
|
"git.eeqj.de/sneak/neoirc/internal/handlers"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/hashcash"
|
|
||||||
"git.eeqj.de/sneak/neoirc/internal/healthcheck"
|
"git.eeqj.de/sneak/neoirc/internal/healthcheck"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/logger"
|
"git.eeqj.de/sneak/neoirc/internal/logger"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/middleware"
|
"git.eeqj.de/sneak/neoirc/internal/middleware"
|
||||||
@@ -2131,6 +2130,121 @@ func TestSessionStillWorks(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestLoginRateLimitExceeded(t *testing.T) {
|
||||||
|
tserver := newTestServer(t)
|
||||||
|
|
||||||
|
// Exhaust the burst (default: 5 per IP) using
|
||||||
|
// nonexistent users. These fail fast (no bcrypt),
|
||||||
|
// preventing token replenishment between requests.
|
||||||
|
for range 5 {
|
||||||
|
loginBody, mErr := json.Marshal(
|
||||||
|
map[string]string{
|
||||||
|
"nick": "nosuchuser",
|
||||||
|
"password": "doesnotmatter",
|
||||||
|
},
|
||||||
|
)
|
||||||
|
if mErr != nil {
|
||||||
|
t.Fatal(mErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
loginResp, rErr := doRequest(
|
||||||
|
t,
|
||||||
|
http.MethodPost,
|
||||||
|
tserver.url("/api/v1/login"),
|
||||||
|
bytes.NewReader(loginBody),
|
||||||
|
)
|
||||||
|
if rErr != nil {
|
||||||
|
t.Fatal(rErr)
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = loginResp.Body.Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// The next request should be rate-limited.
|
||||||
|
loginBody, err := json.Marshal(map[string]string{
|
||||||
|
"nick": "nosuchuser", "password": "doesnotmatter",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := doRequest(
|
||||||
|
t,
|
||||||
|
http.MethodPost,
|
||||||
|
tserver.url("/api/v1/login"),
|
||||||
|
bytes.NewReader(loginBody),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() { _ = resp.Body.Close() }()
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusTooManyRequests {
|
||||||
|
t.Fatalf(
|
||||||
|
"expected 429, got %d",
|
||||||
|
resp.StatusCode,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
retryAfter := resp.Header.Get("Retry-After")
|
||||||
|
if retryAfter == "" {
|
||||||
|
t.Fatal("expected Retry-After header")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLoginRateLimitAllowsNormalUse(t *testing.T) {
|
||||||
|
tserver := newTestServer(t)
|
||||||
|
|
||||||
|
// Register a user.
|
||||||
|
regBody, err := json.Marshal(map[string]string{
|
||||||
|
"nick": "normaluser", "password": "password123",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := doRequest(
|
||||||
|
t,
|
||||||
|
http.MethodPost,
|
||||||
|
tserver.url("/api/v1/register"),
|
||||||
|
bytes.NewReader(regBody),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_ = resp.Body.Close()
|
||||||
|
|
||||||
|
// A single login should succeed without rate limiting.
|
||||||
|
loginBody, err := json.Marshal(map[string]string{
|
||||||
|
"nick": "normaluser", "password": "password123",
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp2, err := doRequest(
|
||||||
|
t,
|
||||||
|
http.MethodPost,
|
||||||
|
tserver.url("/api/v1/login"),
|
||||||
|
bytes.NewReader(loginBody),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatal(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
defer func() { _ = resp2.Body.Close() }()
|
||||||
|
|
||||||
|
if resp2.StatusCode != http.StatusOK {
|
||||||
|
respBody, _ := io.ReadAll(resp2.Body)
|
||||||
|
t.Fatalf(
|
||||||
|
"expected 200, got %d: %s",
|
||||||
|
resp2.StatusCode, respBody,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func TestNickBroadcastToChannels(t *testing.T) {
|
func TestNickBroadcastToChannels(t *testing.T) {
|
||||||
tserver := newTestServer(t)
|
tserver := newTestServer(t)
|
||||||
aliceToken := tserver.createSession("nick_a")
|
aliceToken := tserver.createSession("nick_a")
|
||||||
@@ -2158,397 +2272,3 @@ func TestNickBroadcastToChannels(t *testing.T) {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// --- Channel Hashcash Tests ---
|
|
||||||
|
|
||||||
const (
|
|
||||||
metaKey = "meta"
|
|
||||||
modeCmd = "MODE"
|
|
||||||
hashcashKey = "hashcash"
|
|
||||||
)
|
|
||||||
|
|
||||||
func mintTestChannelHashcash(
|
|
||||||
tb testing.TB,
|
|
||||||
bits int,
|
|
||||||
channel string,
|
|
||||||
body json.RawMessage,
|
|
||||||
) string {
|
|
||||||
tb.Helper()
|
|
||||||
|
|
||||||
bodyHash := hashcash.BodyHash(body)
|
|
||||||
|
|
||||||
return hashcash.MintChannelStamp(bits, channel, bodyHash)
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashSetMode(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcmode_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hctest",
|
|
||||||
})
|
|
||||||
|
|
||||||
_, lastID := tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Set hashcash bits to 2 via MODE +H.
|
|
||||||
status, _ := tserver.sendCommand(
|
|
||||||
token,
|
|
||||||
map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hctest",
|
|
||||||
bodyKey: []string{"+H", "2"},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf("expected 200, got %d", status)
|
|
||||||
}
|
|
||||||
|
|
||||||
msgs, _ := tserver.pollMessages(token, lastID)
|
|
||||||
|
|
||||||
// Should get RPL_CHANNELMODEIS (324) confirming +H.
|
|
||||||
if !findNumeric(msgs, "324") {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected RPL_CHANNELMODEIS (324), got %v",
|
|
||||||
msgs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashQueryMode(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcquery_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hcquery",
|
|
||||||
})
|
|
||||||
|
|
||||||
// Set hashcash bits.
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcquery",
|
|
||||||
bodyKey: []string{"+H", "5"},
|
|
||||||
})
|
|
||||||
|
|
||||||
_, lastID := tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Query mode — should show +nH.
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcquery",
|
|
||||||
})
|
|
||||||
|
|
||||||
msgs, _ := tserver.pollMessages(token, lastID)
|
|
||||||
|
|
||||||
found := false
|
|
||||||
|
|
||||||
for _, msg := range msgs {
|
|
||||||
code, ok := msg["code"].(float64)
|
|
||||||
if ok && int(code) == 324 {
|
|
||||||
found = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if !found {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected RPL_CHANNELMODEIS (324), got %v",
|
|
||||||
msgs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashClearMode(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcclear_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hcclear",
|
|
||||||
})
|
|
||||||
|
|
||||||
// Set hashcash bits.
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcclear",
|
|
||||||
bodyKey: []string{"+H", "5"},
|
|
||||||
})
|
|
||||||
|
|
||||||
// Clear hashcash bits.
|
|
||||||
status, _ := tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcclear",
|
|
||||||
bodyKey: []string{"-H"},
|
|
||||||
})
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf("expected 200, got %d", status)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Now message should succeed without hashcash.
|
|
||||||
status, result := tserver.sendCommand(
|
|
||||||
token,
|
|
||||||
map[string]any{
|
|
||||||
commandKey: privmsgCmd,
|
|
||||||
toKey: "#hcclear",
|
|
||||||
bodyKey: []string{"test message"},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected 200, got %d: %v", status, result,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashRejectNoStamp(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcreject_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hcreject",
|
|
||||||
})
|
|
||||||
|
|
||||||
// Set hashcash requirement.
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcreject",
|
|
||||||
bodyKey: []string{"+H", "2"},
|
|
||||||
})
|
|
||||||
|
|
||||||
_, lastID := tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Send message without hashcash — should fail.
|
|
||||||
status, _ := tserver.sendCommand(
|
|
||||||
token,
|
|
||||||
map[string]any{
|
|
||||||
commandKey: privmsgCmd,
|
|
||||||
toKey: "#hcreject",
|
|
||||||
bodyKey: []string{"spam message"},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf("expected 200, got %d", status)
|
|
||||||
}
|
|
||||||
|
|
||||||
msgs, _ := tserver.pollMessages(token, lastID)
|
|
||||||
|
|
||||||
// Should get ERR_CANNOTSENDTOCHAN (404).
|
|
||||||
if !findNumeric(msgs, "404") {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected ERR_CANNOTSENDTOCHAN (404), got %v",
|
|
||||||
msgs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashAcceptValidStamp(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcaccept_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hcaccept",
|
|
||||||
})
|
|
||||||
|
|
||||||
// Set hashcash requirement (2 bits = fast to mint).
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcaccept",
|
|
||||||
bodyKey: []string{"+H", "2"},
|
|
||||||
})
|
|
||||||
|
|
||||||
_, lastID := tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Mint a valid hashcash stamp.
|
|
||||||
msgBody, marshalErr := json.Marshal(
|
|
||||||
[]string{"hello world"},
|
|
||||||
)
|
|
||||||
if marshalErr != nil {
|
|
||||||
t.Fatal(marshalErr)
|
|
||||||
}
|
|
||||||
|
|
||||||
stamp := mintTestChannelHashcash(
|
|
||||||
t, 2, "#hcaccept", msgBody,
|
|
||||||
)
|
|
||||||
|
|
||||||
// Send message with valid hashcash.
|
|
||||||
status, result := tserver.sendCommand(
|
|
||||||
token,
|
|
||||||
map[string]any{
|
|
||||||
commandKey: privmsgCmd,
|
|
||||||
toKey: "#hcaccept",
|
|
||||||
bodyKey: []string{"hello world"},
|
|
||||||
metaKey: map[string]any{
|
|
||||||
hashcashKey: stamp,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected 200, got %d: %v", status, result,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
if result["id"] == nil || result["id"] == "" {
|
|
||||||
t.Fatal("expected message id for valid hashcash")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Verify the message was delivered.
|
|
||||||
msgs, _ := tserver.pollMessages(token, lastID)
|
|
||||||
if !findMessage(msgs, privmsgCmd, "hcaccept_user") {
|
|
||||||
t.Fatalf(
|
|
||||||
"message not received: %v", msgs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashRejectReplayedStamp(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcreplay_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hcreplay",
|
|
||||||
})
|
|
||||||
|
|
||||||
// Set hashcash requirement.
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcreplay",
|
|
||||||
bodyKey: []string{"+H", "2"},
|
|
||||||
})
|
|
||||||
|
|
||||||
_, _ = tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Mint and send once — should succeed.
|
|
||||||
msgBody, marshalErr := json.Marshal(
|
|
||||||
[]string{"unique msg"},
|
|
||||||
)
|
|
||||||
if marshalErr != nil {
|
|
||||||
t.Fatal(marshalErr)
|
|
||||||
}
|
|
||||||
|
|
||||||
stamp := mintTestChannelHashcash(
|
|
||||||
t, 2, "#hcreplay", msgBody,
|
|
||||||
)
|
|
||||||
|
|
||||||
status, _ := tserver.sendCommand(
|
|
||||||
token,
|
|
||||||
map[string]any{
|
|
||||||
commandKey: privmsgCmd,
|
|
||||||
toKey: "#hcreplay",
|
|
||||||
bodyKey: []string{"unique msg"},
|
|
||||||
metaKey: map[string]any{
|
|
||||||
hashcashKey: stamp,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf("expected 200, got %d", status)
|
|
||||||
}
|
|
||||||
|
|
||||||
_, lastID := tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Replay the same stamp — should fail.
|
|
||||||
status, _ = tserver.sendCommand(
|
|
||||||
token,
|
|
||||||
map[string]any{
|
|
||||||
commandKey: privmsgCmd,
|
|
||||||
toKey: "#hcreplay",
|
|
||||||
bodyKey: []string{"unique msg"},
|
|
||||||
metaKey: map[string]any{
|
|
||||||
hashcashKey: stamp,
|
|
||||||
},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf("expected 200, got %d", status)
|
|
||||||
}
|
|
||||||
|
|
||||||
msgs, _ := tserver.pollMessages(token, lastID)
|
|
||||||
|
|
||||||
// Should get ERR_CANNOTSENDTOCHAN (404).
|
|
||||||
if !findNumeric(msgs, "404") {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected replay rejection (404), got %v",
|
|
||||||
msgs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashNoRequirementWorks(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcnone_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#nohashcash",
|
|
||||||
})
|
|
||||||
|
|
||||||
// No hashcash set — message should work.
|
|
||||||
status, result := tserver.sendCommand(
|
|
||||||
token,
|
|
||||||
map[string]any{
|
|
||||||
commandKey: privmsgCmd,
|
|
||||||
toKey: "#nohashcash",
|
|
||||||
bodyKey: []string{"free message"},
|
|
||||||
},
|
|
||||||
)
|
|
||||||
if status != http.StatusOK {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected 200, got %d: %v", status, result,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
if result["id"] == nil || result["id"] == "" {
|
|
||||||
t.Fatal("expected message id")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashInvalidBitsRange(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcbits_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hcbits",
|
|
||||||
})
|
|
||||||
|
|
||||||
_, lastID := tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Try to set bits to 0 — should fail.
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcbits",
|
|
||||||
bodyKey: []string{"+H", "0"},
|
|
||||||
})
|
|
||||||
|
|
||||||
msgs, _ := tserver.pollMessages(token, lastID)
|
|
||||||
|
|
||||||
if !findNumeric(msgs, "472") {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected ERR_UNKNOWNMODE (472), got %v",
|
|
||||||
msgs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelHashcashMissingBitsArg(t *testing.T) {
|
|
||||||
tserver := newTestServer(t)
|
|
||||||
token := tserver.createSession("hcnoarg_user")
|
|
||||||
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: joinCmd, toKey: "#hcnoarg",
|
|
||||||
})
|
|
||||||
|
|
||||||
_, lastID := tserver.pollMessages(token, 0)
|
|
||||||
|
|
||||||
// Try to set +H without bits argument.
|
|
||||||
tserver.sendCommand(token, map[string]any{
|
|
||||||
commandKey: modeCmd,
|
|
||||||
toKey: "#hcnoarg",
|
|
||||||
bodyKey: []string{"+H"},
|
|
||||||
})
|
|
||||||
|
|
||||||
msgs, _ := tserver.pollMessages(token, lastID)
|
|
||||||
|
|
||||||
if !findNumeric(msgs, "461") {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected ERR_NEEDMOREPARAMS (461), got %v",
|
|
||||||
msgs,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -2,6 +2,7 @@ package handlers
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
|
"net"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
@@ -10,6 +11,33 @@ import (
|
|||||||
|
|
||||||
const minPasswordLength = 8
|
const minPasswordLength = 8
|
||||||
|
|
||||||
|
// clientIP extracts the client IP address from the request.
|
||||||
|
// It checks X-Forwarded-For and X-Real-IP headers before
|
||||||
|
// falling back to RemoteAddr.
|
||||||
|
func clientIP(request *http.Request) string {
|
||||||
|
if forwarded := request.Header.Get("X-Forwarded-For"); forwarded != "" {
|
||||||
|
// X-Forwarded-For may contain a comma-separated list;
|
||||||
|
// the first entry is the original client.
|
||||||
|
parts := strings.SplitN(forwarded, ",", 2) //nolint:mnd // split into two parts
|
||||||
|
ip := strings.TrimSpace(parts[0])
|
||||||
|
|
||||||
|
if ip != "" {
|
||||||
|
return ip
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if realIP := request.Header.Get("X-Real-IP"); realIP != "" {
|
||||||
|
return strings.TrimSpace(realIP)
|
||||||
|
}
|
||||||
|
|
||||||
|
host, _, err := net.SplitHostPort(request.RemoteAddr)
|
||||||
|
if err != nil {
|
||||||
|
return request.RemoteAddr
|
||||||
|
}
|
||||||
|
|
||||||
|
return host
|
||||||
|
}
|
||||||
|
|
||||||
// HandleRegister creates a new user with a password.
|
// HandleRegister creates a new user with a password.
|
||||||
func (hdlr *Handlers) HandleRegister() http.HandlerFunc {
|
func (hdlr *Handlers) HandleRegister() http.HandlerFunc {
|
||||||
return func(
|
return func(
|
||||||
@@ -137,6 +165,21 @@ func (hdlr *Handlers) handleLogin(
|
|||||||
writer http.ResponseWriter,
|
writer http.ResponseWriter,
|
||||||
request *http.Request,
|
request *http.Request,
|
||||||
) {
|
) {
|
||||||
|
ip := clientIP(request)
|
||||||
|
|
||||||
|
if !hdlr.loginLimiter.Allow(ip) {
|
||||||
|
writer.Header().Set(
|
||||||
|
"Retry-After", "1",
|
||||||
|
)
|
||||||
|
hdlr.respondError(
|
||||||
|
writer, request,
|
||||||
|
"too many login attempts, try again later",
|
||||||
|
http.StatusTooManyRequests,
|
||||||
|
)
|
||||||
|
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
type loginRequest struct {
|
type loginRequest struct {
|
||||||
Nick string `json:"nick"`
|
Nick string `json:"nick"`
|
||||||
Password string `json:"password"`
|
Password string `json:"password"`
|
||||||
|
|||||||
@@ -16,6 +16,7 @@ import (
|
|||||||
"git.eeqj.de/sneak/neoirc/internal/hashcash"
|
"git.eeqj.de/sneak/neoirc/internal/hashcash"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/healthcheck"
|
"git.eeqj.de/sneak/neoirc/internal/healthcheck"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/logger"
|
"git.eeqj.de/sneak/neoirc/internal/logger"
|
||||||
|
"git.eeqj.de/sneak/neoirc/internal/ratelimit"
|
||||||
"git.eeqj.de/sneak/neoirc/internal/stats"
|
"git.eeqj.de/sneak/neoirc/internal/stats"
|
||||||
"go.uber.org/fx"
|
"go.uber.org/fx"
|
||||||
)
|
)
|
||||||
@@ -36,11 +37,6 @@ type Params struct {
|
|||||||
|
|
||||||
const defaultIdleTimeout = 30 * 24 * time.Hour
|
const defaultIdleTimeout = 30 * 24 * time.Hour
|
||||||
|
|
||||||
// spentHashcashTTL is how long spent hashcash tokens are
|
|
||||||
// retained for replay prevention. Per issue requirements,
|
|
||||||
// this is 1 year.
|
|
||||||
const spentHashcashTTL = 365 * 24 * time.Hour
|
|
||||||
|
|
||||||
// Handlers manages HTTP request handling.
|
// Handlers manages HTTP request handling.
|
||||||
type Handlers struct {
|
type Handlers struct {
|
||||||
params *Params
|
params *Params
|
||||||
@@ -48,7 +44,7 @@ type Handlers struct {
|
|||||||
hc *healthcheck.Healthcheck
|
hc *healthcheck.Healthcheck
|
||||||
broker *broker.Broker
|
broker *broker.Broker
|
||||||
hashcashVal *hashcash.Validator
|
hashcashVal *hashcash.Validator
|
||||||
channelHashcash *hashcash.ChannelValidator
|
loginLimiter *ratelimit.Limiter
|
||||||
stats *stats.Tracker
|
stats *stats.Tracker
|
||||||
cancelCleanup context.CancelFunc
|
cancelCleanup context.CancelFunc
|
||||||
}
|
}
|
||||||
@@ -63,13 +59,23 @@ func New(
|
|||||||
resource = "neoirc"
|
resource = "neoirc"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
loginRate := params.Config.LoginRateLimit
|
||||||
|
if loginRate <= 0 {
|
||||||
|
loginRate = ratelimit.DefaultRate
|
||||||
|
}
|
||||||
|
|
||||||
|
loginBurst := params.Config.LoginRateBurst
|
||||||
|
if loginBurst <= 0 {
|
||||||
|
loginBurst = ratelimit.DefaultBurst
|
||||||
|
}
|
||||||
|
|
||||||
hdlr := &Handlers{ //nolint:exhaustruct // cancelCleanup set in startCleanup
|
hdlr := &Handlers{ //nolint:exhaustruct // cancelCleanup set in startCleanup
|
||||||
params: ¶ms,
|
params: ¶ms,
|
||||||
log: params.Logger.Get(),
|
log: params.Logger.Get(),
|
||||||
hc: params.Healthcheck,
|
hc: params.Healthcheck,
|
||||||
broker: broker.New(),
|
broker: broker.New(),
|
||||||
hashcashVal: hashcash.NewValidator(resource),
|
hashcashVal: hashcash.NewValidator(resource),
|
||||||
channelHashcash: hashcash.NewChannelValidator(),
|
loginLimiter: ratelimit.New(loginRate, loginBurst),
|
||||||
stats: params.Stats,
|
stats: params.Stats,
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -162,6 +168,10 @@ func (hdlr *Handlers) stopCleanup() {
|
|||||||
if hdlr.cancelCleanup != nil {
|
if hdlr.cancelCleanup != nil {
|
||||||
hdlr.cancelCleanup()
|
hdlr.cancelCleanup()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if hdlr.loginLimiter != nil {
|
||||||
|
hdlr.loginLimiter.Stop()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (hdlr *Handlers) cleanupLoop(ctx context.Context) {
|
func (hdlr *Handlers) cleanupLoop(ctx context.Context) {
|
||||||
@@ -292,20 +302,4 @@ func (hdlr *Handlers) pruneQueuesAndMessages(
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Prune spent hashcash tokens older than 1 year.
|
|
||||||
hashcashCutoff := time.Now().Add(-spentHashcashTTL)
|
|
||||||
|
|
||||||
pruned, err := hdlr.params.Database.
|
|
||||||
PruneSpentHashcash(ctx, hashcashCutoff)
|
|
||||||
if err != nil {
|
|
||||||
hdlr.log.Error(
|
|
||||||
"spent hashcash pruning failed", "error", err,
|
|
||||||
)
|
|
||||||
} else if pruned > 0 {
|
|
||||||
hdlr.log.Info(
|
|
||||||
"pruned spent hashcash tokens",
|
|
||||||
"deleted", pruned,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,186 +0,0 @@
|
|||||||
package hashcash
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/sha256"
|
|
||||||
"encoding/hex"
|
|
||||||
"errors"
|
|
||||||
"fmt"
|
|
||||||
"strconv"
|
|
||||||
"strings"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
errBodyHashMismatch = errors.New(
|
|
||||||
"body hash mismatch",
|
|
||||||
)
|
|
||||||
errBodyHashMissing = errors.New(
|
|
||||||
"body hash missing",
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
// ChannelValidator checks hashcash stamps for
|
|
||||||
// per-channel PRIVMSG validation. It verifies that
|
|
||||||
// stamps are bound to a specific channel and message
|
|
||||||
// body. Replay prevention is handled externally via
|
|
||||||
// the database spent_hashcash table for persistence
|
|
||||||
// across server restarts (1-year TTL).
|
|
||||||
type ChannelValidator struct{}
|
|
||||||
|
|
||||||
// NewChannelValidator creates a ChannelValidator.
|
|
||||||
func NewChannelValidator() *ChannelValidator {
|
|
||||||
return &ChannelValidator{}
|
|
||||||
}
|
|
||||||
|
|
||||||
// BodyHash computes the hex-encoded SHA-256 hash of a
|
|
||||||
// message body for use in hashcash stamp validation.
|
|
||||||
func BodyHash(body []byte) string {
|
|
||||||
hash := sha256.Sum256(body)
|
|
||||||
|
|
||||||
return hex.EncodeToString(hash[:])
|
|
||||||
}
|
|
||||||
|
|
||||||
// ValidateStamp checks a channel hashcash stamp. It
|
|
||||||
// verifies the stamp format, difficulty, date, channel
|
|
||||||
// binding, body hash binding, and proof-of-work. Replay
|
|
||||||
// detection is NOT performed here — callers must check
|
|
||||||
// the spent_hashcash table separately.
|
|
||||||
//
|
|
||||||
// Stamp format: 1:bits:YYMMDD:channel:bodyhash:counter.
|
|
||||||
func (cv *ChannelValidator) ValidateStamp(
|
|
||||||
stamp string,
|
|
||||||
requiredBits int,
|
|
||||||
channel string,
|
|
||||||
bodyHash string,
|
|
||||||
) error {
|
|
||||||
if requiredBits <= 0 {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
parts := strings.Split(stamp, ":")
|
|
||||||
if len(parts) != stampFields {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"%w: expected %d, got %d",
|
|
||||||
errInvalidFields, stampFields, len(parts),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
version := parts[0]
|
|
||||||
bitsStr := parts[1]
|
|
||||||
dateStr := parts[2]
|
|
||||||
resource := parts[3]
|
|
||||||
stampBodyHash := parts[4]
|
|
||||||
|
|
||||||
headerErr := validateChannelHeader(
|
|
||||||
version, bitsStr, resource,
|
|
||||||
requiredBits, channel,
|
|
||||||
)
|
|
||||||
if headerErr != nil {
|
|
||||||
return headerErr
|
|
||||||
}
|
|
||||||
|
|
||||||
stampTime, parseErr := parseStampDate(dateStr)
|
|
||||||
if parseErr != nil {
|
|
||||||
return parseErr
|
|
||||||
}
|
|
||||||
|
|
||||||
timeErr := validateTime(stampTime)
|
|
||||||
if timeErr != nil {
|
|
||||||
return timeErr
|
|
||||||
}
|
|
||||||
|
|
||||||
bodyErr := validateBodyHash(
|
|
||||||
stampBodyHash, bodyHash,
|
|
||||||
)
|
|
||||||
if bodyErr != nil {
|
|
||||||
return bodyErr
|
|
||||||
}
|
|
||||||
|
|
||||||
return validateProof(stamp, requiredBits)
|
|
||||||
}
|
|
||||||
|
|
||||||
// StampHash returns a deterministic hash of a stamp
|
|
||||||
// string for use as a spent-token key.
|
|
||||||
func StampHash(stamp string) string {
|
|
||||||
hash := sha256.Sum256([]byte(stamp))
|
|
||||||
|
|
||||||
return hex.EncodeToString(hash[:])
|
|
||||||
}
|
|
||||||
|
|
||||||
func validateChannelHeader(
|
|
||||||
version, bitsStr, resource string,
|
|
||||||
requiredBits int,
|
|
||||||
channel string,
|
|
||||||
) error {
|
|
||||||
if version != stampVersion {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"%w: %s", errBadVersion, version,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
claimedBits, err := strconv.Atoi(bitsStr)
|
|
||||||
if err != nil || claimedBits < requiredBits {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"%w: need %d bits",
|
|
||||||
errInsufficientBits, requiredBits,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
if resource != channel {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"%w: got %q, want %q",
|
|
||||||
errWrongResource, resource, channel,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func validateBodyHash(
|
|
||||||
stampBodyHash, expectedBodyHash string,
|
|
||||||
) error {
|
|
||||||
if stampBodyHash == "" {
|
|
||||||
return errBodyHashMissing
|
|
||||||
}
|
|
||||||
|
|
||||||
if stampBodyHash != expectedBodyHash {
|
|
||||||
return fmt.Errorf(
|
|
||||||
"%w: got %q, want %q",
|
|
||||||
errBodyHashMismatch,
|
|
||||||
stampBodyHash, expectedBodyHash,
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// MintChannelStamp computes a channel hashcash stamp
|
|
||||||
// with the given difficulty, channel name, and body hash.
|
|
||||||
// This is intended for clients to generate stamps before
|
|
||||||
// sending PRIVMSG to hashcash-protected channels.
|
|
||||||
//
|
|
||||||
// Stamp format: 1:bits:YYMMDD:channel:bodyhash:counter.
|
|
||||||
func MintChannelStamp(
|
|
||||||
bits int,
|
|
||||||
channel string,
|
|
||||||
bodyHash string,
|
|
||||||
) string {
|
|
||||||
date := time.Now().UTC().Format(dateFormatShort)
|
|
||||||
prefix := fmt.Sprintf(
|
|
||||||
"1:%d:%s:%s:%s:",
|
|
||||||
bits, date, channel, bodyHash,
|
|
||||||
)
|
|
||||||
|
|
||||||
counter := uint64(0)
|
|
||||||
|
|
||||||
for {
|
|
||||||
stamp := prefix + strconv.FormatUint(counter, 16)
|
|
||||||
hash := sha256.Sum256([]byte(stamp))
|
|
||||||
|
|
||||||
if hasLeadingZeroBits(hash[:], bits) {
|
|
||||||
return stamp
|
|
||||||
}
|
|
||||||
|
|
||||||
counter++
|
|
||||||
}
|
|
||||||
}
|
|
||||||
@@ -1,244 +0,0 @@
|
|||||||
package hashcash_test
|
|
||||||
|
|
||||||
import (
|
|
||||||
"crypto/sha256"
|
|
||||||
"encoding/hex"
|
|
||||||
"testing"
|
|
||||||
|
|
||||||
"git.eeqj.de/sneak/neoirc/internal/hashcash"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
testChannel = "#general"
|
|
||||||
testBodyText = `["hello world"]`
|
|
||||||
)
|
|
||||||
|
|
||||||
func testBodyHash() string {
|
|
||||||
hash := sha256.Sum256([]byte(testBodyText))
|
|
||||||
|
|
||||||
return hex.EncodeToString(hash[:])
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateHappyPath(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
|
|
||||||
stamp := hashcash.MintChannelStamp(
|
|
||||||
testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("valid channel stamp rejected: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateWrongChannel(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
|
|
||||||
stamp := hashcash.MintChannelStamp(
|
|
||||||
testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, testBits, "#other", bodyHash,
|
|
||||||
)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected channel mismatch error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateWrongBodyHash(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
|
|
||||||
stamp := hashcash.MintChannelStamp(
|
|
||||||
testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
|
|
||||||
wrongHash := sha256.Sum256([]byte("different body"))
|
|
||||||
wrongBodyHash := hex.EncodeToString(wrongHash[:])
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, testBits, testChannel, wrongBodyHash,
|
|
||||||
)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected body hash mismatch error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateInsufficientBits(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
|
|
||||||
// Mint with 2 bits but require 4.
|
|
||||||
stamp := hashcash.MintChannelStamp(
|
|
||||||
testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, 4, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected insufficient bits error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateZeroBitsSkips(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
"garbage", 0, "#ch", "abc",
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("zero bits should skip: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateBadFormat(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
"not:valid", testBits, testChannel, "abc",
|
|
||||||
)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected bad format error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateBadVersion(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
|
|
||||||
stamp := "2:2:260317:#general:" + bodyHash + ":counter"
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected bad version error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateExpiredStamp(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
|
|
||||||
// Mint with a very old date by manually constructing.
|
|
||||||
stamp := mintStampWithDate(
|
|
||||||
t, testBits, testChannel, "200101",
|
|
||||||
)
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected expired stamp error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestChannelValidateMissingBodyHash(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
|
|
||||||
// Construct a stamp with empty body hash field.
|
|
||||||
stamp := mintStampWithDate(
|
|
||||||
t, testBits, testChannel, todayDate(),
|
|
||||||
)
|
|
||||||
|
|
||||||
// This uses the session-style stamp which has empty
|
|
||||||
// ext field — body hash is missing.
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
if err == nil {
|
|
||||||
t.Fatal("expected missing body hash error")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestBodyHash(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
body := []byte(`["hello world"]`)
|
|
||||||
bodyHash := hashcash.BodyHash(body)
|
|
||||||
|
|
||||||
if len(bodyHash) != 64 {
|
|
||||||
t.Fatalf(
|
|
||||||
"expected 64-char hex hash, got %d",
|
|
||||||
len(bodyHash),
|
|
||||||
)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Same input should produce same hash.
|
|
||||||
bodyHash2 := hashcash.BodyHash(body)
|
|
||||||
if bodyHash != bodyHash2 {
|
|
||||||
t.Fatal("body hash not deterministic")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Different input should produce different hash.
|
|
||||||
bodyHash3 := hashcash.BodyHash([]byte("different"))
|
|
||||||
if bodyHash == bodyHash3 {
|
|
||||||
t.Fatal("different inputs produced same hash")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestStampHash(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
hash1 := hashcash.StampHash("stamp1")
|
|
||||||
hash2 := hashcash.StampHash("stamp2")
|
|
||||||
|
|
||||||
if hash1 == hash2 {
|
|
||||||
t.Fatal("different stamps produced same hash")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Same input should be deterministic.
|
|
||||||
hash1b := hashcash.StampHash("stamp1")
|
|
||||||
if hash1 != hash1b {
|
|
||||||
t.Fatal("stamp hash not deterministic")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func TestMintChannelStamp(t *testing.T) {
|
|
||||||
t.Parallel()
|
|
||||||
|
|
||||||
bodyHash := testBodyHash()
|
|
||||||
stamp := hashcash.MintChannelStamp(
|
|
||||||
testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
|
|
||||||
if stamp == "" {
|
|
||||||
t.Fatal("expected non-empty stamp")
|
|
||||||
}
|
|
||||||
|
|
||||||
// Validate the minted stamp.
|
|
||||||
validator := hashcash.NewChannelValidator()
|
|
||||||
|
|
||||||
err := validator.ValidateStamp(
|
|
||||||
stamp, testBits, testChannel, bodyHash,
|
|
||||||
)
|
|
||||||
if err != nil {
|
|
||||||
t.Fatalf("minted stamp failed validation: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
122
internal/ratelimit/ratelimit.go
Normal file
122
internal/ratelimit/ratelimit.go
Normal file
@@ -0,0 +1,122 @@
|
|||||||
|
// Package ratelimit provides per-IP rate limiting for HTTP endpoints.
|
||||||
|
package ratelimit
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"golang.org/x/time/rate"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// DefaultRate is the default number of allowed requests per second.
|
||||||
|
DefaultRate = 1.0
|
||||||
|
|
||||||
|
// DefaultBurst is the default maximum burst size.
|
||||||
|
DefaultBurst = 5
|
||||||
|
|
||||||
|
// DefaultSweepInterval controls how often stale entries are pruned.
|
||||||
|
DefaultSweepInterval = 10 * time.Minute
|
||||||
|
|
||||||
|
// DefaultEntryTTL is how long an unused entry lives before eviction.
|
||||||
|
DefaultEntryTTL = 15 * time.Minute
|
||||||
|
)
|
||||||
|
|
||||||
|
// entry tracks a per-IP rate limiter and when it was last used.
|
||||||
|
type entry struct {
|
||||||
|
limiter *rate.Limiter
|
||||||
|
lastSeen time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
// Limiter manages per-key rate limiters with automatic cleanup
|
||||||
|
// of stale entries.
|
||||||
|
type Limiter struct {
|
||||||
|
mu sync.Mutex
|
||||||
|
entries map[string]*entry
|
||||||
|
rate rate.Limit
|
||||||
|
burst int
|
||||||
|
entryTTL time.Duration
|
||||||
|
stopCh chan struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
// New creates a new per-key rate Limiter.
|
||||||
|
// The ratePerSec parameter sets how many requests per second are
|
||||||
|
// allowed per key. The burst parameter sets the maximum number of
|
||||||
|
// requests that can be made in a single burst.
|
||||||
|
func New(ratePerSec float64, burst int) *Limiter {
|
||||||
|
limiter := &Limiter{
|
||||||
|
mu: sync.Mutex{},
|
||||||
|
entries: make(map[string]*entry),
|
||||||
|
rate: rate.Limit(ratePerSec),
|
||||||
|
burst: burst,
|
||||||
|
entryTTL: DefaultEntryTTL,
|
||||||
|
stopCh: make(chan struct{}),
|
||||||
|
}
|
||||||
|
|
||||||
|
go limiter.sweepLoop()
|
||||||
|
|
||||||
|
return limiter
|
||||||
|
}
|
||||||
|
|
||||||
|
// Allow reports whether a request from the given key should be
|
||||||
|
// allowed. It consumes one token from the key's rate limiter.
|
||||||
|
func (l *Limiter) Allow(key string) bool {
|
||||||
|
l.mu.Lock()
|
||||||
|
ent, exists := l.entries[key]
|
||||||
|
|
||||||
|
if !exists {
|
||||||
|
ent = &entry{
|
||||||
|
limiter: rate.NewLimiter(l.rate, l.burst),
|
||||||
|
lastSeen: time.Now(),
|
||||||
|
}
|
||||||
|
l.entries[key] = ent
|
||||||
|
} else {
|
||||||
|
ent.lastSeen = time.Now()
|
||||||
|
}
|
||||||
|
l.mu.Unlock()
|
||||||
|
|
||||||
|
return ent.limiter.Allow()
|
||||||
|
}
|
||||||
|
|
||||||
|
// Stop terminates the background sweep goroutine.
|
||||||
|
func (l *Limiter) Stop() {
|
||||||
|
close(l.stopCh)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Len returns the number of tracked keys (for testing).
|
||||||
|
func (l *Limiter) Len() int {
|
||||||
|
l.mu.Lock()
|
||||||
|
defer l.mu.Unlock()
|
||||||
|
|
||||||
|
return len(l.entries)
|
||||||
|
}
|
||||||
|
|
||||||
|
// sweepLoop periodically removes entries that haven't been seen
|
||||||
|
// within the TTL.
|
||||||
|
func (l *Limiter) sweepLoop() {
|
||||||
|
ticker := time.NewTicker(DefaultSweepInterval)
|
||||||
|
defer ticker.Stop()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case <-ticker.C:
|
||||||
|
l.sweep()
|
||||||
|
case <-l.stopCh:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// sweep removes stale entries.
|
||||||
|
func (l *Limiter) sweep() {
|
||||||
|
l.mu.Lock()
|
||||||
|
defer l.mu.Unlock()
|
||||||
|
|
||||||
|
cutoff := time.Now().Add(-l.entryTTL)
|
||||||
|
|
||||||
|
for key, ent := range l.entries {
|
||||||
|
if ent.lastSeen.Before(cutoff) {
|
||||||
|
delete(l.entries, key)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
106
internal/ratelimit/ratelimit_test.go
Normal file
106
internal/ratelimit/ratelimit_test.go
Normal file
@@ -0,0 +1,106 @@
|
|||||||
|
package ratelimit_test
|
||||||
|
|
||||||
|
import (
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"git.eeqj.de/sneak/neoirc/internal/ratelimit"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestNewCreatesLimiter(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
limiter := ratelimit.New(1.0, 5)
|
||||||
|
defer limiter.Stop()
|
||||||
|
|
||||||
|
if limiter == nil {
|
||||||
|
t.Fatal("expected non-nil limiter")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAllowWithinBurst(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
limiter := ratelimit.New(1.0, 3)
|
||||||
|
defer limiter.Stop()
|
||||||
|
|
||||||
|
for i := range 3 {
|
||||||
|
if !limiter.Allow("192.168.1.1") {
|
||||||
|
t.Fatalf(
|
||||||
|
"request %d should be allowed within burst",
|
||||||
|
i+1,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAllowExceedsBurst(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
// Rate of 0 means no token replenishment, only burst.
|
||||||
|
limiter := ratelimit.New(0, 3)
|
||||||
|
defer limiter.Stop()
|
||||||
|
|
||||||
|
for range 3 {
|
||||||
|
limiter.Allow("10.0.0.1")
|
||||||
|
}
|
||||||
|
|
||||||
|
if limiter.Allow("10.0.0.1") {
|
||||||
|
t.Fatal("fourth request should be denied after burst exhausted")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAllowSeparateKeys(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
// Rate of 0, burst of 1 — only one request per key.
|
||||||
|
limiter := ratelimit.New(0, 1)
|
||||||
|
defer limiter.Stop()
|
||||||
|
|
||||||
|
if !limiter.Allow("10.0.0.1") {
|
||||||
|
t.Fatal("first request for key A should be allowed")
|
||||||
|
}
|
||||||
|
|
||||||
|
if !limiter.Allow("10.0.0.2") {
|
||||||
|
t.Fatal("first request for key B should be allowed")
|
||||||
|
}
|
||||||
|
|
||||||
|
if limiter.Allow("10.0.0.1") {
|
||||||
|
t.Fatal("second request for key A should be denied")
|
||||||
|
}
|
||||||
|
|
||||||
|
if limiter.Allow("10.0.0.2") {
|
||||||
|
t.Fatal("second request for key B should be denied")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestLenTracksKeys(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
limiter := ratelimit.New(1.0, 5)
|
||||||
|
defer limiter.Stop()
|
||||||
|
|
||||||
|
if limiter.Len() != 0 {
|
||||||
|
t.Fatalf("expected 0 entries, got %d", limiter.Len())
|
||||||
|
}
|
||||||
|
|
||||||
|
limiter.Allow("10.0.0.1")
|
||||||
|
limiter.Allow("10.0.0.2")
|
||||||
|
|
||||||
|
if limiter.Len() != 2 {
|
||||||
|
t.Fatalf("expected 2 entries, got %d", limiter.Len())
|
||||||
|
}
|
||||||
|
|
||||||
|
// Same key again should not increase count.
|
||||||
|
limiter.Allow("10.0.0.1")
|
||||||
|
|
||||||
|
if limiter.Len() != 2 {
|
||||||
|
t.Fatalf("expected 2 entries, got %d", limiter.Len())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestStopDoesNotPanic(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
limiter := ratelimit.New(1.0, 5)
|
||||||
|
limiter.Stop()
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user