示例#1
0
func (t authorizedKeyType) CreateKey(raw utils.RawMessage) (KeyLocator, error) {
	var value string
	if err := json.Unmarshal(raw, &value); err != nil {
		return nil, errors.New("The key value must be a string in the authorized_keys format.")
	}

	pk, _, _, _, ok := key.ParseAuthorizedKey([]byte(value))
	if !ok {
		return nil, errors.New("Unable to parse the provided key")
	}

	contents := key.MarshalAuthorizedKey(pk)
	fingerprint := KeyFingerprint(pk)
	path := publicKeyPathFor(fingerprint)

	if err := utils.AtomicWriteToContentPath(path, 0664, contents); err != nil {
		return nil, err
	}
	return &SimpleKeyLocator{path, fingerprint.ToShortName()}, nil
}
示例#2
0
文件: cmd.go 项目: jhadvig/geard
func readAuthorizedKeysFile(keyFile string) ([]jobs.KeyData, error) {
	var (
		data []byte
		keys []jobs.KeyData
		err  error
	)

	// keyFile - contains the sshd AuthorizedKeysFile location
	// Stdin - contains the AuthorizedKeysFile if keyFile is not specified
	if len(keyFile) != 0 {
		absPath, _ := filepath.Abs(keyFile)
		data, err = ioutil.ReadFile(absPath)
		if err != nil {
			return keys, err
		}
	} else {
		data, _ = ioutil.ReadAll(os.Stdin)
	}

	bytesReader := bytes.NewReader(data)
	scanner := bufio.NewScanner(bytesReader)
	for scanner.Scan() {
		// Parse the AuthorizedKeys line
		pk, _, _, _, ok := sshkey.ParseAuthorizedKey(scanner.Bytes())
		if !ok {
			return keys, errors.New("Unable to parse authorized key from input source, invalid format")
		}
		value := sshkey.MarshalAuthorizedKey(pk)
		key, err := jobs.NewKeyData("authorized_keys", string(value))
		if err != nil {
			return keys, err
		}
		keys = append(keys, *key)
	}

	return keys, nil
}