Compare commits
22 Commits
0bb7ce12bd
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
|
de0589761e
|
|||
|
6880401ac8
|
|||
| 4ee522c26c | |||
| b19eb8f422 | |||
| 817a14704b | |||
| a3f2a5ee6d | |||
| 069fdf8215 | |||
| e6d05290fc | |||
| ca6a51bf36 | |||
| aea7050a72 | |||
| c2f301a6b1 | |||
| 950e4fb30e | |||
| 7cb3636328 | |||
| f2e25cb402 | |||
| 528faedfb4 | |||
| 0419af16b0 | |||
| 873f653b86 | |||
| 74a3fc3900 | |||
| 73459571d5 | |||
| 18b2a65885 | |||
| 9968d69fdd | |||
| 7f667c0cf7 |
@@ -1,25 +1,36 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
RPC "net/rpc"
|
||||
|
||||
logging "github.com/ipfs/go-log/v2"
|
||||
"github.com/spf13/cobra"
|
||||
|
||||
"github.com/k4lipso/pentapass/rpc"
|
||||
"github.com/k4lipso/pentapass/crypto"
|
||||
"github.com/k4lipso/pentapass/internal/rpc"
|
||||
"github.com/k4lipso/pentapass/internal/crypto"
|
||||
. "github.com/k4lipso/pentapass/internal/log"
|
||||
)
|
||||
|
||||
var (
|
||||
dbPath = flag.String("db", "./db", "db file path")
|
||||
logger = logging.Logger("globaldb")
|
||||
// topicName = "globaldb-example"
|
||||
// netTopic = "globaldb-example-net"
|
||||
// config = "globaldb-example"
|
||||
dbPath string
|
||||
debug bool
|
||||
rpcClient *RPC.Client
|
||||
)
|
||||
|
||||
|
||||
func parseSingleArgWithOptionalVault(args []string) (vault string, arg string){
|
||||
if len(args) == 1 {
|
||||
vault = "root"
|
||||
arg = args[0]
|
||||
} else {
|
||||
vault = args[0]
|
||||
arg = args[1]
|
||||
}
|
||||
|
||||
return vault, arg
|
||||
}
|
||||
|
||||
|
||||
// Create the root command
|
||||
var rootCmd = &cobra.Command{
|
||||
Use: "ppass",
|
||||
@@ -28,29 +39,25 @@ var rootCmd = &cobra.Command{
|
||||
|
||||
// Create the 'list' subcommand
|
||||
var listCmd = &cobra.Command{
|
||||
Use: "list",
|
||||
Use: "list [vault]",
|
||||
Short: "List all passwords",
|
||||
Args: cobra.RangeArgs(0, 1),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
//all, _ := cmd.Flags().GetBool("all")
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
vault := "root"
|
||||
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
if len(args) == 1 {
|
||||
vault = args[0]
|
||||
}
|
||||
|
||||
var names []string
|
||||
namespace := "root"
|
||||
err = client.Call("Query.GetAllNames", &namespace, &names)
|
||||
err := rpcClient.Call("Query.GetAllNames", &vault, &names)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
}
|
||||
|
||||
for _, name := range names {
|
||||
fmt.Println(name)
|
||||
Logger.Info(name)
|
||||
}
|
||||
|
||||
},
|
||||
@@ -59,140 +66,100 @@ var listCmd = &cobra.Command{
|
||||
var generateCmd = &cobra.Command{
|
||||
Use: "generate",
|
||||
Short: "Generate a Password",
|
||||
Args: cobra.ExactArgs(1),
|
||||
Args: cobra.RangeArgs(1, 2),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
serviceName := args[0]
|
||||
vault, serviceName := parseSingleArgWithOptionalVault(args)
|
||||
|
||||
var password *crypto.Password
|
||||
np := rpc.NamespaceService{ Namespace: "root", Service: serviceName }
|
||||
err = client.Call("Query.Generate", &np, &password)
|
||||
np := rpc.VaultService{ Vault: vault, Service: serviceName }
|
||||
err := rpcClient.Call("Query.Generate", &np, &password)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
}
|
||||
|
||||
fmt.Println(*password)
|
||||
Logger.Info(password.Password)
|
||||
},
|
||||
}
|
||||
|
||||
var showCmd = &cobra.Command{
|
||||
Use: "show",
|
||||
Short: "show a Password",
|
||||
Args: cobra.ExactArgs(1),
|
||||
Args: cobra.RangeArgs(1, 2),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
serviceName := args[0]
|
||||
vault, serviceName := parseSingleArgWithOptionalVault(args)
|
||||
|
||||
var password *crypto.Password
|
||||
np := rpc.NamespaceService{ Namespace: "root", Service: serviceName }
|
||||
err = client.Call("Query.Get", &np, &password)
|
||||
np := rpc.VaultService{ Vault: vault, Service: serviceName }
|
||||
err := rpcClient.Call("Query.Get", &np, &password)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Println(password.Password)
|
||||
Logger.Info(password.Password)
|
||||
},
|
||||
}
|
||||
|
||||
var namespaceCmd = &cobra.Command{
|
||||
Use: "namespace",
|
||||
Short: "Add, delete or list namespaces",
|
||||
var vaultCmd = &cobra.Command{
|
||||
Use: "vault",
|
||||
Short: "Add, delete or list vaults",
|
||||
}
|
||||
|
||||
var addNamespaceCmd = &cobra.Command{
|
||||
var addVaultCmd = &cobra.Command{
|
||||
Use: "add",
|
||||
Short: "add a namespace",
|
||||
Short: "add a vault",
|
||||
Args: cobra.ExactArgs(1),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
namespace := args[0]
|
||||
vault := args[0]
|
||||
|
||||
var placeholder int
|
||||
err = client.Call("Query.AddNamespace", &namespace, &placeholder)
|
||||
err := rpcClient.Call("Query.AddVault", &vault, &placeholder)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Printf("Namespace %s was added\n", namespace)
|
||||
Logger.Infof("Vault %s was added\n", vault)
|
||||
},
|
||||
}
|
||||
|
||||
var deleteNamespaceCmd = &cobra.Command{
|
||||
var deleteVaultCmd = &cobra.Command{
|
||||
Use: "delete",
|
||||
Short: "delete a namespace",
|
||||
Short: "delete a vault",
|
||||
Args: cobra.ExactArgs(1),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
namespace := args[0]
|
||||
vault := args[0]
|
||||
|
||||
var placeholder int
|
||||
err = client.Call("Query.DeleteNamespace", &namespace, &placeholder)
|
||||
err := rpcClient.Call("Query.DeleteVault", &vault, &placeholder)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Printf("Namespace %s was deleted\n", namespace)
|
||||
Logger.Infof("Vault %s was deleted\n", vault)
|
||||
},
|
||||
}
|
||||
|
||||
var listNamespacesCmd = &cobra.Command{
|
||||
var listVaultsCmd = &cobra.Command{
|
||||
Use: "list",
|
||||
Short: "list all namespaces",
|
||||
Short: "list all vaults",
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
var reply []string
|
||||
err = client.Call("Query.ListNamespaces", 0, &reply)
|
||||
err := rpcClient.Call("Query.ListVaults", 0, &reply)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Printf("Namespaces:\n")
|
||||
Logger.Info("Vaults:")
|
||||
for _, ns := range reply {
|
||||
fmt.Println(ns)
|
||||
Logger.Info(ns)
|
||||
}
|
||||
},
|
||||
}
|
||||
@@ -207,17 +174,15 @@ var infoCmd = &cobra.Command{
|
||||
Use: "info",
|
||||
Short: "print your own peerstring",
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
var result *string
|
||||
err := rpcClient.Call("Query.GetPeerString", 0, &result)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
var result *string
|
||||
err = client.Call("Query.GetPeerString", 0, &result)
|
||||
fmt.Println(*result)
|
||||
Logger.Info(*result)
|
||||
},
|
||||
}
|
||||
|
||||
@@ -227,35 +192,22 @@ var addPeerCmd = &cobra.Command{
|
||||
Short: "add a peer",
|
||||
Args: cobra.RangeArgs(1, 2),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
var namespace string
|
||||
var peerString string
|
||||
|
||||
if len(args) == 1 {
|
||||
namespace = "root"
|
||||
peerString = args[0]
|
||||
} else {
|
||||
namespace = args[0]
|
||||
peerString = args[1]
|
||||
}
|
||||
vault, peerString := parseSingleArgWithOptionalVault(args)
|
||||
|
||||
var success *bool
|
||||
np := rpc.NamespacePeer{ Namespace: namespace, Peer: peerString }
|
||||
err = client.Call("Query.AddPeer", &np, &success)
|
||||
np := rpc.VaultPeer{ Vault: vault, Peer: peerString }
|
||||
err := rpcClient.Call("Query.AddPeer", &np, &success)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Println(*success)
|
||||
if *success {
|
||||
Logger.Infof("Added peer: %s", peerString)
|
||||
} else {
|
||||
Logger.Infof("Could not add peer: %s", peerString)
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
@@ -264,35 +216,22 @@ var removePeerCmd = &cobra.Command{
|
||||
Short: "remove a peer",
|
||||
Args: cobra.RangeArgs(1, 2),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
var namespace string
|
||||
var peerString string
|
||||
|
||||
if len(args) == 1 {
|
||||
namespace = "root"
|
||||
peerString = args[0]
|
||||
} else {
|
||||
namespace = args[0]
|
||||
peerString = args[1]
|
||||
}
|
||||
vault, peerString := parseSingleArgWithOptionalVault(args)
|
||||
|
||||
var success *bool
|
||||
np := rpc.NamespacePeer{ Namespace: namespace, Peer: peerString }
|
||||
err = client.Call("Query.DeletePeer", &np, &success)
|
||||
np := rpc.VaultPeer{ Vault: vault, Peer: peerString }
|
||||
err := rpcClient.Call("Query.DeletePeer", &np, &success)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
fmt.Println(*success)
|
||||
if *success {
|
||||
Logger.Infof("Removed peer: %s", peerString)
|
||||
} else {
|
||||
Logger.Infof("Could not find peer: %s", peerString)
|
||||
}
|
||||
},
|
||||
}
|
||||
|
||||
@@ -301,62 +240,61 @@ var deleteCmd = &cobra.Command{
|
||||
Short: "delete a Password",
|
||||
Args: cobra.RangeArgs(1, 2),
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
dbPath, _ := cmd.Flags().GetString("db")
|
||||
client, err := rpc.Receive(dbPath)
|
||||
vault, serviceName := parseSingleArgWithOptionalVault(args)
|
||||
|
||||
var success *bool
|
||||
np := rpc.VaultService{ Vault: vault, Service: serviceName }
|
||||
err := rpcClient.Call("Query.Delete", &np, &success)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("dialing: %s\n", err)
|
||||
Logger.Error(err)
|
||||
return
|
||||
}
|
||||
|
||||
var namespace string
|
||||
var serviceName string
|
||||
|
||||
if len(args) == 1 {
|
||||
namespace = "root"
|
||||
serviceName = args[0]
|
||||
if *success {
|
||||
Logger.Infof("Deleted password: %s", serviceName)
|
||||
} else {
|
||||
namespace = args[0]
|
||||
serviceName = args[1]
|
||||
Logger.Infof("Could not find password: %s", serviceName)
|
||||
}
|
||||
|
||||
var success *bool
|
||||
np := rpc.NamespaceService{ Namespace: namespace, Service: serviceName }
|
||||
err = client.Call("Query.Delete", &np, &success)
|
||||
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
|
||||
fmt.Println(*success)
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
func init() {
|
||||
//listCmd.Flags().BoolP("all", "a", false, "List all items")
|
||||
rootCmd.PersistentFlags().String("db", "", "db path")
|
||||
rootCmd.PersistentFlags().StringVar(&dbPath, "db", "", "db path")
|
||||
rootCmd.PersistentFlags().BoolVar(&debug, "debug", false, "enable debug mode")
|
||||
rootCmd.MarkPersistentFlagRequired("db")
|
||||
|
||||
peerCmd.AddCommand(addPeerCmd)
|
||||
peerCmd.AddCommand(removePeerCmd)
|
||||
peerCmd.AddCommand(infoCmd)
|
||||
|
||||
namespaceCmd.AddCommand(addNamespaceCmd)
|
||||
namespaceCmd.AddCommand(deleteNamespaceCmd)
|
||||
namespaceCmd.AddCommand(listNamespacesCmd)
|
||||
vaultCmd.AddCommand(addVaultCmd)
|
||||
vaultCmd.AddCommand(deleteVaultCmd)
|
||||
vaultCmd.AddCommand(listVaultsCmd)
|
||||
|
||||
rootCmd.AddCommand(listCmd)
|
||||
rootCmd.AddCommand(generateCmd)
|
||||
rootCmd.AddCommand(showCmd)
|
||||
rootCmd.AddCommand(deleteCmd)
|
||||
rootCmd.AddCommand(peerCmd)
|
||||
rootCmd.AddCommand(namespaceCmd)
|
||||
rootCmd.AddCommand(vaultCmd)
|
||||
}
|
||||
|
||||
func main() {
|
||||
cobra.OnInitialize(func() {
|
||||
var tmpClient *RPC.Client
|
||||
tmpClient, err := rpc.Receive(dbPath)
|
||||
rpcClient = tmpClient
|
||||
|
||||
if err != nil {
|
||||
Logger.Fatalf("dialing: %s\n", err)
|
||||
return
|
||||
}
|
||||
|
||||
InitLogger(debug)
|
||||
})
|
||||
if err := rootCmd.Execute(); err != nil {
|
||||
fmt.Println(err)
|
||||
Logger.Error(err)
|
||||
os.Exit(1)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -3,76 +3,63 @@ package main
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
|
||||
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
||||
ipfslite "github.com/hsanjuan/ipfs-lite"
|
||||
badger "github.com/ipfs/go-ds-badger2"
|
||||
logging "github.com/ipfs/go-log/v2"
|
||||
|
||||
"github.com/k4lipso/pentapass/storage"
|
||||
"github.com/k4lipso/pentapass/rpc"
|
||||
"github.com/k4lipso/pentapass/crypto/age"
|
||||
"github.com/k4lipso/pentapass/internal/storage"
|
||||
"github.com/k4lipso/pentapass/internal/rpc"
|
||||
"github.com/k4lipso/pentapass/internal/crypto/age"
|
||||
. "github.com/k4lipso/pentapass/internal/log"
|
||||
)
|
||||
|
||||
var (
|
||||
topicNameFlag = flag.String("topicName", "pentapass", "name of topic to join")
|
||||
dbPath = flag.String("db", "./db", "db file path")
|
||||
nameSpace = flag.String("namespace", "crdt", "namespace")
|
||||
logger = logging.Logger("globaldb")
|
||||
// netTopic = "globaldb-example-net"
|
||||
// config = "globaldb-example"
|
||||
debug = flag.Bool("debug", false, "log debug information")
|
||||
)
|
||||
|
||||
|
||||
func main() {
|
||||
flag.Parse()
|
||||
InitLogger(*debug)
|
||||
|
||||
ctx := context.Background()
|
||||
data := *dbPath
|
||||
|
||||
Logger.Info("Initializing passd")
|
||||
Logger.Info("Looking for Keys...")
|
||||
key, err := age.LoadOrGenerateKeys(*dbPath + "/age.key")
|
||||
|
||||
if err != nil {
|
||||
panic(err)
|
||||
Logger.Panic(err)
|
||||
}
|
||||
|
||||
fmt.Printf("AgeKey: %s\n", key.String())
|
||||
fmt.Printf("AgePublicKey: %s\n", key.Recipient().String())
|
||||
|
||||
//cipher, err := age.Encrypt([]byte("Test Message"), []string{key.Recipient().String()})
|
||||
//fmt.Printf("Encrypted: %s\n", cipher)
|
||||
//decrypted, err := age.Decrypt(cipher, key)
|
||||
//fmt.Printf("Decrypted: %s\n", decrypted)
|
||||
Logger.Infof("Found Key: %s", key.Recipient().String())
|
||||
Logger.Info("Setting up DHT...")
|
||||
|
||||
h, dht, err := storage.SetupLibp2pHost(ctx, *dbPath)
|
||||
|
||||
fmt.Println(h.ID().String())
|
||||
|
||||
if err != nil {
|
||||
panic(err)
|
||||
Logger.Panic(err)
|
||||
}
|
||||
|
||||
Logger.Infof("Own ID: %s", h.ID().String())
|
||||
|
||||
ps, err := pubsub.NewGossipSub(ctx, h)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
Logger.Panic(err)
|
||||
}
|
||||
|
||||
//_, err = ps.Join(*topicNameFlag)
|
||||
//if err != nil {
|
||||
// panic(err)
|
||||
//}
|
||||
|
||||
go storage.DiscoverPeers(ctx, h, dht)
|
||||
|
||||
Logger.Info("Initializing datastore...")
|
||||
store, err := badger.NewDatastore(data, &badger.DefaultOptions)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
defer store.Close()
|
||||
|
||||
ipfs, err := ipfslite.New(ctx, store, nil, h, dht, nil)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
|
||||
storageHandler := storage.StorageHandler{
|
||||
@@ -84,20 +71,29 @@ func main() {
|
||||
Key: key,
|
||||
}
|
||||
|
||||
Cfg, err := storageHandler.NewConfig(*dbPath + "/config.json")
|
||||
Logger.Info("Starting peer discovery...")
|
||||
go storageHandler.RunBackground(ctx, h, dht)
|
||||
|
||||
configPath := *dbPath + "/config.json"
|
||||
Logger.Infof("Loading config from: %s", configPath)
|
||||
Cfg, err := storageHandler.NewConfig(configPath)
|
||||
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
|
||||
storageHandler.Config = Cfg
|
||||
storageHandler.ConfigPath = *dbPath + "/config.json"
|
||||
storageHandler.InitNamespaces()
|
||||
storageHandler.ConfigPath = configPath
|
||||
|
||||
for _, val := range storageHandler.Namespaces {
|
||||
Logger.Infof("Setting up Vaults...")
|
||||
storageHandler.InitVaults()
|
||||
|
||||
for _, val := range storageHandler.Vaults {
|
||||
defer val.Close()
|
||||
}
|
||||
|
||||
Logger.Infof("Initialization complete!")
|
||||
Logger.Infof("Serving RPC for ppass cli")
|
||||
rpc.StorageHandler = &storageHandler
|
||||
rpc.Serve(*dbPath)
|
||||
}
|
||||
|
||||
@@ -1,147 +0,0 @@
|
||||
package rsa
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"errors"
|
||||
"os"
|
||||
"log"
|
||||
|
||||
"crypto/rand"
|
||||
"crypto/rsa"
|
||||
"crypto/x509"
|
||||
"encoding/pem"
|
||||
)
|
||||
|
||||
|
||||
var (
|
||||
privateKeyName = "rsa_private_key.pem"
|
||||
publicKeyName = "rsa_public_key.pem"
|
||||
)
|
||||
|
||||
func GenerateKeys(path string) (*rsa.PrivateKey, *rsa.PublicKey, error) {
|
||||
// Generate a new RSA key pair
|
||||
privateKey, err := rsa.GenerateKey(rand.Reader, 2048)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
// Save the private key to a file
|
||||
privateKeyPath := path + privateKeyName
|
||||
privateKeyFile, err := os.Create(privateKeyPath)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
defer privateKeyFile.Close()
|
||||
|
||||
privateKeyPEM := pem.EncodeToMemory(&pem.Block{
|
||||
Type: "RSA PRIVATE KEY",
|
||||
Bytes: x509.MarshalPKCS1PrivateKey(privateKey),
|
||||
})
|
||||
_, err = privateKeyFile.Write(privateKeyPEM)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
fmt.Printf("Private key saved to %s\n", privateKeyPath)
|
||||
|
||||
// Extract and save the public key
|
||||
publicKeyPath := path + publicKeyName
|
||||
publicKeyFile, err := os.Create(publicKeyPath)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
defer publicKeyFile.Close()
|
||||
|
||||
publicKey := &privateKey.PublicKey
|
||||
publicKeyBytes, err := x509.MarshalPKIXPublicKey(publicKey)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
publicKeyPEM := pem.EncodeToMemory(&pem.Block{
|
||||
Type: "RSA PUBLIC KEY",
|
||||
Bytes: publicKeyBytes,
|
||||
})
|
||||
_, err = publicKeyFile.Write(publicKeyPEM)
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
fmt.Printf("Public key saved to %s\n", publicKeyPath)
|
||||
|
||||
return privateKey, publicKey, nil
|
||||
}
|
||||
|
||||
func LoadKeys(path string) (*rsa.PrivateKey, *rsa.PublicKey, error) {
|
||||
// Load private key from file
|
||||
privateKeyPath := path + privateKeyName
|
||||
privateKeyData, err := os.ReadFile(privateKeyPath)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to read private key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
block, _ := pem.Decode(privateKeyData)
|
||||
if block == nil || block.Type != "RSA PRIVATE KEY" {
|
||||
log.Fatalf("Failed to decode PEM block containing private key")
|
||||
return nil, nil, err
|
||||
}
|
||||
privateKey, err := x509.ParsePKCS1PrivateKey(block.Bytes)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to parse private key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
// Load public key from file
|
||||
publicKeyPath := path + publicKeyName
|
||||
publicKeyData, err := os.ReadFile(publicKeyPath)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to read public key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
block, _ = pem.Decode(publicKeyData)
|
||||
if block == nil || block.Type != "RSA PUBLIC KEY" {
|
||||
log.Fatalf("Failed to decode PEM block containing public key")
|
||||
return nil, nil, err
|
||||
}
|
||||
publicKeyInterface, err := x509.ParsePKIXPublicKey(block.Bytes)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to parse public key: %v", err)
|
||||
return nil, nil, err
|
||||
}
|
||||
publicKey := publicKeyInterface.(*rsa.PublicKey)
|
||||
|
||||
return privateKey, publicKey, nil
|
||||
}
|
||||
|
||||
|
||||
func LoadOrGenerateKeys(path string) (*rsa.PrivateKey, *rsa.PublicKey, error) {
|
||||
_, err := os.Open(path + privateKeyName)
|
||||
if errors.Is(err, os.ErrNotExist) {
|
||||
fmt.Println("Not Exists - Generate Keys")
|
||||
return GenerateKeys(path)
|
||||
}
|
||||
|
||||
return LoadKeys(path)
|
||||
}
|
||||
|
||||
|
||||
func Encrypt(data []byte, key *rsa.PublicKey) ([]byte, error) {
|
||||
ciphertext, err := rsa.EncryptPKCS1v15(rand.Reader, key, data)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to encrypt: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return ciphertext, nil
|
||||
}
|
||||
|
||||
func Decrypt(data []byte, key *rsa.PrivateKey) ([]byte, error) {
|
||||
decrypted, err := rsa.DecryptPKCS1v15(rand.Reader, key, data)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to decrypt: %v", err)
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return decrypted, nil
|
||||
|
||||
}
|
||||
24
go.mod
24
go.mod
@@ -9,11 +9,12 @@ require (
|
||||
github.com/ipfs/go-datastore v0.6.0
|
||||
github.com/ipfs/go-ds-badger2 v0.1.3
|
||||
github.com/ipfs/go-ds-crdt v0.5.2
|
||||
github.com/ipfs/go-log/v2 v2.5.1
|
||||
github.com/libp2p/go-libp2p v0.35.4
|
||||
github.com/libp2p/go-libp2p-kad-dht v0.25.2
|
||||
github.com/libp2p/go-libp2p-pubsub v0.11.0
|
||||
github.com/mudler/edgevpn v0.28.3
|
||||
github.com/multiformats/go-multiaddr v0.13.0
|
||||
github.com/sirupsen/logrus v1.8.1
|
||||
github.com/spf13/cobra v0.0.5
|
||||
)
|
||||
|
||||
@@ -23,6 +24,8 @@ require (
|
||||
github.com/alecthomas/units v0.0.0-20240626203959-61d1e3462e30 // indirect
|
||||
github.com/benbjohnson/clock v1.3.5 // indirect
|
||||
github.com/beorn7/perks v1.0.1 // indirect
|
||||
github.com/c-robinson/iplib v1.0.8 // indirect
|
||||
github.com/cenkalti/backoff/v4 v4.3.0 // indirect
|
||||
github.com/cespare/xxhash v1.1.0 // indirect
|
||||
github.com/cespare/xxhash/v2 v2.3.0 // indirect
|
||||
github.com/containerd/cgroups v1.1.0 // indirect
|
||||
@@ -69,6 +72,7 @@ require (
|
||||
github.com/ipfs/go-ipld-format v0.6.0 // indirect
|
||||
github.com/ipfs/go-ipld-legacy v0.2.1 // indirect
|
||||
github.com/ipfs/go-log v1.0.5 // indirect
|
||||
github.com/ipfs/go-log/v2 v2.5.1 // indirect
|
||||
github.com/ipfs/go-metrics-interface v0.0.1 // indirect
|
||||
github.com/ipfs/go-peertaskqueue v0.8.1 // indirect
|
||||
github.com/ipld/go-codec-dagpb v1.6.0 // indirect
|
||||
@@ -93,7 +97,7 @@ require (
|
||||
github.com/libp2p/go-yamux/v4 v4.0.1 // indirect
|
||||
github.com/marten-seemann/tcp v0.0.0-20210406111302-dfbc87cc63fd // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/miekg/dns v1.1.61 // indirect
|
||||
github.com/miekg/dns v1.1.62 // indirect
|
||||
github.com/mikioh/tcpinfo v0.0.0-20190314235526-30a79bb1804b // indirect
|
||||
github.com/mikioh/tcpopt v0.0.0-20190314235656-172688c1accc // indirect
|
||||
github.com/minio/sha256-simd v1.0.1 // indirect
|
||||
@@ -108,7 +112,7 @@ require (
|
||||
github.com/multiformats/go-multistream v0.5.0 // indirect
|
||||
github.com/multiformats/go-varint v0.0.7 // indirect
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
|
||||
github.com/onsi/ginkgo/v2 v2.19.1 // indirect
|
||||
github.com/onsi/ginkgo/v2 v2.20.1 // indirect
|
||||
github.com/opencontainers/runtime-spec v1.2.0 // indirect
|
||||
github.com/opentracing/opentracing-go v1.2.0 // indirect
|
||||
github.com/pbnjay/memory v0.0.0-20210728143218-7b4eea64cf58 // indirect
|
||||
@@ -154,14 +158,14 @@ require (
|
||||
go.uber.org/mock v0.4.0 // indirect
|
||||
go.uber.org/multierr v1.11.0 // indirect
|
||||
go.uber.org/zap v1.27.0 // indirect
|
||||
golang.org/x/crypto v0.25.0 // indirect
|
||||
golang.org/x/crypto v0.26.0 // indirect
|
||||
golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56 // indirect
|
||||
golang.org/x/mod v0.19.0 // indirect
|
||||
golang.org/x/net v0.27.0 // indirect
|
||||
golang.org/x/sync v0.7.0 // indirect
|
||||
golang.org/x/sys v0.22.0 // indirect
|
||||
golang.org/x/text v0.16.0 // indirect
|
||||
golang.org/x/tools v0.23.0 // indirect
|
||||
golang.org/x/mod v0.20.0 // indirect
|
||||
golang.org/x/net v0.28.0 // indirect
|
||||
golang.org/x/sync v0.8.0 // indirect
|
||||
golang.org/x/sys v0.24.0 // indirect
|
||||
golang.org/x/text v0.17.0 // indirect
|
||||
golang.org/x/tools v0.24.0 // indirect
|
||||
gonum.org/v1/gonum v0.15.0 // indirect
|
||||
google.golang.org/protobuf v1.34.2 // indirect
|
||||
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||
|
||||
51
go.sum
51
go.sum
@@ -33,6 +33,10 @@ github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM=
|
||||
github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw=
|
||||
github.com/bradfitz/go-smtpd v0.0.0-20170404230938-deb6d6237625/go.mod h1:HYsPBTaaSFSlLx/70C2HPIMNZpVV8+vt/A+FMnYP11g=
|
||||
github.com/buger/jsonparser v0.0.0-20181115193947-bf1c66bbce23/go.mod h1:bbYlZJ7hK1yFx9hf58LP0zeX7UjIGs20ufpu3evjr+s=
|
||||
github.com/c-robinson/iplib v1.0.8 h1:exDRViDyL9UBLcfmlxxkY5odWX5092nPsQIykHXhIn4=
|
||||
github.com/c-robinson/iplib v1.0.8/go.mod h1:i3LuuFL1hRT5gFpBRnEydzw8R6yhGkF4szNDIbF8pgo=
|
||||
github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
|
||||
github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
|
||||
github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU=
|
||||
github.com/cespare/xxhash v1.1.0 h1:a6HrQnmkObjyL+Gs60czilIUGqrzKutQD6XZog3p+ko=
|
||||
github.com/cespare/xxhash v1.1.0/go.mod h1:XrSqR1VqqWfGrhpAt58auRo0WTKS1nRRg3ghfAqPWnc=
|
||||
@@ -316,8 +320,8 @@ github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D
|
||||
github.com/matttproud/golang_protobuf_extensions v1.0.1/go.mod h1:D8He9yQNgCq6Z5Ld7szi9bcBfOoFv/3dc6xSMkL2PC0=
|
||||
github.com/microcosm-cc/bluemonday v1.0.1/go.mod h1:hsXNsILzKxV+sX77C5b8FSuKF00vh2OMYv+xgHpAMF4=
|
||||
github.com/miekg/dns v1.1.41/go.mod h1:p6aan82bvRIyn+zDIv9xYNUpwa73JcSh9BKwknJysuI=
|
||||
github.com/miekg/dns v1.1.61 h1:nLxbwF3XxhwVSm8g9Dghm9MHPaUZuqhPiGL+675ZmEs=
|
||||
github.com/miekg/dns v1.1.61/go.mod h1:mnAarhS3nWaW+NVP2wTkYVIZyHNJ098SJZUki3eykwQ=
|
||||
github.com/miekg/dns v1.1.62 h1:cN8OuEF1/x5Rq6Np+h1epln8OiyPWV+lROx9LxcGgIQ=
|
||||
github.com/miekg/dns v1.1.62/go.mod h1:mvDlcItzm+br7MToIKqkglaGhlFMHJ9DTNNWONWXbNQ=
|
||||
github.com/mikioh/tcp v0.0.0-20190314235350-803a9b46060c h1:bzE/A84HN25pxAuk9Eej1Kz9OUelF97nAc82bDquQI8=
|
||||
github.com/mikioh/tcp v0.0.0-20190314235350-803a9b46060c/go.mod h1:0SQS9kMwD2VsyFEB++InYyBJroV/FRmBgcydeSUcJms=
|
||||
github.com/mikioh/tcpinfo v0.0.0-20190314235526-30a79bb1804b h1:z78hV3sbSMAUoyUMM0I83AUIT6Hu17AWfgjzIbtrYFc=
|
||||
@@ -335,6 +339,8 @@ github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3Rllmb
|
||||
github.com/mr-tron/base58 v1.1.2/go.mod h1:BinMc/sQntlIE1frQmRFPUoPA1Zkr8VRgBdjWI2mNwc=
|
||||
github.com/mr-tron/base58 v1.2.0 h1:T/HDJBh4ZCPbU39/+c3rRvE0uKBQlU27+QI8LJ4t64o=
|
||||
github.com/mr-tron/base58 v1.2.0/go.mod h1:BinMc/sQntlIE1frQmRFPUoPA1Zkr8VRgBdjWI2mNwc=
|
||||
github.com/mudler/edgevpn v0.28.3 h1:yIuoMExwKHy/mNMBXIsm6FUFbnB9ELIxw9KXrK9KHDk=
|
||||
github.com/mudler/edgevpn v0.28.3/go.mod h1:HWcdIwj5zBgOD04Hn3I+J5E5Yb3kK1CwwWaEe6/QERo=
|
||||
github.com/multiformats/go-base32 v0.1.0 h1:pVx9xoSPqEIQG8o+UbAe7DNi51oej1NtK+aGkbLYxPE=
|
||||
github.com/multiformats/go-base32 v0.1.0/go.mod h1:Kj3tFY6zNr+ABYMqeUNeGvkIC/UYgtWibDcT0rExnbI=
|
||||
github.com/multiformats/go-base36 v0.2.0 h1:lFsAbNOGeKtuKozrtBsAkSVhv1p9D0/qedU9rQyccr0=
|
||||
@@ -363,10 +369,10 @@ github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 h1:C3w9PqII01/Oq
|
||||
github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ=
|
||||
github.com/neelance/astrewrite v0.0.0-20160511093645-99348263ae86/go.mod h1:kHJEU3ofeGjhHklVoIGuVj85JJwZ6kWPaJwCIxgnFmo=
|
||||
github.com/neelance/sourcemap v0.0.0-20151028013722-8c68805598ab/go.mod h1:Qr6/a/Q4r9LP1IltGz7tA7iOK1WonHEYhu1HRBA7ZiM=
|
||||
github.com/onsi/ginkgo/v2 v2.19.1 h1:QXgq3Z8Crl5EL1WBAC98A5sEBHARrAJNzAmMxzLcRF0=
|
||||
github.com/onsi/ginkgo/v2 v2.19.1/go.mod h1:O3DtEWQkPa/F7fBMgmZQKKsluAy8pd3rEQdrjkPb9zA=
|
||||
github.com/onsi/gomega v1.34.0 h1:eSSPsPNp6ZpsG8X1OVmOTxig+CblTc4AxpPBykhe2Os=
|
||||
github.com/onsi/gomega v1.34.0/go.mod h1:MIKI8c+f+QLWk+hxbePD4i0LMJSExPaZOVfkoex4cAo=
|
||||
github.com/onsi/ginkgo/v2 v2.20.1 h1:YlVIbqct+ZmnEph770q9Q7NVAz4wwIiVNahee6JyUzo=
|
||||
github.com/onsi/ginkgo/v2 v2.20.1/go.mod h1:lG9ey2Z29hR41WMVthyJBGUBcBhGOtoPF2VFMvBXFCI=
|
||||
github.com/onsi/gomega v1.34.1 h1:EUMJIKUjM8sKjYbtxQI9A4z2o+rruxnzNvpknOXie6k=
|
||||
github.com/onsi/gomega v1.34.1/go.mod h1:kU1QgUvBDLXBJq618Xvm2LUX6rSAfRaFRTcdOeDLwwY=
|
||||
github.com/opencontainers/runtime-spec v1.0.2/go.mod h1:jwyrGlmzljRJv/Fgzds9SsS/C5hL+LL3ko9hs6T5lQ0=
|
||||
github.com/opencontainers/runtime-spec v1.2.0 h1:z97+pHb3uELt/yiAWD691HNHQIF07bE7dzrbT927iTk=
|
||||
github.com/opencontainers/runtime-spec v1.2.0/go.mod h1:jwyrGlmzljRJv/Fgzds9SsS/C5hL+LL3ko9hs6T5lQ0=
|
||||
@@ -477,8 +483,11 @@ github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeV
|
||||
github.com/shurcooL/users v0.0.0-20180125191416-49c67e49c537/go.mod h1:QJTqeLYEDaXHZDBsXlPCDqdhQuJkuw4NOtaxYe3xii4=
|
||||
github.com/shurcooL/webdavfs v0.0.0-20170829043945-18c3829fa133/go.mod h1:hKmq5kWdCj2z2KEozexVbfEZIWiTjhE0+UjmZgPqehw=
|
||||
github.com/sirupsen/logrus v1.7.0/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
|
||||
github.com/smartystreets/assertions v1.2.0 h1:42S6lae5dvLc7BrLu/0ugRtcFVjoJNMC/N3yZFZkDFs=
|
||||
github.com/sirupsen/logrus v1.8.1 h1:dJKuHgqk1NNQlqoA6BTlM1Wf9DOH3NBjQyu0h9+AZZE=
|
||||
github.com/sirupsen/logrus v1.8.1/go.mod h1:yWOB1SBYBC5VeMP7gHvWumXLIWorT60ONWic61uBYv0=
|
||||
github.com/smartystreets/assertions v1.2.0/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo=
|
||||
github.com/smartystreets/assertions v1.13.0 h1:Dx1kYM01xsSqKPno3aqLnrwac2LetPvN23diwyr69Qs=
|
||||
github.com/smartystreets/assertions v1.13.0/go.mod h1:wDmR7qL282YbGsPy6H/yAsesrxfxaaSlJazyFLYVFx8=
|
||||
github.com/smartystreets/goconvey v1.7.2 h1:9RBaZCeXEQ3UselpuwUQHltGVXvdwm6cv1hgR6gDIPg=
|
||||
github.com/smartystreets/goconvey v1.7.2/go.mod h1:Vw0tHAZW6lzCRk3xgdin6fKYcG+G3Pg9vgXWeJpQFMM=
|
||||
github.com/sourcegraph/annotate v0.0.0-20160123013949-f4cad6c6324d/go.mod h1:UdhH50NIW0fCiwBSr0co2m7BnFLdv4fQTgdqdJTHFeE=
|
||||
@@ -577,8 +586,8 @@ golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5y
|
||||
golang.org/x/crypto v0.8.0/go.mod h1:mRqEX+O9/h5TFCrQhkgjo2yKi0yYA+9ecGkdQoHrywE=
|
||||
golang.org/x/crypto v0.12.0/go.mod h1:NF0Gs7EO5K4qLn+Ylc+fih8BSTeIjAP05siRnAh98yw=
|
||||
golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg=
|
||||
golang.org/x/crypto v0.25.0 h1:ypSNr+bnYL2YhwoMt2zPxHFmbAN1KZs/njMG3hxUp30=
|
||||
golang.org/x/crypto v0.25.0/go.mod h1:T+wALwcMOSE0kXgUAnPAHqTLW+XHgcELELW8VaDgm/M=
|
||||
golang.org/x/crypto v0.26.0 h1:RrRspgV4mU+YwB4FYnuBoKsUapNIL5cohGAmSH3azsw=
|
||||
golang.org/x/crypto v0.26.0/go.mod h1:GY7jblb9wI+FOo5y8/S2oY4zWP07AkOJ4+jxCqdqn54=
|
||||
golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA=
|
||||
golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56 h1:2dVuKD2vS7b0QIHQbpyTISPd0LeHDbnYEryqj5Q1ug8=
|
||||
golang.org/x/exp v0.0.0-20240719175910-8a7402abbf56/go.mod h1:M4RDyNAINzryxdtnbRXRL/OHtkFuWGRjvuhBJpk2IlY=
|
||||
@@ -595,8 +604,8 @@ golang.org/x/mod v0.3.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||
golang.org/x/mod v0.4.2/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA=
|
||||
golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4=
|
||||
golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs=
|
||||
golang.org/x/mod v0.19.0 h1:fEdghXQSo20giMthA7cd28ZC+jts4amQ3YMXiP5oMQ8=
|
||||
golang.org/x/mod v0.19.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
|
||||
golang.org/x/mod v0.20.0 h1:utOm6MM3R3dnawAiJgn0y+xvuYRsm1RKM/4giyfDgV0=
|
||||
golang.org/x/mod v0.20.0/go.mod h1:hTbmBsO62+eylJbnUtE2MGJUyE7QWk4xUqPFrRgJ+7c=
|
||||
golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||
golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||
golang.org/x/net v0.0.0-20180906233101-161cd47e91fd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4=
|
||||
@@ -620,8 +629,8 @@ golang.org/x/net v0.9.0/go.mod h1:d48xBJpPfHeWQsugry2m+kC02ZBRGRgulfHnEXEuWns=
|
||||
golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg=
|
||||
golang.org/x/net v0.14.0/go.mod h1:PpSgVXXLK0OxS0F31C1/tv6XNguvCrnXIDrFMspZIUI=
|
||||
golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY=
|
||||
golang.org/x/net v0.27.0 h1:5K3Njcw06/l2y9vpGCSdcxWOYHOUk3dVNGDXN+FvAys=
|
||||
golang.org/x/net v0.27.0/go.mod h1:dDi0PyhWNoiUOrAS8uXv/vnScO4wnHQO4mj9fn/RytE=
|
||||
golang.org/x/net v0.28.0 h1:a9JDOJc5GMUJ0+UDqmLT86WiEy7iWyIhz8gz8E4e5hE=
|
||||
golang.org/x/net v0.28.0/go.mod h1:yqtgsTWOOnlGLG9GFRrK3++bGOUEkNBoHZc8MEDWPNg=
|
||||
golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
||||
golang.org/x/oauth2 v0.0.0-20181017192945-9dcd33a902f4/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
||||
golang.org/x/oauth2 v0.0.0-20181203162652-d668ce993890/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U=
|
||||
@@ -637,8 +646,8 @@ golang.org/x/sync v0.0.0-20201020160332-67f06af15bc9/go.mod h1:RxMgew5VJxzue5/jJ
|
||||
golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.7.0 h1:YsImfSBoP9QPYL0xyKJPq0gcaJdG3rInoqxTWbfQu9M=
|
||||
golang.org/x/sync v0.7.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
|
||||
golang.org/x/sync v0.8.0 h1:3NFvSEYkUoMifnESzZl15y791HH1qU2xm6eCJU5ZPXQ=
|
||||
golang.org/x/sync v0.8.0/go.mod h1:Czt+wKu1gCyEFDUtn0jG5QVvpJ6rzVqr5aXyt9drQfk=
|
||||
golang.org/x/sys v0.0.0-20180810173357-98c5dad5d1a0/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
golang.org/x/sys v0.0.0-20180909124046-d0be0721c37e/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
||||
@@ -669,8 +678,8 @@ golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.9.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.11.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/sys v0.22.0 h1:RI27ohtqKCnwULzJLqkv897zojh5/DwS/ENaMzUOaWI=
|
||||
golang.org/x/sys v0.22.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/sys v0.24.0 h1:Twjiwq9dn6R1fQcyiK+wQyHWfaz/BJB+YIpzU/Cv3Xg=
|
||||
golang.org/x/sys v0.24.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
||||
golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
|
||||
golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
|
||||
golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
|
||||
@@ -686,8 +695,8 @@ golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8=
|
||||
golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8=
|
||||
golang.org/x/text v0.12.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE=
|
||||
golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU=
|
||||
golang.org/x/text v0.16.0 h1:a94ExnEXNtEwYLGJSIUxnWoxoRz/ZcCsV63ROupILh4=
|
||||
golang.org/x/text v0.16.0/go.mod h1:GhwF1Be+LQoKShO3cGOHzqOgRrGaYc9AvblQOmPVHnI=
|
||||
golang.org/x/text v0.17.0 h1:XtiM5bkSOt+ewxlOE/aE/AKEHibwj/6gvWMl9Rsh0Qc=
|
||||
golang.org/x/text v0.17.0/go.mod h1:BuEKDfySbSR4drPmRPG/7iBdf8hvFMuRexcpahXilzY=
|
||||
golang.org/x/time v0.0.0-20180412165947-fbb02b2291d2/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
|
||||
golang.org/x/time v0.0.0-20181108054448-85acf8d2951c/go.mod h1:tRJNPiyCQ0inRvYxbN9jk5I+vvW/OXSQhTDSoE431IQ=
|
||||
golang.org/x/time v0.5.0 h1:o7cqy6amK/52YcAKIPlM3a+Fpj35zvRj2TP+e1xFSfk=
|
||||
@@ -711,8 +720,8 @@ golang.org/x/tools v0.0.0-20210106214847-113979e3529a/go.mod h1:emZCQorbCU4vsT4f
|
||||
golang.org/x/tools v0.1.5/go.mod h1:o0xws9oXOQQZyjljx8fwUC0k7L1pTE6eaCbjGeHmOkk=
|
||||
golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc=
|
||||
golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU=
|
||||
golang.org/x/tools v0.23.0 h1:SGsXPZ+2l4JsgaCKkx+FQ9YZ5XEtA1GZYuoDjenLjvg=
|
||||
golang.org/x/tools v0.23.0/go.mod h1:pnu6ufv6vQkll6szChhK3C3L/ruaIv5eBeztNG8wtsI=
|
||||
golang.org/x/tools v0.24.0 h1:J1shsA93PJUEVaUSaay7UXAyE8aimq3GW0pjlolpa24=
|
||||
golang.org/x/tools v0.24.0/go.mod h1:YhNqVBIfWHdzvTLs0d8LCuMhkKUgSUKldakyV7W/WDQ=
|
||||
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
|
||||
|
||||
@@ -8,6 +8,8 @@ import (
|
||||
"errors"
|
||||
|
||||
"filippo.io/age"
|
||||
|
||||
. "github.com/k4lipso/pentapass/internal/log"
|
||||
)
|
||||
|
||||
var (
|
||||
@@ -30,7 +32,7 @@ func GenerateAgeKey(filename string) (*age.X25519Identity, error) {
|
||||
return nil, fmt.Errorf("failed to save private key to file: %w", err)
|
||||
}
|
||||
|
||||
fmt.Printf("Private key saved to %s\n", filename)
|
||||
Logger.Infof("Private key saved to %s\n", filename)
|
||||
return identity, nil
|
||||
}
|
||||
|
||||
@@ -53,7 +55,7 @@ func LoadAgeKey(filename string) (*age.X25519Identity, error) {
|
||||
func LoadOrGenerateKeys(filename string) (*age.X25519Identity, error) {
|
||||
_, err := os.Open(filename)
|
||||
if errors.Is(err, os.ErrNotExist) {
|
||||
fmt.Println("Not Exists - Generate Keys")
|
||||
Logger.Info("No Key found. Generating Key")
|
||||
return GenerateAgeKey(filename)
|
||||
}
|
||||
|
||||
48
internal/log/log.go
Normal file
48
internal/log/log.go
Normal file
@@ -0,0 +1,48 @@
|
||||
package log
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"os"
|
||||
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
var Logger = logrus.New()
|
||||
|
||||
type CustomTextFormatter struct {
|
||||
logrus.TextFormatter
|
||||
}
|
||||
|
||||
func (f *CustomTextFormatter) Format(entry *logrus.Entry) ([]byte, error) {
|
||||
if entry.Level != logrus.InfoLevel {
|
||||
return f.TextFormatter.Format(entry)
|
||||
}
|
||||
|
||||
var b bytes.Buffer
|
||||
b.WriteString(entry.Message)
|
||||
b.WriteByte('\n')
|
||||
return b.Bytes(), nil
|
||||
}
|
||||
|
||||
|
||||
func InitLogger(debug bool) {
|
||||
if debug {
|
||||
Logger.SetFormatter(&logrus.TextFormatter{
|
||||
DisableLevelTruncation: true,
|
||||
PadLevelText: true,
|
||||
DisableTimestamp: true,
|
||||
})
|
||||
Logger.SetLevel(logrus.DebugLevel)
|
||||
} else {
|
||||
Logger.SetFormatter(&CustomTextFormatter{
|
||||
logrus.TextFormatter{
|
||||
DisableLevelTruncation: true,
|
||||
PadLevelText: true,
|
||||
DisableTimestamp: true,
|
||||
},
|
||||
})
|
||||
Logger.SetLevel(logrus.InfoLevel)
|
||||
}
|
||||
|
||||
Logger.SetOutput(os.Stdout)
|
||||
}
|
||||
@@ -10,30 +10,31 @@ import (
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"github.com/k4lipso/pentapass/storage"
|
||||
"github.com/k4lipso/pentapass/crypto"
|
||||
"github.com/k4lipso/pentapass/crypto/age"
|
||||
"github.com/k4lipso/pentapass/internal/storage"
|
||||
"github.com/k4lipso/pentapass/internal/crypto"
|
||||
"github.com/k4lipso/pentapass/internal/crypto/age"
|
||||
. "github.com/k4lipso/pentapass/internal/log"
|
||||
)
|
||||
|
||||
var StorageHandler *storage.StorageHandler
|
||||
|
||||
type Query int
|
||||
|
||||
type NamespaceService struct {
|
||||
Namespace string
|
||||
type VaultService struct {
|
||||
Vault string
|
||||
Service string
|
||||
}
|
||||
|
||||
type NamespacePeer struct {
|
||||
Namespace string
|
||||
type VaultPeer struct {
|
||||
Vault string
|
||||
Peer string
|
||||
}
|
||||
|
||||
func (t *Query) Generate(np *NamespaceService, reply *crypto.Password) error {
|
||||
val, ok := StorageHandler.Namespaces[np.Namespace]
|
||||
func (t *Query) Generate(np *VaultService, reply *crypto.Password) error {
|
||||
val, ok := StorageHandler.Vaults[np.Vault]
|
||||
|
||||
if !ok {
|
||||
return fmt.Errorf("Namespace does not exist")
|
||||
return fmt.Errorf("Vault does not exist")
|
||||
}
|
||||
|
||||
password := crypto.NewPassword(crypto.DEFAULT_LENGTH)
|
||||
@@ -58,12 +59,12 @@ func (t *Query) Generate(np *NamespaceService, reply *crypto.Password) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *Query) Get(np *NamespaceService, reply *crypto.Password) error {
|
||||
namespace := np.Namespace
|
||||
val, ok := StorageHandler.Namespaces[namespace]
|
||||
func (t *Query) Get(np *VaultService, reply *crypto.Password) error {
|
||||
vault := np.Vault
|
||||
val, ok := StorageHandler.Vaults[vault]
|
||||
|
||||
if !ok {
|
||||
return fmt.Errorf("Namespace does not exist")
|
||||
return fmt.Errorf("Vault does not exist")
|
||||
}
|
||||
|
||||
v, err := val.GetPassword(np.Service)
|
||||
@@ -84,18 +85,18 @@ func (t *Query) GetPeerString(_ *int, result *string) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *Query) AddPeer(np *NamespacePeer, success *bool) error {
|
||||
namespace := np.Namespace
|
||||
val, ok := StorageHandler.Namespaces[namespace]
|
||||
func (t *Query) AddPeer(np *VaultPeer, success *bool) error {
|
||||
vault := np.Vault
|
||||
val, ok := StorageHandler.Vaults[vault]
|
||||
|
||||
if !ok {
|
||||
return fmt.Errorf("Namespace does not exist")
|
||||
return fmt.Errorf("Vault does not exist")
|
||||
}
|
||||
|
||||
peer, err := storage.PeerFromString(np.Peer)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error parsing peer string: %s\n", err)
|
||||
Logger.Infof("Error parsing peer string: %s\n", err)
|
||||
*success = false
|
||||
return err
|
||||
}
|
||||
@@ -106,18 +107,18 @@ func (t *Query) AddPeer(np *NamespacePeer, success *bool) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *Query) DeletePeer(np *NamespacePeer, success *bool) error {
|
||||
namespace := np.Namespace
|
||||
val, ok := StorageHandler.Namespaces[namespace]
|
||||
func (t *Query) DeletePeer(np *VaultPeer, success *bool) error {
|
||||
vault := np.Vault
|
||||
val, ok := StorageHandler.Vaults[vault]
|
||||
|
||||
if !ok {
|
||||
return fmt.Errorf("Namespace does not exist")
|
||||
return fmt.Errorf("Vault does not exist")
|
||||
}
|
||||
|
||||
peer, err := storage.PeerFromString(np.Peer)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error parsing peer string: %s\n", err)
|
||||
Logger.Infof("Error parsing peer string: %s\n", err)
|
||||
*success = false
|
||||
return err
|
||||
}
|
||||
@@ -128,17 +129,20 @@ func (t *Query) DeletePeer(np *NamespacePeer, success *bool) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *Query) Delete(np *NamespaceService, success *bool) error {
|
||||
namespace := np.Namespace
|
||||
val, ok := StorageHandler.Namespaces[namespace]
|
||||
func (t *Query) Delete(np *VaultService, success *bool) error {
|
||||
vault := np.Vault
|
||||
val, ok := StorageHandler.Vaults[vault]
|
||||
|
||||
if !ok {
|
||||
return fmt.Errorf("Namespace does not exist")
|
||||
return fmt.Errorf("Vault does not exist")
|
||||
}
|
||||
|
||||
err := val.Delete(np.Service)
|
||||
if err != nil {
|
||||
*success = false
|
||||
if err == storage.ErrNotFound {
|
||||
return nil
|
||||
}
|
||||
return err
|
||||
}
|
||||
|
||||
@@ -146,26 +150,25 @@ func (t *Query) Delete(np *NamespaceService, success *bool) error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *Query) AddNamespace(namespace *string, _ *int) error {
|
||||
_, err := StorageHandler.AddNamespace(*namespace)
|
||||
func (t *Query) AddVault(vault *string, _ *int) error {
|
||||
_, err := StorageHandler.AddVault(*vault)
|
||||
return err
|
||||
}
|
||||
|
||||
func (t *Query) DeleteNamespace(namespace *string, _ *int) error {
|
||||
err := StorageHandler.DeleteNamespace(*namespace)
|
||||
func (t *Query) DeleteVault(vault *string, _ *int) error {
|
||||
err := StorageHandler.DeleteVault(*vault)
|
||||
return err
|
||||
}
|
||||
|
||||
func (t *Query) ListNamespaces(_ *int, reply *[]string) error {
|
||||
*reply = StorageHandler.ListNamespaces()
|
||||
func (t *Query) ListVaults(_ *int, reply *[]string) error {
|
||||
*reply = StorageHandler.ListVaults()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (t *Query) GetAllNames(namespace *string, reply *[]string) error {
|
||||
fmt.Println("RPC Request: Query::LoadedTriggers")
|
||||
fmt.Printf("Listing content of %s", *namespace)
|
||||
func (t *Query) GetAllNames(vault *string, reply *[]string) error {
|
||||
Logger.Infof("Listing content of %s", *vault)
|
||||
|
||||
val, ok := StorageHandler.Namespaces[*namespace]
|
||||
val, ok := StorageHandler.Vaults[*vault]
|
||||
|
||||
if !ok {
|
||||
return fmt.Errorf("Namesapce does not exist")
|
||||
@@ -182,7 +185,7 @@ func Serve(path string) {
|
||||
l, err := net.Listen("unix", path + "/rpc_test.socket")
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error while listening on unix socket: %s\n", err)
|
||||
Logger.Errorf("Error while listening on unix socket: %s\n", err)
|
||||
}
|
||||
|
||||
go http.Serve(l, nil)
|
||||
@@ -191,7 +194,7 @@ func Serve(path string) {
|
||||
signal.Notify(sigc, os.Interrupt, syscall.SIGTERM)
|
||||
func(ln net.Listener, c chan os.Signal) {
|
||||
sig := <-c
|
||||
fmt.Printf("Caught signal %s: shutting down.\n", sig)
|
||||
Logger.Infof("Caught signal %s: shutting down.\n", sig)
|
||||
ln.Close()
|
||||
os.Exit(0)
|
||||
}(l, sigc)
|
||||
@@ -201,7 +204,7 @@ func Receive(path string) (*rpc.Client, error) {
|
||||
client, err := rpc.DialHTTP("unix", path + "/rpc_test.socket")
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Cant connect to RPC server: %s\n", err)
|
||||
Logger.Errorf("Cant connect to RPC server: %s\n", err)
|
||||
}
|
||||
|
||||
return client, err
|
||||
469
internal/storage/storage.go
Normal file
469
internal/storage/storage.go
Normal file
@@ -0,0 +1,469 @@
|
||||
package storage
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"os"
|
||||
"fmt"
|
||||
"encoding/json"
|
||||
"path/filepath"
|
||||
"time"
|
||||
"sync"
|
||||
"strings"
|
||||
|
||||
"github.com/mudler/edgevpn/pkg/utils"
|
||||
"github.com/libp2p/go-libp2p"
|
||||
dht "github.com/libp2p/go-libp2p-kad-dht"
|
||||
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
||||
"github.com/libp2p/go-libp2p/core/host"
|
||||
"github.com/libp2p/go-libp2p/core/peer"
|
||||
"github.com/libp2p/go-libp2p/core/control"
|
||||
"github.com/libp2p/go-libp2p/core/network"
|
||||
//"github.com/libp2p/go-libp2p/core/peerstore"
|
||||
discovery "github.com/libp2p/go-libp2p/p2p/discovery/routing"
|
||||
ipfslite "github.com/hsanjuan/ipfs-lite"
|
||||
"github.com/google/uuid"
|
||||
|
||||
multiaddr "github.com/multiformats/go-multiaddr"
|
||||
|
||||
badger "github.com/ipfs/go-ds-badger2"
|
||||
dsq "github.com/ipfs/go-datastore/query"
|
||||
crypto "github.com/libp2p/go-libp2p/core/crypto"
|
||||
routed "github.com/libp2p/go-libp2p/p2p/host/routed"
|
||||
|
||||
agelib "filippo.io/age"
|
||||
. "github.com/k4lipso/pentapass/internal/log"
|
||||
)
|
||||
|
||||
var (
|
||||
Listen = libp2p.ListenAddrStrings("/ip4/0.0.0.0/tcp/0")
|
||||
)
|
||||
|
||||
func SetupLibp2pHost(ctx context.Context, dbPath string) (host host.Host, dht *dht.IpfsDHT, err error) {
|
||||
data := dbPath
|
||||
keyPath := filepath.Join(data, "key")
|
||||
var priv crypto.PrivKey
|
||||
_, err = os.Stat(keyPath)
|
||||
if os.IsNotExist(err) {
|
||||
priv, _, err = crypto.GenerateKeyPair(crypto.Ed25519, 1)
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
data, err := crypto.MarshalPrivateKey(priv)
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
err = os.WriteFile(keyPath, data, 0400)
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
} else if err != nil {
|
||||
Logger.Fatal(err)
|
||||
} else {
|
||||
key, err := os.ReadFile(keyPath)
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
priv, err = crypto.UnmarshalPrivateKey(key)
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
|
||||
host, err = libp2p.New(libp2p.Identity(priv), Listen)
|
||||
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
dht = initDHT(ctx, host)
|
||||
host = routed.Wrap(host, dht)
|
||||
|
||||
return host, dht, nil
|
||||
}
|
||||
|
||||
type Peer struct {
|
||||
Id string `json:"Id"`
|
||||
Key string `json:"Key"`
|
||||
}
|
||||
|
||||
type VaultConfig struct {
|
||||
Name string `json:"Name"`
|
||||
Id string `json:"Id"`
|
||||
Peers []Peer `json:"Peers"`
|
||||
}
|
||||
|
||||
type Config []VaultConfig
|
||||
|
||||
|
||||
type WhitelistConnectionGater struct {
|
||||
whitelistedPeers map[peer.ID]struct{}
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptPeerDial(p peer.ID) (allowed bool) {
|
||||
//_, allowed = wg.whitelistedPeers[p]
|
||||
return true
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptAddrDial(p peer.ID, addr multiaddr.Multiaddr) (bool) {
|
||||
return wg.InterceptPeerDial(p)
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptAccept(conn network.ConnMultiaddrs) (bool) {
|
||||
addr, err := peer.AddrInfoFromP2pAddr(conn.RemoteMultiaddr())
|
||||
|
||||
if err != nil {
|
||||
Logger.Debugf("Error InterceptAccept: %s\n", err)
|
||||
return false
|
||||
}
|
||||
|
||||
return wg.InterceptPeerDial(addr.ID)
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptSecured(direction network.Direction, p peer.ID, conn network.ConnMultiaddrs) (allow bool) {
|
||||
return wg.InterceptPeerDial(p)
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptUpgraded(conn network.Conn) (allow bool, reason control.DisconnectReason) {
|
||||
return wg.InterceptPeerDial(conn.RemotePeer()), 0
|
||||
}
|
||||
|
||||
func GetTrustedPeers(config []VaultConfig) map[string][]Peer {
|
||||
result := make(map[string][]Peer)
|
||||
for _, c := range config {
|
||||
result[c.Id] = c.Peers
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func InitRootNs() {
|
||||
//TODO: check if "SharedKeyRegistry" key exists, if not create
|
||||
}
|
||||
|
||||
func PeerFromString(str string) (Peer, error) {
|
||||
parts := strings.Split(str, "/")
|
||||
|
||||
if len(parts) != 2 {
|
||||
return Peer{}, fmt.Errorf("Invalid Peer String")
|
||||
}
|
||||
//TODO: validate each part
|
||||
|
||||
return Peer{ Id: parts[0], Key: parts[1] }, nil
|
||||
}
|
||||
|
||||
type StorageHandler struct {
|
||||
Ctx context.Context
|
||||
Store *badger.Datastore
|
||||
Host host.Host
|
||||
Ipfs *ipfslite.Peer
|
||||
PubSub *pubsub.PubSub
|
||||
Key *agelib.X25519Identity
|
||||
Config []VaultConfig
|
||||
Vaults map[string]*Vault
|
||||
ConfigPath string
|
||||
}
|
||||
|
||||
func (s *StorageHandler) GetSelfPeer() Peer {
|
||||
return Peer {
|
||||
Id: s.Host.ID().String(),
|
||||
Key: s.Key.Recipient().String(),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *StorageHandler) UpdateConfig() {
|
||||
Logger.Debug("Updating Config...")
|
||||
s.recreateConfig()
|
||||
s.writeConfig(s.ConfigPath, s.Config)
|
||||
}
|
||||
|
||||
func (s *StorageHandler) recreateConfig() {
|
||||
var newCfg []VaultConfig
|
||||
for key, val := range s.Vaults {
|
||||
newCfg = append(newCfg, VaultConfig{
|
||||
Name: key,
|
||||
Id: val.ID,
|
||||
Peers: val.TrustedPeers,
|
||||
})
|
||||
}
|
||||
s.Config = newCfg
|
||||
//for idx, vaultConfig := range s.Config {
|
||||
// s.Config[idx].Peers = s.Vaults[vaultConfig.Name].TrustedPeers
|
||||
//}
|
||||
}
|
||||
|
||||
func (s *StorageHandler) writeConfig(filename string, config []VaultConfig) error {
|
||||
jsonData, err := json.Marshal(config)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
|
||||
err = os.WriteFile(filename, jsonData, 0644)
|
||||
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *StorageHandler) NewConfig(filename string) ([]VaultConfig, error) {
|
||||
if _, err := os.Stat(filename); errors.Is(err, os.ErrNotExist) {
|
||||
err := s.writeConfig(filename, []VaultConfig{
|
||||
{
|
||||
Name: "root",
|
||||
Id: uuid.New().String(),
|
||||
Peers: []Peer{
|
||||
{
|
||||
Id: s.Host.ID().String(),
|
||||
Key: s.Key.Recipient().String(),
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Could not create config file: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
content, err := os.ReadFile(filename)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Could not read config file: %s", err)
|
||||
}
|
||||
|
||||
var result []VaultConfig
|
||||
err = json.Unmarshal(content, &result)
|
||||
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Could not parse config file: %s", err)
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (s *StorageHandler) GetDefaultVault(Name string) *Vault {
|
||||
return s.Vaults["root"]
|
||||
}
|
||||
|
||||
func (s *StorageHandler) InitVaults() {
|
||||
VaultMap := make(map[string]*Vault)
|
||||
for _, nsCfg := range s.Config {
|
||||
ns1, err := CreateVault(nsCfg.Id, s)
|
||||
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
|
||||
VaultMap[nsCfg.Name] = ns1
|
||||
}
|
||||
|
||||
s.Vaults = VaultMap
|
||||
}
|
||||
|
||||
func IsTrustedPeer(ctx context.Context, id peer.ID, vault string, config []VaultConfig) bool {
|
||||
peerMap := GetTrustedPeers(config)
|
||||
|
||||
val, ok := peerMap[vault]
|
||||
|
||||
if ok {
|
||||
for _, v := range val {
|
||||
if v.Id == id.String() {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func PrintDBContent(ctx context.Context, store *badger.Datastore) {
|
||||
q := dsq.Query{}
|
||||
result, _ := store.Query(ctx, q)
|
||||
|
||||
entries, _ := result.Rest()
|
||||
for _, entry := range entries {
|
||||
Logger.Infof("Key: %s, Value: %s\n", entry.Key, string(entry.Value))
|
||||
}
|
||||
}
|
||||
|
||||
func (s *StorageHandler) ListVaults() []string {
|
||||
var result []string
|
||||
for k := range s.Vaults {
|
||||
result = append(result, k)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (s *StorageHandler) DeleteVault(ID string) error {
|
||||
ns, ok := s.Vaults[ID]
|
||||
|
||||
if !ok {
|
||||
Logger.Debug("DeleteVault that does not exists")
|
||||
return nil
|
||||
}
|
||||
|
||||
delete(s.Vaults, ID)
|
||||
ns.Close()
|
||||
s.UpdateConfig()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *StorageHandler) AddVault(Name string) (*Vault, error) {
|
||||
ns, ok := s.Vaults[Name]
|
||||
|
||||
if ok {
|
||||
return ns, nil
|
||||
}
|
||||
|
||||
result, err := CreateVault(uuid.New().String(), s)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
result.TrustedPeers = append(result.TrustedPeers, s.GetSelfPeer())
|
||||
s.Vaults[Name] = result
|
||||
s.UpdateConfig()
|
||||
return result, nil
|
||||
}
|
||||
|
||||
|
||||
func initDHT(ctx context.Context, h host.Host) *dht.IpfsDHT {
|
||||
// Start a DHT, for use in peer discovery. We can't just make a new DHT
|
||||
// client because we want each peer to maintain its own local copy of the
|
||||
// DHT, so that the bootstrapping node of the DHT can go down without
|
||||
// inhibiting future peer discovery.
|
||||
kademliaDHT, err := dht.New(ctx, h)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
// if err = kademliaDHT.Bootstrap(ctx); err != nil {
|
||||
// panic(err)
|
||||
// }
|
||||
|
||||
return kademliaDHT
|
||||
}
|
||||
|
||||
func (s *StorageHandler) bootstrapPeers(ctx context.Context, h host.Host) {
|
||||
Logger.Info("Bootstrapping DHT")
|
||||
|
||||
var wg sync.WaitGroup
|
||||
for _, peerAddr := range dht.DefaultBootstrapPeers {
|
||||
peerinfo, _ := peer.AddrInfoFromP2pAddr(peerAddr)
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
if err := h.Connect(ctx, *peerinfo); err != nil {
|
||||
Logger.Debugf("Bootstrap warning: %s", err)
|
||||
} else {
|
||||
Logger.Debugf("Connection established with bootstrap node: %q\n", *peerinfo)
|
||||
}
|
||||
}()
|
||||
}
|
||||
wg.Wait()
|
||||
}
|
||||
|
||||
func (s *StorageHandler) RunBackground(ctx context.Context, h host.Host, dht *dht.IpfsDHT) {
|
||||
s.discoverPeers(ctx, h, dht)
|
||||
t := utils.NewBackoffTicker(utils.BackoffInitialInterval(2 * time.Minute),
|
||||
utils.BackoffMaxInterval(6 * time.Minute))
|
||||
defer t.Stop()
|
||||
for {
|
||||
select {
|
||||
case <-t.C:
|
||||
// We announce ourselves to the rendezvous point for all the peers.
|
||||
// We have a safeguard of 1 hour to avoid blocking the main loop
|
||||
// in case of network issues.
|
||||
// The TTL of DHT is by default no longer than 3 hours, so we should
|
||||
// be safe by having an entry less than that.
|
||||
safeTimeout, cancel := context.WithTimeout(ctx, time.Hour)
|
||||
|
||||
endChan := make(chan struct{})
|
||||
go func() {
|
||||
s.discoverPeers(safeTimeout, h, dht)
|
||||
endChan <- struct{}{}
|
||||
}()
|
||||
|
||||
select {
|
||||
case <-endChan:
|
||||
cancel()
|
||||
case <-safeTimeout.Done():
|
||||
Logger.Error("Timeout while peer discovery")
|
||||
cancel()
|
||||
}
|
||||
case <-ctx.Done():
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *StorageHandler) discoverPeers(ctx context.Context, h host.Host, dht *dht.IpfsDHT) error {
|
||||
s.bootstrapPeers(ctx, h)
|
||||
time.Sleep(2 * time.Second)
|
||||
|
||||
for vaultName, v := range s.Vaults {
|
||||
Logger.Debugf("Announcing vault \"%s\" with id: %s", vaultName, v.ID)
|
||||
routingDiscovery := discovery.NewRoutingDiscovery(dht)
|
||||
routingDiscovery.Advertise(ctx, v.ID)
|
||||
|
||||
Logger.Debugf("Start peer discovery...")
|
||||
|
||||
timedCtx, cf := context.WithTimeout(ctx, time.Second*120)
|
||||
defer cf()
|
||||
|
||||
peerChan, err := routingDiscovery.FindPeers(timedCtx, v.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
for peer := range peerChan {
|
||||
if peer.ID == h.ID() || len(peer.Addrs) == 0 {
|
||||
continue // No self connection
|
||||
}
|
||||
|
||||
Logger.Debugf("Found peer with id %s", peer.ID.String())
|
||||
if h.Network().Connectedness(peer.ID) == network.Connected {
|
||||
Logger.Debugf("Already connected to %s", peer.ID.String())
|
||||
continue
|
||||
}
|
||||
|
||||
timeoutCtx, cancel := context.WithTimeout(ctx, time.Second*120)
|
||||
defer cancel()
|
||||
err := h.Connect(timeoutCtx, peer)
|
||||
if err != nil {
|
||||
Logger.Debugf("Failed connecting to %s, error: %s\n", peer.ID, err)
|
||||
} else {
|
||||
Logger.Debugf("Connected to: %s", peer.ID)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Logger.Debug("Peer discovery complete")
|
||||
return nil
|
||||
}
|
||||
|
||||
func printErr(err error) {
|
||||
Logger.Errorf("error: %s", err)
|
||||
}
|
||||
|
||||
func ConnectedPeers(h host.Host) []*peer.AddrInfo {
|
||||
var pinfos []*peer.AddrInfo
|
||||
for _, c := range h.Network().Conns() {
|
||||
pinfos = append(pinfos, &peer.AddrInfo{
|
||||
ID: c.RemotePeer(),
|
||||
Addrs: []multiaddr.Multiaddr{c.RemoteMultiaddr()},
|
||||
})
|
||||
}
|
||||
return pinfos
|
||||
}
|
||||
272
internal/storage/vault.go
Normal file
272
internal/storage/vault.go
Normal file
@@ -0,0 +1,272 @@
|
||||
package storage
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"time"
|
||||
"errors"
|
||||
|
||||
"github.com/ipfs/go-datastore/query"
|
||||
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
||||
"github.com/libp2p/go-libp2p/core/peer"
|
||||
ds "github.com/ipfs/go-datastore"
|
||||
|
||||
crdt "github.com/ipfs/go-ds-crdt"
|
||||
|
||||
agelib "filippo.io/age"
|
||||
password "github.com/k4lipso/pentapass/internal/crypto"
|
||||
"github.com/k4lipso/pentapass/internal/crypto/age"
|
||||
. "github.com/k4lipso/pentapass/internal/log"
|
||||
)
|
||||
|
||||
var ErrNotFound error = errors.New("Not found")
|
||||
|
||||
type Vault struct {
|
||||
ID string
|
||||
Datastore *crdt.Datastore
|
||||
//Registry *sharedKeyRegistry
|
||||
CancelFunc context.CancelFunc
|
||||
ctx context.Context
|
||||
Key *agelib.X25519Identity
|
||||
TrustedPeers []Peer
|
||||
}
|
||||
|
||||
func (n *Vault) AddPeer(peer Peer) {
|
||||
for _, CurrentPeer := range n.TrustedPeers {
|
||||
if CurrentPeer.Id == peer.Id && CurrentPeer.Key == peer.Key {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
n.TrustedPeers = append(n.TrustedPeers, peer)
|
||||
}
|
||||
|
||||
func (n *Vault) RemovePeer(peer Peer) {
|
||||
var Peers []Peer
|
||||
for _, CurrentPeer := range n.TrustedPeers {
|
||||
if CurrentPeer.Id == peer.Id && CurrentPeer.Key == peer.Key {
|
||||
continue
|
||||
}
|
||||
|
||||
Peers = append(Peers, CurrentPeer)
|
||||
}
|
||||
|
||||
n.TrustedPeers = Peers
|
||||
}
|
||||
|
||||
func (n *Vault) GetRecipients() []string {
|
||||
var result []string
|
||||
|
||||
for _, peer := range n.TrustedPeers {
|
||||
result = append(result, peer.Key)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (n *Vault) Put(k string, v string) error {
|
||||
key := ds.NewKey(k)
|
||||
err := n.Datastore.Put(n.ctx, key, []byte(v))
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (n *Vault) Delete(k string) error {
|
||||
_, err := n.GetPassword(k)
|
||||
|
||||
if err != nil {
|
||||
return ErrNotFound
|
||||
}
|
||||
|
||||
key := ds.NewKey(k)
|
||||
|
||||
err = n.Datastore.Delete(n.ctx, key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (n *Vault) GetPassword(k string) (password.Password, error) {
|
||||
v, err := n.Datastore.Get(n.ctx, ds.NewKey(k))
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return password.Password{}, err
|
||||
}
|
||||
|
||||
val, err := age.Decrypt(v, n.Key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return password.Password{}, err
|
||||
}
|
||||
|
||||
pw, err := password.GetPasswordFromJson(val)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return password.Password{}, err
|
||||
}
|
||||
|
||||
return pw, nil
|
||||
}
|
||||
|
||||
|
||||
func (n *Vault) Get(k string) (string, error) {
|
||||
v, err := n.Datastore.Get(n.ctx, ds.NewKey(k))
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return "", err
|
||||
}
|
||||
|
||||
return string(v), nil
|
||||
}
|
||||
|
||||
func (n *Vault) GetAllNames() []string {
|
||||
q := query.Query{}
|
||||
results, err := n.Datastore.Query(n.ctx, q)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
var result []string
|
||||
for r := range results.Next() {
|
||||
if r.Error != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
result = append(result, r.Key)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (n *Vault) List() {
|
||||
q := query.Query{}
|
||||
results, err := n.Datastore.Query(n.ctx, q)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
for r := range results.Next() {
|
||||
if r.Error != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
val, err := age.Decrypt(r.Value, n.Key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
Logger.Infof("[%s] -> %s\n", r.Key, string(val))
|
||||
}
|
||||
}
|
||||
|
||||
func (n *Vault) GetAllPasswords() ([]password.Password, error) {
|
||||
q := query.Query{}
|
||||
results, err := n.Datastore.Query(n.ctx, q)
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Error during GetAllPasswords: %s", err)
|
||||
}
|
||||
|
||||
var result []password.Password
|
||||
|
||||
for r := range results.Next() {
|
||||
if r.Error != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
val, err := age.Decrypt(r.Value, n.Key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
pw, err := password.GetPasswordFromJson(val)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
result = append(result, pw)
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (n *Vault) Close() {
|
||||
n.CancelFunc()
|
||||
n.Datastore.Close()
|
||||
}
|
||||
|
||||
func CreateVault(ID string, storageHandler *StorageHandler) (*Vault, error) {
|
||||
Logger.Infof("Creating Vault %s\n", ID)
|
||||
err := storageHandler.PubSub.RegisterTopicValidator(
|
||||
ID, //== topicName
|
||||
func(ctx context.Context, id peer.ID, msg *pubsub.Message) bool {
|
||||
if id == storageHandler.Host.ID() {
|
||||
return true
|
||||
}
|
||||
|
||||
Logger.Debugf("PubSubmsg TOPIC: %s, PEER: %s\n", ID, id)
|
||||
trusted := IsTrustedPeer(ctx, id, ID, storageHandler.Config)
|
||||
if !trusted {
|
||||
Logger.Debugf("discarded pubsub message from non trusted source %s\n", id)
|
||||
}
|
||||
return trusted
|
||||
},
|
||||
)
|
||||
if err != nil {
|
||||
Logger.Errorf("error registering topic validator: %s", err)
|
||||
}
|
||||
|
||||
psubCtx, psubCancel := context.WithCancel(storageHandler.Ctx)
|
||||
pubsubBC, err := crdt.NewPubSubBroadcaster(psubCtx, storageHandler.PubSub, ID)
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
}
|
||||
|
||||
opts := crdt.DefaultOptions()
|
||||
//opts.Logger = Logger
|
||||
opts.RebroadcastInterval = 5 * time.Second
|
||||
opts.PutHook = func(k ds.Key, v []byte) {
|
||||
Logger.Infof("Added: [%s]\n", k)
|
||||
}
|
||||
opts.DeleteHook = func(k ds.Key) {
|
||||
Logger.Infof("Removed: [%s]\n", k)
|
||||
}
|
||||
|
||||
crdt, err := crdt.New(storageHandler.Store, ds.NewKey(ID), storageHandler.Ipfs, pubsubBC, opts)
|
||||
if err != nil {
|
||||
Logger.Fatal(err)
|
||||
psubCancel()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
PeerMap := GetTrustedPeers(storageHandler.Config)
|
||||
|
||||
val, ok := PeerMap[ID]
|
||||
|
||||
if !ok {
|
||||
Logger.Debug("vault config does not contain any peers")
|
||||
}
|
||||
|
||||
return &Vault{ID: ID, Datastore: crdt, CancelFunc: psubCancel, ctx: storageHandler.Ctx, Key: storageHandler.Key, TrustedPeers: val}, nil
|
||||
}
|
||||
|
||||
@@ -1,684 +0,0 @@
|
||||
package storage
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"os"
|
||||
"encoding/json"
|
||||
"path/filepath"
|
||||
"time"
|
||||
"sync"
|
||||
"strings"
|
||||
|
||||
"github.com/ipfs/go-datastore/query"
|
||||
"github.com/libp2p/go-libp2p"
|
||||
dht "github.com/libp2p/go-libp2p-kad-dht"
|
||||
pubsub "github.com/libp2p/go-libp2p-pubsub"
|
||||
"github.com/libp2p/go-libp2p/core/host"
|
||||
"github.com/libp2p/go-libp2p/core/peer"
|
||||
"github.com/libp2p/go-libp2p/core/control"
|
||||
"github.com/libp2p/go-libp2p/core/network"
|
||||
//"github.com/libp2p/go-libp2p/core/peerstore"
|
||||
discovery "github.com/libp2p/go-libp2p/p2p/discovery/routing"
|
||||
ds "github.com/ipfs/go-datastore"
|
||||
ipfslite "github.com/hsanjuan/ipfs-lite"
|
||||
"github.com/google/uuid"
|
||||
|
||||
multiaddr "github.com/multiformats/go-multiaddr"
|
||||
|
||||
badger "github.com/ipfs/go-ds-badger2"
|
||||
dsq "github.com/ipfs/go-datastore/query"
|
||||
crdt "github.com/ipfs/go-ds-crdt"
|
||||
crypto "github.com/libp2p/go-libp2p/core/crypto"
|
||||
routed "github.com/libp2p/go-libp2p/p2p/host/routed"
|
||||
logging "github.com/ipfs/go-log/v2"
|
||||
|
||||
agelib "filippo.io/age"
|
||||
password "github.com/k4lipso/pentapass/crypto"
|
||||
"github.com/k4lipso/pentapass/crypto/age"
|
||||
)
|
||||
|
||||
var (
|
||||
topicNameFlag = "pentapass"
|
||||
logger = logging.Logger("globaldb")
|
||||
Listen = libp2p.ListenAddrStrings("/ip4/0.0.0.0/tcp/0")
|
||||
)
|
||||
|
||||
func SetupLibp2pHost(ctx context.Context, dbPath string) (host host.Host, dht *dht.IpfsDHT, err error) {
|
||||
data := dbPath
|
||||
keyPath := filepath.Join(data, "key")
|
||||
var priv crypto.PrivKey
|
||||
_, err = os.Stat(keyPath)
|
||||
if os.IsNotExist(err) {
|
||||
priv, _, err = crypto.GenerateKeyPair(crypto.Ed25519, 1)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
data, err := crypto.MarshalPrivateKey(priv)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
err = os.WriteFile(keyPath, data, 0400)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
} else if err != nil {
|
||||
logger.Fatal(err)
|
||||
} else {
|
||||
key, err := os.ReadFile(keyPath)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
priv, err = crypto.UnmarshalPrivateKey(key)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
|
||||
//whitelistedPeers := map[peer.ID]struct{} {
|
||||
// "12D3KooWLF7BU5VgpqWdS1XwSTFCLphENozhYQAj6i5LqU8BPZZZ": {},
|
||||
// "12D3KooWBRvtW83QYnPgJCyVyAgMXtg71wjkGefVB2fBnm1A36kS": {},
|
||||
// "12D3KooWMmc4kYy78vSumqWtPkUNAoPeCpJ66ysFv1U8S554B7e2": {},
|
||||
//}
|
||||
|
||||
//connectionGater := &WhitelistConnectionGater{whitelistedPeers: whitelistedPeers}
|
||||
|
||||
//host, err = libp2p.New(libp2p.Identity(priv), libp2p.ConnectionGater(connectionGater), Listen)
|
||||
host, err = libp2p.New(libp2p.Identity(priv), Listen)
|
||||
|
||||
if err != nil {
|
||||
return nil, nil, err
|
||||
}
|
||||
|
||||
dht = initDHT(ctx, host)
|
||||
host = routed.Wrap(host, dht)
|
||||
|
||||
|
||||
return host, dht, nil
|
||||
}
|
||||
|
||||
type Peer struct {
|
||||
Id string `json:"Id"`
|
||||
Key string `json:"Key"`
|
||||
}
|
||||
|
||||
type NamespaceConfig struct {
|
||||
Name string `json:"Name"`
|
||||
Id string `json:"Id"`
|
||||
Peers []Peer `json:"Peers"`
|
||||
}
|
||||
|
||||
type Config []NamespaceConfig
|
||||
|
||||
|
||||
type WhitelistConnectionGater struct {
|
||||
whitelistedPeers map[peer.ID]struct{}
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptPeerDial(p peer.ID) (allowed bool) {
|
||||
//fmt.Println("PeerDial")
|
||||
//_, allowed = wg.whitelistedPeers[p]
|
||||
return true
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptAddrDial(p peer.ID, addr multiaddr.Multiaddr) (bool) {
|
||||
//fmt.Println("AddrDial")
|
||||
return wg.InterceptPeerDial(p)
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptAccept(conn network.ConnMultiaddrs) (bool) {
|
||||
//fmt.Println("InterceptAccept")
|
||||
addr, err := peer.AddrInfoFromP2pAddr(conn.RemoteMultiaddr())
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error InterceptAccept: %s\n", err)
|
||||
return false
|
||||
}
|
||||
|
||||
return wg.InterceptPeerDial(addr.ID)
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptSecured(direction network.Direction, p peer.ID, conn network.ConnMultiaddrs) (allow bool) {
|
||||
//fmt.Println("InterceptSecured")
|
||||
return wg.InterceptPeerDial(p)
|
||||
}
|
||||
|
||||
func (wg *WhitelistConnectionGater) InterceptUpgraded(conn network.Conn) (allow bool, reason control.DisconnectReason) {
|
||||
//fmt.Println("InterceptUpgraded")
|
||||
return wg.InterceptPeerDial(conn.RemotePeer()), 0
|
||||
}
|
||||
|
||||
func GetTrustedPeers(config []NamespaceConfig) map[string][]Peer {
|
||||
result := make(map[string][]Peer)
|
||||
for _, c := range config {
|
||||
result[c.Id] = c.Peers
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func InitRootNs() {
|
||||
//TODO: check if "SharedKeyRegistry" key exists, if not create
|
||||
}
|
||||
|
||||
type Namespace struct {
|
||||
ID string
|
||||
Datastore *crdt.Datastore
|
||||
//Registry *sharedKeyRegistry
|
||||
CancelFunc context.CancelFunc
|
||||
ctx context.Context
|
||||
Key *agelib.X25519Identity
|
||||
TrustedPeers []Peer
|
||||
}
|
||||
|
||||
func PeerFromString(str string) (Peer, error) {
|
||||
parts := strings.Split(str, "/")
|
||||
|
||||
fmt.Println(str)
|
||||
fmt.Println(parts)
|
||||
if len(parts) != 2 {
|
||||
return Peer{}, fmt.Errorf("Invalid Peer String")
|
||||
}
|
||||
//TODO: validate each part
|
||||
|
||||
return Peer{ Id: parts[0], Key: parts[1] }, nil
|
||||
}
|
||||
|
||||
func (n *Namespace) AddPeer(peer Peer) {
|
||||
for _, CurrentPeer := range n.TrustedPeers {
|
||||
if CurrentPeer.Id == peer.Id && CurrentPeer.Key == peer.Key {
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
n.TrustedPeers = append(n.TrustedPeers, peer)
|
||||
}
|
||||
|
||||
func (n *Namespace) RemovePeer(peer Peer) {
|
||||
var Peers []Peer
|
||||
for _, CurrentPeer := range n.TrustedPeers {
|
||||
if CurrentPeer.Id == peer.Id && CurrentPeer.Key == peer.Key {
|
||||
continue
|
||||
}
|
||||
|
||||
Peers = append(Peers, CurrentPeer)
|
||||
}
|
||||
|
||||
n.TrustedPeers = Peers
|
||||
}
|
||||
|
||||
func (n *Namespace) GetRecipients() []string {
|
||||
var result []string
|
||||
|
||||
for _, peer := range n.TrustedPeers {
|
||||
result = append(result, peer.Key)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (n *Namespace) Put(k string, v string) error {
|
||||
key := ds.NewKey(k)
|
||||
err := n.Datastore.Put(n.ctx, key, []byte(v))
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (n *Namespace) Delete(k string) error {
|
||||
key := ds.NewKey(k)
|
||||
err := n.Datastore.Delete(n.ctx, key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
return err
|
||||
}
|
||||
|
||||
func (n *Namespace) GetPassword(k string) (password.Password, error) {
|
||||
v, err := n.Datastore.Get(n.ctx, ds.NewKey(k))
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return password.Password{}, err
|
||||
}
|
||||
|
||||
val, err := age.Decrypt(v, n.Key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return password.Password{}, err
|
||||
}
|
||||
|
||||
pw, err := password.GetPasswordFromJson(val)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return password.Password{}, err
|
||||
}
|
||||
|
||||
return pw, nil
|
||||
}
|
||||
|
||||
|
||||
func (n *Namespace) Get(k string) (string, error) {
|
||||
v, err := n.Datastore.Get(n.ctx, ds.NewKey(k))
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
return "", err
|
||||
}
|
||||
|
||||
return string(v), nil
|
||||
}
|
||||
|
||||
func (n *Namespace) GetAllNames() []string {
|
||||
q := query.Query{}
|
||||
results, err := n.Datastore.Query(n.ctx, q)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
var result []string
|
||||
for r := range results.Next() {
|
||||
if r.Error != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
result = append(result, r.Key)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (n *Namespace) List() {
|
||||
q := query.Query{}
|
||||
results, err := n.Datastore.Query(n.ctx, q)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
}
|
||||
|
||||
for r := range results.Next() {
|
||||
if r.Error != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
val, err := age.Decrypt(r.Value, n.Key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
fmt.Printf("[%s] -> %s\n", r.Key, string(val))
|
||||
}
|
||||
}
|
||||
|
||||
func (n *Namespace) GetAllPasswords() ([]password.Password, error) {
|
||||
q := query.Query{}
|
||||
results, err := n.Datastore.Query(n.ctx, q)
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Error during GetAllPasswords: %s", err)
|
||||
}
|
||||
|
||||
var result []password.Password
|
||||
|
||||
for r := range results.Next() {
|
||||
if r.Error != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
val, err := age.Decrypt(r.Value, n.Key)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
pw, err := password.GetPasswordFromJson(val)
|
||||
|
||||
if err != nil {
|
||||
printErr(err)
|
||||
continue
|
||||
}
|
||||
|
||||
result = append(result, pw)
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (n *Namespace) Close() {
|
||||
n.CancelFunc()
|
||||
n.Datastore.Close()
|
||||
}
|
||||
|
||||
type StorageHandler struct {
|
||||
Ctx context.Context
|
||||
Store *badger.Datastore
|
||||
Host host.Host
|
||||
Ipfs *ipfslite.Peer
|
||||
PubSub *pubsub.PubSub
|
||||
Key *agelib.X25519Identity
|
||||
Config []NamespaceConfig
|
||||
Namespaces map[string]*Namespace
|
||||
ConfigPath string
|
||||
}
|
||||
|
||||
func (s *StorageHandler) GetSelfPeer() Peer {
|
||||
return Peer {
|
||||
Id: s.Host.ID().String(),
|
||||
Key: s.Key.Recipient().String(),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *StorageHandler) UpdateConfig() {
|
||||
fmt.Println("Updating Config...")
|
||||
s.recreateConfig()
|
||||
s.writeConfig(s.ConfigPath, s.Config)
|
||||
}
|
||||
|
||||
func (s *StorageHandler) recreateConfig() {
|
||||
var newCfg []NamespaceConfig
|
||||
for key, val := range s.Namespaces {
|
||||
newCfg = append(newCfg, NamespaceConfig{
|
||||
Name: key,
|
||||
Id: val.ID,
|
||||
Peers: val.TrustedPeers,
|
||||
})
|
||||
}
|
||||
s.Config = newCfg
|
||||
//for idx, namespaceConfig := range s.Config {
|
||||
// s.Config[idx].Peers = s.Namespaces[namespaceConfig.Name].TrustedPeers
|
||||
//}
|
||||
}
|
||||
|
||||
func (s *StorageHandler) writeConfig(filename string, config []NamespaceConfig) error {
|
||||
jsonData, err := json.Marshal(config)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error during config initialization")
|
||||
return err
|
||||
}
|
||||
|
||||
|
||||
err = os.WriteFile(filename, jsonData, 0644)
|
||||
|
||||
if err != nil {
|
||||
fmt.Printf("Error during config initialization")
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *StorageHandler) NewConfig(filename string) ([]NamespaceConfig, error) {
|
||||
if _, err := os.Stat(filename); errors.Is(err, os.ErrNotExist) {
|
||||
err := s.writeConfig(filename, []NamespaceConfig{
|
||||
{
|
||||
Name: "root",
|
||||
Id: uuid.New().String(),
|
||||
Peers: []Peer{
|
||||
{
|
||||
Id: s.Host.ID().String(),
|
||||
Key: s.Key.Recipient().String(),
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
)
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Could not create config file: %s", err)
|
||||
}
|
||||
}
|
||||
|
||||
content, err := os.ReadFile(filename)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Could not read config file: %s", err)
|
||||
}
|
||||
|
||||
var result []NamespaceConfig
|
||||
err = json.Unmarshal(content, &result)
|
||||
|
||||
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("Could not parse config file: %s", err)
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (s *StorageHandler) GetDefaultNamespace(Name string) *Namespace {
|
||||
return s.Namespaces["root"]
|
||||
}
|
||||
|
||||
func (s *StorageHandler) InitNamespaces() {
|
||||
NamespaceMap := make(map[string]*Namespace)
|
||||
for _, nsCfg := range s.Config {
|
||||
ns1, err := CreateNamespace(nsCfg.Id, s)
|
||||
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
|
||||
NamespaceMap[nsCfg.Name] = ns1
|
||||
}
|
||||
|
||||
s.Namespaces = NamespaceMap
|
||||
}
|
||||
|
||||
func IsTrustedPeer(ctx context.Context, id peer.ID, namespace string, config []NamespaceConfig) bool {
|
||||
peerMap := GetTrustedPeers(config)
|
||||
|
||||
val, ok := peerMap[namespace]
|
||||
|
||||
if ok {
|
||||
for _, v := range val {
|
||||
if v.Id == id.String() {
|
||||
return true
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
|
||||
func PrintDBContent(ctx context.Context, store *badger.Datastore) {
|
||||
q := dsq.Query{}
|
||||
result, _ := store.Query(ctx, q)
|
||||
|
||||
entries, _ := result.Rest()
|
||||
for _, entry := range entries {
|
||||
fmt.Printf("Key: %s, Value: %s\n", entry.Key, string(entry.Value))
|
||||
}
|
||||
}
|
||||
|
||||
func (s *StorageHandler) ListNamespaces() []string {
|
||||
var result []string
|
||||
for k := range s.Namespaces {
|
||||
result = append(result, k)
|
||||
}
|
||||
|
||||
return result
|
||||
}
|
||||
|
||||
func (s *StorageHandler) DeleteNamespace(ID string) error {
|
||||
ns, ok := s.Namespaces[ID]
|
||||
|
||||
if !ok {
|
||||
fmt.Print("DeleteNamespace that does not exists")
|
||||
return nil
|
||||
}
|
||||
|
||||
delete(s.Namespaces, ID)
|
||||
ns.Close()
|
||||
s.UpdateConfig()
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *StorageHandler) AddNamespace(Name string) (*Namespace, error) {
|
||||
ns, ok := s.Namespaces[Name]
|
||||
|
||||
if ok {
|
||||
return ns, nil
|
||||
}
|
||||
|
||||
result, err := CreateNamespace(uuid.New().String(), s)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
result.TrustedPeers = append(result.TrustedPeers, s.GetSelfPeer())
|
||||
s.Namespaces[Name] = result
|
||||
s.UpdateConfig()
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func CreateNamespace(ID string, storageHandler *StorageHandler) (*Namespace, error) {
|
||||
fmt.Printf("Creating Namespace %s\n", ID)
|
||||
err := storageHandler.PubSub.RegisterTopicValidator(
|
||||
ID, //== topicName
|
||||
func(ctx context.Context, id peer.ID, msg *pubsub.Message) bool {
|
||||
fmt.Printf("PubSubmsg TOPIC: %s, PEER: %s\n", ID, id)
|
||||
signer := msg.GetFrom()
|
||||
trusted := IsTrustedPeer(ctx, signer, ID, storageHandler.Config)
|
||||
if !trusted {
|
||||
fmt.Printf("discarded pubsub message from non trusted source %s\n", signer)
|
||||
}
|
||||
return trusted
|
||||
},
|
||||
)
|
||||
if err != nil {
|
||||
logger.Errorf("error registering topic validator: %s", err)
|
||||
}
|
||||
|
||||
psubCtx, psubCancel := context.WithCancel(storageHandler.Ctx)
|
||||
pubsubBC, err := crdt.NewPubSubBroadcaster(psubCtx, storageHandler.PubSub, ID)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
}
|
||||
|
||||
opts := crdt.DefaultOptions()
|
||||
opts.Logger = logger
|
||||
opts.RebroadcastInterval = 5 * time.Second
|
||||
opts.PutHook = func(k ds.Key, v []byte) {
|
||||
fmt.Printf("Added: [%s]\n", k)
|
||||
}
|
||||
opts.DeleteHook = func(k ds.Key) {
|
||||
fmt.Printf("Removed: [%s]\n", k)
|
||||
}
|
||||
|
||||
crdt, err := crdt.New(storageHandler.Store, ds.NewKey(ID), storageHandler.Ipfs, pubsubBC, opts)
|
||||
if err != nil {
|
||||
logger.Fatal(err)
|
||||
psubCancel()
|
||||
return nil, err
|
||||
}
|
||||
|
||||
PeerMap := GetTrustedPeers(storageHandler.Config)
|
||||
|
||||
val, ok := PeerMap[ID]
|
||||
|
||||
if !ok {
|
||||
fmt.Println("namespace config does not contain any peers")
|
||||
}
|
||||
|
||||
return &Namespace{ID: ID, Datastore: crdt, CancelFunc: psubCancel, ctx: storageHandler.Ctx, Key: storageHandler.Key, TrustedPeers: val}, nil
|
||||
}
|
||||
|
||||
|
||||
func initDHT(ctx context.Context, h host.Host) *dht.IpfsDHT {
|
||||
// Start a DHT, for use in peer discovery. We can't just make a new DHT
|
||||
// client because we want each peer to maintain its own local copy of the
|
||||
// DHT, so that the bootstrapping node of the DHT can go down without
|
||||
// inhibiting future peer discovery.
|
||||
kademliaDHT, err := dht.New(ctx, h)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
if err = kademliaDHT.Bootstrap(ctx); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
var wg sync.WaitGroup
|
||||
for _, peerAddr := range dht.DefaultBootstrapPeers {
|
||||
peerinfo, _ := peer.AddrInfoFromP2pAddr(peerAddr)
|
||||
wg.Add(1)
|
||||
go func() {
|
||||
defer wg.Done()
|
||||
if err := h.Connect(ctx, *peerinfo); err != nil {
|
||||
fmt.Println("Bootstrap warning:", err)
|
||||
} else {
|
||||
fmt.Printf("Connection established with bootstrap node: %q\n", *peerinfo)
|
||||
}
|
||||
}()
|
||||
}
|
||||
wg.Wait()
|
||||
|
||||
return kademliaDHT
|
||||
}
|
||||
|
||||
func DiscoverPeers(ctx context.Context, h host.Host, dht *dht.IpfsDHT) {
|
||||
routingDiscovery := discovery.NewRoutingDiscovery(dht)
|
||||
routingDiscovery.Advertise(ctx, topicNameFlag)
|
||||
|
||||
// Look for others who have announced and attempt to connect to them
|
||||
anyConnected := false
|
||||
fmt.Printf("Own Id: %s\n", h.ID())
|
||||
for !anyConnected {
|
||||
time.Sleep(2 * time.Second)
|
||||
fmt.Println("Searching for peers...")
|
||||
peerChan, err := routingDiscovery.FindPeers(ctx, topicNameFlag)
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
for peer := range peerChan {
|
||||
if peer.ID == h.ID() || len(peer.Addrs) == 0 {
|
||||
continue // No self connection
|
||||
}
|
||||
|
||||
if h.Network().Connectedness(peer.ID) == network.Connected {
|
||||
continue
|
||||
}
|
||||
|
||||
fmt.Printf("Connecting to peer with id %s", peer.ID.String())
|
||||
err := h.Connect(ctx, peer)
|
||||
if err != nil {
|
||||
fmt.Printf("Failed connecting to %s, error: %s\n", peer.ID, err)
|
||||
} else {
|
||||
fmt.Println("Connected to:", peer.ID)
|
||||
}
|
||||
}
|
||||
}
|
||||
fmt.Println("Peer discovery complete")
|
||||
}
|
||||
|
||||
func printErr(err error) {
|
||||
fmt.Println("error:", err)
|
||||
}
|
||||
|
||||
func ConnectedPeers(h host.Host) []*peer.AddrInfo {
|
||||
var pinfos []*peer.AddrInfo
|
||||
for _, c := range h.Network().Conns() {
|
||||
pinfos = append(pinfos, &peer.AddrInfo{
|
||||
ID: c.RemotePeer(),
|
||||
Addrs: []multiaddr.Multiaddr{c.RemoteMultiaddr()},
|
||||
})
|
||||
}
|
||||
return pinfos
|
||||
}
|
||||
Reference in New Issue
Block a user