Checks if metaheader.json is valid
parent
73664bd3bb
commit
04f40a364c
|
@ -7,6 +7,7 @@ import (
|
||||||
"crypto/tls"
|
"crypto/tls"
|
||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
"encoding/binary"
|
"encoding/binary"
|
||||||
|
"encoding/json"
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
@ -39,6 +40,9 @@ const (
|
||||||
|
|
||||||
// HDR_SIZE total header size
|
// HDR_SIZE total header size
|
||||||
HDR_SIZE = VERSION_SIZE + TYPE_SIZE + UUID_SIZE + HMAC_SIZE + TIMESTAMP_SIZE + SSIZE
|
HDR_SIZE = VERSION_SIZE + TYPE_SIZE + UUID_SIZE + HMAC_SIZE + TIMESTAMP_SIZE + SSIZE
|
||||||
|
|
||||||
|
// MH_FILE_LIMIT defines in bytes the max size of the json meta header file
|
||||||
|
MH_FILE_LIMIT = 100000
|
||||||
)
|
)
|
||||||
|
|
||||||
type (
|
type (
|
||||||
|
@ -282,12 +286,29 @@ func d4loadConfig(d4 *d4S) bool {
|
||||||
tmp, _ = strconv.ParseUint(string(readConfFile(d4, "type")), 10, 8)
|
tmp, _ = strconv.ParseUint(string(readConfFile(d4, "type")), 10, 8)
|
||||||
(*d4).conf.ttype = uint8(tmp)
|
(*d4).conf.ttype = uint8(tmp)
|
||||||
// parse meta header file
|
// parse meta header file
|
||||||
|
data := make([]byte, MH_FILE_LIMIT)
|
||||||
if tmp == 254 {
|
if tmp == 254 {
|
||||||
file, err := os.Open((*d4).confdir + "/metaheader.json")
|
file, err := os.Open((*d4).confdir + "/metaheader.json")
|
||||||
if err != nil && err != io.EOF {
|
if err != nil {
|
||||||
panic("Failed to open Meta-Header File.")
|
panic("Failed to open Meta-Header File.")
|
||||||
} else {
|
} else {
|
||||||
(*d4).mh = newMetaHeader(file)
|
if count, err := file.Read(data); err != nil {
|
||||||
|
panic("Failed to open Meta-Header File.")
|
||||||
|
} else {
|
||||||
|
if json.Valid(data[:count]) {
|
||||||
|
if checkType(data[:count]) {
|
||||||
|
if off, err := file.Seek(0, 0); err != nil || off != 0 {
|
||||||
|
panic(fmt.Sprintf("Cannot read Meta-Header file: %s", err))
|
||||||
|
} else {
|
||||||
|
(*d4).mh = newMetaHeader(file)
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
panic("A Meta-Header File should at least contain a 'type' field.")
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
panic("Failed to validate open Meta-Header File.")
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
// Add the custom CA cert in D4 certpool
|
// Add the custom CA cert in D4 certpool
|
||||||
|
@ -302,6 +323,25 @@ func d4loadConfig(d4 *d4S) bool {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func checkType(b []byte) bool {
|
||||||
|
var f interface{}
|
||||||
|
if err := json.Unmarshal(b, &f); err != nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
m := f.(map[string]interface{})
|
||||||
|
for k, v := range m {
|
||||||
|
if k == "type" {
|
||||||
|
switch v.(type) {
|
||||||
|
case string:
|
||||||
|
if v != nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
func newMetaHeader(mhr io.Reader) metaHeader {
|
func newMetaHeader(mhr io.Reader) metaHeader {
|
||||||
return metaHeader{r: mhr}
|
return metaHeader{r: mhr}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue