mirror of
https://github.com/R0m1k3/Loki.git
synced 2026-10-11 17:26:57 +02:00
Les deux reprises d'AJEAN arrivées en parallèle du chiffrement écrivaient en clair à côté de discussions chiffrées. - Résultats complets des outils (« voir plus ») : bucket toolres ajouté à encryptedBuckets, écrits et relus par putStoreBytes / getStoreBytesErr. Mémoire verrouillée : rien n'est écrit, le flux porte le résultat entier. Le nettoyage des orphelins ne tourne plus quand la mémoire est verrouillée : l'index des discussions revenait vide et TOUT passait pour orphelin. - Images du fil (chatimg/) : même enveloppe que les pages mémoire ; verrouillée, l'image reste en base64 dans le message. Elles n'étaient jamais effacées : comme un fil rechargé perd ses images (stripImageParts), celles de plus de 24 h partent au démarrage et à chaque bascule de discussion. Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
73 lines
2.6 KiB
Go
73 lines
2.6 KiB
Go
package loki
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
// Un résultat coupé est rangé à part, relu en entier, et part avec sa discussion.
|
|
func TestToolResultsSaveLoadDelete(t *testing.T) {
|
|
testHome(t)
|
|
_ = putStr(bkChat, ckActive, "c42")
|
|
long := strings.Repeat("ligne de sortie\n", 500)
|
|
ev := fillToolResult(&ToolUsedEvent{Name: "bash", Done: true}, long)
|
|
if ev.ResultID == "" || !strings.HasPrefix(ev.ResultID, "c42.") {
|
|
t.Fatalf("id attendu préfixé par la discussion, got %q", ev.ResultID)
|
|
}
|
|
if len([]rune(ev.Result)) != toolPreviewChars || ev.ResultChars != len([]rune(long)) {
|
|
t.Fatalf("aperçu %d car., taille %d — attendu %d et %d", len([]rune(ev.Result)), ev.ResultChars, toolPreviewChars, len([]rune(long)))
|
|
}
|
|
if got, ok := loadToolResult(ev.ResultID); !ok || got != long {
|
|
t.Fatal("le résultat complet doit être relisible")
|
|
}
|
|
short := fillToolResult(&ToolUsedEvent{}, "ok")
|
|
if short.ResultID != "" || short.Result != "ok" {
|
|
t.Fatal("un résultat court part entier, sans id")
|
|
}
|
|
deleteToolResultsFor("c42")
|
|
if _, ok := loadToolResult(ev.ResultID); ok {
|
|
t.Fatal("les résultats doivent partir avec leur discussion")
|
|
}
|
|
}
|
|
|
|
// Une sortie de commande coupée le DIT, avec sa taille réelle.
|
|
func TestTailOutputAnnouncesCut(t *testing.T) {
|
|
if tailOutput("court") != "court" {
|
|
t.Fatal("une sortie courte reste intacte")
|
|
}
|
|
out := tailOutput(strings.Repeat("x", toolMaxOutput+10))
|
|
if !strings.HasPrefix(out, "[sortie tronquée : ") || !strings.Contains(out, "au total") {
|
|
t.Fatalf("mention de coupe absente : %q", out[:60])
|
|
}
|
|
}
|
|
|
|
// Mémoire chiffrée : le résultat complet est chiffré en base et relu intact ;
|
|
// verrouillée, rien n'est écrit (le flux portera le résultat entier) et le
|
|
// nettoyage des orphelins ne tourne pas (l'index illisible ferait tout effacer).
|
|
func TestToolResultsChiffres(t *testing.T) {
|
|
testHome(t)
|
|
clearMemDEK()
|
|
if _, err := EnableMemEncryption("motdepasse-fort"); err != nil {
|
|
t.Fatalf("EnableMemEncryption: %v", err)
|
|
}
|
|
full := strings.Repeat("ligne de sortie\n", 500)
|
|
id := saveToolResult(full)
|
|
if id == "" {
|
|
t.Fatal("résultat non enregistré")
|
|
}
|
|
if !looksEncrypted(getBytes(bkToolRes, id)) {
|
|
t.Fatal("résultat en clair alors que la mémoire est chiffrée")
|
|
}
|
|
if got, ok := loadToolResult(id); !ok || got != full {
|
|
t.Fatal("résultat chiffré non restitué à l'identique")
|
|
}
|
|
clearMemDEK()
|
|
if saveToolResult("autre") != "" {
|
|
t.Fatal("mémoire verrouillée : rien n'aurait dû être écrit")
|
|
}
|
|
pruneToolResults()
|
|
if getBytes(bkToolRes, id) == nil {
|
|
t.Fatal("le nettoyage a effacé un résultat pendant que la mémoire était verrouillée")
|
|
}
|
|
}
|