Ensh/ensh.sh at main · 0xbbuddha/Ensh · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
#!/usr/bin/env bash
#
# ensh.sh — Chargeur principal de la bibliothèque Ensh
#
# Usage :
#   source /path/to/ensh.sh             # Charge uniquement le core
#   source /path/to/ensh.sh --all       # Charge tous les modules
#   source /path/to/ensh.sh --ldap      # Charge la pile LDAP
#   source /path/to/ensh.sh --smb       # Charge la pile SMB/MSRPC
#
# Une fois chargé, on peut importer des modules à la demande :
#   ensh::import crypto/nt_hash
#   ensh::import protocol/ntlm
#
# ─────────────────────────────────────────────────────────────────────────────

# Protection contre le double-chargement
[[ -n "${_ENSH_LOADED:-}" ]] && return 0
readonly _ENSH_LOADED=1

# Résolution du chemin racine de la bibliothèque, même si l'on est sourcé
# depuis un répertoire différent. On ne redéclare pas si déjà défini (ex: par run_tests.sh).
if [[ -z "${ENSH_ROOT:-}" ]]; then
    ENSH_ROOT="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
    readonly ENSH_ROOT
fi
readonly ENSH_LIB="${ENSH_ROOT}/lib"

# Version courante
readonly ENSH_VERSION="0.1.0"

# Bash 5.0 minimum requis (pour les tableaux associatifs améliorés, etc.)
if (( BASH_VERSINFO[0] < 5 )); then
    printf '[ensh] ERREUR : Bash >= 5.0 requis (actuel : %s)\n' "${BASH_VERSION}" >&2
    return 1
fi

# ── Registre des modules chargés ─────────────────────────────────────────────
#
# Clé   : chemin relatif du module (ex: "core/hex")
# Valeur: 1 si chargé
declare -gA _ENSH_MODULES=()

# ── Fonction d'import ────────────────────────────────────────────────────────
#
# ensh::import <module> [module...]
#
# Charge un ou plusieurs modules par leur chemin relatif depuis lib/.
# Les imports redondants sont silencieusement ignorés (idempotent).
#
# Exemples :
#   ensh::import core/hex
#   ensh::import crypto/nt_hash protocol/ntlm
#
ensh::import() {
    local module
    for module in "$@"; do
        # Déjà chargé ? On passe.
        [[ -n "${_ENSH_MODULES[${module}]:-}" ]] && continue

        local path="${ENSH_LIB}/${module}.sh"
        if [[ ! -f "${path}" ]]; then
            printf '[ensh] ERREUR : module introuvable : %s\n' "${module}" >&2
            return 1
        fi

        # Marquer avant le source pour éviter les cycles
        _ENSH_MODULES["${module}"]=1
        # shellcheck source=/dev/null
        source "${path}"
    done
}

# ── Chargement du core (toujours effectué) ────────────────────────────────────
ensh::import \
    core/log   \
    core/hex   \
    core/bytes \
    core/endian

# ── Presets de chargement ────────────────────────────────────────────────────

ensh::preset::ldap() {
    ensh::import \
        protocol/ldap/message           \
        protocol/ldap/bind              \
        protocol/ldap/filter            \
        protocol/ldap/search            \
        protocol/ldap/modify            \
        protocol/ldap/add               \
        protocol/ldap/session
}

ensh::preset::smb() {
    ensh::import \
        protocol/netbios/nbt            \
        protocol/netbios/nbns           \
        protocol/ntlm/flags             \
        protocol/ntlm/negotiate         \
        protocol/ntlm/challenge         \
        protocol/ntlm/authenticate      \
        protocol/smb/spnego             \
        protocol/smb/smb1/header        \
        protocol/smb/smb1/negotiate     \
        protocol/smb/smb1/session_setup \
        protocol/smb/smb1/tree_connect  \
        protocol/smb/smb2/header        \
        protocol/smb/smb2/negotiate     \
        protocol/smb/smb2/session_setup \
        protocol/smb/smb2/tree_connect  \
        protocol/smb/smb2/ioctl         \
        protocol/smb/smb2/signing       \
        protocol/smb/smb2/create        \
        protocol/smb/smb2/read          \
        protocol/smb/smb2/write         \
        protocol/smb/smb2/close         \
        protocol/smb/smb2/query_directory \
        protocol/smb/smb3/signing       \
        protocol/smb/session            \
        protocol/dcerpc/bind            \
        protocol/dcerpc/request         \
        protocol/msrpc/srvsvc           \
        protocol/msrpc/samr             \
        protocol/msrpc/lsarpc
}

ensh::preset::all() {
    ensh::import \
        encoding/utf16          \
        encoding/base64         \
        encoding/asn1           \
        crypto/md4              \
        crypto/hmac_md5         \
        crypto/hmac_sha256      \
        crypto/aes_cmac         \
        crypto/rc4              \
        crypto/nt_hash          \
        crypto/lm_hash          \
        transport/tcp           \
        transport/udp           \
        transport/tls           \
        protocol/llmnr/message  \
        protocol/llmnr/client   \
        protocol/llmnr/server   \
        protocol/netbios/nbt    \
        protocol/netbios/nbns   \
        protocol/kerberos/asreq \
        protocol/kerberos/tgsreq

    ensh::preset::ldap
    ensh::preset::smb
}

# ── Chargement des presets demandés ──────────────────────────────────────────

case "${1:-}" in
    --all)
        ensh::preset::all
        ;;
    --ldap)
        ensh::preset::ldap
        ;;
    --smb)
        ensh::preset::smb
        ;;
esac