Update NCM, QMC Decoder

This commit is contained in:
Emmm Monster 2020-12-26 02:53:28 +08:00
parent fe5403cdb7
commit a92ea82090
No known key found for this signature in database
GPG Key ID: C98279C83FB50DB9
3 changed files with 29 additions and 30 deletions

View File

@ -1,7 +1,7 @@
package common
type Decoder interface {
Validate() bool
Validate() error
Decode() error
GetCoverImage() []byte
GetAudioData() []byte

View File

@ -42,10 +42,10 @@ type Decoder struct {
metaRaw []byte
metaType string
Meta RawMeta
meta RawMeta
Cover []byte
Audio []byte
cover []byte
audio []byte
offsetKey uint32
offsetMeta uint32
@ -53,16 +53,16 @@ type Decoder struct {
offsetAudio uint32
}
func (d *Decoder) Validate() bool {
func (d *Decoder) Validate() error {
if !bytes.Equal(magicHeader, d.file[:len(magicHeader)]) {
return false
return errors.New("ncm magic header not match")
}
/*if status.IsDebug {
logging.Log().Info("the unknown field of the header is: \n" + spew.Sdump(d.file[8:10]))
}*/
d.offsetKey = 8 + 2
return true
return nil
}
//todo: 读取前进行检查长度,防止越界
@ -141,11 +141,11 @@ func (d *Decoder) buildKeyBox() {
func (d *Decoder) parseMeta() error {
switch d.metaType {
case "music":
d.Meta = new(RawMetaMusic)
return json.Unmarshal(d.metaRaw, d.Meta)
d.meta = new(RawMetaMusic)
return json.Unmarshal(d.metaRaw, d.meta)
case "dj":
d.Meta = new(RawMetaDJ)
return json.Unmarshal(d.metaRaw, d.Meta)
d.meta = new(RawMetaDJ)
return json.Unmarshal(d.metaRaw, d.meta)
default:
return errors.New("unknown ncm meta type: " + d.metaType)
}
@ -173,7 +173,7 @@ func (d *Decoder) readCoverData() error {
if iCoverLen == 0 {
return errors.New("no any cover file found")
}
d.Cover = d.file[coverLenStart+4 : 4+coverLenStart+iCoverLen]
d.cover = d.file[coverLenStart+4 : 4+coverLenStart+iCoverLen]
return nil
}
@ -183,9 +183,9 @@ func (d *Decoder) readAudioData() error {
}
audioRaw := d.file[d.offsetAudio:]
audioLen := len(audioRaw)
d.Audio = make([]byte, audioLen)
d.audio = make([]byte, audioLen)
for i := uint32(0); i < uint32(audioLen); i++ {
d.Audio[i] = d.box[i&0xff] ^ audioRaw[i]
d.audio[i] = d.box[i&0xff] ^ audioRaw[i]
}
return nil
}
@ -213,23 +213,23 @@ func (d *Decoder) Decode() error {
}
func (d Decoder) GetAudioExt() string {
if d.Meta != nil {
return d.Meta.GetFormat()
if d.meta != nil {
return d.meta.GetFormat()
}
return ""
}
func (d Decoder) GetAudioData() []byte {
return d.Audio
return d.audio
}
func (d Decoder) GetCoverImage() []byte {
if d.Cover != nil {
return d.Cover
if d.cover != nil {
return d.cover
}
{
imgURL := d.Meta.GetAlbumImageURL()
if d.Meta != nil && !strings.HasPrefix(imgURL, "http") {
imgURL := d.meta.GetAlbumImageURL()
if d.meta != nil && !strings.HasPrefix(imgURL, "http") {
return nil
}
resp, err := http.Get(imgURL)
@ -253,5 +253,5 @@ func (d Decoder) GetCoverImage() []byte {
}
func (d Decoder) GetMeta() common.Meta {
return d.Meta
return d.meta
}

View File

@ -5,8 +5,6 @@ import (
"encoding/binary"
"errors"
"github.com/umlock-music/cli/algo/common"
"github.com/umlock-music/cli/internal/logging"
"go.uber.org/zap"
)
var (
@ -36,18 +34,19 @@ func NewMgg256Decoder(data []byte) *Decoder {
return &Decoder{file: data, maskDetector: detectMgg256Mask, audioExt: "ogg"}
}
func (d *Decoder) Validate() bool {
func (d *Decoder) Validate() error {
if nil != d.mask {
return true
return nil
}
if nil != d.maskDetector {
if err := d.validateKey(); err != nil {
logging.Log().Error("detect file failed", zap.Error(err))
return false
return err
}
d.mask, _ = d.maskDetector(d.file)
var err error
d.mask, err = d.maskDetector(d.file)
return err
}
return d.mask != nil
return errors.New("no mask or mask detector found")
}
func (d *Decoder) validateKey() error {