1
0
forked from um/cli
cli/algo/ncm/ncm.go

243 lines
5.6 KiB
Go
Raw Permalink Normal View History

2020-12-25 14:41:04 +00:00
package ncm
import (
"bytes"
2022-11-18 23:25:40 +00:00
"context"
2020-12-25 14:41:04 +00:00
"encoding/base64"
"encoding/binary"
"encoding/json"
"errors"
2022-11-18 23:25:40 +00:00
"fmt"
"io"
2020-12-25 14:41:04 +00:00
"net/http"
"strings"
2022-11-18 23:25:40 +00:00
2022-11-18 23:44:44 +00:00
"unlock-music.dev/cli/algo/common"
"unlock-music.dev/cli/internal/utils"
2020-12-25 14:41:04 +00:00
)
const magicHeader = "CTENFDAM"
2020-12-25 14:41:04 +00:00
var (
keyCore = []byte{
0x68, 0x7a, 0x48, 0x52, 0x41, 0x6d, 0x73, 0x6f,
0x35, 0x6b, 0x49, 0x6e, 0x62, 0x61, 0x78, 0x57,
}
2020-12-25 14:41:04 +00:00
keyMeta = []byte{
0x23, 0x31, 0x34, 0x6C, 0x6A, 0x6B, 0x5F, 0x21,
0x5C, 0x5D, 0x26, 0x30, 0x55, 0x3C, 0x27, 0x28,
}
2020-12-25 14:41:04 +00:00
)
func NewDecoder(p *common.DecoderParams) common.Decoder {
return &Decoder{rd: p.Reader}
2020-12-25 14:41:04 +00:00
}
type Decoder struct {
2022-11-19 18:47:28 +00:00
rd io.ReadSeeker // rd is the original file reader
2022-11-19 18:47:28 +00:00
offset int
cipher common.StreamDecoder
2020-12-25 14:41:04 +00:00
metaRaw []byte
metaType string
2020-12-25 18:53:28 +00:00
meta RawMeta
2022-11-19 18:47:28 +00:00
cover []byte
2020-12-25 14:41:04 +00:00
}
2022-11-19 18:47:28 +00:00
// Validate checks if the file is a valid Netease .ncm file.
// rd will be seeked to the beginning of the encrypted audio.
2020-12-25 18:53:28 +00:00
func (d *Decoder) Validate() error {
if err := d.validateMagicHeader(); err != nil {
return err
}
if _, err := d.rd.Seek(2, io.SeekCurrent); err != nil { // 2 bytes gap
return fmt.Errorf("ncm seek file: %w", err)
}
keyData, err := d.readKeyData()
if err != nil {
return err
}
if err := d.readMetaData(); err != nil {
return fmt.Errorf("read meta date failed: %w", err)
}
if _, err := d.rd.Seek(5, io.SeekCurrent); err != nil { // 5 bytes gap
return fmt.Errorf("ncm seek gap: %w", err)
}
if err := d.readCoverData(); err != nil {
return fmt.Errorf("parse ncm cover file failed: %w", err)
}
if err := d.parseMeta(); err != nil {
return fmt.Errorf("parse meta failed: %w", err)
}
d.cipher = newNcmCipher(keyData)
return nil
}
func (d *Decoder) validateMagicHeader() error {
header := make([]byte, len(magicHeader)) // 0x00 - 0x07
if _, err := d.rd.Read(header); err != nil {
return fmt.Errorf("ncm read magic header: %w", err)
}
if !bytes.Equal([]byte(magicHeader), header) {
2020-12-25 18:53:28 +00:00
return errors.New("ncm magic header not match")
2020-12-25 14:41:04 +00:00
}
2020-12-25 18:53:28 +00:00
return nil
2020-12-25 14:41:04 +00:00
}
func (d *Decoder) readKeyData() ([]byte, error) {
bKeyLen := make([]byte, 4) //
if _, err := io.ReadFull(d.rd, bKeyLen); err != nil {
return nil, fmt.Errorf("ncm read key length: %w", err)
2020-12-25 14:41:04 +00:00
}
iKeyLen := binary.LittleEndian.Uint32(bKeyLen)
bKeyRaw := make([]byte, iKeyLen)
if _, err := io.ReadFull(d.rd, bKeyRaw); err != nil {
return nil, fmt.Errorf("ncm read key data: %w", err)
}
2020-12-25 14:41:04 +00:00
for i := uint32(0); i < iKeyLen; i++ {
bKeyRaw[i] ^= 0x64
2020-12-25 14:41:04 +00:00
}
return utils.PKCS7UnPadding(utils.DecryptAES128ECB(bKeyRaw, keyCore))[17:], nil
2020-12-25 14:41:04 +00:00
}
2020-12-25 16:49:03 +00:00
func (d *Decoder) readMetaData() error {
bMetaLen := make([]byte, 4) //
if _, err := io.ReadFull(d.rd, bMetaLen); err != nil {
return fmt.Errorf("ncm read key length: %w", err)
2020-12-25 14:41:04 +00:00
}
iMetaLen := binary.LittleEndian.Uint32(bMetaLen)
2020-12-25 14:41:04 +00:00
if iMetaLen == 0 {
return nil // no meta data
2020-12-25 14:41:04 +00:00
}
bMetaRaw := make([]byte, iMetaLen)
if _, err := io.ReadFull(d.rd, bMetaRaw); err != nil {
return fmt.Errorf("ncm read meta data: %w", err)
}
bMetaRaw = bMetaRaw[22:] // skip "163 key(Don't modify):"
for i := 0; i < len(bMetaRaw); i++ {
bMetaRaw[i] ^= 0x63
2020-12-25 14:41:04 +00:00
}
cipherText, err := base64.StdEncoding.DecodeString(string(bMetaRaw))
2020-12-25 14:41:04 +00:00
if err != nil {
return errors.New("decode ncm meta failed: " + err.Error())
}
metaRaw := utils.PKCS7UnPadding(utils.DecryptAES128ECB(cipherText, keyMeta))
sep := bytes.IndexByte(metaRaw, ':')
if sep == -1 {
2020-12-25 16:49:03 +00:00
return errors.New("invalid ncm meta file")
2020-12-25 14:41:04 +00:00
}
d.metaType = string(metaRaw[:sep])
d.metaRaw = metaRaw[sep+1:]
2020-12-25 14:41:04 +00:00
return nil
}
func (d *Decoder) readCoverData() error {
bCoverCRC := make([]byte, 4)
if _, err := io.ReadFull(d.rd, bCoverCRC); err != nil {
return fmt.Errorf("ncm read cover crc: %w", err)
2020-12-25 14:41:04 +00:00
}
bCoverLen := make([]byte, 4) //
if _, err := io.ReadFull(d.rd, bCoverLen); err != nil {
return fmt.Errorf("ncm read cover length: %w", err)
2020-12-25 14:41:04 +00:00
}
iCoverLen := binary.LittleEndian.Uint32(bCoverLen)
2020-12-25 14:41:04 +00:00
coverBuf := make([]byte, iCoverLen)
if _, err := io.ReadFull(d.rd, coverBuf); err != nil {
return fmt.Errorf("ncm read cover data: %w", err)
2020-12-25 14:41:04 +00:00
}
d.cover = coverBuf
return nil
2020-12-25 14:41:04 +00:00
}
2020-12-25 16:49:03 +00:00
func (d *Decoder) parseMeta() error {
switch d.metaType {
2020-12-25 14:41:04 +00:00
case "music":
2020-12-25 18:53:28 +00:00
d.meta = new(RawMetaMusic)
return json.Unmarshal(d.metaRaw, d.meta)
2020-12-25 14:41:04 +00:00
case "dj":
2020-12-25 18:53:28 +00:00
d.meta = new(RawMetaDJ)
return json.Unmarshal(d.metaRaw, d.meta)
2020-12-25 14:41:04 +00:00
default:
2020-12-25 16:49:03 +00:00
return errors.New("unknown ncm meta type: " + d.metaType)
2020-12-25 14:41:04 +00:00
}
}
func (d *Decoder) Read(buf []byte) (int, error) {
n, err := d.rd.Read(buf)
if n > 0 {
d.cipher.Decrypt(buf[:n], d.offset)
d.offset += n
2020-12-25 14:41:04 +00:00
}
return n, err
2020-12-25 14:41:04 +00:00
}
2022-11-18 23:25:40 +00:00
func (d *Decoder) GetAudioExt() string {
2020-12-25 18:53:28 +00:00
if d.meta != nil {
if format := d.meta.GetFormat(); format != "" {
return "." + d.meta.GetFormat()
}
2020-12-25 14:41:04 +00:00
}
return ""
}
2022-11-18 23:25:40 +00:00
func (d *Decoder) GetCoverImage(ctx context.Context) ([]byte, error) {
2020-12-25 18:53:28 +00:00
if d.cover != nil {
2022-11-18 23:25:40 +00:00
return d.cover, nil
2020-12-25 14:41:04 +00:00
}
if d.meta == nil {
return nil, errors.New("ncm meta not found")
}
2022-11-18 23:25:40 +00:00
imgURL := d.meta.GetAlbumImageURL()
if !strings.HasPrefix(imgURL, "http") {
2022-11-18 23:25:40 +00:00
return nil, nil // no cover image
}
// fetch cover image
req, err := http.NewRequestWithContext(ctx, http.MethodGet, imgURL, nil)
resp, err := http.DefaultClient.Do(req)
if err != nil {
return nil, fmt.Errorf("ncm download image failed: %w", err)
2022-11-18 23:25:40 +00:00
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return nil, fmt.Errorf("ncm download image failed: unexpected http status %s", resp.Status)
2022-11-18 23:25:40 +00:00
}
d.cover, err = io.ReadAll(resp.Body)
2022-11-18 23:25:40 +00:00
if err != nil {
return nil, fmt.Errorf("ncm download image failed: %w", err)
2020-12-25 14:41:04 +00:00
}
return d.cover, nil
2020-12-25 14:41:04 +00:00
}
2022-11-18 23:25:40 +00:00
func (d *Decoder) GetMeta() common.Meta {
2020-12-25 18:53:28 +00:00
return d.meta
2020-12-25 14:41:04 +00:00
}
2020-12-26 07:47:10 +00:00
func init() {
// Netease Mp3/Flac
2021-11-11 15:43:20 +00:00
common.RegisterDecoder("ncm", false, NewDecoder)
2020-12-26 07:47:10 +00:00
}