-
Notifications
You must be signed in to change notification settings - Fork 115
/
stegify.go
144 lines (124 loc) · 4.39 KB
/
stegify.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
//Command line tool capable of steganography encoding and decoding any file within given images as carriers
package main
import (
"flag"
"fmt"
"github.com/DimitarPetrov/stegify/steg"
"os"
"strings"
)
const encode = "encode"
const decode = "decode"
type sliceFlag []string
func (sf *sliceFlag) String() string {
return strings.Join(*sf, " ")
}
func (sf *sliceFlag) Set(value string) error {
*sf = append(*sf, value)
return nil
}
var carrierFilesSlice sliceFlag
var carrierFiles = flag.String("carriers", "", "carrier files in which the data is encoded (separated by space)")
var dataFile = flag.String("data", "", "data file which is being encoded in the carrier")
var resultFilesSlice sliceFlag
var resultFiles = flag.String("results", "", "names of the result files (separated by space)")
func init() {
flag.StringVar(carrierFiles, "c", "", "carrier files in which the data is encoded (separated by space, shorthand for --carriers)")
flag.Var(&carrierFilesSlice, "carrier", "carrier file in which the data is encoded (could be used multiple times for multiple carriers)")
flag.StringVar(dataFile, "d", "", "data file which is being encoded in the carrier (shorthand for --data)")
flag.Var(&resultFilesSlice, "result", "name of the result file (could be used multiple times for multiple result file names)")
flag.StringVar(resultFiles, "r", "", "names of the result files (separated by space, shorthand for --results)")
flag.Usage = func() {
fmt.Fprintln(os.Stdout, "Usage: stegify [encode/decode] [flags...]")
flag.PrintDefaults()
fmt.Fprintln(os.Stdout, `NOTE: When multiple carriers are provided with different kinds of flags, the names provided through "carrier" flag are taken first and with "carriers"/"c" flags second. Same goes for the "result"/"results" flags.`)
fmt.Fprintln(os.Stdout, `NOTE: When no results are provided a default values will be used for the names of the results.`)
}
}
func main() {
operation := parseOperation()
flag.Parse()
carriers := parseCarriers()
results := parseResults()
switch operation {
case encode:
if len(results) == 0 { // if no results provided use defaults
for i := range carriers {
results = append(results, fmt.Sprintf("result%d", i))
}
}
if len(results) != len(carriers) {
fmt.Fprintln(os.Stderr, "Carrier and result files count must be equal when encoding.")
os.Exit(1)
}
if dataFile == nil || *dataFile == "" {
fmt.Fprintln(os.Stderr, "Data file must be specified. Use stegify --help for more information.")
os.Exit(1)
}
err := steg.MultiCarrierEncodeByFileNames(carriers, *dataFile, results)
if err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
case decode:
if len(results) == 0 { // if no result provided use default
results = append(results, "result")
}
if len(results) != 1 {
fmt.Fprintln(os.Stderr, "Only one result file expected.")
os.Exit(1)
}
err := steg.MultiCarrierDecodeByFileNames(carriers, results[0])
if err != nil {
fmt.Fprintln(os.Stderr, err)
os.Exit(1)
}
}
}
func parseOperation() string {
if len(os.Args) < 2 {
fmt.Fprintln(os.Stderr, "Operation must be specified [encode/decode]. Use stegify --help for more information.")
os.Exit(1)
}
operation := os.Args[1]
if operation != encode && operation != decode {
helpFlags := map[string]bool{
"--help": true,
"-help": true,
"--h": true,
"-h": true,
}
if helpFlags[operation] {
flag.Parse()
os.Exit(0)
}
fmt.Fprintf(os.Stderr, "Unsupported operation: %s. Only [encode/decode] operations are supported.\n Use stegify --help for more information.", operation)
os.Exit(1)
}
os.Args = append(os.Args[:1], os.Args[2:]...) // needed because go flags implementation stop parsing after first non-flag argument
return operation
}
func parseCarriers() []string {
carriers := make([]string, 0)
if len(carrierFilesSlice) != 0 {
carriers = append(carriers, carrierFilesSlice...)
}
if len(*carrierFiles) != 0 {
carriers = append(carriers, strings.Split(*carrierFiles, " ")...)
}
if len(carriers) == 0 {
fmt.Fprintln(os.Stderr, "Carrier file must be specified. Use stegify --help for more information.")
os.Exit(1)
}
return carriers
}
func parseResults() []string {
results := make([]string, 0)
if len(resultFilesSlice) != 0 {
results = append(results, resultFilesSlice...)
}
if len(*resultFiles) != 0 {
results = append(results, strings.Split(*resultFiles, " ")...)
}
return results
}