Initial commit: hotpcalc CLI
This commit is contained in:
@@ -0,0 +1,210 @@
|
||||
// Package initfile читает и пишет файлы инициализации токенов в родном
|
||||
// XML-формате SafeNet/Aladdin для eToken PASS.
|
||||
//
|
||||
// Структура формата: Tokens → Token[@serial] → Applications → Application →
|
||||
// {Seed, MovingFactor}. Seed записывается шестнадцатеричным текстом, счётчик —
|
||||
// в MovingFactor. Полей для разрядности, режима и периода в формате нет:
|
||||
// он описывает только шестиразрядный HOTP, а алгоритм импортирующая сторона
|
||||
// выводит из длины seed (64 hex-символа — SHA256, иначе SHA1).
|
||||
package initfile
|
||||
|
||||
import (
|
||||
"encoding/hex"
|
||||
"encoding/xml"
|
||||
"fmt"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"hotpcalc/internal/otp"
|
||||
"hotpcalc/internal/store"
|
||||
)
|
||||
|
||||
// Значения по умолчанию для полей, которые eToken PASS заполняет на заводе.
|
||||
const (
|
||||
DefaultConnectorID = "{ab1397d2-ddb6-4705-b66e-9f83f322deb9}"
|
||||
DefaultProductName = "eToken PASS"
|
||||
DefaultCaseModel = "5"
|
||||
DefaultModel = "101"
|
||||
)
|
||||
|
||||
// Options задаёт заводские поля выгружаемого файла.
|
||||
type Options struct {
|
||||
ConnectorID string
|
||||
ProductName string
|
||||
CaseModel string
|
||||
Model string
|
||||
ProductionDate time.Time
|
||||
}
|
||||
|
||||
func (o Options) withDefaults() Options {
|
||||
if o.ConnectorID == "" {
|
||||
o.ConnectorID = DefaultConnectorID
|
||||
}
|
||||
if o.ProductName == "" {
|
||||
o.ProductName = DefaultProductName
|
||||
}
|
||||
if o.CaseModel == "" {
|
||||
o.CaseModel = DefaultCaseModel
|
||||
}
|
||||
if o.Model == "" {
|
||||
o.Model = DefaultModel
|
||||
}
|
||||
if o.ProductionDate.IsZero() {
|
||||
o.ProductionDate = time.Now()
|
||||
}
|
||||
return o
|
||||
}
|
||||
|
||||
type xmlTokens struct {
|
||||
XMLName xml.Name `xml:"Tokens"`
|
||||
Tokens []xmlToken `xml:"Token"`
|
||||
}
|
||||
|
||||
type xmlToken struct {
|
||||
Serial string `xml:"serial,attr"`
|
||||
CaseModel string `xml:"CaseModel,omitempty"`
|
||||
Model string `xml:"Model,omitempty"`
|
||||
ProductionDate string `xml:"ProductionDate,omitempty"`
|
||||
ProductName string `xml:"ProductName,omitempty"`
|
||||
Applications xmlApplications `xml:"Applications"`
|
||||
}
|
||||
|
||||
type xmlApplications struct {
|
||||
Applications []xmlApplication `xml:"Application"`
|
||||
}
|
||||
|
||||
type xmlApplication struct {
|
||||
ConnectorID string `xml:"ConnectorID,attr"`
|
||||
Seed string `xml:"Seed"`
|
||||
MovingFactor string `xml:"MovingFactor"`
|
||||
}
|
||||
|
||||
// WriteSafeNet выгружает токены в XML-файл инициализации.
|
||||
func WriteSafeNet(w io.Writer, tokens []store.Token, o Options) error {
|
||||
if len(tokens) == 0 {
|
||||
return fmt.Errorf("нечего выгружать: список токенов пуст")
|
||||
}
|
||||
o = o.withDefaults()
|
||||
|
||||
doc := xmlTokens{Tokens: make([]xmlToken, 0, len(tokens))}
|
||||
for _, t := range tokens {
|
||||
if err := store.ValidateSerial(t.Serial); err != nil {
|
||||
return err
|
||||
}
|
||||
doc.Tokens = append(doc.Tokens, xmlToken{
|
||||
Serial: t.Serial,
|
||||
CaseModel: o.CaseModel,
|
||||
Model: o.Model,
|
||||
ProductionDate: o.ProductionDate.Format("01/02/2006"),
|
||||
ProductName: o.ProductName,
|
||||
Applications: xmlApplications{Applications: []xmlApplication{{
|
||||
ConnectorID: o.ConnectorID,
|
||||
Seed: strings.ToUpper(hex.EncodeToString(t.Secret)),
|
||||
MovingFactor: strconv.FormatUint(t.Counter, 10),
|
||||
}}},
|
||||
})
|
||||
}
|
||||
|
||||
if _, err := io.WriteString(w, xml.Header); err != nil {
|
||||
return err
|
||||
}
|
||||
enc := xml.NewEncoder(w)
|
||||
enc.Indent("", " ")
|
||||
if err := enc.Encode(doc); err != nil {
|
||||
return fmt.Errorf("не удалось сформировать XML: %w", err)
|
||||
}
|
||||
if _, err := io.WriteString(w, "\n"); err != nil {
|
||||
return err
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
// ReadSafeNet читает файл инициализации. Отсутствующие в формате параметры
|
||||
// восстанавливаются по тем же правилам, что применяет импортирующая сторона:
|
||||
// тип hotp, шесть разрядов, алгоритм по длине seed.
|
||||
func ReadSafeNet(r io.Reader) ([]store.Token, error) {
|
||||
var doc xmlTokens
|
||||
dec := xml.NewDecoder(r)
|
||||
if err := dec.Decode(&doc); err != nil {
|
||||
return nil, fmt.Errorf("не удалось разобрать XML: %w", err)
|
||||
}
|
||||
if doc.XMLName.Local != "Tokens" {
|
||||
return nil, fmt.Errorf("корневой элемент %q, ожидался Tokens", doc.XMLName.Local)
|
||||
}
|
||||
if len(doc.Tokens) == 0 {
|
||||
return nil, fmt.Errorf("в файле нет ни одного элемента Token")
|
||||
}
|
||||
|
||||
out := make([]store.Token, 0, len(doc.Tokens))
|
||||
for _, xt := range doc.Tokens {
|
||||
if xt.Serial == "" {
|
||||
return nil, fmt.Errorf("найден Token без атрибута serial")
|
||||
}
|
||||
if len(xt.Applications.Applications) == 0 {
|
||||
return nil, fmt.Errorf("токен %s: нет элемента Application", xt.Serial)
|
||||
}
|
||||
app := xt.Applications.Applications[0]
|
||||
|
||||
seed := strings.TrimSpace(app.Seed)
|
||||
if seed == "" {
|
||||
return nil, fmt.Errorf("токен %s: нет элемента Seed", xt.Serial)
|
||||
}
|
||||
secret, err := hex.DecodeString(seed)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("токен %s: Seed не является корректным hex: %w", xt.Serial, err)
|
||||
}
|
||||
|
||||
var counter uint64
|
||||
if mf := strings.TrimSpace(app.MovingFactor); mf != "" {
|
||||
if counter, err = strconv.ParseUint(mf, 10, 64); err != nil {
|
||||
return nil, fmt.Errorf("токен %s: некорректный MovingFactor %q", xt.Serial, mf)
|
||||
}
|
||||
}
|
||||
|
||||
alg := otp.SHA1
|
||||
if len(seed) == 64 {
|
||||
alg = otp.SHA256
|
||||
}
|
||||
// ProductName у заводских файлов — название модели, а не владелец;
|
||||
// в метку оно попадает, только если несёт что-то своё.
|
||||
label := strings.TrimSpace(xt.ProductName)
|
||||
if label == DefaultProductName {
|
||||
label = ""
|
||||
}
|
||||
t := store.Token{
|
||||
Serial: xt.Serial, Type: store.TypeHOTP, Secret: secret,
|
||||
Digits: 6, Alg: alg, Counter: counter, Label: label,
|
||||
}
|
||||
if err := t.Validate(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
out = append(out, t)
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// Warnings перечисляет параметры токена, которые формат SafeNet передать
|
||||
// не может: сервер импортирует такой токен с настройками по умолчанию,
|
||||
// и коды не сойдутся.
|
||||
func Warnings(t store.Token) []string {
|
||||
var w []string
|
||||
if t.IsTOTP() {
|
||||
w = append(w, "формат описывает только HOTP — режим TOTP и период будут потеряны")
|
||||
}
|
||||
if t.Digits != 6 {
|
||||
w = append(w, fmt.Sprintf("формат не хранит разрядность — сервер импортирует токен как 6-разрядный, а не %d-разрядный", t.Digits))
|
||||
}
|
||||
|
||||
// Импортирующая сторона выводит алгоритм из длины seed.
|
||||
guessed := otp.SHA1
|
||||
if len(t.Secret) == 32 {
|
||||
guessed = otp.SHA256
|
||||
}
|
||||
if t.Alg != guessed {
|
||||
w = append(w, fmt.Sprintf("формат не хранит алгоритм — при seed в %d байт сервер выберет %s вместо %s",
|
||||
len(t.Secret), guessed, t.Alg))
|
||||
}
|
||||
return w
|
||||
}
|
||||
Reference in New Issue
Block a user