forked from axiom-team/jaklis
Refactor project: Python way only
This commit is contained in:
parent
adbc6ecda2
commit
317c666fca
|
@ -1,3 +0,0 @@
|
||||||
dunikey="" # Chemin du fichier de trousseau Ḡ1 de l'émetteur, au format PubSec
|
|
||||||
pod="https://g1.data.duniter.fr" # Noeud Cecium+ utilisé pour l'envoi du message
|
|
||||||
#pod="https://data.gchange.fr" # Noeud Gchange utilisé pour l'envoi du message
|
|
70
deletemsg.sh
70
deletemsg.sh
|
@ -1,70 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
|
|
||||||
# ###
|
|
||||||
# Supprimer un message Cesium+
|
|
||||||
# ###
|
|
||||||
|
|
||||||
[[ ! -f .env ]] && cp .env.template .env
|
|
||||||
source .env
|
|
||||||
|
|
||||||
REGEX_PUBKEYS="[a-zA-Z0-9]{42,44}"
|
|
||||||
|
|
||||||
# Help display
|
|
||||||
helpOpt() {
|
|
||||||
echo -e "Cesium+ messages deleting
|
|
||||||
Default: ID in interactive mode.
|
|
||||||
Advice: Fill your .env file for more fun.
|
|
||||||
Example: $0 <ID du message>
|
|
||||||
|
|
||||||
\rOptions:
|
|
||||||
-id,--id <ID du message>\tDelete the message with ID <id>.
|
|
||||||
-k,--key <key>\t\tPath <key> to the pubsec keychain file of the issuer.
|
|
||||||
-o,--outbox\t\t\tDelete outbox messages instead of inbox
|
|
||||||
-h,--help\t\t\tDisplay this help"
|
|
||||||
}
|
|
||||||
|
|
||||||
# Parse options
|
|
||||||
declare -a args=($@)
|
|
||||||
for ((i=0; i<${#args[*]}; ++i))
|
|
||||||
do
|
|
||||||
case ${args[$i]} in
|
|
||||||
-o|--outbox) type=outbox;;
|
|
||||||
-id|--id) id="${args[$i+1]}"
|
|
||||||
[[ -z $id ]] && echo "Veuillez préciser un ID de message." && exit 1;;
|
|
||||||
-k|--key) dunikey="${args[$i+1]}"
|
|
||||||
[[ -z $dunikey ]] && echo "Veuillez préciser un fichier de trousseau." && exit 1;;
|
|
||||||
-h|--help) helpOpt && exit 0;;
|
|
||||||
*) [[ "${args[$i]}" == "-"* ]] && echo "Option inconnue." && exit 1;;
|
|
||||||
esac
|
|
||||||
done
|
|
||||||
|
|
||||||
if [[ -z $dunikey ]]; then
|
|
||||||
read -p "Fichier de trousseau: " dunikey
|
|
||||||
fi
|
|
||||||
issuer=$(./natools.py pk -f pubsec -k $dunikey)
|
|
||||||
|
|
||||||
if [[ -z $type ]]; then
|
|
||||||
type="inbox"
|
|
||||||
fi
|
|
||||||
[[ -z $id ]] && id=$1
|
|
||||||
if [[ -z $id ]]; then
|
|
||||||
read -p "ID de message: " ID
|
|
||||||
fi
|
|
||||||
|
|
||||||
[[ -z $(grep -Eo $REGEX_PUBKEYS <<<$issuer) ]] && echo "Le format de la clé publique de l'émetteur est invalide." && exit 1
|
|
||||||
|
|
||||||
times=$(date -u +'%s')
|
|
||||||
|
|
||||||
# Fabrication du hash
|
|
||||||
hashBrut="{\"version\":2,\"index\":\"message\",\"type\":\"$type\",\"id\":\"$id\",\"issuer\":\"$issuer\",\"time\":$times}"
|
|
||||||
hash=$(echo -n "$hashBrut" | sha256sum | cut -d ' ' -f1 | awk '{ print toupper($0) }')
|
|
||||||
|
|
||||||
# Fabrication de la signature
|
|
||||||
signature=$(echo -n "$hash" | ./natools.py sign -f pubsec -k $dunikey --noinc -O 64)
|
|
||||||
|
|
||||||
document="{\"hash\":\"$hash\",\"signature\":\"$signature\",${hashBrut:1}"
|
|
||||||
jq . <<<$document
|
|
||||||
|
|
||||||
# Envoi du document
|
|
||||||
curl -s -X POST "$pod/history/delete" -d "$document"
|
|
||||||
echo
|
|
17
dialog.py
17
dialog.py
|
@ -11,7 +11,7 @@ try:
|
||||||
except:
|
except:
|
||||||
sys.stderr.write("Please fill the path to your private key (PubSec), and a Cesium ES address in userEnv.py\n")
|
sys.stderr.write("Please fill the path to your private key (PubSec), and a Cesium ES address in userEnv.py\n")
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
from lib.cesiumMessaging import ReadFromCesium, SendToCesium
|
from lib.cesiumMessaging import ReadFromCesium, SendToCesium, DeleteFromCesium
|
||||||
|
|
||||||
# Parse arguments
|
# Parse arguments
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
|
@ -19,14 +19,15 @@ parser = argparse.ArgumentParser()
|
||||||
subparsers = parser.add_subparsers()
|
subparsers = parser.add_subparsers()
|
||||||
read_cmd = subparsers.add_parser('read', help="Lecture des messages")
|
read_cmd = subparsers.add_parser('read', help="Lecture des messages")
|
||||||
send_cmd = subparsers.add_parser('send', help="Envoi d'un message")
|
send_cmd = subparsers.add_parser('send', help="Envoi d'un message")
|
||||||
|
delete_cmd = subparsers.add_parser('delete', help="Supression d'un message")
|
||||||
|
|
||||||
if len(sys.argv) <= 1 or not sys.argv[1] in ('read','send'):
|
if len(sys.argv) <= 1 or not sys.argv[1] in ('read','send','delete'):
|
||||||
sys.stderr.write("Veuillez indiquer une commande valide:\n\n")
|
sys.stderr.write("Veuillez indiquer une commande valide:\n\n")
|
||||||
parser.print_help()
|
parser.print_help()
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
|
|
||||||
read_cmd.add_argument('-n', '--number',type=int, default=3, help="Affiche les NUMBER derniers messages")
|
read_cmd.add_argument('-n', '--number',type=int, default=3, help="Affiche les NUMBER derniers messages")
|
||||||
read_cmd.add_argument('-o', '--outbox', action='store_true', help="Lit les messages de la boite d'envoi")
|
read_cmd.add_argument('-o', '--outbox', action='store_true', help="Lit les messages envoyés")
|
||||||
|
|
||||||
send_cmd.add_argument('-d', '--destinataire', required=True, help="Destinataire du message")
|
send_cmd.add_argument('-d', '--destinataire', required=True, help="Destinataire du message")
|
||||||
send_cmd.add_argument('-t', '--titre', help="Titre du message à envoyer")
|
send_cmd.add_argument('-t', '--titre', help="Titre du message à envoyer")
|
||||||
|
@ -34,6 +35,8 @@ send_cmd.add_argument('-m', '--message', help="Message à envoyer")
|
||||||
send_cmd.add_argument('-f', '--fichier', help="Envoyer le message contenu dans le fichier 'FICHIER'")
|
send_cmd.add_argument('-f', '--fichier', help="Envoyer le message contenu dans le fichier 'FICHIER'")
|
||||||
send_cmd.add_argument('-o', '--outbox', action='store_true', help="Envoi le message sur la boite d'envoi")
|
send_cmd.add_argument('-o', '--outbox', action='store_true', help="Envoi le message sur la boite d'envoi")
|
||||||
|
|
||||||
|
delete_cmd.add_argument('-i', '--id', required=True, help="ID du message à supprimer")
|
||||||
|
delete_cmd.add_argument('-o', '--outbox', action='store_true', help="Suppression d'un message envoyé")
|
||||||
|
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
@ -42,8 +45,7 @@ args = parser.parse_args()
|
||||||
if sys.argv[1] == "read":
|
if sys.argv[1] == "read":
|
||||||
messages = ReadFromCesium(dunikey, pod)
|
messages = ReadFromCesium(dunikey, pod)
|
||||||
messages.read(args.number, args.outbox)
|
messages.read(args.number, args.outbox)
|
||||||
# print(messages.sendDocument(args.number, args.outbox)) # For debug, print complete JSON answer
|
elif sys.argv[1] == "send":
|
||||||
elif sys.argv[1] == "send":
|
|
||||||
if args.fichier:
|
if args.fichier:
|
||||||
with open(args.fichier, 'r') as f:
|
with open(args.fichier, 'r') as f:
|
||||||
titre = f.readline()
|
titre = f.readline()
|
||||||
|
@ -54,7 +56,10 @@ elif sys.argv[1] == "send":
|
||||||
else:
|
else:
|
||||||
titre = input("Indiquez le titre du message: ")
|
titre = input("Indiquez le titre du message: ")
|
||||||
msg = input("Indiquez le contenu du message: ")
|
msg = input("Indiquez le contenu du message: ")
|
||||||
|
|
||||||
messages = SendToCesium(dunikey, pod, args.destinataire, args.outbox)
|
messages = SendToCesium(dunikey, pod, args.destinataire, args.outbox)
|
||||||
messages.send(titre, msg)
|
messages.send(titre, msg)
|
||||||
|
|
||||||
|
elif sys.argv[1] == "delete":
|
||||||
|
messages = DeleteFromCesium(dunikey, pod, args.outbox)
|
||||||
|
messages.delete(args.id)
|
||||||
|
|
||||||
|
|
|
@ -1,6 +1,6 @@
|
||||||
#!/usr/bin/env python3
|
#!/usr/bin/env python3
|
||||||
|
|
||||||
import os, sys, ast, requests, json, base58, base64, time, string, random
|
import os, sys, ast, requests, json, base58, base64, time, string, random, re
|
||||||
from natools import fmt, sign, get_privkey, box_decrypt, box_encrypt
|
from natools import fmt, sign, get_privkey, box_decrypt, box_encrypt
|
||||||
from hashlib import sha256
|
from hashlib import sha256
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
|
@ -21,6 +21,10 @@ class ReadFromCesium:
|
||||||
self.recipient = get_privkey(dunikey, "pubsec").pubkey
|
self.recipient = get_privkey(dunikey, "pubsec").pubkey
|
||||||
self.pod = pod
|
self.pod = pod
|
||||||
|
|
||||||
|
if not re.match(r"(?![OIl])[1-9A-Za-z]{42,45}", self.recipient):
|
||||||
|
sys.stderr.write("La clé publique n'est pas au bon format.\n")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
# Configure JSON document to send
|
# Configure JSON document to send
|
||||||
def configDoc(self, nbrMsg, outbox):
|
def configDoc(self, nbrMsg, outbox):
|
||||||
if outbox:
|
if outbox:
|
||||||
|
@ -128,6 +132,10 @@ class SendToCesium:
|
||||||
nonce.append(random.choice(string.ascii_letters + string.digits))
|
nonce.append(random.choice(string.ascii_letters + string.digits))
|
||||||
self.nonce = base64.b64decode(''.join(nonce))
|
self.nonce = base64.b64decode(''.join(nonce))
|
||||||
|
|
||||||
|
if not re.match(r"(?![OIl])[1-9A-Za-z]{42,45}", recipient):
|
||||||
|
sys.stderr.write("La clé publique n'est pas au bon format.\n")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
|
||||||
def encryptMsg(self, msg):
|
def encryptMsg(self, msg):
|
||||||
return fmt["64"](box_encrypt(msg.encode(), get_privkey(self.dunikey, "pubsec"), self.issuer, self.nonce)).decode()
|
return fmt["64"](box_encrypt(msg.encode(), get_privkey(self.dunikey, "pubsec"), self.issuer, self.nonce)).decode()
|
||||||
|
@ -149,7 +157,6 @@ class SendToCesium:
|
||||||
|
|
||||||
# Build final document
|
# Build final document
|
||||||
finalDoc = '{' + '"hash":"{0}","signature":"{1}",'.format(hashDoc, signature) + document[1:]
|
finalDoc = '{' + '"hash":"{0}","signature":"{1}",'.format(hashDoc, signature) + document[1:]
|
||||||
# document="{\"hash\":\"$hash\",\"signature\":\"$signature\",${hashBrut:1}"
|
|
||||||
|
|
||||||
return finalDoc
|
return finalDoc
|
||||||
|
|
||||||
|
@ -168,10 +175,11 @@ class SendToCesium:
|
||||||
try:
|
try:
|
||||||
result = requests.post('{0}/message/{1}?pubkey={2}'.format(self.pod, boxType, self.recipient), headers=headers, data=document)
|
result = requests.post('{0}/message/{1}?pubkey={2}'.format(self.pod, boxType, self.recipient), headers=headers, data=document)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
sys.stderr.write("Impossible d'envyer le message:\n" + str(e))
|
sys.stderr.write("Impossible d'envoyer le message:\n" + str(e))
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
else:
|
else:
|
||||||
print(result.text)
|
print(colored("Message envoyé avec succès !", "green"))
|
||||||
|
print("ID: " + result.text)
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
@ -179,3 +187,71 @@ class SendToCesium:
|
||||||
finalDoc = self.configDoc(self.encryptMsg(title), self.encryptMsg(msg)) # Configure JSON document to send
|
finalDoc = self.configDoc(self.encryptMsg(title), self.encryptMsg(msg)) # Configure JSON document to send
|
||||||
self.sendDocument(finalDoc) # Send final signed document
|
self.sendDocument(finalDoc) # Send final signed document
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
#################### Deleting class ####################
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class DeleteFromCesium:
|
||||||
|
def __init__(self, dunikey, pod, outbox):
|
||||||
|
# Get my pubkey from my private key
|
||||||
|
try:
|
||||||
|
self.dunikey = dunikey
|
||||||
|
except:
|
||||||
|
sys.stderr.write("Please fill the path to your private key (PubSec)\n")
|
||||||
|
sys.exit(1)
|
||||||
|
|
||||||
|
self.issuer = get_privkey(dunikey, "pubsec").pubkey
|
||||||
|
self.pod = pod
|
||||||
|
self.outbox = outbox
|
||||||
|
|
||||||
|
|
||||||
|
def configDoc(self, idMsg):
|
||||||
|
# Get current timestamp
|
||||||
|
timeSent = int(time.time())
|
||||||
|
|
||||||
|
# Generate document to customize
|
||||||
|
|
||||||
|
if self.outbox:
|
||||||
|
boxType = "outbox"
|
||||||
|
else:
|
||||||
|
boxType = "inbox"
|
||||||
|
|
||||||
|
document = str({"version":2,"index":"message","type":boxType,"id":idMsg,"issuer":self.issuer,"time":timeSent}).replace("'",'"')
|
||||||
|
# "{\"version\":2,\"index\":\"message\",\"type\":\"$type\",\"id\":\"$id\",\"issuer\":\"$issuer\",\"time\":$times}"
|
||||||
|
|
||||||
|
# Generate hash of document
|
||||||
|
hashDoc = sha256(document.encode()).hexdigest().upper()
|
||||||
|
|
||||||
|
# Generate signature of document
|
||||||
|
signature = fmt["64"](sign(hashDoc.encode(), get_privkey(self.dunikey, "pubsec"))[:-len(hashDoc.encode())]).decode()
|
||||||
|
|
||||||
|
# Build final document
|
||||||
|
finalDoc = '{' + '"hash":"{0}","signature":"{1}",'.format(hashDoc, signature) + document[1:]
|
||||||
|
|
||||||
|
return finalDoc
|
||||||
|
|
||||||
|
def sendDocument(self, document):
|
||||||
|
headers = {
|
||||||
|
'Content-type': 'application/json',
|
||||||
|
}
|
||||||
|
|
||||||
|
# Send JSON document and get result
|
||||||
|
try:
|
||||||
|
result = requests.post('{0}/history/delete'.format(self.pod), headers=headers, data=document)
|
||||||
|
if result.status_code == 404:
|
||||||
|
raise ValueError("Message introuvable")
|
||||||
|
except Exception as e:
|
||||||
|
sys.stderr.write("Impossible de supprimer le message:\n" + str(e) + "\n")
|
||||||
|
sys.exit(1)
|
||||||
|
else:
|
||||||
|
print(colored("Message supprimé avec succès !", "green"))
|
||||||
|
return result
|
||||||
|
|
||||||
|
def delete(self, idMsg):
|
||||||
|
finalDoc = self.configDoc(idMsg)
|
||||||
|
self.sendDocument(finalDoc)
|
||||||
|
|
||||||
|
|
101
sendmsg.sh
101
sendmsg.sh
|
@ -1,101 +0,0 @@
|
||||||
#!/bin/bash
|
|
||||||
|
|
||||||
# ###
|
|
||||||
# Simple testeur d'envoi de message via la messagerie de Cesium ou de Gchange.
|
|
||||||
# ###
|
|
||||||
|
|
||||||
[[ -z $(which jq) || -z $(which curl) ]] && echo "Installation de jq et curl ..." && sudo apt update && sudo apt install jq curl -y
|
|
||||||
|
|
||||||
[[ ! -f .env ]] && cp .env.template .env
|
|
||||||
source .env
|
|
||||||
|
|
||||||
# Help display
|
|
||||||
helpOpt() {
|
|
||||||
echo -e "Cesium+ messages sender
|
|
||||||
Default: ask title, content and recipient in interactive mode.
|
|
||||||
Advice: Fill your .env file for more fun.
|
|
||||||
Example: $0 -f <Path of file content message> -r <recipient pubkey> -k <path of pubsec keychain of issuer>
|
|
||||||
|
|
||||||
\rOptions:
|
|
||||||
-t\t\t\t\tTest mode: Uses the \"test.txt\" file as well as the same recipient as the sender.
|
|
||||||
-f,--file <file>\t\tRead the file <file> with title in first line and content in rest of the file for the message.
|
|
||||||
-r,--recipient <pubkey>\tUses <pubkey> as recipient of the message.
|
|
||||||
-k,--key <key>\t\tPath <key> to the pubsec keychain file of the issuer.
|
|
||||||
-h,--help\t\t\tDisplay this help"
|
|
||||||
}
|
|
||||||
|
|
||||||
REGEX_PUBKEYS="[a-zA-Z0-9]{42,44}"
|
|
||||||
|
|
||||||
# Parse options
|
|
||||||
declare -a args=($@)
|
|
||||||
for ((i=0; i<${#args[*]}; ++i))
|
|
||||||
do
|
|
||||||
case ${args[$i]} in
|
|
||||||
-f|--file) file="${args[$i+1]}"
|
|
||||||
[[ ! -f $file ]] && echo "Le fichier $file n'existe pas." && exit 1;;
|
|
||||||
-t|--test) file="test.txt"
|
|
||||||
issuer=$(./natools.py pk -f pubsec -k $dunikey)
|
|
||||||
recipient=$issuer;;
|
|
||||||
-r|--recipient) recipient="${args[$i+1]}"
|
|
||||||
[[ -z $recipient ]] && echo "Veuillez préciser un destinataire." && exit 1;;
|
|
||||||
-k|--key) dunikey="${args[$i+1]}"
|
|
||||||
[[ -z $dunikey ]] && echo "Veuillez préciser un fichier de trousseau." && exit 1;;
|
|
||||||
-h|--help) helpOpt && exit 0;;
|
|
||||||
*) [[ "${args[$i]}" == "-"* ]] && echo "Option inconnue." && exit 1;;
|
|
||||||
esac
|
|
||||||
done
|
|
||||||
|
|
||||||
if [[ -z $dunikey ]]; then
|
|
||||||
read -p "Fichier de trousseau: " dunikey
|
|
||||||
fi
|
|
||||||
issuer=$(./natools.py pk -f pubsec -k $dunikey)
|
|
||||||
|
|
||||||
if [[ -z $file ]]; then
|
|
||||||
read -p "Objet du message: " title
|
|
||||||
read -p "Corps du message: " content
|
|
||||||
message="$title"$'\n'"$content"
|
|
||||||
else
|
|
||||||
message=$(cat $file)
|
|
||||||
fi
|
|
||||||
if [[ -z $recipient ]]; then
|
|
||||||
read -p "Destinataire: " recipient
|
|
||||||
fi
|
|
||||||
|
|
||||||
[[ -z $(grep -Eo $REGEX_PUBKEYS <<<$recipient) ]] && echo "Le format de la clé publique du destinataire est invalide." && exit 1
|
|
||||||
[[ -z $(grep -Eo $REGEX_PUBKEYS <<<$issuer) ]] && echo "Le format de la clé publique de l'émetteur est invalide." && exit 1
|
|
||||||
|
|
||||||
# Récupération et chiffrement du titre et du message
|
|
||||||
nonce=$(cat /dev/urandom | tr -dc 'a-zA-Z0-9' | fold -w 32 | head -n 1)
|
|
||||||
b58nonce=$(echo $nonce | base64 -d | base58)
|
|
||||||
title=$(head -n1 <<<$message | ./natools.py box-encrypt -n $nonce -f pubsec -k $dunikey -p $recipient -O 64)
|
|
||||||
content=$(tail -n+2 <<<$message | ./natools.py box-encrypt -n $nonce -f pubsec -k $dunikey -p $recipient -O 64)
|
|
||||||
|
|
||||||
times=$(date -u +'%s')
|
|
||||||
|
|
||||||
# Fabrication du hash
|
|
||||||
hashBrut="{\"issuer\":\"$issuer\",\"recipient\":\"$recipient\",\"title\":\"$title\",\"content\":\"$content\",\"time\":$times,\"nonce\":\"$b58nonce\",\"version\":2}"
|
|
||||||
hash=$(echo -n "$hashBrut" | sha256sum | cut -d ' ' -f1 | awk '{ print toupper($0) }')
|
|
||||||
|
|
||||||
# Fabrication de la signature
|
|
||||||
signature=$(echo -n "$hash" | ./natools.py sign -f pubsec -k $dunikey --noinc -O 64)
|
|
||||||
|
|
||||||
# Affichage du JSON final
|
|
||||||
document="{\"hash\":\"$hash\",\"signature\":\"$signature\",${hashBrut:1}"
|
|
||||||
jq . <<<$document
|
|
||||||
|
|
||||||
# Envoi du document
|
|
||||||
#curl -s -i -X OPTIONS "$pod/message/inbox?pubkey=$issuer" -d "pubkey=$issuer"
|
|
||||||
msgID=$(curl -s -X POST "$pod/message/inbox?pubkey=$recipient" -d "$document")
|
|
||||||
echo -e "\nMessage ID: $msgID"
|
|
||||||
|
|
||||||
|
|
||||||
### Tests mode ###
|
|
||||||
|
|
||||||
# Delete the message 1 second later, just for test
|
|
||||||
#sleep 1 && ./deletemsg.sh -id $msgID
|
|
||||||
|
|
||||||
# To put the message in outbox too
|
|
||||||
#curl -s -X POST "$pod/message/outbox?pubkey=$issuer" -d "$document"
|
|
||||||
|
|
||||||
# To put the message as read, add this at the end of document
|
|
||||||
#,\"read_signature\":\"$signature\"
|
|
Loading…
Reference in New Issue