File: dumpscts.go

package info (click to toggle)
golang-github-google-certificate-transparency 1.3.2-2
  • links: PTS, VCS
  • area: main
  • in suites: sid
  • size: 4,764 kB
  • sloc: sh: 606; makefile: 103; sql: 16
file content (73 lines) | stat: -rw-r--r-- 1,871 bytes parent folder | download | duplicates (3)
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
// Copyright 2015 Google LLC. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

// Dumpscts prints out SCTs written to a file by the preloader command in
// the ../preloader directory.
package main

import (
	"compress/zlib"
	"encoding/gob"
	"flag"
	"io"
	"log"
	"os"

	"github.com/google/certificate-transparency-go/preload"
)

var sctFile = flag.String("sct_file", "", "File to load SCTs & leaf data from")

func main() {
	flag.Parse()
	var sctReader io.ReadCloser
	if *sctFile == "" {
		log.Fatal("Must specify --sct_file")
	}

	sctFileReader, err := os.Open(*sctFile)
	if err != nil {
		log.Fatal(err)
	}
	sctReader, err = zlib.NewReader(sctFileReader)
	if err != nil {
		log.Fatal(err)
	}
	defer func() {
		err := sctReader.Close()
		if err != nil && err != io.EOF {
			log.Fatalf("Error closing file: %s", err)
		}
	}()

	// TODO(alcutter) should probably store this stuff in a protobuf really.
	decoder := gob.NewDecoder(sctReader)
	var addedCert preload.AddedCert
	numAdded := 0
	numFailed := 0
	for {
		err = decoder.Decode(&addedCert)
		if err != nil {
			break
		}
		if addedCert.AddedOk {
			log.Println(addedCert.SignedCertificateTimestamp)
			numAdded++
		} else {
			log.Printf("Cert was not added: %s", addedCert.ErrorMessage)
			numFailed++
		}
	}
	log.Printf("Num certs added: %d, num failed: %d\n", numAdded, numFailed)
}