73 Commits

Author SHA1 Message Date
kpcyrd
6cd97d980d Release v0.24.2 2022-03-29 02:38:09 +02:00
kpcyrd
b4f2591378 Merge pull request #225 from kpcyrd/bump
Update dependencies
2022-03-29 00:27:34 +00:00
kpcyrd
3a204a92a5 Add docker release github action 2022-03-29 01:16:52 +02:00
kpcyrd
b65c72d090 Update Dockerfile to alpine 3.15 and buildkit 2022-03-28 20:40:15 +02:00
kpcyrd
035ef9aa76 Add missing seccomp syscall (rseq) 2022-03-28 19:25:22 +02:00
kpcyrd
35c6501623 Update dependencies 2022-03-28 18:44:32 +02:00
kpcyrd
c0cb5840cc Merge pull request #224 from ysf/patch-1
Fixed typo
2022-03-21 17:57:30 +00:00
ysf
bba152d560 Fixed typo 2022-03-21 09:18:52 +01:00
kpcyrd
53a39d54bc Merge pull request #223 from definitepotato/issue-222
Adding syscall to seccomp sandbox.
2022-01-19 11:19:33 +00:00
definitepotato
455403baaf Adding syscall to seccomp sandbox. 2022-01-15 13:14:16 -05:00
kpcyrd
ec07344a0b Merge pull request #221 from kpcyrd/clippy
Fix clippy warnings
2022-01-14 15:56:14 +00:00
kpcyrd
c2d95659dc Fix clippy warnings 2022-01-13 20:05:34 +01:00
kpcyrd
cd99ac3911 Release v0.24.1 2021-12-05 16:17:51 +01:00
kpcyrd
f2a5dbc60c Merge pull request #219 from kpcyrd/seccomp
seccomp: Allow fstat call
2021-12-05 12:05:12 +00:00
kpcyrd
6425483c2b seccomp: Allow fstat call 2021-12-05 12:37:20 +01:00
kpcyrd
25f940788f Release v0.24.0 2021-12-05 01:15:41 +01:00
kpcyrd
c232b23b1e Merge pull request #217 from kpcyrd/rescope
Add rescope command to run autonoscope rules on existing db
2021-12-04 23:40:51 +00:00
kpcyrd
5b19c8c4b3 Add basic completions for interactive shell 2021-12-03 18:38:36 +01:00
kpcyrd
aeba3f4574 Allow statx and lseek syscalls 2021-12-03 18:38:36 +01:00
kpcyrd
1f5ea402ea Support rescoping with filters 2021-12-03 14:27:06 +01:00
kpcyrd
d54ffd1eba Add advanced interactive mode for rescope 2021-12-03 13:49:03 +01:00
kpcyrd
5bca5677b6 Add rescope command to run autonoscope rules on existing db 2021-12-03 04:11:53 +01:00
kpcyrd
af021cb6be Allow invoking sn0int autonoscope and sn0int autoscope from cli 2021-12-03 01:05:25 +01:00
kpcyrd
52891cf6b1 Fix compile warning about dead code 2021-12-03 00:52:29 +01:00
kpcyrd
6e76c035df Bump dependencies 2021-12-03 00:43:41 +01:00
kpcyrd
850d628a93 Release v0.23.0 2021-10-23 20:02:24 +02:00
kpcyrd
5bb03d39bc Merge pull request #215 from kpcyrd/user-agent
Allow setting a different default user agent
2021-10-23 12:23:24 +00:00
kpcyrd
f5660570d2 Update dependencies 2021-10-23 14:03:12 +02:00
kpcyrd
f1b0608daa Allow setting a different default user agent 2021-10-23 13:50:32 +02:00
kpcyrd
c775ae1dee Add patreon link 2021-10-23 13:15:50 +02:00
kpcyrd
36b5219008 Update readme 2021-10-02 15:03:25 +02:00
kpcyrd
79982fd5f0 Merge pull request #214 from kpcyrd/img-hash
Add perceptual image hashing
2021-09-22 01:58:21 +00:00
kpcyrd
5fe71feec3 Fix regression for sn0int run -f ./foo.lua 2021-09-22 01:21:03 +02:00
kpcyrd
b8d4eb0f51 Add functions for perceptual image hashing 2021-09-22 01:19:59 +02:00
kpcyrd
e6444db009 Merge pull request #213 from kpcyrd/cli-proxy
Allow setting a proxy with `run -X <proxy>`
2021-09-13 10:19:38 +00:00
kpcyrd
7dcb6b81dc Allow setting a proxy with run -X <proxy> 2021-09-13 11:51:32 +02:00
kpcyrd
4fb56d91c2 Allow using run <module> interactively 2021-09-13 11:29:13 +02:00
kpcyrd
8c486b7e6e Release v0.22.0 2021-09-08 18:44:54 +02:00
kpcyrd
e939a28469 Merge pull request #210 from kpcyrd/bump
Update dependencies
2021-09-08 12:12:28 +00:00
kpcyrd
2dec9dd28c Update dependencies 2021-09-08 13:50:57 +02:00
kpcyrd
06459098b1 Merge pull request #209 from kpcyrd/update
Update dependencies, fix clippy warnings
2021-09-06 15:07:53 +00:00
kpcyrd
fc0447725c Fix clippy warnings 2021-09-06 16:26:45 +02:00
kpcyrd
8312695e46 Update dependencies 2021-09-06 15:44:45 +02:00
kpcyrd
6c8e2df632 Update to nom7 2021-08-21 13:31:09 +02:00
kpcyrd
b1ee2d4ce7 Merge pull request #208 from kpcyrd/update
Update dependencies
2021-08-17 20:48:13 +00:00
kpcyrd
4cbb72e4c8 Update dependencies 2021-08-17 21:22:48 +02:00
kpcyrd
9e76935f55 Merge pull request #207 from kpcyrd/readline-exec
Support invoking shell commands with `!` from readline
2021-08-01 17:00:10 +00:00
kpcyrd
8b5ad635fa Support invoking shell commands with ! from readline 2021-07-29 14:15:00 +02:00
kpcyrd
9f4914f419 Release v0.21.2 2021-06-23 23:12:19 +02:00
kpcyrd
f24b1b2b9b Update dockerfile baseimage 2021-06-23 23:07:37 +02:00
kpcyrd
735e2dbaf8 Merge pull request #205 from kpcyrd/update
Update dependencies, fix seccomp issue on x86_64 musl
2021-06-23 21:03:19 +00:00
kpcyrd
5409ed8cdb Fix integration tests 2021-06-23 17:33:09 +02:00
kpcyrd
d0a36be95a Update dependencies 2021-06-23 17:00:11 +02:00
kpcyrd
0ab8aace70 Fix seccomp segfault with open on x86_64 musl 2021-06-23 17:00:11 +02:00
kpcyrd
699ebad23a registry: Set user agent when authenticating with github 2021-06-23 16:22:31 +02:00
kpcyrd
84e0e62753 Release v0.21.1 2021-05-18 00:49:10 +02:00
kpcyrd
dc8aac1b0f Merge pull request #202 from kpcyrd/aarch64
Update dependencies to fix aarch64 on musl
2021-05-17 22:46:55 +00:00
kpcyrd
7b4d599951 Update dependencies to fix aarch64 on musl 2021-05-18 00:30:39 +02:00
kpcyrd
d88c722a4b Release v0.21.0 2021-05-12 19:19:30 +02:00
kpcyrd
d70fc8f4c0 Merge pull request #200 from kpcyrd/update
Update dependencies
2021-05-12 16:54:46 +00:00
kpcyrd
da08f3a4bb Update dependencies 2021-05-12 18:27:44 +02:00
kpcyrd
892fa0d3e4 Merge pull request #199 from kpcyrd/stats
Improve stats command
2021-04-07 01:03:21 +00:00
kpcyrd
605d691b28 Add select --values as shorthand for jq -r .value 2021-04-07 02:34:41 +02:00
kpcyrd
efb458a725 Allow deleting multiple workspaces at once 2021-04-07 02:25:58 +02:00
kpcyrd
84e147f709 Add stats -q for more compact output 2021-04-07 02:11:29 +02:00
kpcyrd
1876e9ac8d Add stats --all to show stats for all workspaces 2021-04-07 01:51:57 +02:00
kpcyrd
8525883b91 README: Add packaging status 2021-03-23 01:29:53 +00:00
kpcyrd
02c537c253 Merge pull request #197 from kpcyrd/stats
Add sn0int stats
2021-03-20 13:38:12 +00:00
kpcyrd
d84038dcc3 Add irc notifications for github actions 2021-03-18 03:31:07 +01:00
kpcyrd
becb44a5d7 Update dependencies 2021-03-18 02:38:47 +01:00
kpcyrd
6e1904eaf9 Extend interactive completions for pkg list 2021-03-14 19:38:44 +01:00
kpcyrd
d493857011 Add stats --json output 2021-03-14 18:40:47 +01:00
kpcyrd
3af9abde1b Add stats command 2021-03-14 01:34:42 +01:00
99 changed files with 2778 additions and 1566 deletions

1
.github/FUNDING.yml vendored
View File

@@ -1 +1,2 @@
github: [kpcyrd]
patreon: kpcyrd

58
.github/workflows/docker-release.yml vendored Normal file
View File

@@ -0,0 +1,58 @@
name: Publish Docker image
on:
release:
types: [ published ]
jobs:
push_to_registry:
name: Push Docker image to GitHub Registry
runs-on: ubuntu-latest
steps:
-
name: Checkout
uses: actions/checkout@v2
-
name: Docker meta
id: meta
uses: docker/metadata-action@v3
with:
images: |
ghcr.io/kpcyrd/sn0int
tags: |
type=semver,pattern={{raw}}
-
name: Set up Docker Buildx
uses: docker/setup-buildx-action@v1
-
name: Cache Docker layers
uses: actions/cache@v2
with:
path: /tmp/.buildx-cache
key: ${{ runner.os }}-buildx-${{ github.sha }}
restore-keys: |
${{ runner.os }}-buildx-
-
name: Login to Registry
uses: docker/login-action@v1
with:
registry: ghcr.io
username: ${{ github.repository_owner }}
password: ${{ secrets.GITHUB_TOKEN }}
-
name: Build and push Docker images
uses: docker/build-push-action@v2
with:
push: true
tags: ${{ steps.meta.outputs.tags }}
file: Dockerfile
cache-from: type=local,src=/tmp/.buildx-cache
cache-to: type=local,dest=/tmp/.buildx-cache-new
-
# Temp fix
# https://github.com/docker/build-push-action/issues/252
# https://github.com/moby/buildkit/issues/1896
name: Move cache
run: |
rm -rf /tmp/.buildx-cache
mv /tmp/.buildx-cache-new /tmp/.buildx-cache

View File

@@ -2,9 +2,9 @@ name: Docker
on:
push:
branches: [ master ]
branches: [ main ]
pull_request:
branches: [ master ]
branches: [ main ]
jobs:
build:
@@ -22,7 +22,7 @@ jobs:
- uses: actions/checkout@v2
- name: Build the Docker image
run: docker build -t ${{ matrix.build.name }} -f ${{ matrix.build.file }} .
run: DOCKER_BUILDKIT=1 docker build -t ${{ matrix.build.name }} -f ${{ matrix.build.file }} .
- name: Test the Docker image
run: docker run --rm ${{ matrix.build.name }} --help

View File

@@ -2,9 +2,11 @@ name: Rust
on:
push:
branches: [ master ]
branches: [ main ]
pull_request:
branches: [ master ]
branches: [ main ]
schedule:
- cron: '0 9 * * 1'
env:
CARGO_TERM_COLOR: always
@@ -49,3 +51,19 @@ jobs:
run: cd sn0int-std && cargo test --verbose
- name: Run tests (std, --ignored)
run: cd sn0int-std && cargo test --verbose -- --ignored
notify:
# forks shouldn't notify
if: github.repository == 'kpcyrd/sn0int'
runs-on: ubuntu-latest
needs:
- build
steps:
- name: irc notify
uses: rectalogic/notify-irc@v1
with:
server: irc.hackint.org
channel: "#sn0int"
nickname: github-ci
message: '${{ github.repository }}#${{ github.run_id }}(${{ github.event_name }}): ${{ github.ref }}: tests completed: ${{ needs.build.result }} (https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }})'

View File

@@ -1,61 +0,0 @@
language: rust
# upload takes too long on windows and gets killed due to inactivity
#cache: cargo
matrix:
include:
- os: linux
rust: stable
env:
- BUILD_MODE=test
- os: linux
rust: stable
env:
- BUILD_MODE=common
- os: linux
rust: stable
env:
- BUILD_MODE=std
- os: linux
rust: stable
env:
- BUILD_MODE=boxxy
- os: osx
rust: stable
env:
- BUILD_MODE=test
- os: osx
rust: stable
env:
- BUILD_MODE=common
- os: osx
rust: stable
env:
- BUILD_MODE=std
#- os: windows
# rust: stable
# env:
# - BUILD_MODE="windows test"
#- os: windows
# rust: stable
# env:
# - BUILD_MODE="windows common"
before_install:
- ci/setup.sh "$TRAVIS_OS_NAME"
script:
- df -h
- ci/run.sh $BUILD_MODE
- df -h
notifications:
irc:
template:
- "%{repository_slug}#%{build_number} (%{branch} - %{commit} : %{author}): %{message} (%{build_url})"
channels:
- "ircs://irc.hackint.org:6697/#sn0int"
use_notice: true
skip_join: true
#on_success: change # default: always
#on_failure: always # default: always

2205
Cargo.lock generated

File diff suppressed because it is too large Load Diff

View File

