From b6f99bad2cf69080df8a42890abedf0b1cc3e295 Mon Sep 17 00:00:00 2001
From: Andrew Ayer <andrew@sslmate.com>
Date: Sat, 4 Feb 2017 16:03:48 -0800
Subject: [PATCH] logclient: add support for add-chain request

---
 ct/client/logclient.go | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/ct/client/logclient.go b/ct/client/logclient.go
index 4344fdd..73da904 100644
--- a/ct/client/logclient.go
+++ b/ct/client/logclient.go
@@ -26,6 +26,7 @@ const (
 	GetEntriesPath        = "/ct/v1/get-entries"
 	GetSTHConsistencyPath = "/ct/v1/get-sth-consistency"
 	GetProofByHashPath    = "/ct/v1/get-proof-by-hash"
+	AddChainPath          = "/ct/v1/add-chain"
 )
 
 // LogClient represents a client for a given CT Log instance
@@ -69,6 +70,18 @@ type getAuditProofResponse struct {
 	AuditPath [][]byte `json:"audit_path"`
 }
 
+type addChainRequest struct {
+	Chain [][]byte `json:"chain"`
+}
+
+type addChainResponse struct {
+	SCTVersion uint8  `json:"sct_version"`
+	ID         []byte `json:"id"`
+	Timestamp  uint64 `json:"timestamp"`
+	Extensions []byte `json:"extensions"`
+	Signature  []byte `json:"signature"`
+}
+
 // New constructs a new LogClient instance.
 // |uri| is the base URI of the CT log instance to interact with, e.g.
 // http://ct.googleapis.com/pilot
@@ -240,3 +253,30 @@ func (c *LogClient) GetAuditProof(hash ct.MerkleTreeNode, treeSize uint64) (ct.A
 	}
 	return path, resp.LeafIndex, nil
 }
+
+func (c *LogClient) AddChain(chain [][]byte) (*ct.SignedCertificateTimestamp, error) {
+	req := addChainRequest{Chain: chain}
+
+	var resp addChainResponse
+	if err := c.postAndParse(c.uri+AddChainPath, &req, &resp); err != nil {
+		return nil, err
+	}
+
+	sct := &ct.SignedCertificateTimestamp{
+		SCTVersion: ct.Version(resp.SCTVersion),
+		Timestamp:  resp.Timestamp,
+		Extensions: resp.Extensions,
+	}
+
+	if len(resp.ID) != sha256.Size {
+		return nil, fmt.Errorf("SCT returned by server has invalid id (expected length %d got %d)", sha256.Size, len(resp.ID))
+	}
+	copy(sct.LogID[:], resp.ID)
+
+	ds, err := ct.UnmarshalDigitallySigned(bytes.NewReader(resp.Signature))
+	if err != nil {
+		return nil, err
+	}
+	sct.Signature = *ds
+	return sct, nil
+}