Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add some example binaries #55

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 6 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
@@ -1 +1,7 @@
examples/activate-volume/activate-volume
examples/change-pin/change-pin
examples/provision-status/provision-status
examples/provision-tpm/provision-tpm
examples/seal-key/seal-key
examples/unseal-key/unseal-key
vendor/*/
139 changes: 139 additions & 0 deletions examples/activate-volume/main.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,139 @@
// -*- Mode: Go; indent-tabs-mode: t -*-

/*
* Copyright (C) 2019 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/

package main

import (
"flag"
"fmt"
"io"
"os"
"strconv"
"strings"

"github.com/snapcore/secboot"
)

func run() int {

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think it would be more idiomatic to return an error instead of the OS error code directly, printing the error message and exiting with the appropriate code in main().

args := flag.Args()
if len(args) == 0 {
fmt.Printf("Usage: activate-volume VOLUME SOURCE-DEVICE SEALED-KEY-FILE [AUTH-FILE] [OPTIONS]\n")
return 0
}

if len(args) < 3 {
fmt.Fprintf(os.Stderr, "Cannot activate device: insufficient arguments\n")
return 1
}

volume := args[0]
sourceDevice := args[1]

var keyFilePath string
if args[2] != "" && args[2] != "-" && args[2] != "none" {
keyFilePath = args[2]
}

var authFilePath string
if len(args) >= 4 && args[3] != "" && args[3] != "-" && args[3] != "none" {
authFilePath = args[3]
}

var lock bool
var forceRecovery bool
pinTries := 1
recoveryTries := 1
var activateOptions []string

if len(args) >= 5 && args[4] != "" && args[4] != "-" && args[4] != "none" {

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This command line argument parsing looks familiar, possibly the same one used in the old unlock utility? :)

opts := strings.Split(args[4], ",")
for _, opt := range opts {
switch {
case opt == "lock":
lock = true
case opt == "force-recovery":
forceRecovery = true
case strings.HasPrefix(opt, "pin-tries="):
u, err := strconv.ParseUint(strings.TrimPrefix(opt, "pin-tries="), 10, 8)
if err != nil {
fmt.Fprintf(os.Stderr, "Cannot activate device %s: invalid value for \"recovery-tries=\"\n", sourceDevice)
return 1
}
pinTries = int(u)
case strings.HasPrefix(opt, "recovery-tries="):
u, err := strconv.ParseUint(strings.TrimPrefix(opt, "recovery-tries="), 10, 8)
if err != nil {
fmt.Fprintf(os.Stderr, "Cannot activate device %s: invalid value for \"recovery-tries=\"\n", sourceDevice)
return 1
}
recoveryTries = int(u)
default:
activateOptions = append(activateOptions, opt)
}
}
}

var authReader io.Reader
if authFilePath != "" {
f, err := os.Open(authFilePath)
if err != nil {
fmt.Fprintf(os.Stderr, "Cannot open auth file: %v\n", err)
return 1
}
defer f.Close()
authReader = f
}

if !forceRecovery {

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Instead of the if/else we could invert the test, process the forced recovery and exit, leaving the non-forced in the normal execution flow.

tpm, err := secboot.ConnectToDefaultTPM()
if err != nil {
fmt.Fprintf(os.Stderr, "Cannot connect to TPM: %v\n", err)
return 1
}
defer tpm.Close()

options := secboot.ActivateWithTPMSealedKeyOptions{
PINTries: pinTries,
RecoveryKeyTries: recoveryTries,
ActivateOptions: activateOptions,
LockSealedKeyAccess: lock}
if success, err := secboot.ActivateVolumeWithTPMSealedKey(tpm, volume, sourceDevice, keyFilePath, authReader, &options); err != nil {
if !success {
fmt.Fprintf(os.Stderr, "Activation failed: %v\n", err)
return 1
}
fmt.Printf("Activation succeeded with fallback recovery key: %v\n", err)
}
} else {
options := secboot.ActivateWithRecoveryKeyOptions{
Tries: recoveryTries,
ActivateOptions: activateOptions}
if err := secboot.ActivateVolumeWithRecoveryKey(volume, sourceDevice, authReader, &options); err != nil {
fmt.Fprintf(os.Stderr, "Activation with recovery key failed: %v\n", err)
return 1
}
}

return 0
}

func main() {
flag.Parse()
os.Exit(run())
}
68 changes: 68 additions & 0 deletions examples/change-pin/main.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,68 @@
// -*- Mode: Go; indent-tabs-mode: t -*-

/*
* Copyright (C) 2019 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/

package main

import (
"flag"
"fmt"
"os"

"github.com/snapcore/secboot"
)

var keyFile string
var currentPin string

func init() {
flag.StringVar(&currentPin, "current-pin", "", "")
flag.StringVar(&keyFile, "key-file", "", "")
}

func run() int {
if keyFile == "" {
fmt.Fprintf(os.Stderr, "Cannot change PIN: missing -key-file\n")
return 1
}

args := flag.Args()
var pin string
if len(args) > 0 {
pin = args[0]
}

tpm, err := secboot.ConnectToDefaultTPM()
if err != nil {
fmt.Fprintf(os.Stderr, "Cannot connect to TPM: %v\n", err)
return 1
}
defer tpm.Close()

if err := secboot.ChangePIN(tpm, keyFile, currentPin, pin); err != nil {
fmt.Fprintf(os.Stderr, "Cannot change PIN: %v\n", err)
return 1
}

return 0
}

func main() {
flag.Parse()
os.Exit(run())
}
84 changes: 84 additions & 0 deletions examples/provision-status/main.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,84 @@
// -*- Mode: Go; indent-tabs-mode: t -*-

/*
* Copyright (C) 2019 Canonical Ltd
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License version 3 as
* published by the Free Software Foundation.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*
*/

