aboutsummaryrefslogtreecommitdiffstats
path: root/vendor/golang.org/x/crypto/openpgp/packet/userattribute.go
blob: 96a2b382a1de036fd13217c846b5b6bd795330c9 (plain) (blame)
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
// Copyright 2013 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package packet

import (
    "bytes"
    "image"
    "image/jpeg"
    "io"
    "io/ioutil"
)

const UserAttrImageSubpacket = 1

// UserAttribute is capable of storing other types of data about a user
// beyond name, email and a text comment. In practice, user attributes are typically used
// to store a signed thumbnail photo JPEG image of the user.
// See RFC 4880, section 5.12.
type UserAttribute struct {
    Contents []*OpaqueSubpacket
}

// NewUserAttributePhoto creates a user attribute packet
// containing the given images.
func NewUserAttributePhoto(photos ...image.Image) (uat *UserAttribute, err error) {
    uat = new(UserAttribute)
    for _, photo := range photos {
        var buf bytes.Buffer
        // RFC 4880, Section 5.12.1.
        data := []byte{
            0x10, 0x00, // Little-endian image header length (16 bytes)
            0x01,       // Image header version 1
            0x01,       // JPEG
            0, 0, 0, 0, // 12 reserved octets, must be all zero.
            0, 0, 0, 0,
            0, 0, 0, 0}
        if _, err = buf.Write(data); err != nil {
            return
        }
        if err = jpeg.Encode(&buf, photo, nil); err != nil {
            return
        }
        uat.Contents = append(uat.Contents, &OpaqueSubpacket{
            SubType:  UserAttrImageSubpacket,
            Contents: buf.Bytes()})
    }
    return
}

// NewUserAttribute creates a new user attribute packet containing the given subpackets.
func NewUserAttribute(contents ...*OpaqueSubpacket) *UserAttribute {
    return &UserAttribute{Contents: contents}
}

func (uat *UserAttribute) parse(r io.Reader) (err error) {
    // RFC 4880, section 5.13
    b, err := ioutil.ReadAll(r)
    if err != nil {
        return
    }
    uat.Contents, err = OpaqueSubpackets(b)
    return
}

// Serialize marshals the user attribute to w in the form of an OpenPGP packet, including
// header.
func (uat *UserAttribute) Serialize(w io.Writer) (err error) {
    var buf bytes.Buffer
    for _, sp := range uat.Contents {
        sp.Serialize(&buf)
    }
    if err = serializeHeader(w, packetTypeUserAttribute, buf.Len()); err != nil {
        return err
    }
    _, err = w.Write(buf.Bytes())
    return
}

// ImageData returns zero or more byte slices, each containing
// JPEG File Interchange Format (JFIF), for each photo in the
// the user attribute packet.
func (uat *UserAttribute) ImageData() (imageData [][]byte) {
    for _, sp := range uat.Contents {
        if sp.SubType == UserAttrImageSubpacket && len(sp.Contents) > 16 {
            imageData = append(imageData, sp.Contents[16:])
        }
    }
    return
}