mirror of
https://github.com/superseriousbusiness/gotosocial.git
synced 2024-11-13 21:11:15 +00:00
98263a7de6
* start fixing up tests * fix up tests + automate with drone * fiddle with linting * messing about with drone.yml * some more fiddling * hmmm * add cache * add vendor directory * verbose * ci updates * update some little things * update sig
1521 lines
41 KiB
Go
1521 lines
41 KiB
Go
package exif
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"encoding/binary"
|
|
|
|
"github.com/dsoprea/go-logging"
|
|
|
|
"github.com/dsoprea/go-exif/v2/common"
|
|
"github.com/dsoprea/go-exif/v2/undefined"
|
|
)
|
|
|
|
var (
|
|
ifdEnumerateLogger = log.NewLogger("exif.ifd_enumerate")
|
|
)
|
|
|
|
var (
|
|
// ErrNoThumbnail means that no thumbnail was found.
|
|
ErrNoThumbnail = errors.New("no thumbnail")
|
|
|
|
// ErrNoGpsTags means that no GPS info was found.
|
|
ErrNoGpsTags = errors.New("no gps tags")
|
|
|
|
// ErrTagTypeNotValid means that the tag-type is not valid.
|
|
ErrTagTypeNotValid = errors.New("tag type invalid")
|
|
|
|
// ErrOffsetInvalid means that the file offset is not valid.
|
|
ErrOffsetInvalid = errors.New("file offset invalid")
|
|
)
|
|
|
|
var (
|
|
// ValidGpsVersions is the list of recognized EXIF GPS versions/signatures.
|
|
ValidGpsVersions = [][4]byte{
|
|
// 2.0.0.0 appears to have a very similar format to 2.2.0.0, so enabling
|
|
// it under that assumption.
|
|
//
|
|
// IFD-PATH=[IFD] ID=(0x8825) NAME=[GPSTag] COUNT=(1) TYPE=[LONG] VALUE=[114]
|
|
// IFD-PATH=[IFD/GPSInfo] ID=(0x0000) NAME=[GPSVersionID] COUNT=(4) TYPE=[BYTE] VALUE=[02 00 00 00]
|
|
// IFD-PATH=[IFD/GPSInfo] ID=(0x0001) NAME=[GPSLatitudeRef] COUNT=(2) TYPE=[ASCII] VALUE=[S]
|
|
// IFD-PATH=[IFD/GPSInfo] ID=(0x0002) NAME=[GPSLatitude] COUNT=(3) TYPE=[RATIONAL] VALUE=[38/1...]
|
|
// IFD-PATH=[IFD/GPSInfo] ID=(0x0003) NAME=[GPSLongitudeRef] COUNT=(2) TYPE=[ASCII] VALUE=[E]
|
|
// IFD-PATH=[IFD/GPSInfo] ID=(0x0004) NAME=[GPSLongitude] COUNT=(3) TYPE=[RATIONAL] VALUE=[144/1...]
|
|
// IFD-PATH=[IFD/GPSInfo] ID=(0x0012) NAME=[GPSMapDatum] COUNT=(7) TYPE=[ASCII] VALUE=[WGS-84]
|
|
//
|
|
{2, 0, 0, 0},
|
|
|
|
{2, 2, 0, 0},
|
|
|
|
// Suddenly appeared at the default in 2.31: https://home.jeita.or.jp/tsc/std-pdf/CP-3451D.pdf
|
|
//
|
|
// Note that the presence of 2.3.0.0 doesn't seem to guarantee
|
|
// coordinates. In some cases, we seen just the following:
|
|
//
|
|
// GPS Tag Version |2.3.0.0
|
|
// GPS Receiver Status |V
|
|
// Geodetic Survey Data|WGS-84
|
|
// GPS Differential Cor|0
|
|
//
|
|
{2, 3, 0, 0},
|
|
}
|
|
)
|
|
|
|
// byteParser knows how to decode an IFD and all of the tags it
|
|
// describes.
|
|
//
|
|
// The IFDs and the actual values can float throughout the EXIF block, but the
|
|
// IFD itself is just a minor header followed by a set of repeating,
|
|
// statically-sized records. So, the tags (though notnecessarily their values)
|
|
// are fairly simple to enumerate.
|
|
type byteParser struct {
|
|
byteOrder binary.ByteOrder
|
|
addressableData []byte
|
|
ifdOffset uint32
|
|
currentOffset uint32
|
|
}
|
|
|
|
func newByteParser(addressableData []byte, byteOrder binary.ByteOrder, ifdOffset uint32) (bp *byteParser, err error) {
|
|
if ifdOffset >= uint32(len(addressableData)) {
|
|
return nil, ErrOffsetInvalid
|
|
}
|
|
|
|
// TODO(dustin): Add test
|
|
|
|
bp = &byteParser{
|
|
addressableData: addressableData,
|
|
byteOrder: byteOrder,
|
|
currentOffset: ifdOffset,
|
|
}
|
|
|
|
return bp, nil
|
|
}
|
|
|
|
// getUint16 reads a uint16 and advances both our current and our current
|
|
// accumulator (which allows us to know how far to seek to the beginning of the
|
|
// next IFD when it's time to jump).
|
|
func (bp *byteParser) getUint16() (value uint16, raw []byte, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): Add test
|
|
|
|
needBytes := uint32(2)
|
|
|
|
if bp.currentOffset+needBytes > uint32(len(bp.addressableData)) {
|
|
return 0, nil, io.EOF
|
|
}
|
|
|
|
raw = bp.addressableData[bp.currentOffset : bp.currentOffset+needBytes]
|
|
value = bp.byteOrder.Uint16(raw)
|
|
|
|
bp.currentOffset += uint32(needBytes)
|
|
|
|
return value, raw, nil
|
|
}
|
|
|
|
// getUint32 reads a uint32 and advances both our current and our current
|
|
// accumulator (which allows us to know how far to seek to the beginning of the
|
|
// next IFD when it's time to jump).
|
|
func (bp *byteParser) getUint32() (value uint32, raw []byte, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): Add test
|
|
|
|
needBytes := uint32(4)
|
|
|
|
if bp.currentOffset+needBytes > uint32(len(bp.addressableData)) {
|
|
return 0, nil, io.EOF
|
|
}
|
|
|
|
raw = bp.addressableData[bp.currentOffset : bp.currentOffset+needBytes]
|
|
value = bp.byteOrder.Uint32(raw)
|
|
|
|
bp.currentOffset += uint32(needBytes)
|
|
|
|
return value, raw, nil
|
|
}
|
|
|
|
// CurrentOffset returns the starting offset but the number of bytes that we
|
|
// have parsed. This is arithmetic-based tracking, not a seek(0) operation.
|
|
func (bp *byteParser) CurrentOffset() uint32 {
|
|
return bp.currentOffset
|
|
}
|
|
|
|
// IfdEnumerate is the main enumeration type. It knows how to parse the IFD
|
|
// containers in the EXIF blob.
|
|
type IfdEnumerate struct {
|
|
exifData []byte
|
|
byteOrder binary.ByteOrder
|
|
tagIndex *TagIndex
|
|
ifdMapping *exifcommon.IfdMapping
|
|
furthestOffset uint32
|
|
}
|
|
|
|
// NewIfdEnumerate returns a new instance of IfdEnumerate.
|
|
func NewIfdEnumerate(ifdMapping *exifcommon.IfdMapping, tagIndex *TagIndex, exifData []byte, byteOrder binary.ByteOrder) *IfdEnumerate {
|
|
return &IfdEnumerate{
|
|
exifData: exifData,
|
|
byteOrder: byteOrder,
|
|
ifdMapping: ifdMapping,
|
|
tagIndex: tagIndex,
|
|
}
|
|
}
|
|
|
|
func (ie *IfdEnumerate) getByteParser(ifdOffset uint32) (bp *byteParser, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
bp, err =
|
|
newByteParser(
|
|
ie.exifData[ExifAddressableAreaStart:],
|
|
ie.byteOrder,
|
|
ifdOffset)
|
|
|
|
if err != nil {
|
|
if err == ErrOffsetInvalid {
|
|
return nil, err
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
|
|
return bp, nil
|
|
}
|
|
|
|
func (ie *IfdEnumerate) parseTag(ii *exifcommon.IfdIdentity, tagPosition int, bp *byteParser) (ite *IfdTagEntry, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
tagId, _, err := bp.getUint16()
|
|
log.PanicIf(err)
|
|
|
|
tagTypeRaw, _, err := bp.getUint16()
|
|
log.PanicIf(err)
|
|
|
|
tagType := exifcommon.TagTypePrimitive(tagTypeRaw)
|
|
|
|
unitCount, _, err := bp.getUint32()
|
|
log.PanicIf(err)
|
|
|
|
valueOffset, rawValueOffset, err := bp.getUint32()
|
|
log.PanicIf(err)
|
|
|
|
if tagType.IsValid() == false {
|
|
ite = &IfdTagEntry{
|
|
tagId: tagId,
|
|
tagType: tagType,
|
|
}
|
|
|
|
log.Panic(ErrTagTypeNotValid)
|
|
}
|
|
|
|
ite = newIfdTagEntry(
|
|
ii,
|
|
tagId,
|
|
tagPosition,
|
|
tagType,
|
|
unitCount,
|
|
valueOffset,
|
|
rawValueOffset,
|
|
ie.exifData[ExifAddressableAreaStart:],
|
|
ie.byteOrder)
|
|
|
|
ifdPath := ii.UnindexedString()
|
|
|
|
// If it's an IFD but not a standard one, it'll just be seen as a LONG
|
|
// (the standard IFD tag type), later, unless we skip it because it's
|
|
// [likely] not even in the standard list of known tags.
|
|
mi, err := ie.ifdMapping.GetChild(ifdPath, tagId)
|
|
if err == nil {
|
|
currentIfdTag := ii.IfdTag()
|
|
|
|
childIt := exifcommon.NewIfdTag(¤tIfdTag, tagId, mi.Name)
|
|
iiChild := ii.NewChild(childIt, 0)
|
|
ite.SetChildIfd(iiChild)
|
|
|
|
// We also need to set `tag.ChildFqIfdPath` but can't do it here
|
|
// because we don't have the IFD index.
|
|
} else if log.Is(err, exifcommon.ErrChildIfdNotMapped) == false {
|
|
log.Panic(err)
|
|
}
|
|
|
|
return ite, nil
|
|
}
|
|
|
|
// TagVisitorFn is called for each tag when enumerating through the EXIF.
|
|
type TagVisitorFn func(fqIfdPath string, ifdIndex int, ite *IfdTagEntry) (err error)
|
|
|
|
// postparseTag do some tag-level processing here following the parse of each.
|
|
func (ie *IfdEnumerate) postparseTag(ite *IfdTagEntry, med *MiscellaneousExifData) (err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): Add test
|
|
|
|
ii := ite.IfdIdentity()
|
|
|
|
tagId := ite.TagId()
|
|
tagType := ite.TagType()
|
|
|
|
it, err := ie.tagIndex.Get(ii, tagId)
|
|
if err == nil {
|
|
ite.setTagName(it.Name)
|
|
} else {
|
|
if err != ErrTagNotFound {
|
|
log.Panic(err)
|
|
}
|
|
|
|
// This is an unknown tag.
|
|
|
|
originalBt := exifcommon.BasicTag{
|
|
FqIfdPath: ii.String(),
|
|
IfdPath: ii.UnindexedString(),
|
|
TagId: tagId,
|
|
}
|
|
|
|
if med != nil {
|
|
med.unknownTags[originalBt] = exifcommon.BasicTag{}
|
|
}
|
|
|
|
utilityLogger.Debugf(nil,
|
|
"Tag (0x%04x) is not valid for IFD [%s]. Attempting secondary "+
|
|
"lookup.", tagId, ii.String())
|
|
|
|
// This will overwrite the existing `it` and `err`. Since `FindFirst()`
|
|
// might generate different Errors than `Get()`, the log message above
|
|
// is import to try and mitigate confusion in that case.
|
|
it, err = ie.tagIndex.FindFirst(tagId, tagType, nil)
|
|
if err != nil {
|
|
if err != ErrTagNotFound {
|
|
log.Panic(err)
|
|
}
|
|
|
|
// This is supposed to be a convenience function and if we were
|
|
// to keep the name empty or set it to some placeholder, it
|
|
// might be mismanaged by the package that is calling us. If
|
|
// they want to specifically manage these types of tags, they
|
|
// can use more advanced functionality to specifically -handle
|
|
// unknown tags.
|
|
utilityLogger.Warningf(nil,
|
|
"Tag with ID (0x%04x) in IFD [%s] is not recognized and "+
|
|
"will be ignored.", tagId, ii.String())
|
|
|
|
return ErrTagNotFound
|
|
}
|
|
|
|
ite.setTagName(it.Name)
|
|
|
|
utilityLogger.Warningf(nil,
|
|
"Tag with ID (0x%04x) is not valid for IFD [%s], but it *is* "+
|
|
"valid as tag [%s] under IFD [%s] and has the same type "+
|
|
"[%s], so we will use that. This EXIF blob was probably "+
|
|
"written by a buggy implementation.",
|
|
tagId, ii.UnindexedString(), it.Name, it.IfdPath,
|
|
tagType)
|
|
|
|
if med != nil {
|
|
med.unknownTags[originalBt] = exifcommon.BasicTag{
|
|
IfdPath: it.IfdPath,
|
|
TagId: tagId,
|
|
}
|
|
}
|
|
}
|
|
|
|
// This is a known tag (from the standard, unless the user did
|
|
// something different).
|
|
|
|
// Skip any tags that have a type that doesn't match the type in the
|
|
// index (which is loaded with the standard and accept tag
|
|
// information unless configured otherwise).
|
|
//
|
|
// We've run into multiple instances of the same tag, where a) no
|
|
// tag should ever be repeated, and b) all but one had an incorrect
|
|
// type and caused parsing/conversion woes. So, this is a quick fix
|
|
// for those scenarios.
|
|
if it.DoesSupportType(tagType) == false {
|
|
ifdEnumerateLogger.Warningf(nil,
|
|
"Skipping tag [%s] (0x%04x) [%s] with an unexpected type: %v ∉ %v",
|
|
ii.UnindexedString(), tagId, it.Name,
|
|
tagType, it.SupportedTypes)
|
|
|
|
return ErrTagNotFound
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// parseIfd decodes the IFD block that we're currently sitting on the first
|
|
// byte of.
|
|
func (ie *IfdEnumerate) parseIfd(ii *exifcommon.IfdIdentity, bp *byteParser, visitor TagVisitorFn, doDescend bool, med *MiscellaneousExifData) (nextIfdOffset uint32, entries []*IfdTagEntry, thumbnailData []byte, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
tagCount, _, err := bp.getUint16()
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "IFD [%s] tag-count: (%d)", ii.String(), tagCount)
|
|
|
|
entries = make([]*IfdTagEntry, 0)
|
|
|
|
var enumeratorThumbnailOffset *IfdTagEntry
|
|
var enumeratorThumbnailSize *IfdTagEntry
|
|
|
|
for i := 0; i < int(tagCount); i++ {
|
|
ite, err := ie.parseTag(ii, i, bp)
|
|
if err != nil {
|
|
if log.Is(err, ErrTagTypeNotValid) == true {
|
|
// Technically, we have the type on-file in the tags-index, but
|
|
// if the type stored alongside the data disagrees with it,
|
|
// which it apparently does, all bets are off.
|
|
ifdEnumerateLogger.Warningf(nil, "Tag (0x%04x) in IFD [%s] at position (%d) has invalid type (%d) and will be skipped.", ite.tagId, ii, i, ite.tagType)
|
|
continue
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
|
|
err = ie.postparseTag(ite, med)
|
|
if err == nil {
|
|
if err == ErrTagNotFound {
|
|
continue
|
|
}
|
|
|
|
log.PanicIf(err)
|
|
}
|
|
|
|
tagId := ite.TagId()
|
|
|
|
if visitor != nil {
|
|
err := visitor(ii.String(), ii.Index(), ite)
|
|
log.PanicIf(err)
|
|
}
|
|
|
|
if ite.IsThumbnailOffset() == true {
|
|
ifdEnumerateLogger.Debugf(nil, "Skipping the thumbnail offset tag (0x%04x). Use accessors to get it or set it.", tagId)
|
|
|
|
enumeratorThumbnailOffset = ite
|
|
entries = append(entries, ite)
|
|
|
|
continue
|
|
} else if ite.IsThumbnailSize() == true {
|
|
ifdEnumerateLogger.Debugf(nil, "Skipping the thumbnail size tag (0x%04x). Use accessors to get it or set it.", tagId)
|
|
|
|
enumeratorThumbnailSize = ite
|
|
entries = append(entries, ite)
|
|
|
|
continue
|
|
}
|
|
|
|
if ite.TagType() != exifcommon.TypeUndefined {
|
|
// If this tag's value is an offset, bump our max-offset value to
|
|
// what that offset is plus however large that value is.
|
|
|
|
vc := ite.getValueContext()
|
|
|
|
farOffset, err := vc.GetFarOffset()
|
|
if err == nil {
|
|
candidateOffset := farOffset + uint32(vc.SizeInBytes())
|
|
if candidateOffset > ie.furthestOffset {
|
|
ie.furthestOffset = candidateOffset
|
|
}
|
|
} else if err != exifcommon.ErrNotFarValue {
|
|
log.PanicIf(err)
|
|
}
|
|
}
|
|
|
|
// If it's an IFD but not a standard one, it'll just be seen as a LONG
|
|
// (the standard IFD tag type), later, unless we skip it because it's
|
|
// [likely] not even in the standard list of known tags.
|
|
if ite.ChildIfdPath() != "" {
|
|
if doDescend == true {
|
|
ifdEnumerateLogger.Debugf(nil, "Descending from IFD [%s] to IFD [%s].", ii, ite.ChildIfdPath())
|
|
|
|
currentIfdTag := ii.IfdTag()
|
|
|
|
childIfdTag :=
|
|
exifcommon.NewIfdTag(
|
|
¤tIfdTag,
|
|
ite.TagId(),
|
|
ite.ChildIfdName())
|
|
|
|
iiChild := ii.NewChild(childIfdTag, 0)
|
|
|
|
err := ie.scan(iiChild, ite.getValueOffset(), visitor, med)
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Ascending from IFD [%s] to IFD [%s].", ite.ChildIfdPath(), ii)
|
|
}
|
|
}
|
|
|
|
entries = append(entries, ite)
|
|
}
|
|
|
|
if enumeratorThumbnailOffset != nil && enumeratorThumbnailSize != nil {
|
|
thumbnailData, err = ie.parseThumbnail(enumeratorThumbnailOffset, enumeratorThumbnailSize)
|
|
log.PanicIf(err)
|
|
|
|
// In this case, the value is always an offset.
|
|
offset := enumeratorThumbnailOffset.getValueOffset()
|
|
|
|
// This this case, the value is always a length.
|
|
length := enumeratorThumbnailSize.getValueOffset()
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Found thumbnail in IFD [%s]. Its offset is (%d) and is (%d) bytes.", ii, offset, length)
|
|
|
|
furthestOffset := offset + length
|
|
|
|
if furthestOffset > ie.furthestOffset {
|
|
ie.furthestOffset = furthestOffset
|
|
}
|
|
}
|
|
|
|
nextIfdOffset, _, err = bp.getUint32()
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Next IFD at offset: (%08x)", nextIfdOffset)
|
|
|
|
return nextIfdOffset, entries, thumbnailData, nil
|
|
}
|
|
|
|
func (ie *IfdEnumerate) parseThumbnail(offsetIte, lengthIte *IfdTagEntry) (thumbnailData []byte, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
vRaw, err := lengthIte.Value()
|
|
log.PanicIf(err)
|
|
|
|
vList := vRaw.([]uint32)
|
|
if len(vList) != 1 {
|
|
log.Panicf("not exactly one long: (%d)", len(vList))
|
|
}
|
|
|
|
length := vList[0]
|
|
|
|
// The tag is official a LONG type, but it's actually an offset to a blob of bytes.
|
|
offsetIte.updateTagType(exifcommon.TypeByte)
|
|
offsetIte.updateUnitCount(length)
|
|
|
|
thumbnailData, err = offsetIte.GetRawBytes()
|
|
log.PanicIf(err)
|
|
|
|
return thumbnailData, nil
|
|
}
|
|
|
|
// scan parses and enumerates the different IFD blocks and invokes a visitor
|
|
// callback for each tag. No information is kept or returned.
|
|
func (ie *IfdEnumerate) scan(iiGeneral *exifcommon.IfdIdentity, ifdOffset uint32, visitor TagVisitorFn, med *MiscellaneousExifData) (err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): Add test
|
|
|
|
for ifdIndex := 0; ; ifdIndex++ {
|
|
iiSibling := iiGeneral.NewSibling(ifdIndex)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Parsing IFD [%s] at offset (0x%04x) (scan).", iiSibling.String(), ifdOffset)
|
|
|
|
bp, err := ie.getByteParser(ifdOffset)
|
|
if err != nil {
|
|
if err == ErrOffsetInvalid {
|
|
ifdEnumerateLogger.Errorf(nil, nil, "IFD [%s] at offset (0x%04x) is unreachable. Terminating scan.", iiSibling.String(), ifdOffset)
|
|
break
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
|
|
nextIfdOffset, _, _, err := ie.parseIfd(iiSibling, bp, visitor, true, med)
|
|
log.PanicIf(err)
|
|
|
|
currentOffset := bp.CurrentOffset()
|
|
if currentOffset > ie.furthestOffset {
|
|
ie.furthestOffset = currentOffset
|
|
}
|
|
|
|
if nextIfdOffset == 0 {
|
|
break
|
|
}
|
|
|
|
ifdOffset = nextIfdOffset
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// MiscellaneousExifData is reports additional data collected during the parse.
|
|
type MiscellaneousExifData struct {
|
|
// UnknownTags contains all tags that were invalid for their containing
|
|
// IFDs. The values represent alternative IFDs that were correctly matched
|
|
// to those tags and used instead.
|
|
unknownTags map[exifcommon.BasicTag]exifcommon.BasicTag
|
|
}
|
|
|
|
// UnknownTags returns the unknown tags encountered during the scan.
|
|
func (med *MiscellaneousExifData) UnknownTags() map[exifcommon.BasicTag]exifcommon.BasicTag {
|
|
return med.unknownTags
|
|
}
|
|
|
|
// Scan enumerates the different EXIF blocks (called IFDs). `rootIfdName` will
|
|
// be "IFD" in the TIFF standard.
|
|
func (ie *IfdEnumerate) Scan(iiRoot *exifcommon.IfdIdentity, ifdOffset uint32, visitor TagVisitorFn) (med *MiscellaneousExifData, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): Add test
|
|
|
|
med = &MiscellaneousExifData{
|
|
unknownTags: make(map[exifcommon.BasicTag]exifcommon.BasicTag),
|
|
}
|
|
|
|
err = ie.scan(iiRoot, ifdOffset, visitor, med)
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Scan: It looks like the furthest offset that contained EXIF data in the EXIF blob was (%d) (Scan).", ie.FurthestOffset())
|
|
|
|
return med, nil
|
|
}
|
|
|
|
// Ifd represents a single, parsed IFD.
|
|
type Ifd struct {
|
|
|
|
// TODO(dustin): Add NextIfd().
|
|
|
|
ifdIdentity *exifcommon.IfdIdentity
|
|
|
|
ByteOrder binary.ByteOrder
|
|
|
|
Id int
|
|
|
|
ParentIfd *Ifd
|
|
|
|
// ParentTagIndex is our tag position in the parent IFD, if we had a parent
|
|
// (if `ParentIfd` is not nil and we weren't an IFD referenced as a sibling
|
|
// instead of as a child).
|
|
ParentTagIndex int
|
|
|
|
Offset uint32
|
|
|
|
Entries []*IfdTagEntry
|
|
EntriesByTagId map[uint16][]*IfdTagEntry
|
|
|
|
Children []*Ifd
|
|
|
|
ChildIfdIndex map[string]*Ifd
|
|
|
|
NextIfdOffset uint32
|
|
NextIfd *Ifd
|
|
|
|
thumbnailData []byte
|
|
|
|
ifdMapping *exifcommon.IfdMapping
|
|
tagIndex *TagIndex
|
|
}
|
|
|
|
// IfdIdentity returns IFD identity that this struct represents.
|
|
func (ifd *Ifd) IfdIdentity() *exifcommon.IfdIdentity {
|
|
return ifd.ifdIdentity
|
|
}
|
|
|
|
// ChildWithIfdPath returns an `Ifd` struct for the given child of the current
|
|
// IFD.
|
|
func (ifd *Ifd) ChildWithIfdPath(iiChild *exifcommon.IfdIdentity) (childIfd *Ifd, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): This is a bridge while we're introducing the IFD type-system. We should be able to use the (IfdIdentity).Equals() method for this.
|
|
ifdPath := iiChild.UnindexedString()
|
|
|
|
for _, childIfd := range ifd.Children {
|
|
if childIfd.ifdIdentity.UnindexedString() == ifdPath {
|
|
return childIfd, nil
|
|
}
|
|
}
|
|
|
|
log.Panic(ErrTagNotFound)
|
|
return nil, nil
|
|
}
|
|
|
|
// FindTagWithId returns a list of tags (usually just zero or one) that match
|
|
// the given tag ID. This is efficient.
|
|
func (ifd *Ifd) FindTagWithId(tagId uint16) (results []*IfdTagEntry, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
results, found := ifd.EntriesByTagId[tagId]
|
|
if found != true {
|
|
log.Panic(ErrTagNotFound)
|
|
}
|
|
|
|
return results, nil
|
|
}
|
|
|
|
// FindTagWithName returns a list of tags (usually just zero or one) that match
|
|
// the given tag name. This is not efficient (though the labor is trivial).
|
|
func (ifd *Ifd) FindTagWithName(tagName string) (results []*IfdTagEntry, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
it, err := ifd.tagIndex.GetWithName(ifd.ifdIdentity, tagName)
|
|
if log.Is(err, ErrTagNotFound) == true {
|
|
log.Panic(ErrTagNotKnown)
|
|
} else if err != nil {
|
|
log.Panic(err)
|
|
}
|
|
|
|
results = make([]*IfdTagEntry, 0)
|
|
for _, ite := range ifd.Entries {
|
|
if ite.TagId() == it.Id {
|
|
results = append(results, ite)
|
|
}
|
|
}
|
|
|
|
if len(results) == 0 {
|
|
log.Panic(ErrTagNotFound)
|
|
}
|
|
|
|
return results, nil
|
|
}
|
|
|
|
// String returns a description string.
|
|
func (ifd *Ifd) String() string {
|
|
parentOffset := uint32(0)
|
|
if ifd.ParentIfd != nil {
|
|
parentOffset = ifd.ParentIfd.Offset
|
|
}
|
|
|
|
return fmt.Sprintf("Ifd<ID=(%d) IFD-PATH=[%s] INDEX=(%d) COUNT=(%d) OFF=(0x%04x) CHILDREN=(%d) PARENT=(0x%04x) NEXT-IFD=(0x%04x)>", ifd.Id, ifd.ifdIdentity.UnindexedString(), ifd.ifdIdentity.Index(), len(ifd.Entries), ifd.Offset, len(ifd.Children), parentOffset, ifd.NextIfdOffset)
|
|
}
|
|
|
|
// Thumbnail returns the raw thumbnail bytes. This is typically directly
|
|
// readable by any standard image viewer.
|
|
func (ifd *Ifd) Thumbnail() (data []byte, err error) {
|
|
|
|
if ifd.thumbnailData == nil {
|
|
return nil, ErrNoThumbnail
|
|
}
|
|
|
|
return ifd.thumbnailData, nil
|
|
}
|
|
|
|
// dumpTags recursively builds a list of tags from an IFD.
|
|
func (ifd *Ifd) dumpTags(tags []*IfdTagEntry) []*IfdTagEntry {
|
|
if tags == nil {
|
|
tags = make([]*IfdTagEntry, 0)
|
|
}
|
|
|
|
// Now, print the tags while also descending to child-IFDS as we encounter them.
|
|
|
|
ifdsFoundCount := 0
|
|
|
|
for _, ite := range ifd.Entries {
|
|
tags = append(tags, ite)
|
|
|
|
childIfdPath := ite.ChildIfdPath()
|
|
if childIfdPath != "" {
|
|
ifdsFoundCount++
|
|
|
|
childIfd, found := ifd.ChildIfdIndex[childIfdPath]
|
|
if found != true {
|
|
log.Panicf("alien child IFD referenced by a tag: [%s]", childIfdPath)
|
|
}
|
|
|
|
tags = childIfd.dumpTags(tags)
|
|
}
|
|
}
|
|
|
|
if len(ifd.Children) != ifdsFoundCount {
|
|
log.Panicf("have one or more dangling child IFDs: (%d) != (%d)", len(ifd.Children), ifdsFoundCount)
|
|
}
|
|
|
|
if ifd.NextIfd != nil {
|
|
tags = ifd.NextIfd.dumpTags(tags)
|
|
}
|
|
|
|
return tags
|
|
}
|
|
|
|
// DumpTags prints the IFD hierarchy.
|
|
func (ifd *Ifd) DumpTags() []*IfdTagEntry {
|
|
return ifd.dumpTags(nil)
|
|
}
|
|
|
|
func (ifd *Ifd) printTagTree(populateValues bool, index, level int, nextLink bool) {
|
|
indent := strings.Repeat(" ", level*2)
|
|
|
|
prefix := " "
|
|
if nextLink {
|
|
prefix = ">"
|
|
}
|
|
|
|
fmt.Printf("%s%sIFD: %s\n", indent, prefix, ifd)
|
|
|
|
// Now, print the tags while also descending to child-IFDS as we encounter them.
|
|
|
|
ifdsFoundCount := 0
|
|
|
|
for _, ite := range ifd.Entries {
|
|
if ite.ChildIfdPath() != "" {
|
|
fmt.Printf("%s - TAG: %s\n", indent, ite)
|
|
} else {
|
|
// This will just add noise to the output (byte-tags are fully
|
|
// dumped).
|
|
if ite.IsThumbnailOffset() == true || ite.IsThumbnailSize() == true {
|
|
continue
|
|
}
|
|
|
|
it, err := ifd.tagIndex.Get(ifd.ifdIdentity, ite.TagId())
|
|
|
|
tagName := ""
|
|
if err == nil {
|
|
tagName = it.Name
|
|
}
|
|
|
|
var valuePhrase string
|
|
if populateValues == true {
|
|
var err error
|
|
|
|
valuePhrase, err = ite.Format()
|
|
if err != nil {
|
|
if log.Is(err, exifcommon.ErrUnhandledUndefinedTypedTag) == true {
|
|
ifdEnumerateLogger.Warningf(nil, "Skipping non-standard undefined tag: [%s] (%04x)", ifd.ifdIdentity.UnindexedString(), ite.TagId())
|
|
continue
|
|
} else if err == exifundefined.ErrUnparseableValue {
|
|
ifdEnumerateLogger.Warningf(nil, "Skipping unparseable undefined tag: [%s] (%04x) [%s]", ifd.ifdIdentity.UnindexedString(), ite.TagId(), it.Name)
|
|
continue
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
} else {
|
|
valuePhrase = "!UNRESOLVED"
|
|
}
|
|
|
|
fmt.Printf("%s - TAG: %s NAME=[%s] VALUE=[%v]\n", indent, ite, tagName, valuePhrase)
|
|
}
|
|
|
|
childIfdPath := ite.ChildIfdPath()
|
|
if childIfdPath != "" {
|
|
ifdsFoundCount++
|
|
|
|
childIfd, found := ifd.ChildIfdIndex[childIfdPath]
|
|
if found != true {
|
|
log.Panicf("alien child IFD referenced by a tag: [%s]", childIfdPath)
|
|
}
|
|
|
|
childIfd.printTagTree(populateValues, 0, level+1, false)
|
|
}
|
|
}
|
|
|
|
if len(ifd.Children) != ifdsFoundCount {
|
|
log.Panicf("have one or more dangling child IFDs: (%d) != (%d)", len(ifd.Children), ifdsFoundCount)
|
|
}
|
|
|
|
if ifd.NextIfd != nil {
|
|
ifd.NextIfd.printTagTree(populateValues, index+1, level, true)
|
|
}
|
|
}
|
|
|
|
// PrintTagTree prints the IFD hierarchy.
|
|
func (ifd *Ifd) PrintTagTree(populateValues bool) {
|
|
ifd.printTagTree(populateValues, 0, 0, false)
|
|
}
|
|
|
|
func (ifd *Ifd) printIfdTree(level int, nextLink bool) {
|
|
indent := strings.Repeat(" ", level*2)
|
|
|
|
prefix := " "
|
|
if nextLink {
|
|
prefix = ">"
|
|
}
|
|
|
|
fmt.Printf("%s%s%s\n", indent, prefix, ifd)
|
|
|
|
// Now, print the tags while also descending to child-IFDS as we encounter them.
|
|
|
|
ifdsFoundCount := 0
|
|
|
|
for _, ite := range ifd.Entries {
|
|
childIfdPath := ite.ChildIfdPath()
|
|
if childIfdPath != "" {
|
|
ifdsFoundCount++
|
|
|
|
childIfd, found := ifd.ChildIfdIndex[childIfdPath]
|
|
if found != true {
|
|
log.Panicf("alien child IFD referenced by a tag: [%s]", childIfdPath)
|
|
}
|
|
|
|
childIfd.printIfdTree(level+1, false)
|
|
}
|
|
}
|
|
|
|
if len(ifd.Children) != ifdsFoundCount {
|
|
log.Panicf("have one or more dangling child IFDs: (%d) != (%d)", len(ifd.Children), ifdsFoundCount)
|
|
}
|
|
|
|
if ifd.NextIfd != nil {
|
|
ifd.NextIfd.printIfdTree(level, true)
|
|
}
|
|
}
|
|
|
|
// PrintIfdTree prints the IFD hierarchy.
|
|
func (ifd *Ifd) PrintIfdTree() {
|
|
ifd.printIfdTree(0, false)
|
|
}
|
|
|
|
func (ifd *Ifd) dumpTree(tagsDump []string, level int) []string {
|
|
if tagsDump == nil {
|
|
tagsDump = make([]string, 0)
|
|
}
|
|
|
|
indent := strings.Repeat(" ", level*2)
|
|
|
|
var ifdPhrase string
|
|
if ifd.ParentIfd != nil {
|
|
ifdPhrase = fmt.Sprintf("[%s]->[%s]:(%d)", ifd.ParentIfd.ifdIdentity.UnindexedString(), ifd.ifdIdentity.UnindexedString(), ifd.ifdIdentity.Index())
|
|
} else {
|
|
ifdPhrase = fmt.Sprintf("[ROOT]->[%s]:(%d)", ifd.ifdIdentity.UnindexedString(), ifd.ifdIdentity.Index())
|
|
}
|
|
|
|
startBlurb := fmt.Sprintf("%s> IFD %s TOP", indent, ifdPhrase)
|
|
tagsDump = append(tagsDump, startBlurb)
|
|
|
|
ifdsFoundCount := 0
|
|
for _, ite := range ifd.Entries {
|
|
tagsDump = append(tagsDump, fmt.Sprintf("%s - (0x%04x)", indent, ite.TagId()))
|
|
|
|
childIfdPath := ite.ChildIfdPath()
|
|
if childIfdPath != "" {
|
|
ifdsFoundCount++
|
|
|
|
childIfd, found := ifd.ChildIfdIndex[childIfdPath]
|
|
if found != true {
|
|
log.Panicf("alien child IFD referenced by a tag: [%s]", childIfdPath)
|
|
}
|
|
|
|
tagsDump = childIfd.dumpTree(tagsDump, level+1)
|
|
}
|
|
}
|
|
|
|
if len(ifd.Children) != ifdsFoundCount {
|
|
log.Panicf("have one or more dangling child IFDs: (%d) != (%d)", len(ifd.Children), ifdsFoundCount)
|
|
}
|
|
|
|
finishBlurb := fmt.Sprintf("%s< IFD %s BOTTOM", indent, ifdPhrase)
|
|
tagsDump = append(tagsDump, finishBlurb)
|
|
|
|
if ifd.NextIfd != nil {
|
|
siblingBlurb := fmt.Sprintf("%s* LINKING TO SIBLING IFD [%s]:(%d)", indent, ifd.NextIfd.ifdIdentity.UnindexedString(), ifd.NextIfd.ifdIdentity.Index())
|
|
tagsDump = append(tagsDump, siblingBlurb)
|
|
|
|
tagsDump = ifd.NextIfd.dumpTree(tagsDump, level)
|
|
}
|
|
|
|
return tagsDump
|
|
}
|
|
|
|
// DumpTree returns a list of strings describing the IFD hierarchy.
|
|
func (ifd *Ifd) DumpTree() []string {
|
|
return ifd.dumpTree(nil, 0)
|
|
}
|
|
|
|
// GpsInfo parses and consolidates the GPS info. This can only be called on the
|
|
// GPS IFD.
|
|
func (ifd *Ifd) GpsInfo() (gi *GpsInfo, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
gi = new(GpsInfo)
|
|
|
|
if ifd.ifdIdentity.UnindexedString() != exifcommon.IfdGpsInfoStandardIfdIdentity.UnindexedString() {
|
|
log.Panicf("GPS can only be read on GPS IFD: [%s] != [%s]", ifd.ifdIdentity.UnindexedString(), exifcommon.IfdGpsInfoStandardIfdIdentity.UnindexedString())
|
|
}
|
|
|
|
if tags, found := ifd.EntriesByTagId[TagGpsVersionId]; found == false {
|
|
// We've seen this. We'll just have to default to assuming we're in a
|
|
// 2.2.0.0 format.
|
|
ifdEnumerateLogger.Warningf(nil, "No GPS version tag (0x%04x) found.", TagGpsVersionId)
|
|
} else {
|
|
versionBytes, err := tags[0].GetRawBytes()
|
|
log.PanicIf(err)
|
|
|
|
hit := false
|
|
for _, acceptedGpsVersion := range ValidGpsVersions {
|
|
if bytes.Compare(versionBytes, acceptedGpsVersion[:]) == 0 {
|
|
hit = true
|
|
break
|
|
}
|
|
}
|
|
|
|
if hit != true {
|
|
ifdEnumerateLogger.Warningf(nil, "GPS version not supported: %v", versionBytes)
|
|
log.Panic(ErrNoGpsTags)
|
|
}
|
|
}
|
|
|
|
tags, found := ifd.EntriesByTagId[TagLatitudeId]
|
|
if found == false {
|
|
ifdEnumerateLogger.Warningf(nil, "latitude not found")
|
|
log.Panic(ErrNoGpsTags)
|
|
}
|
|
|
|
latitudeValue, err := tags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
// Look for whether North or South.
|
|
tags, found = ifd.EntriesByTagId[TagLatitudeRefId]
|
|
if found == false {
|
|
ifdEnumerateLogger.Warningf(nil, "latitude-ref not found")
|
|
log.Panic(ErrNoGpsTags)
|
|
}
|
|
|
|
latitudeRefValue, err := tags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
tags, found = ifd.EntriesByTagId[TagLongitudeId]
|
|
if found == false {
|
|
ifdEnumerateLogger.Warningf(nil, "longitude not found")
|
|
log.Panic(ErrNoGpsTags)
|
|
}
|
|
|
|
longitudeValue, err := tags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
// Look for whether West or East.
|
|
tags, found = ifd.EntriesByTagId[TagLongitudeRefId]
|
|
if found == false {
|
|
ifdEnumerateLogger.Warningf(nil, "longitude-ref not found")
|
|
log.Panic(ErrNoGpsTags)
|
|
}
|
|
|
|
longitudeRefValue, err := tags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
// Parse location.
|
|
|
|
latitudeRaw := latitudeValue.([]exifcommon.Rational)
|
|
|
|
gi.Latitude, err = NewGpsDegreesFromRationals(latitudeRefValue.(string), latitudeRaw)
|
|
log.PanicIf(err)
|
|
|
|
longitudeRaw := longitudeValue.([]exifcommon.Rational)
|
|
|
|
gi.Longitude, err = NewGpsDegreesFromRationals(longitudeRefValue.(string), longitudeRaw)
|
|
log.PanicIf(err)
|
|
|
|
// Parse altitude.
|
|
|
|
altitudeTags, foundAltitude := ifd.EntriesByTagId[TagAltitudeId]
|
|
altitudeRefTags, foundAltitudeRef := ifd.EntriesByTagId[TagAltitudeRefId]
|
|
|
|
if foundAltitude == true && foundAltitudeRef == true {
|
|
altitudePhrase, err := altitudeTags[0].Format()
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Altitude is [%s].", altitudePhrase)
|
|
|
|
altitudeValue, err := altitudeTags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
altitudeRefPhrase, err := altitudeRefTags[0].Format()
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Altitude-reference is [%s].", altitudeRefPhrase)
|
|
|
|
altitudeRefValue, err := altitudeRefTags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
altitudeRaw := altitudeValue.([]exifcommon.Rational)
|
|
if altitudeRaw[0].Denominator > 0 {
|
|
altitude := int(altitudeRaw[0].Numerator / altitudeRaw[0].Denominator)
|
|
|
|
if altitudeRefValue.([]byte)[0] == 1 {
|
|
altitude *= -1
|
|
}
|
|
|
|
gi.Altitude = altitude
|
|
}
|
|
}
|
|
|
|
// Parse timestamp from separate date and time tags.
|
|
|
|
timestampTags, foundTimestamp := ifd.EntriesByTagId[TagTimestampId]
|
|
datestampTags, foundDatestamp := ifd.EntriesByTagId[TagDatestampId]
|
|
|
|
if foundTimestamp == true && foundDatestamp == true {
|
|
datestampValue, err := datestampTags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
datePhrase := datestampValue.(string)
|
|
ifdEnumerateLogger.Debugf(nil, "Date tag value is [%s].", datePhrase)
|
|
|
|
// Normalize the separators.
|
|
datePhrase = strings.ReplaceAll(datePhrase, "-", ":")
|
|
|
|
dateParts := strings.Split(datePhrase, ":")
|
|
|
|
year, err1 := strconv.ParseUint(dateParts[0], 10, 16)
|
|
month, err2 := strconv.ParseUint(dateParts[1], 10, 8)
|
|
day, err3 := strconv.ParseUint(dateParts[2], 10, 8)
|
|
|
|
if err1 == nil && err2 == nil && err3 == nil {
|
|
timestampValue, err := timestampTags[0].Value()
|
|
log.PanicIf(err)
|
|
|
|
timePhrase, err := timestampTags[0].Format()
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Time tag value is [%s].", timePhrase)
|
|
|
|
timestampRaw := timestampValue.([]exifcommon.Rational)
|
|
|
|
hour := int(timestampRaw[0].Numerator / timestampRaw[0].Denominator)
|
|
minute := int(timestampRaw[1].Numerator / timestampRaw[1].Denominator)
|
|
second := int(timestampRaw[2].Numerator / timestampRaw[2].Denominator)
|
|
|
|
gi.Timestamp = time.Date(int(year), time.Month(month), int(day), hour, minute, second, 0, time.UTC)
|
|
}
|
|
}
|
|
|
|
return gi, nil
|
|
}
|
|
|
|
// ParsedTagVisitor is a callback used if wanting to visit through all tags and
|
|
// child IFDs from the current IFD and going down.
|
|
type ParsedTagVisitor func(*Ifd, *IfdTagEntry) error
|
|
|
|
// EnumerateTagsRecursively calls the given visitor function for every tag and
|
|
// IFD in the current IFD, recursively.
|
|
func (ifd *Ifd) EnumerateTagsRecursively(visitor ParsedTagVisitor) (err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
for ptr := ifd; ptr != nil; ptr = ptr.NextIfd {
|
|
for _, ite := range ifd.Entries {
|
|
childIfdPath := ite.ChildIfdPath()
|
|
if childIfdPath != "" {
|
|
childIfd := ifd.ChildIfdIndex[childIfdPath]
|
|
|
|
err := childIfd.EnumerateTagsRecursively(visitor)
|
|
log.PanicIf(err)
|
|
} else {
|
|
err := visitor(ifd, ite)
|
|
log.PanicIf(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// QueuedIfd is one IFD that has been identified but yet to be processed.
|
|
type QueuedIfd struct {
|
|
IfdIdentity *exifcommon.IfdIdentity
|
|
|
|
Offset uint32
|
|
Parent *Ifd
|
|
|
|
// ParentTagIndex is our tag position in the parent IFD, if we had a parent
|
|
// (if `ParentIfd` is not nil and we weren't an IFD referenced as a sibling
|
|
// instead of as a child).
|
|
ParentTagIndex int
|
|
}
|
|
|
|
// IfdIndex collects a bunch of IFD and tag information stored in several
|
|
// different ways in order to provide convenient lookups.
|
|
type IfdIndex struct {
|
|
RootIfd *Ifd
|
|
Ifds []*Ifd
|
|
Tree map[int]*Ifd
|
|
Lookup map[string]*Ifd
|
|
}
|
|
|
|
// Collect enumerates the different EXIF blocks (called IFDs) and builds out an
|
|
// index struct for referencing all of the parsed data.
|
|
func (ie *IfdEnumerate) Collect(rootIfdOffset uint32) (index IfdIndex, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): Add MiscellaneousExifData to IfdIndex
|
|
|
|
tree := make(map[int]*Ifd)
|
|
ifds := make([]*Ifd, 0)
|
|
lookup := make(map[string]*Ifd)
|
|
|
|
queue := []QueuedIfd{
|
|
{
|
|
IfdIdentity: exifcommon.IfdStandardIfdIdentity,
|
|
Offset: rootIfdOffset,
|
|
},
|
|
}
|
|
|
|
edges := make(map[uint32]*Ifd)
|
|
|
|
for {
|
|
if len(queue) == 0 {
|
|
break
|
|
}
|
|
|
|
qi := queue[0]
|
|
ii := qi.IfdIdentity
|
|
|
|
offset := qi.Offset
|
|
parentIfd := qi.Parent
|
|
|
|
queue = queue[1:]
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Parsing IFD [%s] (%d) at offset (0x%04x) (Collect).", ii.String(), ii.Index(), offset)
|
|
|
|
bp, err := ie.getByteParser(offset)
|
|
if err != nil {
|
|
if err == ErrOffsetInvalid {
|
|
return index, err
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
|
|
// TODO(dustin): We don't need to pass the index in as a separate argument. Get from the II.
|
|
|
|
nextIfdOffset, entries, thumbnailData, err := ie.parseIfd(ii, bp, nil, false, nil)
|
|
log.PanicIf(err)
|
|
|
|
currentOffset := bp.CurrentOffset()
|
|
if currentOffset > ie.furthestOffset {
|
|
ie.furthestOffset = currentOffset
|
|
}
|
|
|
|
id := len(ifds)
|
|
|
|
entriesByTagId := make(map[uint16][]*IfdTagEntry)
|
|
for _, ite := range entries {
|
|
tagId := ite.TagId()
|
|
|
|
tags, found := entriesByTagId[tagId]
|
|
if found == false {
|
|
tags = make([]*IfdTagEntry, 0)
|
|
}
|
|
|
|
entriesByTagId[tagId] = append(tags, ite)
|
|
}
|
|
|
|
ifd := &Ifd{
|
|
ifdIdentity: ii,
|
|
|
|
ByteOrder: ie.byteOrder,
|
|
|
|
Id: id,
|
|
|
|
ParentIfd: parentIfd,
|
|
ParentTagIndex: qi.ParentTagIndex,
|
|
|
|
Offset: offset,
|
|
Entries: entries,
|
|
EntriesByTagId: entriesByTagId,
|
|
|
|
// This is populated as each child is processed.
|
|
Children: make([]*Ifd, 0),
|
|
|
|
NextIfdOffset: nextIfdOffset,
|
|
thumbnailData: thumbnailData,
|
|
|
|
ifdMapping: ie.ifdMapping,
|
|
tagIndex: ie.tagIndex,
|
|
}
|
|
|
|
// Add ourselves to a big list of IFDs.
|
|
ifds = append(ifds, ifd)
|
|
|
|
// Install ourselves into a by-id lookup table (keys are unique).
|
|
tree[id] = ifd
|
|
|
|
// Install into by-name buckets.
|
|
lookup[ii.String()] = ifd
|
|
|
|
// Add a link from the previous IFD in the chain to us.
|
|
if previousIfd, found := edges[offset]; found == true {
|
|
previousIfd.NextIfd = ifd
|
|
}
|
|
|
|
// Attach as a child to our parent (where we appeared as a tag in
|
|
// that IFD).
|
|
if parentIfd != nil {
|
|
parentIfd.Children = append(parentIfd.Children, ifd)
|
|
}
|
|
|
|
// Determine if any of our entries is a child IFD and queue it.
|
|
for i, ite := range entries {
|
|
if ite.ChildIfdPath() == "" {
|
|
continue
|
|
}
|
|
|
|
tagId := ite.TagId()
|
|
childIfdName := ite.ChildIfdName()
|
|
|
|
currentIfdTag := ii.IfdTag()
|
|
|
|
childIfdTag :=
|
|
exifcommon.NewIfdTag(
|
|
¤tIfdTag,
|
|
tagId,
|
|
childIfdName)
|
|
|
|
iiChild := ii.NewChild(childIfdTag, 0)
|
|
|
|
qi := QueuedIfd{
|
|
IfdIdentity: iiChild,
|
|
|
|
Offset: ite.getValueOffset(),
|
|
Parent: ifd,
|
|
ParentTagIndex: i,
|
|
}
|
|
|
|
queue = append(queue, qi)
|
|
}
|
|
|
|
// If there's another IFD in the chain.
|
|
if nextIfdOffset != 0 {
|
|
iiSibling := ii.NewSibling(ii.Index() + 1)
|
|
|
|
// Allow the next link to know what the previous link was.
|
|
edges[nextIfdOffset] = ifd
|
|
|
|
qi := QueuedIfd{
|
|
IfdIdentity: iiSibling,
|
|
Offset: nextIfdOffset,
|
|
}
|
|
|
|
queue = append(queue, qi)
|
|
}
|
|
}
|
|
|
|
index.RootIfd = tree[0]
|
|
index.Ifds = ifds
|
|
index.Tree = tree
|
|
index.Lookup = lookup
|
|
|
|
err = ie.setChildrenIndex(index.RootIfd)
|
|
log.PanicIf(err)
|
|
|
|
ifdEnumerateLogger.Debugf(nil, "Collect: It looks like the furthest offset that contained EXIF data in the EXIF blob was (%d).", ie.FurthestOffset())
|
|
|
|
return index, nil
|
|
}
|
|
|
|
func (ie *IfdEnumerate) setChildrenIndex(ifd *Ifd) (err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
childIfdIndex := make(map[string]*Ifd)
|
|
for _, childIfd := range ifd.Children {
|
|
childIfdIndex[childIfd.ifdIdentity.UnindexedString()] = childIfd
|
|
}
|
|
|
|
ifd.ChildIfdIndex = childIfdIndex
|
|
|
|
for _, childIfd := range ifd.Children {
|
|
err := ie.setChildrenIndex(childIfd)
|
|
log.PanicIf(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// FurthestOffset returns the furthest offset visited in the EXIF blob. This
|
|
// *does not* account for the locations of any undefined tags since we always
|
|
// evaluate the furthest offset, whether or not the user wants to know it.
|
|
//
|
|
// We are not willing to incur the cost of actually parsing those tags just to
|
|
// know their length when there are still undefined tags that are out there
|
|
// that we still won't have any idea how to parse, thus making this an
|
|
// approximation regardless of how clever we get.
|
|
func (ie *IfdEnumerate) FurthestOffset() uint32 {
|
|
|
|
// TODO(dustin): Add test
|
|
|
|
return ie.furthestOffset
|
|
}
|
|
|
|
// ParseOneIfd is a hack to use an IE to parse a raw IFD block. Can be used for
|
|
// testing. The fqIfdPath ("fully-qualified IFD path") will be less qualified
|
|
// in that the numeric index will always be zero (the zeroth child) rather than
|
|
// the proper number (if its actually a sibling to the first child, for
|
|
// instance).
|
|
func ParseOneIfd(ifdMapping *exifcommon.IfdMapping, tagIndex *TagIndex, ii *exifcommon.IfdIdentity, byteOrder binary.ByteOrder, ifdBlock []byte, visitor TagVisitorFn) (nextIfdOffset uint32, entries []*IfdTagEntry, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
ie := NewIfdEnumerate(ifdMapping, tagIndex, make([]byte, 0), byteOrder)
|
|
|
|
bp, err := newByteParser(ifdBlock, byteOrder, 0)
|
|
if err != nil {
|
|
if err == ErrOffsetInvalid {
|
|
return 0, nil, err
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
|
|
nextIfdOffset, entries, _, err = ie.parseIfd(ii, bp, visitor, true, nil)
|
|
log.PanicIf(err)
|
|
|
|
return nextIfdOffset, entries, nil
|
|
}
|
|
|
|
// ParseOneTag is a hack to use an IE to parse a raw tag block.
|
|
func ParseOneTag(ifdMapping *exifcommon.IfdMapping, tagIndex *TagIndex, ii *exifcommon.IfdIdentity, byteOrder binary.ByteOrder, tagBlock []byte) (ite *IfdTagEntry, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
ie := NewIfdEnumerate(ifdMapping, tagIndex, make([]byte, 0), byteOrder)
|
|
|
|
bp, err := newByteParser(tagBlock, byteOrder, 0)
|
|
if err != nil {
|
|
if err == ErrOffsetInvalid {
|
|
return nil, err
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
|
|
ite, err = ie.parseTag(ii, 0, bp)
|
|
log.PanicIf(err)
|
|
|
|
err = ie.postparseTag(ite, nil)
|
|
if err != nil {
|
|
if err == ErrTagNotFound {
|
|
return nil, err
|
|
}
|
|
|
|
log.Panic(err)
|
|
}
|
|
|
|
return ite, nil
|
|
}
|
|
|
|
// FindIfdFromRootIfd returns the given `Ifd` given the root-IFD and path of the
|
|
// desired IFD.
|
|
func FindIfdFromRootIfd(rootIfd *Ifd, ifdPath string) (ifd *Ifd, err error) {
|
|
defer func() {
|
|
if state := recover(); state != nil {
|
|
err = log.Wrap(state.(error))
|
|
}
|
|
}()
|
|
|
|
// TODO(dustin): !! Add test.
|
|
|
|
lineage, err := rootIfd.ifdMapping.ResolvePath(ifdPath)
|
|
log.PanicIf(err)
|
|
|
|
// Confirm the first IFD is our root IFD type, and then prune it because
|
|
// from then on we'll be searching down through our children.
|
|
|
|
if len(lineage) == 0 {
|
|
log.Panicf("IFD path must be non-empty.")
|
|
} else if lineage[0].Name != exifcommon.IfdStandardIfdIdentity.Name() {
|
|
log.Panicf("First IFD path item must be [%s].", exifcommon.IfdStandardIfdIdentity.Name())
|
|
}
|
|
|
|
desiredRootIndex := lineage[0].Index
|
|
lineage = lineage[1:]
|
|
|
|
// TODO(dustin): !! This is a poorly conceived fix that just doubles the work we already have to do below, which then interacts badly with the indices not being properly represented in the IFD-phrase.
|
|
// TODO(dustin): !! <-- However, we're not sure whether we shouldn't store a secondary IFD-path with the indices. Some IFDs may not necessarily restrict which IFD indices they can be a child of (only the IFD itself matters). Validation should be delegated to the caller.
|
|
thisIfd := rootIfd
|
|
for currentRootIndex := 0; currentRootIndex < desiredRootIndex; currentRootIndex++ {
|
|
if thisIfd.NextIfd == nil {
|
|
log.Panicf("Root-IFD index (%d) does not exist in the data.", currentRootIndex)
|
|
}
|
|
|
|
thisIfd = thisIfd.NextIfd
|
|
}
|
|
|
|
for _, itii := range lineage {
|
|
var hit *Ifd
|
|
for _, childIfd := range thisIfd.Children {
|
|
if childIfd.ifdIdentity.TagId() == itii.TagId {
|
|
hit = childIfd
|
|
break
|
|
}
|
|
}
|
|
|
|
// If we didn't find the child, add it.
|
|
if hit == nil {
|
|
log.Panicf("IFD [%s] in [%s] not found: %s", itii.Name, ifdPath, thisIfd.Children)
|
|
}
|
|
|
|
thisIfd = hit
|
|
|
|
// If we didn't find the sibling, add it.
|
|
for i := 0; i < itii.Index; i++ {
|
|
if thisIfd.NextIfd == nil {
|
|
log.Panicf("IFD [%s] does not have (%d) occurrences/siblings", thisIfd.ifdIdentity.UnindexedString(), itii.Index)
|
|
}
|
|
|
|
thisIfd = thisIfd.NextIfd
|
|
}
|
|
}
|
|
|
|
return thisIfd, nil
|
|
}
|