rolling xor in go

xor.go · 1.1 KiB · Go Raw

package main

import (
	"container/ring"
	"flag"
	"io/ioutil"
	"log"
)

var keyFile string
var keyString string
var inFile string
var outFile string
var keyData []byte
var err error

func init() {
	flag.StringVar(&keyFile, "k", "key", "key file to use")
	flag.StringVar(&keyString, "s", "", "key to use, as string")
	flag.StringVar(&inFile, "f", "file", "file encrypt/decrypt")
	flag.StringVar(&outFile, "o", "out", "output file")
}

func check(err error) {
	if err != nil {
		log.Fatal(err)
	}
}

func main() {
	flag.Parse()
	if keyString == "" {
		// load key file
		keyData, err = ioutil.ReadFile(keyFile)
		check(err)
	} else {
		keyData = []byte(keyString)
	}
	// setup ring buffer with key data
	r := ring.New(len(keyData))
	for i := 0; i < r.Len(); i++ {
		r.Value = keyData[i]
		r = r.Next()
	}
	// load encrypted data from file
	data, err := ioutil.ReadFile(inFile)
	check(err)
	for i := 0; i < len(data); i++ { // foreach byte of data
		data[i] = r.Value.(byte) ^ data[i] // xor data byte with current ring element
		r = r.Next()                       // move ring forward
	}
	ioutil.WriteFile(outFile, data, 0700)
}

1	package main
2
3	import (
4	"container/ring"
5	"flag"
6	"io/ioutil"
7	"log"
8	)
9
10	var keyFile string
11	var keyString string
12	var inFile string
13	var outFile string
14	var keyData []byte
15	var err error
16
17	func init() {
18	flag.StringVar(&keyFile, "k", "key", "key file to use")
19	flag.StringVar(&keyString, "s", "", "key to use, as string")
20	flag.StringVar(&inFile, "f", "file", "file encrypt/decrypt")
21	flag.StringVar(&outFile, "o", "out", "output file")
22	}
23
24	func check(err error) {
25	if err != nil {
26	log.Fatal(err)
27	}
28	}
29
30	func main() {
31	flag.Parse()
32	if keyString == "" {
33	// load key file
34	keyData, err = ioutil.ReadFile(keyFile)
35	check(err)
36	} else {
37	keyData = []byte(keyString)
38	}
39	// setup ring buffer with key data
40	r := ring.New(len(keyData))
41	for i := 0; i < r.Len(); i++ {
42	r.Value = keyData[i]
43	r = r.Next()
44	}
45	// load encrypted data from file
46	data, err := ioutil.ReadFile(inFile)
47	check(err)
48	for i := 0; i < len(data); i++ { // foreach byte of data
49	data[i] = r.Value.(byte) ^ data[i] // xor data byte with current ring element
50	r = r.Next() // move ring forward
51	}
52	ioutil.WriteFile(outFile, data, 0700)
53	}
54