62 lines
860 B
Go
62 lines
860 B
Go
|
package apdu
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ErrTagNotFound struct {
|
||
|
|
tag uint8
|
||
|
|
}
|
||
|
|
|
||
|
|
func (e *ErrTagNotFound) Error() string {
|
||
|
|
return fmt.Sprintf("tag %x not found", e.tag)
|
||
|
|
}
|
||
|
|
|
||
|
|
func FindTag(raw []byte, tags ...uint8) ([]byte, error) {
|
||
|
|
if len(tags) == 0 {
|
||
|
|
return raw, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
target := tags[0]
|
||
|
|
buf := bytes.NewBuffer(raw)
|
||
|
|
|
||
|
|
var (
|
||
|
|
tag uint8
|
||
|
|
length uint8
|
||
|
|
err error
|
||
|
|
)
|
||
|
|
|
||
|
|
for {
|
||
|
|
tag, err = buf.ReadByte()
|
||
|
|
switch {
|
||
|
|
case err == io.EOF:
|
||
|
|
return []byte{}, &ErrTagNotFound{target}
|
||
|
|
case err != nil:
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
length, err = buf.ReadByte()
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
data := make([]byte, length)
|
||
|
|
if length != 0 {
|
||
|
|
_, err = buf.Read(data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if tag == target {
|
||
|
|
if len(tags) == 1 {
|
||
|
|
return data, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
return FindTag(data, tags[1:]...)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|