@@ -1,6 +1,6 @@
[package]
name = "sn0int"
version = "0.20.1"
version = "0.24.2"
description = "Semi-automatic OSINT framework and package manager"
authors = ["kpcyrd <git@rxv.cc>"]
license = "GPL-3.0"
@@ -9,9 +9,6 @@ categories = ["command-line-utilities"]
readme = "README.md"
edition = "2018"
[badges]
travis-ci = { repository = "kpcyrd/sn0int" }
[workspace]
members = ["sn0int-common",
"sn0int-registry",
@@ -27,7 +24,6 @@ the results in a unified format for followup investigations."""
section = "utils"
priority = "optional"
depends = "$auto, publicsuffix"
build-depends = "pkg-config, libsqlite3-dev, libseccomp-dev, libsodium-dev, publicsuffix"
assets = [
["target/release/sn0int", "usr/bin/", "755"],
]
@@ -36,11 +32,11 @@ assets = [
sqlite-bundled = ["libsqlite3-sys/bundled"]
[dependencies]
sn0int-common = { version="0.12.0", path="sn0int-common" }
sn0int-std = { version="=0.20.1", path="sn0int-std" }
rustyline = "7.0"
sn0int-common = { version="0.13.0", path="sn0int-common" }
sn0int-std = { version="=0.24.2", path="sn0int-std" }
rustyline = "9.0"
log = "0.4"
env_logger = "0.8"
env_logger = "0.9"
hlua-badtouch = "0.4"
structopt = "0.3"
failure = "0.1"
@@ -50,7 +46,7 @@ lazy_static = "1.0"
shellwords = "1.0"
diesel = { version = "1.0.0", features = ["sqlite", "chrono"] }
diesel_migrations = { version = "1.3.0", features = ["sqlite"] }
libsqlite3-sys = { version = "0.18.0", features = ["bundled-windows"] }
libsqlite3-sys = { version = "0.22.0", features = ["bundled-windows"] }
chrono = { version = "0.4", features = ["serde"] }
dirs-next = "2.0"
url = "2.0"
@@ -64,7 +60,7 @@ serde_urlencoded = "0.7"
serde_json = "1.0"
crossbeam-channel = "0.5"
ctrlc = "3.1"
opener = "0.4"
opener = "0.5"
separator = "0.4"
maplit = "1.0.1"
sloppy-rfc4880 = "0.2"
@@ -72,37 +68,38 @@ regex = "1.0"
toml = "0.5"
threadpool = "1.7"
atty = "0.2"
semver = "0.11"
semver = "1"
bytes = "0.4"
bytesize = "1.0"
ipnetwork = "0.17"
strum = "0.20"
strum_macros = "0.20"
ipnetwork = "0.18"
strum = "0.24"
strum_macros = "0.24"
embedded-triple = "0.1.0"
humansize = "1.1.0"
digest = "0.9"
md-5 = "0.9"
sha-1 = "0.9"
sha2 = "0.9"
sha3 = "0.9"
hmac = "0.10"
digest = "0.10"
md-5 = "0.10"
sha-1 = "0.10"
sha2 = "0.10"
sha3 = "0.10"
hmac = "0.12"
walkdir = "2.2"
nude = "0.3"
glob = "0.3.0"
os-version = "0.1.1"
os-version = "0.2"
[target.'cfg(target_os="linux")'.dependencies]
caps = "0.5"
#syscallz = { path="../syscallz-rs" }
syscallz = "0.15"
nix = "0.19"
syscallz = "0.16"
nix = "0.23"
[target.'cfg(target_os="openbsd")'.dependencies]
pledge = "0.4"
unveil = "0.2.0"
unveil = "0.3"
[dev-dependencies]
#boxxy = { path = "../boxxy-rs" }
boxxy = "0.11"
boxxy = "0.12"
tempfile = "3.0"

View File

@@ -1,14 +1,18 @@
FROM rust:alpine3.11
FROM rust:alpine3.15
ENV RUSTFLAGS="-C target-feature=-crt-static"
RUN apk add --no-cache musl-dev sqlite-dev libseccomp-dev libsodium-dev
WORKDIR /usr/src/sn0int
COPY . .
RUN cargo build --release --verbose
RUN strip target/release/sn0int
RUN --mount=type=cache,target=/var/cache/buildkit \
CARGO_HOME=/var/cache/buildkit/cargo \
CARGO_TARGET_DIR=/var/cache/buildkit/target \
cargo build --release --locked --verbose && \
cp -v /var/cache/buildkit/target/release/sn0int /
RUN strip /sn0int
FROM alpine:3.11
FROM alpine:3.15
RUN apk add --no-cache libgcc sqlite-libs libseccomp libsodium
COPY --from=0 /usr/src/sn0int/target/release/sn0int /usr/local/bin/sn0int
COPY --from=0 /sn0int /usr/local/bin/sn0int
VOLUME ["/data", "/cache"]
ENV XDG_DATA_HOME=/data \
XDG_CACHE_HOME=/cache

View File

@@ -1,7 +1,5 @@
# sn0int [![Build Status][travis-img]][travis] [![crates.io][crates-img]][crates] [![Documentation Status][docs-img]][docs] [![irc.hackint.org:6697/#sn0int][irc-img]][irc] [![@sn0int][twitter-img]][twitter] [![@sn0int@chaos.social][mastodon-img]][mastodon] [![registry status][registry-img]][registry]
# sn0int [![crates.io][crates-img]][crates] [![Documentation Status][docs-img]][docs] [![irc.hackint.org:6697/#sn0int][irc-img]][irc] [![@sn0int][twitter-img]][twitter] [![@sn0int@chaos.social][mastodon-img]][mastodon] [![registry status][registry-img]][registry]
[travis-img]: https://travis-ci.org/kpcyrd/sn0int.svg?branch=master
[travis]: https://travis-ci.org/kpcyrd/sn0int
[crates-img]: https://img.shields.io/crates/v/sn0int.svg
[crates]: https://crates.io/crates/sn0int
[docs-img]: https://readthedocs.org/projects/sn0int/badge/?version=latest
@@ -16,33 +14,34 @@
[registry]: https://sn0int.com/
sn0int (pronounced [`/snoɪnt/`][ipa]) is a semi-automatic OSINT framework and
package manager. It was built for IT security professionals and bug hunters to
gather intelligence about a given target or about yourself. sn0int is
enumerating attack surface by semi-automatically processing public information
and mapping the results in a unified format for followup investigations.
package manager. It's used by IT security professionals, bug bounty hunters,
law enforcement agencies and in security awareness trainings to gather
intelligence about a given target or about yourself. sn0int is enumerating
attack surface by semi-automatically processing public information and mapping
the results in a unified format for followup investigations.
[ipa]: http://ipa-reader.xyz/?text=sno%C9%AAnt
Among other things, sn0int is currently able to:
- Harvest subdomains from certificate transparency logs and passive dns
- Mass resolve collected subdomains and scan for http or https services
- Enrich ip addresses with asn and geoip info
- Harvest emails from pgp keyservers and whois
- Discover compromised logins in breaches
- Find somebody's profiles across the internet
- Enumerate local networks with unique techniques like passive arp
- Gather information about phonenumbers
- Attempt to bypass cloudflare with shodan
- Harvest data and images from instagram profiles
- Scan images for nudity
- Harvest activity and images from social media profiles
- Basic image processing
sn0int is heavily inspired by recon-ng and maltego, but remains more flexible
and is fully opensource. None of the investigations listed above are hardcoded
in the source, instead those are provided by modules that are executed in a
in the source, instead they are provided by modules that are executed in a
sandbox. You can easily extend sn0int by writing your own modules and share
them with other users by publishing them to the sn0int registry. This allows
you to ship updates for your modules on your own since you don't need to send a
pull request.
you to ship updates for your modules on your own instead of pull-requesting
them into the sn0int codebase.
For questions and support join us on IRC: [irc.hackint.org:6697/#sn0int](https://webirc.hackint.org/#irc://irc.hackint.org/#sn0int)
@@ -50,6 +49,8 @@ For questions and support join us on IRC: [irc.hackint.org:6697/#sn0int](https:/
## Installation
<a href="https://repology.org/project/sn0int/versions"><img align="right" src="https://repology.org/badge/vertical-allrepos/sn0int.svg" alt="Packaging status"></a>
Archlinux
pacman -S sn0int
@@ -226,6 +227,9 @@ For everything else please have a look at the [detailed list][1].
- [http_fetch_json](https://sn0int.readthedocs.io/en/latest/reference.html#http-fetch-json)
- [img_load](https://sn0int.readthedocs.io/en/latest/reference.html#img-load)
- [img_exif](https://sn0int.readthedocs.io/en/latest/reference.html#img-exif)
- [img_ahash](https://sn0int.readthedocs.io/en/latest/reference.html#img-ahash)
- [img_dhash](https://sn0int.readthedocs.io/en/latest/reference.html#img-dhash)
- [img_phash](https://sn0int.readthedocs.io/en/latest/reference.html#img-phash)
- [img_nudity](https://sn0int.readthedocs.io/en/latest/reference.html#img-nudity)
- [info](https://sn0int.readthedocs.io/en/latest/reference.html#info)
- [intval](https://sn0int.readthedocs.io/en/latest/reference.html#intval)

View File

@@ -1,6 +1,8 @@
#!/usr/bin/env python3
import subprocess
from subprocess import DEVNULL, PIPE
import shutil
from pathlib import Path
import tempfile
import json
import sys
@@ -98,6 +100,11 @@ def main(tempdir, binary):
('https://www.example.com/', 200),
}
cache = Path.home() / '.cache' / 'sn0int'
if cache.exists():
print('[*] copying geoip files')
shutil.copytree(cache, tempdir + '/.cache/sn0int', dirs_exist_ok=True)
print('[*] running geoip')
sn0int(tempdir, binary, [
'use geoip',

View File

@@ -1,32 +0,0 @@
#!/bin/sh
set -exu
case "$1" in
build)
cargo build --verbose
cargo build --verbose --examples
;;
test)
cargo test --verbose
cargo test --verbose -- --ignored
;;
common)
cd sn0int-common
cargo test --verbose
;;
std)
cd sn0int-std
cargo test --verbose
cargo test --verbose -- --ignored
;;
windows)
cargo build --verbose
cargo build --verbose --examples
;;
boxxy)
cargo build --verbose --examples
if cat ci/boxxy_stage1.txt | RUST_LOG=boxxy cargo run --example boxxy; then
echo "SANDOX ERROR: should've crashed"
exit 1
fi
;;
esac

View File

@@ -1,11 +0,0 @@
#!/bin/sh
set -exu
case "$1" in
linux)
sudo apt update
sudo apt install libsqlite3-dev libseccomp-dev libsodium-dev
;;
osx)
brew install libsodium
;;
esac

View File

@@ -44,6 +44,13 @@ Alpine
$ apk add sqlite-dev libseccomp-dev libsodium-dev
Docker
~~~~~~
.. code-block:: bash
$ DOCKER_BUILDKIT=1 docker build -t kpcyrd/sn0int .
OpenBSD
~~~~~~~

View File

@@ -504,6 +504,39 @@ Extract exif metadata from an image.
if last_err() then return end
debug(exif)
img_ahash
---------
Calculate the Mean (aHash) perceptual hash.
.. code-block:: lua
hash = img_ahash(blob)
if last_err() then return end
debug(hash)
img_dhash
---------
Calculate the Gradient (dHash) perceptual hash.
.. code-block:: lua
hash = img_dhash(blob)
if last_err() then return end
debug(hash)
img_phash
---------
Calculate the DCT (pHash) perceptual hash.
.. code-block:: lua
hash = img_phash(blob)
if last_err() then return end
debug(hash)
img_nudity
----------

View File

@@ -0,0 +1,11 @@
-- Description: TODO your description here
-- Version: 0.1.0
-- License: GPL-3.0
-- Source: images
function run(arg)
debug(arg)
info(img_ahash(arg['value']))
info(img_dhash(arg['value']))
info(img_phash(arg['value']))
end

View File

@@ -3,7 +3,7 @@
-- License: GPL-3.0
function run()
-- echo 'www.example.com' | sn0int run -vvf --stdin modules/harness/import-subdomains.lu
-- echo 'www.example.com' | sn0int run -vvf --stdin modules/harness/import-subdomains.lua
while true do
local line = stdin_readline()

View File

@@ -1,6 +1,6 @@
[package]
name = "sn0int-common"
version = "0.12.0"
version = "0.13.0"
description = "sn0int - common code"
authors = ["kpcyrd <git@rxv.cc>"]
license = "GPL-3.0"
@@ -12,4 +12,4 @@ serde = { version = "1.0", features=["derive"] }
#rocket_failure_errors = { path = "../../../rocket_failure/rocket_failure_errors" }
rocket_failure_errors = "0.2"
anyhow = "1.0"
nom = "6.0"
nom = "7.0"

View File

@@ -1,5 +1,3 @@
#[macro_use] extern crate nom;
pub mod api;
pub mod errors;
pub use crate::errors::*;

View File

@@ -1,4 +1,8 @@
use crate::errors::*;
use nom::IResult;
use nom::bytes::complete::{tag, take_until};
use nom::combinator::map_res;
use nom::multi::fold_many0;
use serde::{Serialize, Deserialize};
use std::str::FromStr;
@@ -223,23 +227,32 @@ impl<'a> NewMetadata<'a> {
}
}
named!(metaline<&str, (EntryType, &str)>, do_parse!(
tag!("-- ") >>
name: map_res!(take_until!(": "), EntryType::from_str) >>
tag!(": ") >>
value: take_until!("\n") >>
tag!("\n") >>
(name, value)
));
fn metaline(input: &str) -> IResult<&str, (EntryType, &str)> {
let (input, _) = tag("-- ")(input)?;
let (input, name) = map_res(
take_until(": "),
EntryType::from_str
)(input)?;
let (input, _) = tag(": ")(input)?;
let (input, value) = take_until("\n")(input)?;
let (input, _) = tag("\n")(input)?;
named!(metalines<&str, Vec<(EntryType, &str)>>, do_parse!(
lines: fold_many0!(metaline, Vec::new(), |mut acc: Vec<_>, item| {
acc.push(item);
acc
}) >>
tag!("\n") >>
(lines)
));
Ok((input, (name, value)))
}
fn metalines(input: &str) -> IResult<&str, Vec<(EntryType, &str)>> {
let (input, lines) = fold_many0(
metaline,
Vec::new,
|mut acc: Vec<_>, item| {
acc.push(item);
acc
}
)(input)?;
let (input, _) = tag("\n")(input)?;
Ok((input, lines))
}
#[cfg(test)]
mod tests {

View File

@@ -8,7 +8,7 @@ repository = "https://github.com/kpcyrd/sn0int"
edition = "2018"
[dependencies]
sn0int-common = { version="0.12.0", path="../sn0int-common" }
sn0int-common = { version="0.13.0", path="../sn0int-common" }
rocket = { version = "0.4", default-features=false }
#rocket_failure = { path = "../../rocket_failure" }
rocket_failure = { version = "0.2" }
@@ -23,7 +23,7 @@ oauth2 = "2.0.0"
failure = "0.1"
url = "1.0"
log = "0.4"
semver = "0.9.0"
semver = "1"
lazy_static = "1"
blake2 = "0.8.0"
hex = "0.4"
@@ -33,5 +33,5 @@ serde_json = "1.0"
syntect = "3.3"
dotenv = "0.15"
env_logger = "0.7"
env_logger = "0.9"
structopt = "0.3.21"

View File

@@ -5,6 +5,7 @@ pub fn get_username(oauth_token: &str) -> Result<String> {
let client = reqwest::Client::new();
let mut resp = client.get("https://api.github.com/user")
.header("Authorization", format!("token {}", oauth_token))
.header("User-Agent", "sn0int-registry")
.send()
.context("Failed to check access_token")?
.error_for_status()

View File

@@ -1,6 +1,6 @@
[package]
name = "sn0int-std"
version = "0.20.1"
version = "0.24.2"
description = "sn0int - stdlib"
authors = ["kpcyrd <git@rxv.cc>"]
repository = "https://github.com/kpcyrd/sn0int"
@@ -21,35 +21,36 @@ serde = { version = "1.0", features = ["derive"] }
serde_json = "1.0"
serde_urlencoded = "0.7"
webpki = "0.21"
webpki-roots = "0.20"
webpki-roots = "0.21"
ct-logs = "0.7"
#chrootable-https = { path = "../../chrootable-https" }
chrootable-https = "0.16"
http = "0.2"
bufstream = "0.1.4"
pem = "0.8"
pem = "1"
url = "2.0"
tungstenite = { version = "0.11", default-features = false }
tungstenite = { version = "0.13", default-features = false }
kuchiki = "0.8.0"
maxminddb = "0.17"
x509-parser = "0.9"
der-parser = "5"
publicsuffix = { version="1.5", default-features=false }
maxminddb = "0.22"
x509-parser = "0.13"
der-parser = "7"
publicsuffix = { version="2", default-features=false }
xml-rs = "0.8"
geo = "0.16"
geo = "0.19"
bytes = "0.4"
base64 = "0.13"
chrono = { version = "0.4", features = ["serde"] }
mqtt-protocol = "0.10"
mqtt-protocol = "0.11"
sodiumoxide = { version="0.2.5", features=["use-pkg-config"] }
image = "0.23.0"
image = "0.23"
kamadak-exif = "0.5.1"
img_hash_median = "4.0.0"
bs58 = "0.4"
digest = "0.9"
blake2 = "0.9"
digest = "0.10"
blake2 = "0.10"
[dev-dependencies]
env_logger = "0.8"
env_logger = "0.9"
maplit = "1.0.1"

View File

@@ -1,5 +1,5 @@
use bytes::Bytes;
use blake2::VarBlake2b;
use blake2::Blake2bVar;
use digest::{Update, VariableOutput};
use serde::ser::{Serialize, Serializer};
use serde::de::{self, Deserialize, Deserializer};
@@ -21,7 +21,7 @@ impl Blob {
}
pub fn hash(bytes: &[u8]) -> String {
let mut h = VarBlake2b::new(32).unwrap();
let mut h = Blake2bVar::new(32).unwrap();
h.update(bytes);
let output = h.finalize_boxed();
Self::encode_hash(&output)

View File

@@ -5,6 +5,7 @@ use std::net::IpAddr;
use x509_parser::x509::X509Version;
use x509_parser::certificate::X509Certificate;
use x509_parser::extensions::{GeneralName, ParsedExtension};
use x509_parser::prelude::*;
#[derive(Debug, PartialEq, Serialize, Deserialize)]
pub struct Certificate {
@@ -31,7 +32,7 @@ impl Certificate {
}
pub fn from_bytes(crt: &[u8]) -> Result<Certificate> {
let crt = match X509Certificate::from_der(&crt) {
let crt = match X509Certificate::from_der(crt) {
Ok((remaining, der)) => {
if !remaining.is_empty() {
bail!("input cert has trailing garbage");
@@ -55,7 +56,7 @@ impl Certificate {
}
}
for (_oid, ext) in crt.tbs_certificate.extensions {
for ext in crt.tbs_certificate.extensions() {
if let ParsedExtension::SubjectAlternativeName(san) = ext.parsed_extension() {
for name in &san.general_names {
debug!("Certificate is valid for {:?}", name);

View File

@@ -22,7 +22,7 @@ pub fn sodium_secretbox_open(encrypted: &[u8], key: &[u8]) -> Result<Vec<u8>> {
let nonce = Nonce::from_slice(&encrypted[..secretbox::NONCEBYTES])
.ok_or_else(|| format_err!("Nonce has wrong length"))?;
let ciphertext = &encrypted[secretbox::NONCEBYTES..];
let plain = secretbox::open(&ciphertext, &nonce, &key)
let plain = secretbox::open(ciphertext, &nonce, &key)
.map_err(|_| format_err!("Failed to decrypt secretbox"))?;
Ok(plain)
}

View File

@@ -86,16 +86,16 @@ impl From<Vec<(AnyLuaValue, AnyLuaValue)>> for LuaMap {
}
}
impl Into<HashMap<AnyHashableLuaValue, AnyLuaValue>> for LuaMap {
fn into(self: LuaMap) -> HashMap<AnyHashableLuaValue, AnyLuaValue> {
self.0
impl From<LuaMap> for HashMap<AnyHashableLuaValue, AnyLuaValue> {
fn from(map: LuaMap) -> Self {
map.0
}
}
impl Into<AnyLuaValue> for LuaMap {
fn into(self: LuaMap) -> AnyLuaValue {
impl From<LuaMap> for AnyLuaValue {
fn from(map: LuaMap) -> AnyLuaValue {
AnyLuaValue::LuaArray(
self.into_iter()
map.into_iter()
.filter_map(|(k, v)| {
match k {
AnyHashableLuaValue::LuaString(x) => Some((AnyLuaValue::LuaString(x), v)),
@@ -136,9 +136,9 @@ impl LuaList {
}
}
impl Into<AnyLuaValue> for LuaList {
fn into(self: LuaList) -> AnyLuaValue {
AnyLuaValue::LuaArray(self.0)
impl From<LuaList> for AnyLuaValue {
fn from(list: LuaList) -> AnyLuaValue {
AnyLuaValue::LuaArray(list.0)
}
}
@@ -149,7 +149,7 @@ pub fn byte_array(bytes: AnyLuaValue) -> Result<Vec<u8>> {
AnyLuaValue::LuaArray(bytes) => {
Ok(bytes.into_iter()
.map(|num| match num.1 {
AnyLuaValue::LuaNumber(num) if num <= 255.0 && num >= 0.0 && (num % 1.0 == 0.0) =>
AnyLuaValue::LuaNumber(num) if (0.0..=255.0).contains(&num) && (num % 1.0 == 0.0) =>
Ok(num as u8),
AnyLuaValue::LuaNumber(num) =>
Err(format_err!("number is out of range: {:?}", num)),

View File

@@ -72,7 +72,7 @@ pub struct Continent {
}
impl Continent {
fn from_maxmind(continent: geoip2::model::Continent) -> Option<Self> {
fn from_maxmind(continent: geoip2::city::Continent) -> Option<Self> {
let code = continent.code?;
let name = from_geoip_model_names(continent.names)?;
@@ -90,7 +90,7 @@ pub struct Country {
}
impl Country {
fn from_maxmind(country: geoip2::model::Country) -> Option<Self> {
fn from_maxmind(country: geoip2::city::Country) -> Option<Self> {
let code = country.iso_code?;
let name = from_geoip_model_names(country.names)?;
@@ -108,7 +108,7 @@ pub struct Location {
}
impl Location {
fn from_maxmind(location: &geoip2::model::Location) -> Option<Self> {
fn from_maxmind(location: &geoip2::city::Location) -> Option<Self> {
let latitude = match location.latitude {
Some(latitude) => latitude,
_ => return None,

View File

@@ -1,5 +1,6 @@
use crate::errors::*;
use image::{self, DynamicImage, GenericImageView};
pub use img_hash_median::HashAlg;
pub mod exif;
@@ -45,9 +46,9 @@ impl ImageFormat {
}
}
impl Into<image::ImageFormat> for ImageFormat {
fn into(self) -> image::ImageFormat {
match self {
impl From<ImageFormat> for image::ImageFormat {
fn from(format: ImageFormat) -> image::ImageFormat {
match format {
ImageFormat::Png => image::ImageFormat::Png,
ImageFormat::Jpeg => image::ImageFormat::Jpeg,
ImageFormat::Gif => image::ImageFormat::Gif,
@@ -79,6 +80,14 @@ impl Image {
pub fn height(&self) -> u32 {
self.image.height()
}
pub fn perception_hash(&self, hash_alg: HashAlg) -> String {
let hasher = img_hash_median::HasherConfig::new()
.hash_alg(hash_alg)
.to_hasher();
let hash = hasher.hash_image(&self.image);
hash.to_base64()
}
}
impl AsRef<DynamicImage> for Image {
@@ -94,10 +103,10 @@ pub fn guess_format(buf: &[u8]) -> Result<ImageFormat> {
}
pub fn load(buf: &[u8]) -> Result<Image> {
let img_format = image::guess_format(&buf)?;
let img_format = image::guess_format(buf)?;
let format = ImageFormat::try_from(&img_format)?;
let image = image::load_from_memory_with_format(&buf, img_format)?;
let image = image::load_from_memory_with_format(buf, img_format)?;
Ok(Image {
image,
@@ -105,7 +114,6 @@ pub fn load(buf: &[u8]) -> Result<Image> {
})
}
#[cfg(test)]
mod tests {
use super::*;

View File

@@ -11,13 +11,13 @@ pub struct Element {
pub html: String,
}
impl Into<AnyLuaValue> for Element {
fn into(self) -> AnyLuaValue {
impl From<Element> for AnyLuaValue {
fn from(elem: Element) -> AnyLuaValue {
let mut map = LuaMap::new();
map.insert_str("text", self.text);
map.insert("attrs", LuaMap::from(self.attrs));
map.insert_str("html", self.html);
map.insert_str("text", elem.text);
map.insert("attrs", LuaMap::from(elem.attrs));
map.insert_str("html", elem.html);
map.into()
}

View File

@@ -1,6 +1,5 @@
use crate::errors::*;
use std::iter::FromIterator;
use std::collections::HashMap;
use crate::hlua::AnyLuaValue;
use serde_json::{self, Deserializer, Value, Number, Map};
@@ -22,7 +21,7 @@ pub fn encode(v: AnyLuaValue) -> Result<String> {
}
pub fn decode_stream(x: &str) -> Result<Vec<AnyLuaValue>> {
let stream = Deserializer::from_str(&x).into_iter::<Value>();
let stream = Deserializer::from_str(x).into_iter::<Value>();
let list = stream
.filter_map(|x| x.ok())
@@ -52,9 +51,9 @@ pub enum LuaJsonValue {
Object(HashMap<String, LuaJsonValue>),
}
impl Into<AnyLuaValue> for LuaJsonValue {
fn into(self) -> AnyLuaValue {
match self {
impl From<LuaJsonValue> for AnyLuaValue {
fn from(value: LuaJsonValue) -> AnyLuaValue {
match value {
LuaJsonValue::Null => AnyLuaValue::LuaNil,
LuaJsonValue::Bool(v) => AnyLuaValue::LuaBoolean(v),
// TODO: not sure if this might fail
@@ -111,9 +110,9 @@ impl From<AnyLuaValue> for LuaJsonValue {
}
}
impl Into<serde_json::Value> for LuaJsonValue {
fn into(self) -> serde_json::Value {
match self {
impl From<LuaJsonValue> for serde_json::Value {
fn from(value: LuaJsonValue) -> serde_json::Value {
match value {
LuaJsonValue::Null => Value::Null,
LuaJsonValue::Bool(v) => Value::Bool(v),
LuaJsonValue::Number(v) => Value::Number(v),
@@ -122,9 +121,10 @@ impl Into<serde_json::Value> for LuaJsonValue {
.map(|x| x.into())
.collect()
),
LuaJsonValue::Object(v) => Value::Object(Map::from_iter(v.into_iter()
LuaJsonValue::Object(v) => Value::Object(v.into_iter()
.map(|(k, v)| (k, v.into()))
)),
.collect::<Map<_, _>>()
),
}
}
}
@@ -140,9 +140,10 @@ impl From<serde_json::Value> for LuaJsonValue {
.map(|x| x.into())
.collect()
),
Value::Object(v) => LuaJsonValue::Object(HashMap::from_iter(v.into_iter()
Value::Object(v) => LuaJsonValue::Object(v.into_iter()
.map(|(k, v)| (k, v.into()))
)),
.collect::<HashMap<_, _>>()
),
}
}
}

View File

@@ -8,7 +8,7 @@ use mqtt::{TopicFilter, QualityOfService};
use mqtt::control::ConnectReturnCode;
use mqtt::control::fixed_header::FixedHeaderError;
use mqtt::encodable::{Encodable, Decodable};
use mqtt::packet::{Packet, VariablePacket, ConnectPacket, SubscribePacket, PingreqPacket};
use mqtt::packet::{VariablePacket, ConnectPacket, SubscribePacket, PingreqPacket};
use serde::{Serialize, Deserialize};
use std::convert::TryFrom;
use std::io;
@@ -177,7 +177,7 @@ impl TryFrom<VariablePacket> for Pkt {
VariablePacket::ConnackPacket(_) => bail!("Unsupported pkt: {:?}", pkt),
VariablePacket::PublishPacket(pkt) => Ok(Pkt::Publish(Publish {
topic: pkt.topic_name().to_string(),
body: pkt.payload(),
body: pkt.payload().to_vec(),
})),
VariablePacket::PubackPacket(_) => bail!("Unsupported pkt: {:?}", pkt),
VariablePacket::PubrecPacket(_) => bail!("Unsupported pkt: {:?}", pkt),

View File

@@ -1,11 +1,12 @@
use crate::errors::*;
use chrootable_https::Client;
use crate::errors::*;
use crate::lazy::LazyInit;
use publicsuffix::{List, Psl as _};
use std::fs::{self, File};
use std::io::Read;
use std::path::{Path, PathBuf};
use std::sync::Arc;
use crate::lazy::LazyInit;
use std::str::FromStr;
#[derive(Debug, PartialEq)]
pub struct DnsName {
@@ -72,11 +73,15 @@ impl PslReader {
impl LazyInit<Arc<Psl>> for PslReader {
fn initialize(self) -> Result<Arc<Psl>> {
let list = match self {
PslReader::Reader(file) => publicsuffix::List::from_reader(file),
PslReader::String(s) => publicsuffix::List::from_str(&s),
PslReader::Reader(mut file) => {
let mut buf = String::new();
file.read_to_string(&mut buf)?;
buf
},
PslReader::String(s) => s,
};
let list = list
let list = List::from_str(&list)
.map_err(|e| format_err!("Failed to load public suffix list: {}", e))?;
Ok(Arc::new(Psl {
@@ -87,32 +92,28 @@ impl LazyInit<Arc<Psl>> for PslReader {
#[derive(Debug)]
pub struct Psl {
list: publicsuffix::List,
list: List,
}
impl Psl {
pub fn parse_dns_name(&self, name: &str) -> Result<DnsName> {
let dns_name = self.list.parse_dns_name(name)
.map_err(|e| format_err!("Failed to parse dns_name: {}", e))?;
let bytes = name.as_bytes();
let (root, suffix) = if let Some(domain) = dns_name.domain() {
let root = domain.to_string();
if let Some(suffix) = domain.suffix() {
(root, suffix.to_string())
} else {
// XXX: not sure if this is reachable
(root.clone(), root)
}
let suffix = self.list.suffix(bytes)
.ok_or_else(|| format_err!("Failed to detect suffix"))?;
let suffix = String::from_utf8(suffix.as_bytes().to_vec())?;
let root = if let Some(root) = self.list.domain(bytes) {
String::from_utf8(root.as_bytes().to_vec())?
} else {
// this is technically a tld, but support eg. a.prod.fastly.net anyway
// XXX: consider forcing domain to be in self.list.private()
(name.to_string(), name.to_string())
name.to_string()
};
let fulldomain = if root.as_str() != name {
Some(name.to_string())
} else {
let fulldomain = if root == name {
None
} else {
Some(name.to_string())
};
Ok(DnsName {

View File

@@ -107,7 +107,7 @@ impl Stream {
let mut errors = Vec::new();
for addr in addrs {
match Stream::connect_addr(host, (addr, port).into(), &options) {
match Stream::connect_addr(host, (addr, port).into(), options) {
Ok(socket) => {
return Ok(socket);
},
@@ -233,7 +233,7 @@ impl Socket {
}
pub fn send(&mut self, data: &[u8]) -> Result<()> {
match str::from_utf8(&data) {
match str::from_utf8(data) {
Ok(data) => debug!("send: {:?}", data),
Err(_) => debug!("send: {:?}", data),
};

View File

@@ -47,9 +47,9 @@ pub fn wrap(stream: TcpStream, host: &str, options: &SocketOptions) -> Result<(S
}
let dns_name = if let Some(v) = &options.sni_value {
get_dns_name(&mut config, &v)
get_dns_name(&mut config, v)
} else {
get_dns_name(&mut config, &host)
get_dns_name(&mut config, host)
};
let config = Arc::new(config);
@@ -58,7 +58,7 @@ pub fn wrap(stream: TcpStream, host: &str, options: &SocketOptions) -> Result<(S
}
fn get_dns_name(config: &mut ClientConfig, host: &str) -> webpki::DNSName {
if let Ok(name) = webpki::DNSNameRef::try_from_ascii_str(&host) {
if let Ok(name) = webpki::DNSNameRef::try_from_ascii_str(host) {
debug!("setting sni value to: {:?}", host);
name.to_owned()
} else {

View File

@@ -65,21 +65,21 @@ impl HttpSession {
#[derive(Debug, Default, Deserialize)]
pub struct RequestOptions {
query: Option<HashMap<String, String>>,
headers: Option<HashMap<String, String>>,
basic_auth: Option<(String, String)>,
user_agent: Option<String>,
json: Option<serde_json::Value>,
form: Option<serde_json::Value>,
pub query: Option<HashMap<String, String>>,
pub headers: Option<HashMap<String, String>>,
pub basic_auth: Option<(String, String)>,
pub user_agent: Option<String>,
pub json: Option<serde_json::Value>,
pub form: Option<serde_json::Value>,
#[serde(default)]
follow_redirects: usize,
body: Option<String>,
timeout: Option<u64>,
pub follow_redirects: usize,
pub body: Option<String>,
pub timeout: Option<u64>,
#[serde(default)]
into_blob: bool,
proxy: Option<SocketAddr>,
pub into_blob: bool,
pub proxy: Option<SocketAddr>,
#[serde(default)]
binary: bool,
pub binary: bool,
}
impl RequestOptions {
@@ -110,11 +110,8 @@ pub struct HttpRequest {
}
impl HttpRequest {
pub fn new(session: &HttpSession, method: String, url: String, options: RequestOptions, default_agent: fn() -> String) -> HttpRequest {
pub fn new(session: &HttpSession, method: String, url: String, user_agent: String, options: RequestOptions) -> HttpRequest {
let cookies = session.cookies.clone();
let user_agent = options.user_agent.unwrap_or_else(default_agent);
let timeout = options.timeout.map(Duration::from_millis);
let mut request = HttpRequest {
@@ -216,7 +213,7 @@ impl HttpRequest {
if self.follow_redirects > 0 && res.status >= 300 && res.status < 400 {
if let Some(location) = res.headers.get("location") {
let base = Url::parse(&url.to_string())?;
let joined = base.join(&location)?;
let joined = base.join(location)?;
url = joined.to_string().parse()?;
req = self.mkrequest("GET", &url).body(Body::empty())?;
@@ -318,9 +315,9 @@ impl HttpRequest {
}
}
impl Into<AnyLuaValue> for HttpRequest {
fn into(self) -> AnyLuaValue {
let v = serde_json::to_value(&self).unwrap();
impl From<HttpRequest> for AnyLuaValue {
fn from(req: HttpRequest) -> AnyLuaValue {
let v = serde_json::to_value(&req).unwrap();
LuaJsonValue::from(v).into()
}
}

View File

@@ -91,9 +91,11 @@ impl WebSocket {
}
pub fn options(&self, options: &WebSocketOptions) -> Result<()> {
let mut o = SocketOptions::default();
o.read_timeout = options.read_timeout;
o.write_timeout = options.write_timeout;
let o = SocketOptions {
read_timeout: options.read_timeout,
write_timeout: options.write_timeout,
..Default::default()
};
o.apply(self.sock.get_ref())
}

View File

@@ -9,7 +9,7 @@ use sn0int_common::ModuleID;
#[structopt(global_settings = &[AppSettings::ColoredHelp])]
pub struct Args {
/// Select a different workspace instead of the default
#[structopt(short="w", long="workspace")]
#[structopt(short="w", long="workspace", env="SN0INT_WORKSPACE")]
pub workspace: Option<Workspace>,
#[structopt(subcommand)]
@@ -66,6 +66,12 @@ pub enum SubCommand {
/// Exclude entities from scope
#[structopt(name="noscope")]
Noscope(cmd::noscope_cmd::Args),
/// Manage autoscope rules
Autoscope(cmd::autoscope_cmd::Args),
/// Manage autonoscope rules
Autonoscope(cmd::autonoscope_cmd::Args),
/// Rescope all entities based on autonoscope rules
Rescope(cmd::rescope_cmd::Args),
/// Manage workspaces
#[structopt(name="workspace")]
Workspace(cmd::workspace_cmd::Args),
@@ -81,6 +87,9 @@ pub enum SubCommand {
/// Export a workspace for external processing
#[structopt(name="export")]
Export(cmd::export_cmd::Args),
/// Show statistics about your current workspace
#[structopt(name="stats")]
Stats(cmd::stats_cmd::Args),
/// Run a lua repl
#[structopt(name="repl")]
Repl,
@@ -94,18 +103,11 @@ pub enum SubCommand {
#[derive(Debug, StructOpt)]
pub struct Run {
/// Execute a module that has been installed
pub module: String,
#[structopt(flatten)]
pub run: cmd::run_cmd::Args,
/// Run a module from a path
#[structopt(short="f", long="file")]
pub file: bool,
/// Run modules concurrently
#[structopt(short="j", long="threads", default_value="1")]
pub threads: usize,
/// Verbose logging, once to print inserts even if they don't add new
/// data, twice to activate the debug() function
#[structopt(short="v", long="verbose", parse(from_occurrences))]
pub verbose: u64,
/// Expose stdin to modules
#[structopt(long="stdin")]
pub stdin: bool,
@@ -135,7 +137,7 @@ pub struct Run {
#[derive(Debug, StructOpt)]
pub struct Sandbox {
/// This value is only used for process listings
label: String,
_label: String,
}
#[derive(Debug, StructOpt)]

View File

@@ -1,5 +1,5 @@
use crate::errors::*;
use crate::autonoscope::{Autonoscope, IntoRule, AutoRule, RulePrecision};
use crate::autonoscope::{Autonoscope, ToRule, AutoRule, RulePrecision};
use crate::models::*;
use std::convert::TryFrom;
@@ -41,6 +41,13 @@ impl TryFrom<Autonoscope> for DomainRule {
}
}
impl AutoRule<Domain> for DomainRule {
#[inline]
fn matches(&self, domain: &Domain) -> Result<bool> {
self.matches(domain.value.as_str())
}
}
impl AutoRule<NewDomain> for DomainRule {
#[inline]
fn matches(&self, domain: &NewDomain) -> Result<bool> {
@@ -48,6 +55,13 @@ impl AutoRule<NewDomain> for DomainRule {
}
}
impl AutoRule<Subdomain> for DomainRule {
#[inline]
fn matches(&self, domain: &Subdomain) -> Result<bool> {
self.matches(domain.value.as_str())
}
}
impl AutoRule<NewSubdomain> for DomainRule {
#[inline]
fn matches(&self, domain: &NewSubdomain) -> Result<bool> {
@@ -55,6 +69,18 @@ impl AutoRule<NewSubdomain> for DomainRule {
}
}
impl AutoRule<Url> for DomainRule {
#[inline]
fn matches(&self, url: &Url) -> Result<bool> {
let url = url.value.parse::<url::Url>()?;
if let Some(domain) = url.domain() {
self.matches(domain)
} else {
Ok(false)
}
}
}
impl AutoRule<NewUrl> for DomainRule {
#[inline]
fn matches(&self, url: &NewUrl) -> Result<bool> {
@@ -94,8 +120,8 @@ impl RulePrecision for DomainRule {
}
}
impl IntoRule for DomainRule {
fn into_rule(&self) -> (&'static str, String) {
impl ToRule for DomainRule {
fn to_rule(&self) -> (&'static str, String) {
("domain", self.to_string())
}
}

View File

@@ -1,5 +1,5 @@
use crate::errors::*;
use crate::autonoscope::{Autonoscope, IntoRule, AutoRule, RulePrecision};
use crate::autonoscope::{Autonoscope, ToRule, AutoRule, RulePrecision};
use crate::models::*;
use std::convert::TryFrom;
use std::net;
@@ -36,12 +36,25 @@ impl TryFrom<Autonoscope> for IpRule {
}
}
impl AutoRule<IpAddr> for IpRule {
fn matches(&self, ipaddr: &IpAddr) -> Result<bool> {
self.matches(ipaddr.value.as_str())
}
}
impl AutoRule<NewIpAddr> for IpRule {
fn matches(&self, ipaddr: &NewIpAddr) -> Result<bool> {
self.matches(ipaddr.value.as_str())
}
}
impl AutoRule<Port> for IpRule {
fn matches(&self, port: &Port) -> Result<bool> {
let addr = port.value.parse::<net::SocketAddr>()?;
self.matches(&addr.ip())
}
}
impl AutoRule<NewPort> for IpRule {
fn matches(&self, port: &NewPort) -> Result<bool> {
let addr = port.value.parse::<net::SocketAddr>()?;
@@ -49,15 +62,25 @@ impl AutoRule<NewPort> for IpRule {
}
}
fn match_netblock_str(network: &IpNetwork, netblock: &str) -> Result<bool> {
let range = netblock.parse::<ipnetwork::IpNetwork>()?;
if network.prefix() <= range.prefix() {
Ok(network.contains(range.ip()))
} else {
Ok(false)
}
}
impl AutoRule<Netblock> for IpRule {
fn matches(&self, netblock: &Netblock) -> Result<bool> {
match_netblock_str(&self.network, &netblock.value)
}
}
impl AutoRule<NewNetblock> for IpRule {
fn matches(&self, netblock: &NewNetblock) -> Result<bool> {
let range = netblock.value.parse::<ipnetwork::IpNetwork>()?;
if self.network.prefix() <= range.prefix() {
Ok(self.network.contains(range.ip()))
} else {
Ok(false)
}
match_netblock_str(&self.network, &netblock.value)
}
}
@@ -80,8 +103,8 @@ impl RulePrecision for IpRule {
}
}
impl IntoRule for IpRule {
fn into_rule(&self) -> (&'static str, String) {
impl ToRule for IpRule {
fn to_rule(&self) -> (&'static str, String) {
("ip", self.to_string())
}
}

View File

@@ -47,6 +47,18 @@ fn sort_precision_desc<T: RulePrecision>(a: &T, b: &T) -> Ordering {
}
impl RuleSet {
pub fn domains(&self) -> &[Rule<DomainRule>] {
&self.domains
}
pub fn ips(&self) -> &[Rule<IpRule>] {
&self.ips
}
pub fn urls(&self) -> &[Rule<UrlRule>] {
&self.urls
}
pub fn load(db: &DatabaseSock) -> Result<Self> {
use crate::schema::autonoscope::dsl::*;
let rules = autonoscope.load::<Autonoscope>(db)?;
@@ -112,15 +124,15 @@ impl RuleSet {
match obj {
RuleType::Domain => {
self.domains.retain(|x| x.to_string().as_str() != rule);
self.db_delete(db, obj, &rule)?;
self.db_delete(db, obj, rule)?;
},
RuleType::Ip => {
self.ips.retain(|x| x.to_string().as_str() != rule);
self.db_delete(db, obj, &rule)?;
self.db_delete(db, obj, rule)?;
},
RuleType::Url => {
self.urls.retain(|x| x.to_string().as_str() != rule);
self.db_delete(db, obj, &rule)?;
self.db_delete(db, obj, rule)?;
},
}
Ok(())
@@ -143,10 +155,18 @@ impl RuleSet {
rules
}
pub fn is_empty(&self) -> bool {
self.domains.is_empty() && self.ips.is_empty() && self.urls.is_empty()
}
pub fn len(&self) -> usize {
self.domains.len() + self.ips.len() + self.urls.len()
}
#[inline]
fn push_rules_display<T: IntoRule>(output: &mut Vec<(&'static str, String, bool)>, rules: &[Rule<T>]) {
fn push_rules_display<T: ToRule>(output: &mut Vec<(&'static str, String, bool)>, rules: &[Rule<T>]) {
for rule in rules {
let (object, value) = rule.into_rule();
let (object, value) = rule.to_rule();
output.push((object, value, rule.scoped));
}
}
@@ -159,10 +179,8 @@ impl RuleSet {
Insert::Url(url) => {
if let Some(result) = Self::matches_any(&self.domains, url)? {
Some(result)
} else if let Some(result) = Self::matches_any(&self.urls, url)? {
Some(result)
} else {
None
Self::matches_any(&self.urls, url)?
}
},
// Insert::Email(email) => unimplemented!(),
@@ -176,7 +194,7 @@ impl RuleSet {
fn matches_any<T1, T2>(rules: &[Rule<T1>], object: &T2) -> Result<Option<bool>>
where T1: AutoRule<T2>,
T1: IntoRule,
T1: ToRule,
{
for rule in rules {
if rule.matches(object)? {
@@ -238,12 +256,12 @@ pub trait RulePrecision {
}
#[derive(Debug, PartialEq)]
pub struct Rule<T: IntoRule> {
pub struct Rule<T: ToRule> {
rule: T,
scoped: bool,
pub scoped: bool,
}
impl<T: IntoRule> Rule<T> {
impl<T: ToRule> Rule<T> {
pub fn new(rule: T, scoped: bool) -> Rule<T> {
Rule {
rule,
@@ -254,7 +272,7 @@ impl<T: IntoRule> Rule<T> {
// TODO: maybe drop this
use std::ops::Deref;
impl<T: IntoRule> Deref for Rule<T> {
impl<T: ToRule> Deref for Rule<T> {
type Target = T;
fn deref(&self) -> &Self::Target {
@@ -262,23 +280,23 @@ impl<T: IntoRule> Deref for Rule<T> {
}
}
impl<T: IntoRule + RulePrecision> RulePrecision for Rule<T> {
impl<T: ToRule + RulePrecision> RulePrecision for Rule<T> {
fn precision(&self) -> usize {
self.rule.precision()
}
}
pub trait IntoRule {
fn into_rule(&self) -> (&'static str, String);
pub trait ToRule {
fn to_rule(&self) -> (&'static str, String);
}
impl<T: IntoRule> Into<NewAutonoscope> for &Rule<T> {
fn into(self) -> NewAutonoscope {
let (object, value) = self.rule.into_rule();
impl<T: ToRule> From<&Rule<T>> for NewAutonoscope {
fn from(rule: &Rule<T>) -> NewAutonoscope {
let (object, value) = rule.rule.to_rule();
NewAutonoscope {
object: object.to_string(),
value,
scoped: self.scoped,
scoped: rule.scoped,
}
}
}

View File

@@ -1,5 +1,5 @@
use crate::errors::*;
use crate::autonoscope::{Autonoscope, IntoRule, AutoRule, RulePrecision};
use crate::autonoscope::{Autonoscope, ToRule, AutoRule, RulePrecision};
use crate::models::*;
use std::convert::TryFrom;
@@ -47,6 +47,12 @@ impl TryFrom<&str> for UrlRule {
}
// TODO: there is no way to write a rule that matches all urls
impl AutoRule<Url> for UrlRule {
fn matches(&self, url: &Url) -> Result<bool> {
self.matches(url.value.as_str())
}
}
impl AutoRule<NewUrl> for UrlRule {
fn matches(&self, url: &NewUrl) -> Result<bool> {
self.matches(url.value.as_str())
@@ -86,8 +92,8 @@ impl RulePrecision for UrlRule {
}
}
impl IntoRule for UrlRule {
fn into_rule(&self) -> (&'static str, String) {
impl ToRule for UrlRule {
fn to_rule(&self) -> (&'static str, String) {
("url", self.to_string())
}
}

View File

@@ -93,6 +93,14 @@ impl BlobStorage {
}
Ok(blobs)
}
pub fn stat(&self, id: &str) -> Result<u64> {
let path = self.join(id)?;
debug!("Stat-ing blob: {:?}", path);
let md = fs::metadata(path)
.context("Failed to stat blob")?;
Ok(md.len())
}
}

View File

@@ -112,7 +112,7 @@ pub struct DateContext {
impl DateContext {
pub fn new(events: &[Activity], today: NaiveDate) -> DateContext {
let (events, max) = setup_graph_map(&events);
let (events, max) = setup_graph_map(events);
DateContext {
events,
max,
@@ -296,25 +296,25 @@ mod tests {
DateContext {
events: HashMap::new(),
max: 0,
today: NaiveDate::from_ymd(2020, 05, 30),
today: NaiveDate::from_ymd(2020, 5, 30),
}
}
#[test]
fn test_days_in_month_2020_05() {
let days = days_in_month(2020, 05);
let days = days_in_month(2020, 5);
assert_eq!(days, 31);
}
#[test]
fn test_days_in_month_2020_04() {
let days = days_in_month(2020, 04);
let days = days_in_month(2020, 4);
assert_eq!(days, 30);
}
#[test]
fn test_days_in_month_2020_02() {
let days = days_in_month(2020, 02);
let days = days_in_month(2020, 2);
assert_eq!(days, 29);
}
@@ -324,15 +324,15 @@ mod tests {
let ctx = DateContext {
events,
max: 0,
today: NaiveDate::from_ymd(2020, 06, 06),
today: NaiveDate::from_ymd(2020, 6, 6),
};
let grade = ctx.activity_for_day(&NaiveDate::from_ymd(2020, 06, 06));
let grade = ctx.activity_for_day(&NaiveDate::from_ymd(2020, 6, 6));
assert_eq!(grade, ActivityGrade::None);
}
#[test]
fn test_datespec_year_month() {
let ds = DateSpec::YearMonth((2020, 05));
let ds = DateSpec::YearMonth((2020, 5));
let out = ds.to_term_string(&context());
assert_eq!(out, " May 2020
Su Mo Tu We Th Fr Sa

View File

@@ -68,7 +68,7 @@ pub struct DateTimeContext {
impl DateTimeContext {
pub fn new(events: &[Activity], now: NaiveDateTime, slice_width: u32, slice_duration: u32) -> DateTimeContext {
let (events, max) = setup_graph_map(&events, slice_duration);
let (events, max) = setup_graph_map(events, slice_duration);
DateTimeContext {
events,
max,

View File

@@ -290,7 +290,7 @@ impl InsertFromString for AddUrl {
let subdomain = parts.domain()
.ok_or_else(|| format_err!("url doesn't have a domain host"))?;
let dns_name = rl.psl()?.parse_dns_name(&subdomain)
let dns_name = rl.psl()?.parse_dns_name(subdomain)
.map_err(|e| format_err!("Failed to parse dns_name: {}", e))?;
let domain_id = match rl.db().insert_struct(NewDomain {

View File

@@ -1,6 +1,7 @@
use crate::errors::*;
use crate::autonoscope;
use crate::cmd::Cmd;
use crate::fmt::colors::*;
use crate::shell::Shell;
use std::fmt::Write;
@@ -65,7 +66,8 @@ pub fn run_with_scope_param(rl: &mut Shell, args: Args, scoped: bool) -> Result<
}
}
pub fn run(rl: &mut Shell, args: &[String]) -> Result<()> {
let args = Args::from_iter_safe(args)?;
run_with_scope_param(rl, args, false)
impl Cmd for Args {
fn run(self, rl: &mut Shell) -> Result<()> {
run_with_scope_param(rl, self, false)
}
}

View File

@@ -1,13 +1,18 @@
use crate::errors::*;
use crate::cmd::Cmd;
use crate::cmd::autonoscope_cmd;
use crate::shell::Shell;
use structopt::StructOpt;
pub type Args = autonoscope_cmd::Args;
pub fn run(rl: &mut Shell, args: &[String]) -> Result<()> {
let args = Args::from_iter_safe(args)?;
autonoscope_cmd::run_with_scope_param(rl, args, true)
#[derive(Debug, StructOpt)]
pub struct Args {
#[structopt(flatten)]
args: autonoscope_cmd::Args,
}
impl Cmd for Args {
fn run(self, rl: &mut Shell) -> Result<()> {
autonoscope_cmd::run_with_scope_param(rl, self.args, true)
}
}

View File

@@ -19,20 +19,20 @@ pub struct Args {
impl Cmd for Args {
fn run(self, rl: &mut Shell) -> Result<()> {
let rows = match &self.subcommand {
Target::Domains(filter) => delete::<Domain>(rl, &filter),
Target::Subdomains(filter) => delete::<Subdomain>(rl, &filter),
Target::IpAddrs(filter) => delete::<IpAddr>(rl, &filter),
Target::Urls(filter) => delete::<Url>(rl, &filter),
Target::Emails(filter) => delete::<Email>(rl, &filter),
Target::PhoneNumbers(filter) => delete::<PhoneNumber>(rl, &filter),
Target::Devices(filter) => delete::<Device>(rl, &filter),
Target::Networks(filter) => delete::<Network>(rl, &filter),
Target::Accounts(filter) => delete::<Account>(rl, &filter),
Target::Breaches(filter) => delete::<Breach>(rl, &filter),
Target::Images(filter) => delete::<Image>(rl, &filter),
Target::Ports(filter) => delete::<Port>(rl, &filter),
Target::Netblocks(filter) => delete::<Netblock>(rl, &filter),
Target::CryptoAddrs(filter) => delete::<CryptoAddr>(rl, &filter),
Target::Domains(filter) => delete::<Domain>(rl, filter),
Target::Subdomains(filter) => delete::<Subdomain>(rl, filter),
Target::IpAddrs(filter) => delete::<IpAddr>(rl, filter),
Target::Urls(filter) => delete::<Url>(rl, filter),
Target::Emails(filter) => delete::<Email>(rl, filter),
Target::PhoneNumbers(filter) => delete::<PhoneNumber>(rl, filter),
Target::Devices(filter) => delete::<Device>(rl, filter),
Target::Networks(filter) => delete::<Network>(rl, filter),
Target::Accounts(filter) => delete::<Account>(rl, filter),
Target::Breaches(filter) => delete::<Breach>(rl, filter),
Target::Images(filter) => delete::<Image>(rl, filter),
Target::Ports(filter) => delete::<Port>(rl, filter),
Target::Netblocks(filter) => delete::<Netblock>(rl, filter),
Target::CryptoAddrs(filter) => delete::<CryptoAddr>(rl, filter),
}?;
term::info(&format!("Deleted {} rows", rows));
Ok(())

View File

@@ -21,6 +21,7 @@ pub fn run(_rl: &mut Shell, _args: &[String]) -> Result<()> {
help("run", "Run the currently selected module");
help("scope", "Include entities in the scope again");
help("select", "Select entities from the database");
help("stats", "Show statistics about your current workspace");
help("target", "Preview targeted entities or narrow them down");
help("use", "Select a module");
help("workspace", "Switch to a different workspace");

View File

@@ -32,8 +32,10 @@ pub mod keyring_cmd;
pub mod noscope_cmd;
pub mod notify_cmd;
pub mod pkg_cmd;
pub mod rescope_cmd;
pub mod set_cmd;
pub mod scope_cmd;
pub mod stats_cmd;
pub mod target_cmd;
pub mod quickstart_cmd;
pub mod workspace_cmd;

View File

@@ -137,7 +137,7 @@ fn run_subcommand(subcommand: SubCommand, library: &Library, config: &Config) ->
for module in library.list() {
if let Some(source) = &list.source {
if !module.source_equals(&source) {
if !module.source_equals(source) {
continue;
}
}
@@ -148,7 +148,7 @@ fn run_subcommand(subcommand: SubCommand, library: &Library, config: &Config) ->
}
if let Some(stealth) = &list.stealth {
if !module.stealth().equal_or_better(&stealth) {
if !module.stealth().equal_or_better(stealth) {
continue;
}
}
@@ -163,17 +163,17 @@ fn run_subcommand(subcommand: SubCommand, library: &Library, config: &Config) ->
Ok(ModuleReload::No)
},
SubCommand::Install(install) => {
registry::run_install(install, &config)?;
registry::run_install(install, config)?;
// trigger reload
Ok(ModuleReload::Yes)
},
SubCommand::Search(search) => {
registry::run_search(library, &search, &config)?;
registry::run_search(library, &search, config)?;
Ok(ModuleReload::No)
},
SubCommand::Update(_) => {
let mut autoupdate = AutoUpdater::load()?;
let updater = Arc::new(Updater::new(&config)?);
let updater = Arc::new(Updater::new(config)?);
let modules = library.list()
.into_iter()
@@ -199,14 +199,14 @@ fn run_subcommand(subcommand: SubCommand, library: &Library, config: &Config) ->
Ok(ModuleReload::Yes)
},
SubCommand::Uninstall(uninstall) => {
let updater = Updater::new(&config)?;
let updater = Updater::new(config)?;
updater.uninstall(&uninstall.module)?;
// trigger reload
Ok(ModuleReload::Yes)
},
SubCommand::Quickstart => {
let client = Client::new(&config)?;
let updater = Arc::new(Updater::new(&config)?);
let client = Client::new(config)?;
let updater = Arc::new(Updater::new(config)?);
let mut autoupdate = AutoUpdater::load()?;
let installed = library.list()
@@ -250,7 +250,7 @@ fn run_subcommand(subcommand: SubCommand, library: &Library, config: &Config) ->
impl LiteCmd for Args {
fn run(self, config: &Config) -> Result<()> {
let library = Library::new(false, &config)?;
let library = Library::new(false, config)?;
run_subcommand(self.subcommand, &library, config)?;
Ok(())
}

268
src/cmd/rescope_cmd.rs Normal file
View File

@@ -0,0 +1,268 @@
use crate::errors::*;
use crate::autonoscope::{AutoRule, ToRule};
use crate::cmd::Cmd;
use crate::db::Database;
use crate::filters::{Filter, Target};
use crate::shell::Shell;
use std::collections::HashSet;
use std::fmt;
use structopt::StructOpt;
use structopt::clap::AppSettings;
use crate::models::*;
use crate::utils;
use crate::term;
#[derive(Debug, StructOpt)]
#[structopt(global_settings = &[AppSettings::ColoredHelp])]
pub struct Args {
/// Run rules interactively
#[structopt(short, long)]
interactive: bool,
/// Automatically apply changes to database
#[structopt(short="y", long)]
auto_confirm: bool,
/// Only show changes, do not apply them to the database
#[structopt(short="n", long)]
dry_run: bool,
/// Only rescope entities matching specific filter
#[structopt(subcommand)]
target: Option<Target>,
}
enum Entity {
Domain(Domain),
Subdomain(Subdomain),
IpAddr(IpAddr),
Url(Url),
Port(Port),
Netblock(Netblock),
}
impl Entity {
fn set_scoped(&self, db: &Database, value: bool) -> Result<()> {
match self {
Entity::Domain(entity) => entity.set_scoped(db, value),
Entity::Subdomain(entity) => entity.set_scoped(db, value),
Entity::IpAddr(entity) => entity.set_scoped(db, value),
Entity::Url(entity) => entity.set_scoped(db, value),
Entity::Port(entity) => entity.set_scoped(db, value),
Entity::Netblock(entity) => entity.set_scoped(db, value),
}
}
}
enum Input {
Yes,
No,
Done,
Always,
Never,
}
fn get_input() -> Result<Input> {
loop {
let input = utils::question_opt("Update this entity? [Y/n/d/a/x/?]")?;
let input = input.map(|s| s.to_lowercase());
match input.as_deref() {
Some("y") | None => return Ok(Input::Yes),
Some("n") => return Ok(Input::No),
Some("d") => return Ok(Input::Done),
Some("a") => return Ok(Input::Always),
Some("x") => return Ok(Input::Never),
Some("?") => {
term::success("y -> yes, apply this change");
term::success("n -> no, skip this change");
term::success("d -> done, skip this and further changes");
term::success("a -> always, apply every change caused by this specific rule");
term::success("x -> never, skip every change caused by this specific rule");
},
Some(input) => term::error(&format!("Unrecognized input: {:?}", input)),
}
}
}
#[derive(Default)]
struct Context {
update_queue: Vec<(Entity, bool)>,
always_rules: HashSet<(&'static str, String)>,
never_rules: HashSet<(&'static str, String)>,
done: bool,
target: Option<Target>,
}
fn rescope_to_queue<T, F1, F2, F3>(ctx: &mut Context, db: &Database, interactive: bool, get_filter: F1, matches_rule: F2, wrap: F3) -> Result<()>
where
T: Model + Scopable + fmt::Debug,
F1: Fn(&Target) -> Option<&Filter>,
F2: Fn(&T) -> Result<Option<((&'static str, String), bool)>>,
F3: Fn(T) -> Entity,
{
// do nothing if we're already done
if ctx.done {
return Ok(());
}
// check if there are filters to be applied
let filter = if let Some(target) = &ctx.target {
if let Some(filter) = get_filter(target) {
// we've selected this specific entity type and there's a filter
filter.parse_optional()
.context("Filter is invalid")?
} else {
// we do not wish to process this entity type
return Ok(());
}
} else {
// any entity type is fine
Filter::any()
};
let entities = db.filter::<T>(&filter)?;
for entity in entities {
let currently_scoped = entity.scoped();
debug!("rescoping entity: {:?}", entity);
if let Some((rule, should_be)) = matches_rule(&entity)? {
// check if we're actively ignoring this rule
if ctx.never_rules.contains(&rule) {
continue;
}
if currently_scoped != should_be {
let prefix = if should_be {
"\x1b[1m[\x1b[32m+\x1b[0;1m]\x1b[0m"
} else {
"\x1b[1m[\x1b[31m-\x1b[0;1m]\x1b[0m"
};
println!("{} Setting entity {:?} => {:?}: {:?}", prefix, currently_scoped, should_be, entity);
// check if we're auto-accepting this rule
let input = if ctx.always_rules.contains(&rule) {
Input::Yes
} else if interactive {
get_input()?
} else {
Input::Yes
};
// process user input
let input = match input {
Input::Always => {
ctx.always_rules.insert(rule);
Input::Yes
},
Input::Never => {
ctx.never_rules.insert(rule);
Input::No
},
Input::Done => {
ctx.done = true;
break;
},
input => input,
};
if let Input::Yes = input {
ctx.update_queue.push((wrap(entity), should_be));
}
}
}
}
Ok(())
}
impl Cmd for Args {
fn run(self, rl: &mut Shell) -> Result<()> {
let rules = rl.db().autonoscope();
term::success(&format!("Loaded {} rules", rules.len()));
let mut ctx = Context {
target: self.target,
..Default::default()
};
rescope_to_queue::<Domain, _, _, _>(&mut ctx, rl.db(), self.interactive, |t| t.domains(), |entity| {
for rule in rules.domains() {
if rule.matches(entity.value.as_str())? {
return Ok(Some((rule.to_rule(), rule.scoped)));
}
}
Ok(None)
}, Entity::Domain)?;
rescope_to_queue::<Subdomain, _, _, _>(&mut ctx, rl.db(), self.interactive, |t| t.subdomains(), |entity| {
for rule in rules.domains() {
if rule.matches(entity.value.as_str())? {
return Ok(Some((rule.to_rule(), rule.scoped)));
}
}
Ok(None)
}, Entity::Subdomain)?;
rescope_to_queue::<IpAddr, _, _, _>(&mut ctx, rl.db(), self.interactive, |t| t.ipaddrs(), |entity| {
for rule in rules.ips() {
if rule.matches(entity)? {
return Ok(Some((rule.to_rule(), rule.scoped)));
}
}
Ok(None)
}, Entity::IpAddr)?;
rescope_to_queue::<Url, _, _, _>(&mut ctx, rl.db(), self.interactive, |t| t.urls(), |entity| {
for rule in rules.domains() {
if rule.matches(entity)? {
return Ok(Some((rule.to_rule(), rule.scoped)));
}
}
for rule in rules.urls() {
if rule.matches(entity)? {
return Ok(Some((rule.to_rule(), rule.scoped)));
}
}
Ok(None)
}, Entity::Url)?;
rescope_to_queue::<Port, _, _, _>(&mut ctx, rl.db(), self.interactive, |t| t.ports(), |entity| {
for rule in rules.ips() {
if rule.matches(entity)? {
return Ok(Some((rule.to_rule(), rule.scoped)));
}
}
Ok(None)
}, Entity::Port)?;
rescope_to_queue::<Netblock, _, _, _>(&mut ctx, rl.db(), self.interactive, |t| t.netblocks(), |entity| {
for rule in rules.ips() {
if rule.matches(entity)? {
return Ok(Some((rule.to_rule(), rule.scoped)));
}
}
Ok(None)
}, Entity::Netblock)?;
if ctx.update_queue.is_empty() {
term::success("Nothing has changed, not updating database");
} else {
let confirm = if self.dry_run {
false
} else if self.auto_confirm {
true
} else {
utils::no_else_yes(&format!("Apply {} changes to scope now?", ctx.update_queue.len()))?
};
if confirm {
term::info(&format!("Applying {} changes to database", ctx.update_queue.len()));
for (update, value) in ctx.update_queue {
update.set_scoped(rl.db(), value)?;
}
} else {
term::info("Database not updated");
}
}
Ok(())
}
}

View File

@@ -17,6 +17,7 @@ use serde::Serialize;
use sn0int_common::metadata::Source;
use sn0int_std::ratelimits::Ratelimiter;
use std::collections::HashMap;
use std::net::SocketAddr;
use structopt::StructOpt;
use structopt::clap::AppSettings;
@@ -24,14 +25,26 @@ use structopt::clap::AppSettings;
#[derive(Debug, StructOpt)]
#[structopt(global_settings = &[AppSettings::ColoredHelp])]
pub struct Args {
#[structopt(short="j", long="threads", default_value="1")]
threads: usize,
#[structopt(short="v", long="verbose", parse(from_occurrences))]
verbose: u64,
/// Execute a module that has been installed
pub module: Option<String>,
/// Run investigations concurrently
#[structopt(short="j", default_value="1")]
pub threads: usize,
/// Verbose logging, once to print inserts even if they don't add new
/// data, twice to activate the debug() function
#[structopt(short="v", long, parse(from_occurrences))]
pub verbose: u64,
/// Set a specific socks5 proxy to use
#[structopt(short="X", long)]
pub proxy: Option<SocketAddr>,
/// Set a different default user agent
#[structopt(long)]
pub user_agent: Option<String>,
}
#[derive(Debug, Clone)]
pub struct Params<'a> {
pub module: Option<&'a String>,
pub threads: usize,
pub verbose: u64,
pub stdin: bool,
@@ -39,25 +52,61 @@ pub struct Params<'a> {
pub grant_full_keyring: bool,
pub deny_keyring: bool,
pub exit_on_error: bool,
pub proxy: Option<SocketAddr>,
pub user_agent: Option<&'a String>,
}
impl<'a> Params<'a> {
pub fn get_module(&self, rl: &Shell) -> Result<Module> {
let module = if let Some(module) = self.module {
rl.library().get(module)?
.clone()
} else {
rl.module()
.map(|m| m.to_owned())
.ok_or_else(|| format_err!("No module selected"))?
};
Ok(module)
}
pub fn get_proxy(&self, rl: &Shell) -> Option<SocketAddr> {
if self.proxy.is_some() {
self.proxy
} else {
rl.config().network.proxy
}
}
pub fn get_user_agent(&self, rl: &Shell) -> Option<String> {
if let Some(user_agent) = self.user_agent {
Some(user_agent.to_string())
} else {
rl.config().network.user_agent.clone()
}
}
}
impl<'a> From<&'a args::Run> for Params<'a> {
fn from(args: &args::Run) -> Params {
Params {
threads: args.threads,
verbose: args.verbose,
module: args.run.module.as_ref(),
threads: args.run.threads,
verbose: args.run.verbose,
stdin: args.stdin,
grants: &args.grants,
grant_full_keyring: args.grant_full_keyring,
deny_keyring: args.deny_keyring,
exit_on_error: args.exit_on_error,
proxy: args.run.proxy,
user_agent: args.run.user_agent.as_ref(),
}
}
}
impl From<Args> for Params<'static> {
fn from(args: Args) -> Params<'static> {
impl<'a> From<&'a Args> for Params<'a> {
fn from(args: &Args) -> Params {
Params {
module: args.module.as_ref(),
threads: args.threads,
verbose: args.verbose,
stdin: false,
@@ -65,6 +114,8 @@ impl From<Args> for Params<'static> {
grant_full_keyring: false,
deny_keyring: false,
exit_on_error: false,
proxy: args.proxy,
user_agent: args.user_agent.as_ref(),
}
}
}
@@ -93,7 +144,7 @@ fn prepare_args<T: Scopable + Serialize + Model>(rl: &Shell, filter: &Filter, pa
}
pub fn prepare_keyring(keyring: &mut KeyRing, module: &Module, params: &Params) -> Result<()> {
for namespace in keyring.unauthorized_namespaces(&module) {
for namespace in keyring.unauthorized_namespaces(module) {
let grant_access = if params.deny_keyring {
false
} else if params.grant_full_keyring || params.grants.contains(namespace) {
@@ -104,7 +155,7 @@ pub fn prepare_keyring(keyring: &mut KeyRing, module: &Module, params: &Params)
};
if grant_access {
keyring.grant_access(&module, namespace.to_string());
keyring.grant_access(module, namespace.to_string());
term::info(&format!("Granted access to {:?}", namespace));
}
}
@@ -136,8 +187,8 @@ fn get_args(rl: &mut Shell, module: &Module) -> Result<Vec<(serde_json::Value, O
Some(Source::Notifications) => bail!("Notification modules can't be executed like this"),
Some(Source::KeyRing(namespace)) => {
let keyring = rl.keyring();
if keyring.is_access_granted(&module, &namespace) {
keyring.get_all_for(&namespace).into_iter()
if keyring.is_access_granted(module, namespace) {
keyring.get_all_for(namespace).into_iter()
.map(|key| {
let pretty = format!("{}:{}", key.namespace, key.access_key);
let arg = serde_json::to_value(key)?;
@@ -153,15 +204,14 @@ fn get_args(rl: &mut Shell, module: &Module) -> Result<Vec<(serde_json::Value, O
}
pub fn dump_sandbox_init_msg(rl: &mut Shell, params: Params, options: HashMap<String, String>) -> Result<()> {
let module = rl.module()
.map(|m| m.to_owned())
.ok_or_else(|| format_err!("No module selected"))?;
let module = params.get_module(rl)?;
let proxy = params.get_proxy(rl);
let user_agent = params.get_user_agent(rl);
prepare_keyring(rl.keyring_mut(), &module, &params)?;
let keyring = rl.keyring().request_keys(&module);
let dns_config = Resolver::from_system_v4()?;
let proxy = rl.config().network.proxy;
let args = get_args(rl, &module)?;
for (arg, _pretty_arg, blobs) in args {
@@ -169,6 +219,7 @@ pub fn dump_sandbox_init_msg(rl: &mut Shell, params: Params, options: HashMap<St
keyring.clone(),
dns_config.clone(),
proxy,
user_agent.clone(),
options.clone(),
module.clone(),
arg,
@@ -181,15 +232,15 @@ pub fn dump_sandbox_init_msg(rl: &mut Shell, params: Params, options: HashMap<St
}
pub fn execute(rl: &mut Shell, params: Params, options: HashMap<String, String>) -> Result<()> {
let module = rl.module()
.map(|m| m.to_owned())
.ok_or_else(|| format_err!("No module selected"))?;
let module = params.get_module(rl)?;
let proxy = params.get_proxy(rl);
let user_agent = params.get_user_agent(rl);
prepare_keyring(rl.keyring_mut(), &module, &params)?;
let args = get_args(rl, &module)?;
rl.signal_register().catch_ctrl();
let errors = worker::spawn(rl, &module, &mut Ratelimiter::new(), args, &params, rl.config().network.proxy, options);
let errors = worker::spawn(rl, &module, &mut Ratelimiter::new(), args, &params, proxy, user_agent, options);
rl.signal_register().reset_ctrlc();
if errors > 0 {
@@ -212,6 +263,6 @@ impl Cmd for Args {
Some(options) => options.clone(),
_ => HashMap::new(),
};
execute(rl, self.into(), options)
execute(rl, Params::from(&self), options)
}
}

View File

@@ -16,13 +16,16 @@ pub struct Args {
#[structopt(subcommand)]
subcommand: Target,
/// Print json output
#[structopt(long="json", group="output")]
#[structopt(long, group="output")]
json: bool,
/// Only print the value instead of the whole object
#[structopt(long, group="output")]
values: bool,
/// Print paths to blobs
#[structopt(long="paths", group="output")]
#[structopt(long, group="output")]
paths: bool,
/// Count rows returned
#[structopt(short="c", long="count", group="output")]
#[structopt(short="c", group="output")]
count: bool,
}
@@ -30,6 +33,7 @@ pub struct Args {
enum Output {
Normal,
Json,
Values,
Paths,
Count,
}
@@ -43,6 +47,8 @@ impl<'a, 'b> Printer<'a, 'b> {
pub fn new(rl: &'a mut Shell<'b>, args: &Args) -> Printer<'a, 'b> {
let output = if args.json {
Output::Json
} else if args.values {
Output::Values
} else if args.paths {
Output::Paths
} else if args.count {
@@ -70,6 +76,9 @@ impl<'a, 'b> Printer<'a, 'b> {
let v = serde_json::to_string(&obj)?;
println!("{}", v);
},
Output::Values => {
println!("{}", obj.to_string());
},
Output::Paths => {
let blob = obj.blob()
.ok_or_else(|| format_err!("This model isn't linked to blob storage"))?;
@@ -97,20 +106,20 @@ impl Cmd for Args {
let printer = Printer::new(rl, &self);
match &self.subcommand {
Target::Domains(filter) => printer.select::<Domain>(&filter),
Target::Subdomains(filter) => printer.select::<Subdomain>(&filter),
Target::IpAddrs(filter) => printer.select::<IpAddr>(&filter),
Target::Urls(filter) => printer.select::<Url>(&filter),
Target::Emails(filter) => printer.select::<Email>(&filter),
Target::PhoneNumbers(filter) => printer.select::<PhoneNumber>(&filter),
Target::Devices(filter) => printer.select::<Device>(&filter),
Target::Networks(filter) => printer.select::<Network>(&filter),
Target::Accounts(filter) => printer.select::<Account>(&filter),
Target::Breaches(filter) => printer.select::<Breach>(&filter),
Target::Images(filter) => printer.select::<Image>(&filter),
Target::Ports(filter) => printer.select::<Port>(&filter),
Target::Netblocks(filter) => printer.select::<Netblock>(&filter),
Target::CryptoAddrs(filter) => printer.select::<CryptoAddr>(&filter),
Target::Domains(filter) => printer.select::<Domain>(filter),
Target::Subdomains(filter) => printer.select::<Subdomain>(filter),
Target::IpAddrs(filter) => printer.select::<IpAddr>(filter),
Target::Urls(filter) => printer.select::<Url>(filter),
Target::Emails(filter) => printer.select::<Email>(filter),
Target::PhoneNumbers(filter) => printer.select::<PhoneNumber>(filter),
Target::Devices(filter) => printer.select::<Device>(filter),
Target::Networks(filter) => printer.select::<Network>(filter),
Target::Accounts(filter) => printer.select::<Account>(filter),
Target::Breaches(filter) => printer.select::<Breach>(filter),
Target::Images(filter) => printer.select::<Image>(filter),
Target::Ports(filter) => printer.select::<Port>(filter),
Target::Netblocks(filter) => printer.select::<Netblock>(filter),
Target::CryptoAddrs(filter) => printer.select::<CryptoAddr>(filter),
}
}
}

199
src/cmd/stats_cmd.rs Normal file
View File

@@ -0,0 +1,199 @@
use crate::args;
use colored::Colorize;
use crate::blobs::BlobStorage;
use crate::cmd::Cmd;
use crate::db::{ttl, Database};
use crate::errors::*;
use crate::models::*;
use crate::shell::{self, Shell};
use crate::workspaces;
use humansize::{FileSize, file_size_opts};
use separator::Separatable;
use serde::{Serialize, Deserialize};
use structopt::StructOpt;
use structopt::clap::AppSettings;
#[derive(Debug, Clone, StructOpt)]
#[structopt(global_settings = &[AppSettings::ColoredHelp])]
pub struct Args {
/// Exclude blob storage
#[structopt(short, long)]
short: bool,
/// Exclude categories that don't contain any structs
#[structopt(short, long)]
quiet: bool,
/// Show workspace statistics in json
#[structopt(short, long)]
json: bool,
/// Go through all workspaces
#[structopt(short, long)]
all: bool,
}
impl Args {
fn show_amount(&self, label: &str, count: usize, amount: &str) {
if self.quiet && count == 0 {
return;
}
let label = format!("{:20}", label);
let amount = format!("{:>20}", amount);
if count > 0 {
println!("{} {}", label.green(), amount.yellow());
} else {
println!("{} {}", label, amount);
}
}
fn show_count(&self, label: &str, count: usize) {
self.show_amount(label, count, &count.separated_string());
}
}
fn count_models<T: Model>(db: &Database) -> Result<usize> {
let query = db.list::<T>()?;
let count = query.len();
Ok(count)
}
#[derive(Debug, Serialize, Deserialize)]
struct Stats {
workspace: String,
domains: usize,
subdomains: usize,
ipaddrs: usize,
urls: usize,
emails: usize,
phonenumbers: usize,
devices: usize,
networks: usize,
accounts: usize,
breaches: usize,
images: usize,
ports: usize,
netblocks: usize,
cryptoaddrs: usize,
activity: usize,
blobs: Option<BlobStats>,
}
#[derive(Debug, Serialize, Deserialize)]
struct BlobStats {
count: usize,
total_size: u64,
total_human_size: String,
}
impl Stats {
fn count(workspace: String, db: &Database) -> Result<Stats> {
Ok(Stats {
workspace,
domains: count_models::<Domain>(db)?,
subdomains: count_models::<Subdomain>(db)?,
ipaddrs: count_models::<IpAddr>(db)?,
urls: count_models::<Url>(db)?,
emails: count_models::<Email>(db)?,
phonenumbers: count_models::<PhoneNumber>(db)?,
devices: count_models::<Device>(db)?,
networks: count_models::<Network>(db)?,
accounts: count_models::<Account>(db)?,
breaches: count_models::<Breach>(db)?,
images: count_models::<Image>(db)?,
ports: count_models::<Port>(db)?,
netblocks: count_models::<Netblock>(db)?,
cryptoaddrs: count_models::<CryptoAddr>(db)?,
activity: Activity::count(db)?,
blobs: None,
})
}
fn add_blob_usage(&mut self, storage: &BlobStorage) -> Result<()> {
let blobs = storage.list()?;
let mut total_size = 0;
for blob in &blobs {
total_size += storage.stat(blob)?;
}
let total_human_size = total_size.file_size(file_size_opts::CONVENTIONAL)
.map_err(|e| format_err!("Failed to format size: {}", e))?;
self.blobs = Some(BlobStats {
count: blobs.len(),
total_size,
total_human_size,
});
Ok(())
}
}
impl Cmd for Args {
#[inline]
fn run(mut self, rl: &mut Shell) -> Result<()> {
if self.all {
self.all = false;
let mut first = true;
let config = rl.config();
for ws in workspaces::list()? {
if first {
first = false;
} else if !self.json {
println!();
}
let mut rl = shell::init(&args::Args {
workspace: Some(ws),
subcommand: None,
}, config, false)?;
self.clone().run(&mut rl)?;
}
} else {
ttl::reap_expired(rl)?;
let db = rl.db();
let workspace = rl.workspace();
if !self.json {
// print this early in case counting takes a while
println!("{:>41}", workspace.bold());
}
let mut stats = Stats::count(workspace.into(), db)?;
if !self.short {
stats.add_blob_usage(rl.blobs())?;
}
if self.json {
let stats = serde_json::to_string(&stats)?;
println!("{}", stats);
} else {
self.show_count("domains", stats.domains);
self.show_count("subdomains", stats.subdomains);
self.show_count("ipaddrs", stats.ipaddrs);
self.show_count("urls", stats.urls);
self.show_count("emails", stats.emails);
self.show_count("phonenumbers", stats.phonenumbers);
self.show_count("devices", stats.devices);
self.show_count("networks", stats.networks);
self.show_count("accounts", stats.accounts);
self.show_count("breaches", stats.breaches);
self.show_count("images", stats.images);
self.show_count("ports", stats.ports);
self.show_count("netblocks", stats.netblocks);
self.show_count("cryptoaddrs", stats.cryptoaddrs);
self.show_count("activity", stats.activity);
if let Some(blobs) = stats.blobs {
self.show_count("blobs", blobs.count);
self.show_amount("blobs (size)", blobs.total_size as usize, &blobs.total_human_size);
}
}
}
Ok(())
}
}

View File

@@ -81,7 +81,7 @@ fn count_selected(rl: &mut Shell, source: &Source) -> Result<usize> {
Source::Netblocks => db.filter::<Netblock>(&filter)?.len(),
Source::CryptoAddrs(currency) => db.filter_with_param::<CryptoAddr>(&filter, currency.as_ref())?.len(),
Source::Notifications => bail!("Notifications can't be set as target"),
Source::KeyRing(namespace) => rl.keyring().list_for(&namespace).len(),
Source::KeyRing(namespace) => rl.keyring().list_for(namespace).len(),
};
Ok(num)
}

View File

@@ -24,7 +24,7 @@ pub struct Args {
/// Skip confirmation
#[structopt(short = "f", long = "force")]
force: bool,
workspace: Option<Workspace>,
workspaces: Vec<Workspace>,
}
fn delete(workspace: Workspace, force: bool) -> Result<()> {
@@ -38,13 +38,13 @@ fn delete(workspace: Workspace, force: bool) -> Result<()> {
Ok(())
}
fn usage(workspace: Option<Workspace>) -> Result<()> {
if let Some(ws) = workspace {
println!("{}", ws.usage_human()?);
} else {
for ws in workspaces::list()? {
println!("{:50}: {}", ws.as_str(), ws.usage_human()?);
}
fn usage(mut workspaces: Vec<Workspace>) -> Result<()> {
if workspaces.is_empty() {
workspaces = workspaces::list()?;
}
for ws in workspaces {
println!("{:50}: {}", ws.as_str(), ws.usage_human()?);
}
Ok(())
@@ -67,30 +67,34 @@ fn list() -> Result<()> {
Ok(())
}
fn run(args: Args, rl: Option<&mut Shell>) -> Result<()> {
fn run(mut args: Args, rl: Option<&mut Shell>) -> Result<()> {
if args.delete {
if let Some(workspace) = args.workspace {
if let Some(rl) = rl {
if args.workspaces.is_empty() {
bail!("--delete requires workspace");
}
for workspace in args.workspaces {
if let Some(rl) = &rl {
if *rl.db().workspace() == workspace {
bail!("Can't delete current workspace")
}
}
delete(workspace, args.force)
} else {
bail!("--delete requires workspace")
delete(workspace, args.force)?;
}
} else if args.usage {
usage(args.workspace)
} else if let Some(workspace) = args.workspace {
if let Some(rl) = rl {
change(rl, workspace)
} else {
Ok(())
}
usage(args.workspaces)?;
} else {
list()
match args.workspaces.len() {
0 => list()?,
1 => if let Some(rl) = rl {
// we've already tested there's one item in the list
change(rl, args.workspaces.pop().unwrap())?;
},
_ => bail!("Only one argument allowed when switching workspaces"),
}
}
Ok(())
}
impl Cmd for Args {

View File

@@ -75,4 +75,5 @@ fn default_registry() -> String {
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct NetworkConfig {
pub proxy: Option<SocketAddr>,
pub user_agent: Option<String>,
}

View File

@@ -164,6 +164,11 @@ impl Database {
self.autonoscope.rules()
}
#[inline(always)]
pub fn autonoscope(&self) -> &RuleSet {
&self.autonoscope
}
/// Returns true if we didn't have this value yet
pub fn insert_generic(&self, object: Insert) -> Result<Option<(DbChange, i32)>> {
let scoped = self.autonoscope.matches(&object)?;
@@ -198,14 +203,14 @@ impl Database {
let update = obj.upsert(&existing);
if update.is_dirty() {
update.apply(&self)?;
update.apply(self)?;
Ok(Some((DbChange::Update(update.generic()), existing.id())))
} else {
Ok(Some((DbChange::None, existing.id())))
}
} else {
obj.set_scoped(scoped);
obj.insert(&self)?;
obj.insert(self)?;
let id = M::get_id(self, obj.value())?;
Ok(Some((DbChange::Insert, id)))
}
@@ -218,7 +223,7 @@ impl Database {
return Ok(false);
}
}
obj.insert(&self)?;
obj.insert(self)?;
Ok(true)
}
@@ -254,7 +259,7 @@ impl Database {
let update = obj.upsert(&existing);
if update.is_dirty() {
update.apply(&self)?;
update.apply(self)?;
Ok(Some((DbChange::Update(update.generic()), id)))
} else {
Ok(Some((DbChange::None, id)))
@@ -402,7 +407,7 @@ impl Database {
}
fn get_opt_typed<T: Model + Scopable>(&self, value: &T::ID) -> Result<Option<i32>> {
match T::get_opt(self, &value)? {
match T::get_opt(self, value)? {
Some(ref obj) if obj.scoped() => Ok(Some(obj.id())),
_ => Ok(None),
}
@@ -410,23 +415,23 @@ impl Database {
pub fn get_opt(&self, family: &Family, value: &str) -> Result<Option<i32>> {
match family {
Family::Domain => self.get_opt_typed::<Domain>(&value),
Family::Subdomain => self.get_opt_typed::<Subdomain>(&value),
Family::Ipaddr => self.get_opt_typed::<IpAddr>(&value),
Family::Domain => self.get_opt_typed::<Domain>(value),
Family::Subdomain => self.get_opt_typed::<Subdomain>(value),
Family::Ipaddr => self.get_opt_typed::<IpAddr>(value),
Family::SubdomainIpaddr => bail!("Unsupported operation"),
Family::Url => self.get_opt_typed::<Url>(&value),
Family::Email => self.get_opt_typed::<Email>(&value),
Family::Phonenumber => self.get_opt_typed::<PhoneNumber>(&value),
Family::Device => self.get_opt_typed::<Device>(&value),
Family::Network => self.get_opt_typed::<Network>(&value),
Family::Url => self.get_opt_typed::<Url>(value),
Family::Email => self.get_opt_typed::<Email>(value),
Family::Phonenumber => self.get_opt_typed::<PhoneNumber>(value),
Family::Device => self.get_opt_typed::<Device>(value),
Family::Network => self.get_opt_typed::<Network>(value),
Family::NetworkDevice => bail!("Unsupported operation"),
Family::Account => self.get_opt_typed::<Account>(&value),
Family::Breach => self.get_opt_typed::<Breach>(&value),
Family::Account => self.get_opt_typed::<Account>(value),
Family::Breach => self.get_opt_typed::<Breach>(value),
Family::BreachEmail => bail!("Unsupported operation"),
Family::Image => self.get_opt_typed::<Image>(&value),
Family::Port => self.get_opt_typed::<Port>(&value),
Family::Netblock => self.get_opt_typed::<Netblock>(&value),
Family::Cryptoaddr => self.get_opt_typed::<CryptoAddr>(&value),
Family::Image => self.get_opt_typed::<Image>(value),
Family::Port => self.get_opt_typed::<Port>(value),
Family::Netblock => self.get_opt_typed::<Netblock>(value),
Family::Cryptoaddr => self.get_opt_typed::<CryptoAddr>(value),
}
}
@@ -460,6 +465,11 @@ impl Filter {
}
}
#[inline]
pub fn any() -> Filter {
Filter::new("1")
}
fn escape(value: &str) -> String {
let mut out = String::from("'");
for c in value.chars() {
@@ -522,7 +532,7 @@ impl Filter {
if args.is_empty() {
debug!("Using filter with no condition");
return Ok(Filter::new("1"));
return Ok(Filter::any());
}
Self::parse(args)

View File

@@ -135,7 +135,7 @@ pub fn reap_expired(rl: &mut Shell) -> Result<()> {
let subject = format!("Deleted {} {:?}", &expired.family, &expired.value);
let topic = &format!("db:{}:{}:delete", &expired.family, &expired.value);
if let Err(err) = notify::trigger_notify_event(rl, &mut Term, &mut ratelimit, &topic, &Notification {
if let Err(err) = notify::trigger_notify_event(rl, &mut Term, &mut ratelimit, topic, &Notification {
subject,
body: None,
}) {

View File

@@ -215,6 +215,7 @@ pub struct LuaState {
geoip: Option<Mutex<Lazy<MaxmindReader, Arc<GeoIP>>>>,
asn: Option<Mutex<Lazy<MaxmindReader, Arc<AsnDB>>>>,
proxy: Option<SocketAddr>,
user_agent: Option<String>,
options: HashMap<String, String>,
}
@@ -372,7 +373,15 @@ impl State for LuaState {
let mtx = self.http_sessions.lock().unwrap();
let session = mtx.get(session_id).expect("Invalid session reference"); // TODO
HttpRequest::new(&session, method, url, options, || format!("sn0int/{}", env!("CARGO_PKG_VERSION")))
let user_agent = if let Some(user_agent) = &options.user_agent {
user_agent.to_string()
} else if let Some(user_agent) = &self.user_agent {
user_agent.to_string()
} else {
format!("sn0int/{}", env!("CARGO_PKG_VERSION"))
};
HttpRequest::new(session, method, url, user_agent, options)
}
fn get_blob(&self, id: &str) -> Result<Arc<Blob>> {
@@ -472,6 +481,7 @@ pub fn ctx<'a>(env: Environment, logger: Arc<Mutex<Box<dyn IpcChild>>>) -> (hlua
geoip,
asn,
proxy: env.proxy,
user_agent: env.user_agent,
options: env.options,
});
@@ -517,6 +527,9 @@ pub fn ctx<'a>(env: Environment, logger: Arc<Mutex<Box<dyn IpcChild>>>) -> (hlua
runtime::http_fetch_json(&mut lua, state.clone());
runtime::img_exif(&mut lua, state.clone());
runtime::img_load(&mut lua, state.clone());
runtime::img_ahash(&mut lua, state.clone());
runtime::img_dhash(&mut lua, state.clone());
runtime::img_phash(&mut lua, state.clone());
runtime::img_nudity(&mut lua, state.clone());
runtime::info(&mut lua, state.clone());
runtime::intval(&mut lua, state.clone());
@@ -662,6 +675,7 @@ impl Script {
let keyring = Vec::new();
let dns_config = Resolver::from_system_v4()?;
let proxy = None;
let user_agent = None;
let psl = PslReader::String(r#"
// ===BEGIN ICANN DOMAINS===
com
@@ -679,6 +693,7 @@ a.prod.fastly.net
keyring,
dns_config,
proxy,
user_agent,
options: HashMap::new(),
blobs: Vec::new(),
psl,

View File

@@ -34,6 +34,7 @@ pub struct Environment {
pub keyring: Vec<KeyRingEntry>,
pub dns_config: Resolver,
pub proxy: Option<SocketAddr>,
pub user_agent: Option<String>,
pub options: HashMap<String, String>,
pub blobs: Vec<Blob>,
pub psl: PslReader,
@@ -120,14 +121,14 @@ impl<'a> Library<'a> {
folder.to_owned()
} else {
let folder = folder.strip_prefix("~/")
.unwrap_or(&folder);
.unwrap_or(folder);
dirs_next::home_dir()
.ok_or_else(|| format_err!("Failed to find home folder"))?
.join(folder)
};
self.load_module_folder(&folder, &author, true)?;
self.load_module_folder(&folder, author, true)?;
}
Ok(())
@@ -156,7 +157,7 @@ impl<'a> Library<'a> {
continue;
}
if let Err(err) = self.load_single_module(&module.path(), &author_name, &module_name, private_modules) {
if let Err(err) = self.load_single_module(&module.path(), author_name, module_name, private_modules) {
let root = err.find_root_cause();
term::warn(&format!("Failed to load {}/{}: {}", author_name, module_name, root));
}
@@ -167,7 +168,7 @@ impl<'a> Library<'a> {
pub fn load_single_module(&mut self, path: &Path, author_name: &str, module_name: &str, private_module: bool) -> Result<()> {
let module_name = module_name.to_string();
let module = Module::load(path, &author_name, &module_name, private_module)
let module = Module::load(path, author_name, &module_name, private_module)
.context(format!("Failed to parse {}/{}", author_name, module_name))?;
for key in &[&module_name, &format!("{}/{}", author_name, module_name)] {
@@ -332,7 +333,7 @@ impl Module {
pub fn source_equals(&self, other: &str) -> bool {
match self.source() {
Some(source) => source.group_as_str() == other,
None => other == "",
None => other.is_empty(),
}
}
}

View File

@@ -50,12 +50,67 @@ pub enum Target {
CryptoAddrs(Filter),
}
impl Target {
pub fn domains(&self) -> Option<&Filter> {
if let Target::Domains(f) = self {
Some(f)
} else {
None
}
}
pub fn subdomains(&self) -> Option<&Filter> {
if let Target::Subdomains(f) = self {
Some(f)
} else {
None
}
}
pub fn ipaddrs(&self) -> Option<&Filter> {
if let Target::IpAddrs(f) = self {
Some(f)
} else {
None
}
}
pub fn urls(&self) -> Option<&Filter> {
if let Target::Urls(f) = self {
Some(f)
} else {
None
}
}
pub fn ports(&self) -> Option<&Filter> {
if let Target::Ports(f) = self {
Some(f)
} else {
None
}
}
pub fn netblocks(&self) -> Option<&Filter> {
if let Target::Netblocks(f) = self {
Some(f)
} else {
None
}
}
}
#[derive(Debug, StructOpt)]
pub struct Filter {
args: Vec<String>,
}
impl Filter {
#[inline]
pub fn any() -> db::Filter {
db::Filter::any()
}
pub fn parse_optional(&self) -> Result<db::Filter> {
db::Filter::parse_optional(&self.args)
}

View File

@@ -86,6 +86,7 @@ pub fn run(geoip: Option<MaxmindReader>, asn: Option<MaxmindReader>, psl: PslRea
keyring: start.keyring,
dns_config: start.dns_config,
proxy: start.proxy,
user_agent: start.user_agent,
options: start.options,
blobs: start.blobs,
psl,

View File

@@ -12,6 +12,7 @@ pub struct StartCommand {
pub keyring: Vec<KeyRingEntry>,
pub dns_config: Resolver,
pub proxy: Option<SocketAddr>,
pub user_agent: Option<String>,
pub options: HashMap<String, String>,
pub module: Module,
pub arg: serde_json::Value,
@@ -23,6 +24,7 @@ impl StartCommand {
keyring: Vec<KeyRingEntry>,
dns_config: Resolver,
proxy: Option<SocketAddr>,
user_agent: Option<String>,
options: HashMap<String, String>,
module: Module,
arg: serde_json::Value,
@@ -33,6 +35,7 @@ impl StartCommand {
keyring,
dns_config,
proxy,
user_agent,
options,
module,
arg,

View File

@@ -105,6 +105,7 @@ pub fn run(module: Module,
verbose: u64,
has_stdin: bool,
proxy: Option<SocketAddr>,
user_agent: Option<String>,
options: HashMap<String, String>,
blobs: Vec<Blob>,
) -> Result<ExitEvent> {
@@ -117,15 +118,15 @@ pub fn run(module: Module,
};
let mut ipc_parent = IpcParent::setup(&module)?;
ipc_parent.send_start(&StartCommand::new(verbose, keyring, dns_config, proxy, options, module, arg, blobs))?;
ipc_parent.send_start(&StartCommand::new(verbose, keyring, dns_config, proxy, user_agent, options, module, arg, blobs))?;
let exit = loop {
match ipc_parent.recv()? {
Event::Log(event) => tx.send(Event2::Log(event)),
Event::Database(object) => ipc_parent.send_event_callback(*object, &tx),
Event::Database(object) => ipc_parent.send_event_callback(*object, tx),
Event::Stdio(object) => object.apply(&mut ipc_parent, tx, &mut reader),
Event::Ratelimit(req) => ipc_parent.send_event_callback(req, &tx),
Event::Blob(blob) => ipc_parent.send_event_callback(blob, &tx),
Event::Ratelimit(req) => ipc_parent.send_event_callback(req, tx),
Event::Blob(blob) => ipc_parent.send_event_callback(blob, tx),
Event::Exit(event) => {
if let ExitEvent::Err(err) = &event {
tx.send(Event2::Log(LogEvent::Error(err.clone())));

View File

@@ -111,7 +111,7 @@ impl KeyRing {
pub fn insert(&mut self, key: KeyName, secret: Option<String>) -> Result<()> {
// get the namespace or create a new one
let mut x = self.keys.remove(&key.namespace)
.unwrap_or_else(HashMap::new);
.unwrap_or_default();
// insert key into namespace
x.insert(key.name, secret);
// add namespace backinto keyring
@@ -170,13 +170,13 @@ impl KeyRing {
pub fn unauthorized_namespaces<'a>(&self, module: &'a Module) -> Vec<&'a String> {
module.keyring_access().iter()
.filter(|namespace| !self.is_access_granted(&module, &namespace))
.filter(|namespace| !self.is_access_granted(module, namespace))
.collect()
}
pub fn grant_access(&mut self, module: &Module, namespace: String) {
let mut grants = self.grants.remove(&namespace)
.unwrap_or_else(HashSet::new);
.unwrap_or_default();
grants.insert(module.id());
self.grants.insert(namespace, grants);
}
@@ -192,7 +192,7 @@ impl KeyRing {
pub fn request_keys(&self, module: &Module) -> Vec<KeyRingEntry> {
// TODO: we probably want to randomize the order
module.keyring_access().iter()
.filter(|namespace| self.is_access_granted(&module, &namespace))
.filter(|namespace| self.is_access_granted(module, namespace))
.flat_map(|namespace| self.list_for(namespace))
.flat_map(|x| self.get(&x))
.collect()

View File

@@ -1,7 +1,8 @@
use env_logger::{self, Env};
use env_logger::Env;
use sn0int::args::{self, Args, SubCommand};
use sn0int::auth;
use sn0int::cmd::{self, LiteCmd};
use sn0int::cmd::run_cmd::Params;
use sn0int::config::Config;
use sn0int::db;
use sn0int::errors::*;
@@ -23,34 +24,41 @@ use std::path::Path;
fn run_run(gargs: &Args, args: &args::Run, config: &Config) -> Result<()> {
let mut rl = shell::init(gargs, config, false)?;
let module = args.run.module.as_ref()
.ok_or_else(|| format_err!("Module is required"))?;
let module = if args.file {
let path = Path::new(&args.module);
let path = Path::new(&module);
let filename = path.file_stem()
.ok_or_else(|| format_err!("Failed to decode filename"))?
.to_str()
.ok_or_else(|| format_err!("Failed to decode filename"))?;
Module::load(&path.to_path_buf(), "anonymous", &filename, true)
Module::load(path, "anonymous", filename, true)
.context(format!("Failed to parse {:?}", path))?
} else {
rl.library().get(&args.module)?
rl.library().get(module)?
.clone()
};
rl.set_module(module);
if let Some(target) = &args.target {
let target = shellwords::split(&target)
let target = shellwords::split(target)
.map_err(|_| format_err!("Failed to parse target quotes"))?;
let target = db::Filter::parse(&target)?;
rl.set_target(Some(target));
}
let mut params = Params::from(args);
// The module was already set and loaded
params.module = None;
if args.dump_sandbox_init_msg {
cmd::run_cmd::dump_sandbox_init_msg(&mut rl, args.into(), Opt::collect(&args.options))
cmd::run_cmd::dump_sandbox_init_msg(&mut rl, params, Opt::collect(&args.options))
} else {
cmd::run_cmd::execute(&mut rl, args.into(), Opt::collect(&args.options))
cmd::run_cmd::execute(&mut rl, params, Opt::collect(&args.options))
}
}
@@ -123,11 +131,15 @@ fn run() -> Result<()> {
Some(SubCommand::Activity(activity)) => run_cmd(&args, activity, &config),
Some(SubCommand::Scope(scope)) => run_cmd(&args, scope, &config),
Some(SubCommand::Noscope(noscope)) => run_cmd(&args, noscope, &config),
Some(SubCommand::Autoscope(autoscope)) => run_cmd(&args, autoscope, &config),
Some(SubCommand::Autonoscope(autonoscope)) => run_cmd(&args, autonoscope, &config),
Some(SubCommand::Rescope(rescope)) => run_cmd(&args, rescope, &config),
Some(SubCommand::Workspace(workspace)) => workspace.run(&config),
Some(SubCommand::Fsck(fsck)) => run_cmd(&args, fsck, &config),
Some(SubCommand::Export(export)) => run_cmd(&args, export, &config),
Some(SubCommand::Cal(cal)) => run_cmd(&args, cal, &config),
Some(SubCommand::Notify(notify)) => run_cmd(&args, notify, &config),
Some(SubCommand::Stats(stats)) => run_cmd(&args, stats, &config),
Some(SubCommand::Repl) => repl::run(&config),
Some(SubCommand::Paths) => paths::run(&config),
Some(SubCommand::Completions(completions)) => complete::run_generate(&completions),

View File

@@ -115,6 +115,14 @@ impl Scopable for Account {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::accounts::dsl::*;
diesel::update(accounts.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::accounts::dsl::*;
@@ -342,7 +350,7 @@ impl Upsert for AccountUpdate {
}
fn apply(&self, db: &Database) -> Result<i32> {
db.update_account(&self)
db.update_account(self)
}
}

View File

@@ -83,6 +83,14 @@ impl Activity {
.optional()
.map_err(Error::from)
}
pub fn count(db: &Database) -> Result<usize> {
use crate::schema::activity::dsl::*;
activity.count()
.get_result::<i64>(db.db())
.map(|x| x as usize)
.map_err(Error::from)
}
}
pub struct ActivityFilter {

View File

@@ -94,6 +94,14 @@ impl Scopable for Breach {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::breaches::dsl::*;
diesel::update(breaches.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::breaches::dsl::*;

View File

@@ -113,6 +113,14 @@ impl Scopable for CryptoAddr {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::cryptoaddrs::dsl::*;
diesel::update(cryptoaddrs.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::cryptoaddrs::dsl::*;
@@ -326,7 +334,7 @@ impl Upsert for CryptoAddrUpdate {
}
fn apply(&self, db: &Database) -> Result<i32> {
db.update_cryptoaddr(&self)
db.update_cryptoaddr(self)
}
}

View File

@@ -99,6 +99,14 @@ impl Scopable for Device {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::devices::dsl::*;
diesel::update(devices.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::devices::dsl::*;

View File

@@ -94,6 +94,14 @@ impl Scopable for Domain {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::domains::dsl::*;
diesel::update(domains.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::domains::dsl::*;

View File

@@ -96,6 +96,14 @@ impl Scopable for Email {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::emails::dsl::*;
diesel::update(emails.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::emails::dsl::*;

View File

@@ -116,6 +116,14 @@ impl Scopable for Image {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::images::dsl::*;
diesel::update(images.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::images::dsl::*;
@@ -442,7 +450,7 @@ impl Upsert for ImageUpdate {
}
fn apply(&self, db: &Database) -> Result<i32> {
db.update_image(&self)
db.update_image(self)
}
}

View File

@@ -109,6 +109,14 @@ impl Scopable for IpAddr {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::ipaddrs::dsl::*;
diesel::update(ipaddrs.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::ipaddrs::dsl::*;

View File

@@ -243,6 +243,8 @@ pub trait Model: Sized {
pub trait Scopable: Model {
fn scoped(&self) -> bool;
fn set_scoped(&self, _db: &Database, _value: bool) -> Result<()>;
fn scope(db: &Database, filter: &Filter) -> Result<usize>;
fn noscope(db: &Database, filter: &Filter) -> Result<usize>;

View File

@@ -98,6 +98,14 @@ impl Scopable for Netblock {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::netblocks::dsl::*;
diesel::update(netblocks.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::netblocks::dsl::*;

View File

@@ -97,6 +97,14 @@ impl Scopable for Network {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::networks::dsl::*;
diesel::update(networks.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::networks::dsl::*;

View File

@@ -105,6 +105,14 @@ impl Scopable for PhoneNumber {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::phonenumbers::dsl::*;
diesel::update(phonenumbers.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::phonenumbers::dsl::*;

View File

@@ -106,6 +106,14 @@ impl Scopable for Port {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::ports::dsl::*;
diesel::update(ports.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::ports::dsl::*;
@@ -324,7 +332,7 @@ impl Upsert for PortUpdate {
}
fn apply(&self, db: &Database) -> Result<i32> {
db.update_port(&self)
db.update_port(self)
}
}

View File

@@ -98,6 +98,14 @@ impl Scopable for Subdomain {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::subdomains::dsl::*;
diesel::update(subdomains.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::subdomains::dsl::*;

View File

@@ -104,6 +104,14 @@ impl Scopable for Url {
!self.unscoped
}
fn set_scoped(&self, db: &Database, my_value: bool) -> Result<()> {
use crate::schema::urls::dsl::*;
diesel::update(urls.filter(id.eq(self.id)))
.set(unscoped.eq(!my_value))
.execute(db.db())?;
Ok(())
}
fn scope(db: &Database, filter: &Filter) -> Result<usize> {
use crate::schema::urls::dsl::*;
@@ -339,7 +347,7 @@ impl Upsert for UrlChangeset {
}
fn apply(&self, db: &Database) -> Result<i32> {
db.update_url(&self)
db.update_url(self)
}
}

View File

@@ -88,6 +88,7 @@ pub fn exec(rl: &mut Shell, module: &Module, ratelimit: &mut Ratelimiter, option
}
let params = Params {
module: None,
threads: 1,
verbose,
stdin: false,
@@ -95,13 +96,15 @@ pub fn exec(rl: &mut Shell, module: &Module, ratelimit: &mut Ratelimiter, option
grant_full_keyring: false,
deny_keyring: false,
exit_on_error: false,
proxy: None,
user_agent: None,
};
prepare_keyring(rl.keyring_mut(), &module, &params)?;
let args = vec![prepare_arg(&notification)?];
prepare_keyring(rl.keyring_mut(), module, &params)?;
let args = vec![prepare_arg(notification)?];
debug!("Executing notification module {:?}", module_name);
let errors = worker::spawn(rl, &module, ratelimit, args, &params, rl.config().network.proxy, options);
let errors = worker::spawn(rl, module, ratelimit, args, &params, rl.config().network.proxy, None, options);
debug!("Notification module {:?} exited with {:?} errors", module_name, errors);
Ok(errors)

View File

@@ -18,7 +18,7 @@ impl Glob {
loop {
match (filter.next(), topic.next()) {
(Some(filter), Some(topic)) => if !filter.matches(&topic) {
(Some(filter), Some(topic)) => if !filter.matches(topic) {
return None;
},
(None, None) => return Some(!self.inverse),

View File

@@ -23,7 +23,7 @@ pub struct Updater {
impl Updater {
pub fn new(config: &Config) -> Result<Updater> {
let client = Client::new(&config)?;
let client = Client::new(config)?;
Ok(Updater {
client,
})
@@ -97,7 +97,7 @@ pub fn run_publish(_args: &Args, publish: &Publish, config: &Config) -> Result<(
let session = auth::load_token()
.context("Failed to load auth token, login first")?;
let mut client = Client::new(&config)?;
let mut client = Client::new(config)?;
client.authenticate(session);
for path in &publish.paths {
@@ -247,7 +247,7 @@ fn write_tag(out: &mut String, color: Color, txt: &str) -> Result<()> {
}
pub fn run_search(library: &Library, search: &Search, config: &Config) -> Result<()> {
let client = Client::new(&config)?;
let client = Client::new(config)?;
let label = format!("Searching {:?}", search.query);
let modules = worker::spawn_fn(&label, || {

View File

@@ -35,10 +35,8 @@ impl<'a> Repl<'a> {
fn update_globals(&mut self) {
let mut globals = Vec::new();
for item in self.lua.globals_table().iter::<String, AnyLuaValue>() {
if let Some((k, _)) = item {
globals.push(k);
}
for (k, _) in self.lua.globals_table().iter::<String, AnyLuaValue>().flatten() {
globals.push(k);
}
if let Some(helper) = self.rl.helper_mut() {
debug!("updating globals: {:?}", globals);
@@ -84,6 +82,7 @@ pub fn run(config: &Config) -> Result<()> {
let keyring = Vec::new();
let dns_config = Resolver::from_system_v4()?;
let proxy = config.network.proxy;
let user_agent = config.network.user_agent.clone();
let cache_dir = paths::cache_dir()?;
let psl = PslReader::open(&cache_dir)?;
@@ -95,6 +94,7 @@ pub fn run(config: &Config) -> Result<()> {
keyring,
dns_config,
proxy,
user_agent,
options: HashMap::new(),
blobs: Vec::new(),
psl,

View File

@@ -199,7 +199,7 @@ fn gen_changeset2<T: Model, U: UpdateToChangeset<C>, C: Updateable<T>>(object: L
}
fn run_update(state: Arc<dyn State>, family: &str, object: AnyLuaValue, update: AnyLuaValue) -> Result<Option<i32>> {
let family = Family::from_str(&family)?;
let family = Family::from_str(family)?;
let object = LuaJsonValue::from(object);
let update = LuaJsonValue::from(update);

View File

@@ -36,6 +36,48 @@ pub fn img_load(lua: &mut hlua::Lua, state: Arc<dyn State>) {
}))
}
pub fn img_ahash(lua: &mut hlua::Lua, state: Arc<dyn State>) {
lua.set("img_ahash", hlua::function1(move |blob: String| -> Result<String> {
let img = state.get_blob(&blob)
.map_err(|err| state.set_error(err))?;
let img = gfx::load(&img.bytes)
.map_err(|err| state.set_error(err))?;
let hash = img.perception_hash(gfx::HashAlg::Mean);
Ok(hash)
}))
}
pub fn img_dhash(lua: &mut hlua::Lua, state: Arc<dyn State>) {
lua.set("img_dhash", hlua::function1(move |blob: String| -> Result<String> {
let img = state.get_blob(&blob)
.map_err(|err| state.set_error(err))?;
let img = gfx::load(&img.bytes)
.map_err(|err| state.set_error(err))?;
let hash = img.perception_hash(gfx::HashAlg::Gradient);
Ok(hash)
}))
}
pub fn img_phash(lua: &mut hlua::Lua, state: Arc<dyn State>) {
lua.set("img_phash", hlua::function1(move |blob: String| -> Result<String> {
let img = state.get_blob(&blob)
.map_err(|err| state.set_error(err))?;
let img = gfx::load(&img.bytes)
.map_err(|err| state.set_error(err))?;
let hash = img.perception_hash(gfx::HashAlg::Median);
Ok(hash)
}))
}
#[derive(Debug, Serialize)]
pub struct Nudity {
nude: bool,

View File

@@ -2,9 +2,8 @@ use crate::errors::*;
use crate::engine::ctx::State;
use crate::engine::structs::{byte_array, lua_bytes};
use crate::hlua::{self, AnyLuaValue};
use digest::{Digest, Update, BlockInput, FixedOutput, Reset};
use digest::generic_array::ArrayLength;
use hmac::{Hmac, Mac, NewMac};
use digest::{Digest, core_api::BlockSizeUser};
use hmac::{Mac, SimpleHmac};
use std::sync::Arc;
@@ -57,15 +56,12 @@ pub fn sha3_512(lua: &mut hlua::Lua, state: Arc<dyn State>) {
}
fn hmac<D>(secret: AnyLuaValue, msg: AnyLuaValue) -> Result<AnyLuaValue>
where
D: Update + BlockInput + FixedOutput + Reset + Default + Clone,
D::BlockSize: ArrayLength<u8> + Clone,
D::OutputSize: ArrayLength<u8>,
where D: Digest + BlockSizeUser
{
let secret = byte_array(secret)?;
let msg = byte_array(msg)?;
let mut mac = match Hmac::<D>::new_varkey(&secret) {
let mut mac = match SimpleHmac::<D>::new_from_slice(&secret) {
Ok(mac) => mac,
Err(_) => bail!("Invalid key length"),
};

View File

@@ -20,10 +20,10 @@ pub fn format_lua(out: &mut String, x: &AnyLuaValue) {
}
let mut key = String::new();
format_lua(&mut key, &k);
format_lua(&mut key, k);
let mut value = String::new();
format_lua(&mut value, &v);
format_lua(&mut value, v);
out.push_str(&format!("{}: {}", key, value));

View File

@@ -15,7 +15,7 @@ pub fn url_join(lua: &mut hlua::Lua, state: Arc<dyn State>) {
let url = base.join(&update)
.map_err(|err| state.set_error(Error::from(err)))?;
Ok(url.into_string())
Ok(url.into())
}))
}

View File

@@ -59,12 +59,10 @@ fn match_element_name(xml: &AnyLuaValue, name: &str) -> bool {
pub fn xml_named(lua: &mut hlua::Lua, _state: Arc<dyn State>) {
lua.set("xml_named", hlua::function2(move |xml: AnyLuaValue, name: String| -> AnyLuaValue {
if let Some(value) = get_children(xml) {
if let AnyLuaValue::LuaArray(arr) = value {
for (_, value) in arr {
if match_element_name(&value, &name) {
return value;
}
if let Some(AnyLuaValue::LuaArray(arr)) = get_children(xml) {
for (_, value) in arr {
if match_element_name(&value, &name) {
return value;
}
}
}

View File

@@ -54,6 +54,8 @@ pub fn init() -> Result<()> {
ctx.allow_syscall(Syscall::sched_yield)?;
ctx.allow_syscall(Syscall::setsockopt)?;
ctx.allow_syscall(Syscall::madvise)?;
ctx.allow_syscall(Syscall::rt_sigaction)?;
ctx.allow_syscall(Syscall::rseq)?;
#[cfg(target_arch = "x86")]
ctx.allow_syscall(Syscall::time)?;
ctx.allow_syscall(Syscall::clock_gettime)?;
@@ -70,9 +72,14 @@ pub fn init() -> Result<()> {
ctx.allow_syscall(Syscall::getpeername)?;
ctx.allow_syscall(Syscall::gettimeofday)?;
ctx.allow_syscall(Syscall::membarrier)?;
ctx.allow_syscall(Syscall::statx)?;
ctx.allow_syscall(Syscall::fstat)?;
ctx.allow_syscall(Syscall::lseek)?;
#[cfg(target_arch = "arm")]
ctx.allow_syscall(Syscall::_llseek)?;
ctx.set_action_for_syscall(Action::Errno(1), Syscall::openat)?;
#[cfg(not(any(target_arch = "x86_64", target_arch = "aarch64")))]
#[cfg(not(target_arch = "aarch64"))]
ctx.set_action_for_syscall(Action::Errno(1), Syscall::open)?;
ctx.load()?;

View File

@@ -61,7 +61,7 @@ impl Completer for CmdCompleter {
return Ok((0, vec![]));
}
let mut cmd = match shellwords::split(&line) {
let mut cmd = match shellwords::split(line) {
Ok(cmd) => cmd,
Err(_) => return Ok((0, vec![])),
};
@@ -191,18 +191,60 @@ impl Completer for CmdCompleter {
},
Command::Pkg => {
// we can only complete the 2nd argument
if args != 2 {
Ok((0, vec![]))
} else {
Ok(filter_options("pkg", &[
"list",
"install",
"search",
"reload",
"update",
"uninstall",
"quickstart",
], &cmd[1]))
let subcommand = cmd.get(1).map(|x| x.as_str());
let current = &cmd[args - 1];
let prev = &cmd[args - 2];
match (args, subcommand) {
(_, Some("list")) => {
let line = &line[..line.len() - current.len() - 1];
match prev.as_str() {
"--source" => Ok(filter_options(line, &[
"domains",
"subdomains",
"ipaddrs",
"urls",
"emails",
"phonenumbers",
"devices",
"networks",
"accounts",
"breaches",
"images",
"ports",
"netblocks",
"cryptoaddrs",
], current)),
"--stealth" => Ok(filter_options(line, &[
"loud",
"normal",
"passive",
"offline",
], current)),
_ => if current.starts_with('-') {
Ok(filter_options(line, &[
"--source",
"--stealth",
], current))
} else {
Ok((0, vec![]))
},
}
},
(2, Some(subcommand)) => {
Ok(filter_options("pkg", &[
"list",
"install",
"search",
"reload",
"update",
"uninstall",
"quickstart",
], subcommand))
},
(_, _) => Ok((0, vec![])),
}
},
Command::Noscope => self.filter("noscope", &cmd),
@@ -220,6 +262,7 @@ impl Completer for CmdCompleter {
Ok((0, results))
}
},
Command::Rescope => self.filter("rescope", &cmd),
Command::Scope => self.filter("scope", &cmd),
Command::Select => self.filter("select", &cmd),
Command::Workspace => {
@@ -251,13 +294,13 @@ impl Hinter for CmdCompleter {
type Hint = String;
#[inline]
fn hint(&self, _line: &str, _pos: usize, _ctx: &Context<'_>) -> Option<String> {
fn hint(&self, line: &str, pos: usize, ctx: &Context<'_>) -> Option<String> {
// None
match self.complete(_line, _pos, _ctx) {
match self.complete(line, pos, ctx) {
Ok((_, mut cmds)) => if cmds.len() == 1 {
// TODO: this fails if we complete a 2nd argument
let hint = cmds.remove(0);
Some(hint[_pos..].to_string())
Some(hint[pos..].to_string())
} else {
None
},

View File

@@ -41,10 +41,12 @@ pub enum Command {
Mod,
Noscope,
Pkg,
Rescope,
Run,
Scope,
Set,
Select,
Stats,
Target,
Use,
Quickstart,
@@ -55,6 +57,7 @@ pub enum Command {
Quit,
Interrupt,
Exec(String),
}
impl Command {
@@ -72,10 +75,12 @@ impl Command {
Command::Mod => "mod",
Command::Noscope => "noscope",
Command::Pkg => "pkg",
Command::Rescope => "rescope",
Command::Run => "run",
Command::Scope => "scope",
Command::Set => "set",
Command::Select => "select",
Command::Stats => "stats",
Command::Target => "target",
Command::Use => "use",
Command::Quickstart => "quickstart",
@@ -83,6 +88,7 @@ impl Command {
Command::Workspace => "workspace",
Command::Cal => "cal",
Command::Interrupt => unreachable!(),
Command::Exec(_) => unreachable!(),
}
}
@@ -100,10 +106,12 @@ impl Command {
Command::Keyring.as_str(),
Command::Noscope.as_str(),
Command::Pkg.as_str(),
Command::Rescope.as_str(),
Command::Run.as_str(),
Command::Scope.as_str(),
Command::Set.as_str(),
Command::Select.as_str(),
Command::Stats.as_str(),
Command::Target.as_str(),
Command::Use.as_str(),
Command::Quit.as_str(),
@@ -132,14 +140,16 @@ impl FromStr for Command {
"keyring" => Ok(Command::Keyring),
"mod" => Ok(Command::Mod),
"noscope" => Ok(Command::Noscope),
"pkg" => Ok(Command::Pkg),
"run" => Ok(Command::Run),
"scope" => Ok(Command::Scope),
"set" => Ok(Command::Set),
"pkg" => Ok(Command::Pkg),
"rescope" => Ok(Command::Rescope),
"run" => Ok(Command::Run),
"scope" => Ok(Command::Scope),
"set" => Ok(Command::Set),
"select" => Ok(Command::Select),
"target" => Ok(Command::Target),
"use" => Ok(Command::Use),
"quickstart" => Ok(Command::Quickstart),
"stats" => Ok(Command::Stats),
"target" => Ok(Command::Target),
"use" => Ok(Command::Use),
"quickstart" => Ok(Command::Quickstart),
"quit" => Ok(Command::Quit),
"workspace" => Ok(Command::Workspace),
"cal" => Ok(Command::Cal),
@@ -269,7 +279,7 @@ impl<'a> Shell<'a> {
#[inline(always)]
pub fn config(&self) -> &Config {
&self.config
self.config
}
#[inline(always)]
@@ -313,6 +323,10 @@ impl<'a> Shell<'a> {
return None;
}
if let Some(cmd) = line.strip_prefix('!') {
return Some((Command::Exec(cmd.to_string()), vec![]));
}
let cmd = match shellwords::split(&line) {
Ok(cmd) => cmd,
Err(err) => {
@@ -486,8 +500,8 @@ pub fn run_once(rl: &mut Shell) -> Result<bool> {
match line {
Some((Command::Activity, args)) => cmd::<activity_cmd::Args>(rl, &args)?,
Some((Command::Add, args)) => cmd::<add_cmd::Args>(rl, &args)?,
Some((Command::Autonoscope, args)) => autonoscope_cmd::run(rl, &args)?,
Some((Command::Autoscope, args)) => autoscope_cmd::run(rl, &args)?,
Some((Command::Autonoscope, args)) => cmd::<autonoscope_cmd::Args>(rl, &args)?,
Some((Command::Autoscope, args)) => cmd::<autoscope_cmd::Args>(rl, &args)?,
Some((Command::Back, _)) => if rl.take_module().is_none() {
return Ok(true);
},
@@ -500,10 +514,12 @@ pub fn run_once(rl: &mut Shell) -> Result<bool> {
},
Some((Command::Noscope, args)) => noscope_cmd::run(rl, &args)?,
Some((Command::Pkg, args)) => cmd::<pkg_cmd::ArgsInteractive>(rl, &args)?,
Some((Command::Rescope, args)) => cmd::<rescope_cmd::Args>(rl, &args)?,
Some((Command::Run, args)) => cmd::<run_cmd::Args>(rl, &args)?,
Some((Command::Scope, args)) => scope_cmd::run(rl, &args)?,
Some((Command::Set, args)) => set_cmd::run(rl, &args)?,
Some((Command::Select, args)) => cmd::<select_cmd::Args>(rl, &args)?,
Some((Command::Stats, args)) => cmd::<stats_cmd::Args>(rl, &args)?,
Some((Command::Target, args)) => target_cmd::run(rl, &args)?,
Some((Command::Use, args)) => use_cmd::run(rl, &args)?,
Some((Command::Quickstart, args)) => quickstart_cmd::run(rl, &args)?,
@@ -513,6 +529,9 @@ pub fn run_once(rl: &mut Shell) -> Result<bool> {
Some((Command::Exit, _)) => return Ok(true),
Some((Command::Quit, _)) => return Ok(true),
Some((Command::Interrupt, _)) => return Ok(true),
Some((Command::Exec(cmd), _)) => {
shell_exec(&cmd, rl.workspace())?;
},
None => (),
}
@@ -538,7 +557,7 @@ pub fn init<'a>(args: &Args, config: &'a Config, verbose_init: bool) -> Result<S
let psl = PslReader::open_or_download(&cache_dir,
|cb| worker::spawn_fn("Downloading public suffix list", cb, false))
.context("Failed to download public suffix list")?;
let library = Library::new(verbose_init, &config)?;
let library = Library::new(verbose_init, config)?;
let keyring = KeyRing::init()?;
if verbose_init && library.list().is_empty() {
@@ -550,15 +569,35 @@ pub fn init<'a>(args: &Args, config: &'a Config, verbose_init: bool) -> Result<S
if autoupdate.outdated() > 0 {
term::warn(&format!("{} modules are outdated, run: \x1b[1mpkg update\x1b[0m", autoupdate.outdated()));
}
autoupdate.check_background(&config, library.list());
autoupdate.check_background(config, library.list());
let mut rl = Shell::new(&config, db, blobs, psl, library, keyring);
let mut rl = Shell::new(config, db, blobs, psl, library, keyring);
ttl::reap_expired(&mut rl)?;
Ok(rl)
}
pub fn shell_exec(cmd: &str, workspace: &str) -> Result<()> {
use std::process::Command;
let sh = if cfg!(target_os = "windows") {
("cmd", "/C")
} else {
("sh", "-c")
};
Command::new(sh.0)
.arg(sh.1)
.arg(cmd)
.env("SN0INT_WORKSPACE", workspace)
.spawn()
.context("Failed to execute process")?
.wait()?;
Ok(())
}
pub fn run(args: &Args, config: &Config) -> Result<()> {
print_banner();

View File

@@ -199,7 +199,7 @@ impl DatabaseEvent {
}
fn on_activity<T: SpinLogger>(rl: &mut Shell, spinner: &mut T, ratelimit: &mut Ratelimiter, object: &NewActivity, verbose: u64) {
Self::spinner_log_new_activity(spinner, &object, verbose);
Self::spinner_log_new_activity(spinner, object, verbose);
// TODO: we don't want to copy the match arms everywhere
let mut subject = format!("New activity: {:?}", object.topic);
@@ -330,7 +330,7 @@ impl DatabaseEvent {
let result = match result {
Ok(id) => {
Self::on_update(rl, spinner, ratelimit, family, &value, &update);
Self::on_update(rl, spinner, ratelimit, family, value, update);
Ok(DatabaseResponse::Updated(id))
},
Err(err) => {
@@ -437,7 +437,15 @@ impl RatelimitEvent {
}
}
pub fn spawn(rl: &mut Shell, module: &Module, ratelimit: &mut Ratelimiter, args: Vec<(serde_json::Value, Option<String>, Vec<Blob>)>, params: &Params, proxy: Option<SocketAddr>, options: HashMap<String, String>) -> usize {
pub fn spawn(rl: &mut Shell,
module: &Module,
ratelimit: &mut Ratelimiter,
args: Vec<(serde_json::Value, Option<String>, Vec<Blob>)>,
params: &Params,
proxy: Option<SocketAddr>,
user_agent: Option<String>,
options: HashMap<String, String>,
) -> usize {
// This function hangs if args is empty, so return early if that's the case
if args.is_empty() {
return 0;
@@ -445,7 +453,7 @@ pub fn spawn(rl: &mut Shell, module: &Module, ratelimit: &mut Ratelimiter, args:
let verbose = params.verbose;
let has_stdin = params.stdin;
let keyring = rl.keyring().request_keys(&module);
let keyring = rl.keyring().request_keys(module);
let mut stack = StackedSpinners::new();
@@ -463,6 +471,7 @@ pub fn spawn(rl: &mut Shell, module: &Module, ratelimit: &mut Ratelimiter, args:
let tx = tx.clone();
let module = module.clone();
let keyring = keyring.clone();
let user_agent = user_agent.clone();
let options = options.clone();
let signal_register = rl.signal_register().clone();
pool.execute(move || {
@@ -476,7 +485,7 @@ pub fn spawn(rl: &mut Shell, module: &Module, ratelimit: &mut Ratelimiter, args:
}
tx.send(Event2::Start);
let event = match ipc::parent::run(module, &tx, arg, keyring, verbose, has_stdin, proxy, options, blobs) {
let event = match ipc::parent::run(module, &tx, arg, keyring, verbose, has_stdin, proxy, user_agent, options, blobs) {
Ok(exit) => exit,
Err(err) => ExitEvent::SetupFailed(err.to_string()),
};

View File

@@ -19,7 +19,7 @@ pub struct Workspace {
impl Workspace {
#[inline]
pub fn db_path(&self) -> Result<PathBuf> {
Ok(paths::workspace_dir(&self)?
Ok(paths::workspace_dir(self)?
.join("db.sqlite"))
}
@@ -43,7 +43,7 @@ impl Workspace {
pub fn delete(&self) -> Result<()> {
self.migrate()?;
let path = paths::workspace_dir(&self)?;
let path = paths::workspace_dir(self)?;
fs::remove_dir_all(path)?;
Ok(())