Compare commits
3 commits
52a87a41da
...
c4ee8e7b04
Author | SHA1 | Date | |
---|---|---|---|
Lilian Jónsdóttir | c4ee8e7b04 | ||
Lilian Jónsdóttir | 633673200a | ||
Lilian Jónsdóttir | 67ab59ec48 |
|
@ -9,6 +9,7 @@ import (
|
||||||
|
|
||||||
"filippo.io/age"
|
"filippo.io/age"
|
||||||
"git.burning.moe/celediel/agedit/internal/config"
|
"git.burning.moe/celediel/agedit/internal/config"
|
||||||
|
"git.burning.moe/celediel/agedit/pkg/decrypt"
|
||||||
"git.burning.moe/celediel/agedit/pkg/editor"
|
"git.burning.moe/celediel/agedit/pkg/editor"
|
||||||
"git.burning.moe/celediel/agedit/pkg/encrypt"
|
"git.burning.moe/celediel/agedit/pkg/encrypt"
|
||||||
"git.burning.moe/celediel/agedit/pkg/env"
|
"git.burning.moe/celediel/agedit/pkg/env"
|
||||||
|
@ -21,7 +22,7 @@ import (
|
||||||
const (
|
const (
|
||||||
name string = "agedit"
|
name string = "agedit"
|
||||||
usage string = "Edit age encrypted files with your $EDITOR"
|
usage string = "Edit age encrypted files with your $EDITOR"
|
||||||
version string = "0.1.1"
|
version string = "0.2.0"
|
||||||
help_template string = `NAME:
|
help_template string = `NAME:
|
||||||
{{.Name}} {{if .Version}}v{{.Version}}{{end}} - {{.Usage}}
|
{{.Name}} {{if .Version}}v{{.Version}}{{end}} - {{.Usage}}
|
||||||
|
|
||||||
|
@ -242,19 +243,13 @@ func action(ctx *cli.Context) error {
|
||||||
|
|
||||||
// get recipients from specified identities
|
// get recipients from specified identities
|
||||||
for _, id := range identities {
|
for _, id := range identities {
|
||||||
// TODO: figure out how age actually intends for
|
if actual_id, ok := id.(*age.X25519Identity); ok {
|
||||||
// TODO: a recpient to be retrieved from an age.Identity
|
|
||||||
// TODO: beccause this is stupid and I hate it
|
|
||||||
actual_id, err := age.ParseX25519Identity(fmt.Sprint(id))
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("couldn't get recipient? %v", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
recipients = append(recipients, actual_id.Recipient())
|
recipients = append(recipients, actual_id.Recipient())
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// try to decrypt the file
|
// try to decrypt the file
|
||||||
decrypted, err := encrypt.Decrypt(input_file, identities...)
|
decrypted, err := decrypt.Decrypt(input_file, identities...)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
32
pkg/decrypt/decrypt.go
Normal file
32
pkg/decrypt/decrypt.go
Normal file
|
@ -0,0 +1,32 @@
|
||||||
|
package decrypt
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"filippo.io/age"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Decrypt decrypts bytes from filename
|
||||||
|
func Decrypt(filename string, identities ...age.Identity) ([]byte, error) {
|
||||||
|
var (
|
||||||
|
f *os.File
|
||||||
|
r io.Reader
|
||||||
|
err error
|
||||||
|
out = &bytes.Buffer{}
|
||||||
|
)
|
||||||
|
if f, err = os.Open(filename); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if r, err = age.Decrypt(f, identities...); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := io.Copy(out, r); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return out.Bytes(), nil
|
||||||
|
}
|
|
@ -38,61 +38,3 @@ func Encrypt(data []byte, filename string, recipients ...age.Recipient) error {
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// Decrypt decrypts bytes from filename
|
|
||||||
func Decrypt(filename string, identities ...age.Identity) ([]byte, error) {
|
|
||||||
var (
|
|
||||||
f *os.File
|
|
||||||
r io.Reader
|
|
||||||
err error
|
|
||||||
out = &bytes.Buffer{}
|
|
||||||
)
|
|
||||||
if f, err = os.Open(filename); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
if r, err = age.Decrypt(f, identities...); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
if _, err := io.Copy(out, r); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return out.Bytes(), nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// NewIdentity generates a new Age identity
|
|
||||||
func NewIdentity() (*age.X25519Identity, error) {
|
|
||||||
id, err := age.GenerateX25519Identity()
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return id, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// ReadIdentityFromFile reads the identity from the supplied filename
|
|
||||||
func ReadIdentityFromFile(filename string) (*age.X25519Identity, error) {
|
|
||||||
bytes, err := os.ReadFile(filename)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
id, err := age.ParseX25519Identity(string(bytes))
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
return id, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// WriteIdentityToFile writes the supplied identity to the supplied filename
|
|
||||||
func WriteIdentityToFile(id *age.X25519Identity, filename string) error {
|
|
||||||
err := os.WriteFile(filename, []byte(id.String()), fs.FileMode(0600))
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
package encrypt
|
package encrypt_decrypt_test
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"io/fs"
|
"io/fs"
|
||||||
|
@ -6,6 +6,9 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
"filippo.io/age"
|
"filippo.io/age"
|
||||||
|
"git.burning.moe/celediel/agedit/pkg/decrypt"
|
||||||
|
"git.burning.moe/celediel/agedit/pkg/encrypt"
|
||||||
|
"git.burning.moe/celediel/agedit/pkg/identity"
|
||||||
"git.burning.moe/celediel/agedit/pkg/tmpfile"
|
"git.burning.moe/celediel/agedit/pkg/tmpfile"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -49,11 +52,11 @@ func TestEncryptionDecryption(t *testing.T) {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if err = Encrypt(b, encrypted_outname, id.Recipient()); err != nil {
|
if err = encrypt.Encrypt(b, encrypted_outname, id.Recipient()); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if b, err = Decrypt(encrypted_outname, id); err != nil {
|
if b, err = decrypt.Decrypt(encrypted_outname, id); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -105,13 +108,13 @@ func TestMultipleIdentities(t *testing.T) {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if err = Encrypt(b, encrypted_outname, recipients...); err != nil {
|
if err = encrypt.Encrypt(b, encrypted_outname, recipients...); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// try decrypting with each identity
|
// try decrypting with each identity
|
||||||
for _, id := range identities {
|
for _, id := range identities {
|
||||||
if b, err = Decrypt(encrypted_outname, id); err != nil {
|
if b, err = decrypt.Decrypt(encrypted_outname, id); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
if string(b) != str {
|
if string(b) != str {
|
||||||
|
@ -120,7 +123,7 @@ func TestMultipleIdentities(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// then all of them because why not
|
// then all of them because why not
|
||||||
if b, err = Decrypt(encrypted_outname, identities...); err != nil {
|
if b, err = decrypt.Decrypt(encrypted_outname, identities...); err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -145,23 +148,23 @@ func TestNewIdentity(t *testing.T) {
|
||||||
for range 1000 {
|
for range 1000 {
|
||||||
outfile := generator.GenerateFullPath()
|
outfile := generator.GenerateFullPath()
|
||||||
|
|
||||||
identity, err := NewIdentity()
|
id, err := identity.NewIdentity()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
err = WriteIdentityToFile(identity, outfile)
|
err = identity.WriteIdentityToFile(id, outfile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
other_identity, err := ReadIdentityFromFile(outfile)
|
other_identity, err := identity.ReadIdentityFromFile(outfile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatal(err)
|
t.Fatal(err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if identity.Recipient().String() != other_identity.Recipient().String() && identity.String() != other_identity.String() {
|
if id.Recipient().String() != other_identity.Recipient().String() && id.String() != other_identity.String() {
|
||||||
t.Fatal("Identities don't match!", identity.Recipient(), "!=", identity.Recipient())
|
t.Fatal("Identities don't match!", id.Recipient(), "!=", id.Recipient())
|
||||||
}
|
}
|
||||||
os.Remove(outfile)
|
os.Remove(outfile)
|
||||||
}
|
}
|
43
pkg/identity/identity.go
Normal file
43
pkg/identity/identity.go
Normal file
|
@ -0,0 +1,43 @@
|
||||||
|
package identity
|
||||||
|
|
||||||
|
import (
|
||||||
|
"io/fs"
|
||||||
|
"os"
|
||||||
|
|
||||||
|
"filippo.io/age"
|
||||||
|
)
|
||||||
|
|
||||||
|
// NewIdentity generates a new Age identity
|
||||||
|
func NewIdentity() (*age.X25519Identity, error) {
|
||||||
|
id, err := age.GenerateX25519Identity()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return id, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ReadIdentityFromFile reads the identity from the supplied filename
|
||||||
|
func ReadIdentityFromFile(filename string) (*age.X25519Identity, error) {
|
||||||
|
bytes, err := os.ReadFile(filename)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
id, err := age.ParseX25519Identity(string(bytes))
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
return id, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// WriteIdentityToFile writes the supplied identity to the supplied filename
|
||||||
|
func WriteIdentityToFile(id *age.X25519Identity, filename string) error {
|
||||||
|
err := os.WriteFile(filename, []byte(id.String()), fs.FileMode(0600))
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
Loading…
Reference in a new issue