package main

import (
"fmt"
"os"

"github.com/snapcore/secboot"
)

func run() int {
tpm, err := secboot.ConnectToDefaultTPM()
if err != nil {
fmt.Fprintf(os.Stderr, "Cannot connect to TPM: %v\n", err)
return 1
}
defer tpm.Close()

status, err := secboot.ProvisionStatus(tpm)
if err != nil {
fmt.Fprintf(os.Stderr, "Cannot determine status: %v\n", err)
return 1
}

if status&secboot.AttrValidSRK > 0 {
fmt.Println("Valid SRK found in TPM")
} else {
fmt.Println("** ERROR: TPM does not have a valid SRK **")
}

if status&secboot.AttrValidEK > 0 {
fmt.Println("Valid EK found in TPM")
} else {
fmt.Println("** ERROR: TPM does not have a valid EK **")
}

if status&secboot.AttrDAParamsOK > 0 {

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I noticed that on my Lenovo test machine the DA parameters stay configured after clearing the TPM, or at least its status attribute is the only one that remains enabled. Is that a normal/expected TPM behavior? In swtpm all attributes are unset.

fmt.Println("TPM's DA parameters are correct")
} else {
fmt.Println("** ERROR: TPM's DA parameters are not the values set during provisioning **")
}

if status&secboot.AttrOwnerClearDisabled > 0 {
fmt.Println("TPM does not allow clearing with the lockout hierarchy authorization")
} else {
fmt.Println("** ERROR: TPM allows clearing with the lockout hierarchy authorization **")
}

if status&secboot.AttrLockoutAuthSet > 0 {
fmt.Println("The lockout hierarchy authorization is set")
} else {
fmt.Println("** ERROR: The lockout hierarchy authorization is not set **")
}

if status&secboot.AttrValidLockNVIndex > 0 {
fmt.Println("Valid lock NV index found in TPM")
} else {
fmt.Println("** ERROR: TPM does not have a valid lock NV index **")
}

return 0
}

func main() {
os.Exit(run())
}
Loading