1
0
forked from um/cli

feat(qmc): support .mflach on darwin

This commit is contained in:
Unlock Music Dev 2022-12-05 00:06:38 +08:00
parent d2019b04ec
commit 52e986e644
Signed by untrusted user: um-dev
GPG Key ID: 95202E10D3413A1D
2 changed files with 101 additions and 3 deletions

92
algo/qmc/key_mmkv.go Normal file
View File

@ -0,0 +1,92 @@
package qmc
import (
"errors"
"fmt"
"os"
"path/filepath"
"runtime"
"strings"
"unlock-music.dev/mmkv"
)
var streamKeyVault mmkv.MMKV
func readKeyFromMMKV(file string) ([]byte, error) {
if file == "" {
return nil, errors.New("file path is required while reading key from mmkv")
}
//goland:noinspection GoBoolExpressions
if runtime.GOOS != "darwin" {
return nil, errors.New("mmkv vault not supported on this platform")
}
if streamKeyVault == nil {
mmkvDir, err := getRelativeMMKVDir(file)
if err != nil {
mmkvDir, err = getDefaultMMKVDir()
if err != nil {
return nil, fmt.Errorf("mmkv key valut not found: %w", err)
}
}
mmkv.InitializeMMKV(mmkvDir)
streamKeyVault = mmkv.MMKVWithID("MMKVStreamEncryptId")
}
buf := streamKeyVault.GetBytes(file)
if len(buf) == 0 {
_, partName := filepath.Split(file)
keys := streamKeyVault.AllKeys()
for _, key := range keys {
if strings.HasSuffix(key, partName) {
buf = streamKeyVault.GetBytes(key)
break
}
}
}
if len(buf) == 0 {
return nil, errors.New("key not found in mmkv vault")
}
return deriveKey(buf)
}
func getRelativeMMKVDir(file string) (string, error) {
mmkvDir := filepath.Join(filepath.Dir(file), "../mmkv")
if _, err := os.Stat(mmkvDir); err != nil {
return "", fmt.Errorf("stat default mmkv dir: %w", err)
}
keyFile := filepath.Join(mmkvDir, "MMKVStreamEncryptId")
if _, err := os.Stat(keyFile); err != nil {
return "", fmt.Errorf("stat default mmkv file: %w", err)
}
return mmkvDir, nil
}
func getDefaultMMKVDir() (string, error) {
homeDir, err := os.UserHomeDir()
if err != nil {
return "", fmt.Errorf("get user home dir: %w", err)
}
mmkvDir := filepath.Join(
homeDir,
"Library/Containers/com.tencent.QQMusicMac/Data", // todo: make configurable
"Library/Application Support/QQMusicMac/mmkv",
)
if _, err := os.Stat(mmkvDir); err != nil {
return "", fmt.Errorf("stat default mmkv dir: %w", err)
}
keyFile := filepath.Join(mmkvDir, "MMKVStreamEncryptId")
if _, err := os.Stat(keyFile); err != nil {
return "", fmt.Errorf("stat default mmkv file: %w", err)
}
return mmkvDir, nil
}

View File

@ -15,6 +15,7 @@ import (
type Decoder struct { type Decoder struct {
raw io.ReadSeeker // raw is the original file reader raw io.ReadSeeker // raw is the original file reader
params *common.DecoderParams
audio io.Reader // audio is the encrypted audio data audio io.Reader // audio is the encrypted audio data
audioLen int // audioLen is the audio data length audioLen int // audioLen is the audio data length
@ -39,7 +40,7 @@ func (d *Decoder) Read(p []byte) (int, error) {
} }
func NewDecoder(p *common.DecoderParams) common.Decoder { func NewDecoder(p *common.DecoderParams) common.Decoder {
return &Decoder{raw: p.Reader} return &Decoder{raw: p.Reader, params: p}
} }
func (d *Decoder) Validate() error { func (d *Decoder) Validate() error {
@ -97,7 +98,12 @@ func (d *Decoder) validateDecode() error {
return nil return nil
} }
func (d *Decoder) searchKey() error { func (d *Decoder) searchKey() (err error) {
if d.params.Extension == ".mflach" {
d.decodedKey, err = readKeyFromMMKV(d.params.FilePath)
return err
}
fileSizeM4, err := d.raw.Seek(-4, io.SeekEnd) fileSizeM4, err := d.raw.Seek(-4, io.SeekEnd)
if err != nil { if err != nil {
return err return err