diff options
Diffstat (limited to '')
| -rw-r--r-- | state/header.go | 50 |
1 files changed, 25 insertions, 25 deletions
diff --git a/state/header.go b/state/header.go index 3e55899..3becd5d 100644 --- a/state/header.go +++ b/state/header.go @@ -1,57 +1,57 @@ package state import ( + "bareos-zabbix-check/utils" "bytes" "encoding/binary" "fmt" - "log" - "os" + "io" + + "github.com/pkg/errors" ) // c.StateFile()HeaderLength : the length of the state file header struct -const headerLength = 14 + 2 + 4 + 4 + 8 + 8 + 19*8 +const headerLength = 14 + 2 + 4 + 4 + 8 + 8 // + 19*8 -// header : A structure to hold the header of the state file. It is statically aligned for amd64 architecture +// Header is a structure to hold the header of the state file. It is statically aligned for amd64 architecture // This comes from bareos repository file core/src/lib/bsys.cc:525 and core/src/lib/bsys.cc:652 -type header struct { +type Header struct { ID [14]byte _ int16 Version int32 _ int32 LastJobsAddr uint64 EndOfRecentJobResultsList uint64 - Reserved [19]uint64 + //Reserved [19]uint64 } -func (sfh header) String() string { - return fmt.Sprintf("ID: \"%s\", Version: %d, LastJobsAddr: %d, EndOfRecentJobResultsList: %d", sfh.ID[:len(sfh.ID)-2], sfh.Version, sfh.EndOfRecentJobResultsList, sfh.Reserved) +func (sfh *Header) String() string { + return fmt.Sprintf("ID: \"%s\", Version: %d, LastJobsAddr: %d, EndOfRecentJobResultsList: %d", + string(sfh.ID[:utils.Clen(sfh.ID[:])]), sfh.Version, sfh.LastJobsAddr, sfh.EndOfRecentJobResultsList) } -func (s *State) parseHeader(file *os.File) (err error) { +// ParseHeader parses a Header struct +func ParseHeader(handle io.Reader) (h *Header, err error) { // Parsing the state file header - n, data, err := s.readNextBytes(file, headerLength) + data := make([]byte, headerLength) + n, err := handle.Read(data) if err != nil { - return fmt.Errorf("INFO Corrupted state file : %s", err) + return nil, errors.Wrap(err, "Corrupted state file") } if n != headerLength { - return fmt.Errorf("INFO Corrupted state file : invalid header length in %s", s.config.StateFile()) + return nil, fmt.Errorf("Corrupted state file : invalid header length") } buffer := bytes.NewBuffer(data) - err = binary.Read(buffer, binary.LittleEndian, &s.header) - if err != nil { - return fmt.Errorf("INFO Corrupted state file : binary.Read failed on header in %s : %s", s.config.StateFile(), err) - } - if s.config.Verbose() { - log.Printf("Parsed header: %+s\n", s.header) - } - if id := string(s.header.ID[:len(s.header.ID)-1]); id != "Bareos State\n" && id != "Bacula State\n" { - return fmt.Errorf("INFO Corrupted state file : Not a bareos or bacula state file %s", s.config.StateFile()) + h = &Header{} + _ = binary.Read(buffer, binary.LittleEndian, h) // this call cannot fail since we checked the header length + if id := string(h.ID[:utils.Clen(h.ID[:])]); id != "Bareos State\n" && id != "Bacula State\n" { + return nil, fmt.Errorf("Corrupted state file : Not a bareos or bacula state file : %s", id) } - if s.header.Version != 4 { - return fmt.Errorf("INFO Invalid state file : This script only supports bareos state file version 4, got %d", s.header.Version) + if h.Version != 4 { + return nil, fmt.Errorf("Invalid state file : This script only supports bareos state file version 4, got %d", h.Version) } - if s.header.LastJobsAddr == 0 { - return fmt.Errorf("INFO No jobs exist in the state file") + if h.LastJobsAddr == 0 { + return nil, fmt.Errorf("No jobs exist in the state file") } return } |