519 lines
31 KiB
Rust
519 lines
31 KiB
Rust
pub mod deserializer {
|
|
use log::info;
|
|
use crate::artworkdb::aobjects::{ADataSet, ADatabase, AImageItem, AImageName, SecondTypeTag, ThirdTypeTag};
|
|
use crate::artworkdb::objects::{ChunkHeader, ChunkType, DataSet, Database, ImageFile, ImageItem, ImageName, LocationTag};
|
|
|
|
|
|
const SAMPLE_DB: [u8; 2144] = [0x6D,0x68,0x66,0x64,0x84,0x00,0x00,0x00,0x60,0x08,0x00,0x00,0x00,0x00,0x00,0x00,0x06,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x66,0x00,0x00,0x00,0xE9,0x15,0xB6,0x2C,0x03,0x42,0x6A,0xDE,0x9F,0xBE,0x2C,0x6B,0x2F,0xEE,0x85,0x72,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xC5,0xA6,0xEC,0x35,0xAC,0xEB,0x95,0xC5,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x73,0x64,0x60,0x00,0x00,0x00,0x6C,0x05,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6C,0x69,0x5C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x69,0x98,0x00,0x00,0x00,0x58,0x02,0x00,0x00,0x03,0x00,0x00,0x00,0x64,0x00,0x00,0x00,0xDC,0x41,0xBD,0xF9,0xDB,0x98,0x58,0xFC,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x5D,0x4F,0x11,0x00,0x00,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x04,0x04,0x00,0x00,0x00,0x00,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x64,0x00,0x64,0x00,0x00,0x00,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x38,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x05,0x04,0x00,0x00,0x00,0x00,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0xC8,0x00,0xC8,0x00,0x00,0x00,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x39,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x78,0x00,0x00,0x00,0x06,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x61,0x66,0x60,0x00,0x00,0x00,0x3C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x69,0x98,0x00,0x00,0x00,0x58,0x02,0x00,0x00,0x03,0x00,0x00,0x00,0x65,0x00,0x00,0x00,0x1F,0xBE,0xB3,0xD9,0x94,0x3F,0xE5,0x5D,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xBC,0x89,0x03,0x00,0x00,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0xF8,0x7F,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x04,0x04,0x00,0x00,0x20,0x4E,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x64,0x00,0x64,0x00,0x00,0x00,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x38,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0xA4,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6E,0x69,0x4C,0x00,0x00,0x00,0x8C,0x00,0x00,0x00,0x01,0x00,0x00,0x00,0x05,0x04,0x00,0x00,0x80,0x38,0x01,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0xC8,0x00,0xC8,0x00,0x00,0x00,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x40,0x00,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x1C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x3A,0x00,0x46,0x00,0x31,0x00,0x30,0x00,0x32,0x00,0x39,0x00,0x5F,0x00,0x31,0x00,0x2E,0x00,0x69,0x00,0x74,0x00,0x68,0x00,0x6D,0x00,0x62,0x00,0x6D,0x68,0x6F,0x64,0x18,0x00,0x00,0x00,0x78,0x00,0x00,0x00,0x06,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x61,0x66,0x60,0x00,0x00,0x00,0x3C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x73,0x64,0x60,0x00,0x00,0x00,0xBC,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6C,0x61,0x5C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x73,0x64,0x60,0x00,0x00,0x00,0xB4,0x01,0x00,0x00,0x03,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x6C,0x66,0x5C,0x00,0x00,0x00,0x02,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x66,0x7C,0x00,0x00,0x00,0x7C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x04,0x04,0x00,0x00,0x20,0x4E,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x6D,0x68,0x69,0x66,0x7C,0x00,0x00,0x00,0x7C,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x05,0x04,0x00,0x00,0x80,0x38,0x01,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00,0x00];
|
|
|
|
/*
|
|
File system:
|
|
/iPod_Control/Artwork/ArtworkDB
|
|
/iPod_Control/Artwork/[any].ithmb
|
|
*/
|
|
|
|
|
|
enum ChunkState {
|
|
Header,
|
|
Data
|
|
}
|
|
|
|
pub fn new_db() -> ADatabase {
|
|
let mut adb = parse_bytes(&SAMPLE_DB[..]);
|
|
let data_set = adb.find_dataset(3);
|
|
data_set.child = Vec::new();
|
|
let data_set = adb.find_dataset(1);
|
|
data_set.child = Vec::new();
|
|
adb
|
|
}
|
|
|
|
pub fn parse_bytes(data: &[u8]) -> ADatabase {
|
|
let mut adb = ADatabase {data: None, header: None, children: Vec::new() };
|
|
let mut state = ChunkState::Header;
|
|
let mut chunk_header: Option<ChunkHeader> = None;
|
|
let mut last_type: u32 = 0;
|
|
let mut i = 0;
|
|
while i < data.len() {
|
|
state = match state {
|
|
ChunkState::Header => {
|
|
if i + 12 >= data.len() { break; }
|
|
chunk_header = Some(bincode::deserialize(&data[i..i + 12]).unwrap());
|
|
i += 12;
|
|
ChunkState::Data
|
|
},
|
|
ChunkState::Data => {
|
|
let mut u = 0;
|
|
let header = chunk_header.unwrap();
|
|
match ChunkType::from(header.chunk_type) {
|
|
ChunkType::ArtworkDB => {
|
|
u = usize::try_from(header.end_of_chunk).unwrap() - 12;
|
|
let db: Database = bincode::deserialize(&data[i..i+u]).unwrap();
|
|
info!("val: {:?}", db);
|
|
adb.data = Some(db);
|
|
adb.header = Some(header);
|
|
},
|
|
ChunkType::DataSet => {
|
|
u = usize::try_from(header.end_of_chunk).unwrap() - 12;
|
|
let ds: DataSet = bincode::deserialize(&data[i..i + u]).unwrap();
|
|
last_type = ds.data_type;
|
|
adb.children.push(ADataSet { header, data: ds, child: Vec::new()});
|
|
},
|
|
ChunkType::ImageList | ChunkType::AlbumList | ChunkType::FileList => {
|
|
u = usize::try_from(header.end_of_chunk).unwrap() - 12;
|
|
},
|
|
ChunkType::ImageItem => {
|
|
info!("ImageItem tag");
|
|
u = usize::try_from(header.end_of_chunk).unwrap() - 12;
|
|
let ai: ImageItem = bincode::deserialize(&data[i..i+u]).unwrap();
|
|
let images = &mut adb.find_dataset(last_type).child;
|
|
images.push(AImageItem { data: Some(ai), tag: Vec::new(), file: None });
|
|
},
|
|
ChunkType::FileImage => {
|
|
u = usize::try_from(header.end_of_chunk).unwrap() - 12;
|
|
let ai: ImageFile = bincode::deserialize(&data[i..i+u]).unwrap();
|
|
let images = &mut adb.find_dataset(last_type).child;
|
|
images.push(AImageItem { data: None, tag: Vec::new(), file: Some(ai) });
|
|
},
|
|
ChunkType::LocationTag => {
|
|
u = usize::try_from(header.end_of_chunk).unwrap() - 12;
|
|
let ds: LocationTag = bincode::deserialize(&data[i..i + u]).unwrap();
|
|
let images = &mut adb.find_dataset(last_type).child;
|
|
let mut str = None;
|
|
match ds.tag_type {
|
|
3 => {
|
|
let mut bytes = Vec::new();
|
|
let str_end = u32::from_le_bytes(data[i+12..i+16].try_into().unwrap()) as usize;
|
|
let mut h = i+24;
|
|
u += 12 + str_end;
|
|
while h < i+24+str_end {
|
|
bytes.push(u16::from_le_bytes(data[h..h+2].try_into().unwrap()));
|
|
h+=2;
|
|
}
|
|
let g = String::from_utf16(&bytes).unwrap();
|
|
info!("str: {}", g);
|
|
str = Some(g);
|
|
images.last_mut().unwrap().tag.last_mut().unwrap().child.as_mut().unwrap().tag = Some(ThirdTypeTag { data: ds, str });
|
|
},
|
|
2 => {
|
|
images.last_mut().unwrap().tag.push(SecondTypeTag { data: ds, str, child: None });
|
|
},
|
|
_ => { u = usize::try_from(header.children_count).unwrap() - 12; }
|
|
}
|
|
},
|
|
ChunkType::ImageName => {
|
|
info!("ImageName tag");
|
|
u = usize::try_from(header.end_of_chunk).unwrap() - 12;
|
|
let ds: ImageName = bincode::deserialize(&data[i..i + u]).unwrap();
|
|
let images = &mut adb.find_dataset(last_type).child;
|
|
images.last_mut().unwrap().tag.last_mut().unwrap().child = Some(AImageName { iname: ds, tag: None });
|
|
},
|
|
_ => { u = 1; info!("Unknown stuff happened {:X?}", header.chunk_type.to_vec()); }
|
|
}
|
|
i += u;
|
|
chunk_header = None;
|
|
ChunkState::Header
|
|
}
|
|
}
|
|
}
|
|
adb
|
|
}
|
|
}
|
|
|
|
pub mod serializer {
|
|
use log::info;
|
|
use crate::artworkdb::aobjects::ADatabase;
|
|
use crate::artworkdb::objects::{ChunkHeader, ChunkType};
|
|
|
|
pub fn to_bytes(adb: ADatabase) -> Vec<u8> {
|
|
let mut bytes: Vec<u8> = Vec::new();
|
|
for i in 0..(adb.children.len()) {
|
|
let data_set = adb.children.get(i).unwrap();
|
|
let mut entry_bytes = Vec::new();
|
|
|
|
match data_set.data.data_type {
|
|
1 => {
|
|
entry_bytes.append(&mut generate_header_raw(ChunkType::ImageList, 80, data_set.child.len()));
|
|
entry_bytes.append(&mut [0; 80].to_vec());
|
|
}, // Image List
|
|
2 => {
|
|
entry_bytes.append(&mut generate_header_raw(ChunkType::AlbumList, 80, 0));
|
|
entry_bytes.append(&mut [0; 80].to_vec());
|
|
}, // Album list
|
|
3 => {
|
|
entry_bytes.append(&mut generate_header_raw(ChunkType::FileList, 80, data_set.child.len()));
|
|
entry_bytes.append(&mut [0; 80].to_vec());
|
|
}, // File list
|
|
_ => { info!("Unknown data_set type!"); }
|
|
}
|
|
|
|
for img in data_set.child.iter() {
|
|
let mut args = Vec::new();
|
|
if img.data.is_some() {
|
|
for o in &img.tag {
|
|
let mut inameb = Vec::new();
|
|
if let Some(name) = &o.child {
|
|
|
|
let mut nb = Vec::new();
|
|
|
|
let arg = name.tag.as_ref().unwrap();
|
|
let mut str_b = string_to_ipod16(arg.str.as_ref().unwrap());
|
|
str_b = [3u32.to_le_bytes().to_vec(), 0u32.to_le_bytes().to_vec(), 0u32.to_le_bytes().to_vec(), (str_b.len() as u32).to_le_bytes().to_vec(), 2u32.to_le_bytes().to_vec(), 0u32.to_le_bytes().to_vec(), str_b ].concat();
|
|
nb.append(&mut generate_header(ChunkType::LocationTag, 12, str_b.len()));
|
|
nb.append(&mut str_b);
|
|
|
|
inameb.append(&mut generate_header(ChunkType::ImageName, 64, nb.len()));
|
|
inameb.append(&mut [0; 32].to_vec());
|
|
inameb.append(&mut bincode::serialize(&name.iname).unwrap());
|
|
inameb.append(&mut nb);
|
|
}
|
|
|
|
let mut data = [2u32.to_le_bytes(), 0u32.to_le_bytes(), 0u32.to_le_bytes()].concat();
|
|
args.append(&mut generate_header_raw(ChunkType::LocationTag, 12, inameb.len()));
|
|
args.append(&mut data);
|
|
args.append(&mut inameb);
|
|
}
|
|
let mut item = Vec::new();
|
|
item.append(&mut generate_header(ChunkType::ImageItem, 140, args.len()));
|
|
item.append(&mut bincode::serialize(img.data.as_ref().unwrap()).unwrap());
|
|
item.append(&mut [0; 100].to_vec());
|
|
entry_bytes.append(&mut item);
|
|
}
|
|
|
|
if img.file.is_some() {
|
|
entry_bytes.append(&mut generate_header(ChunkType::FileImage, 112, 0));
|
|
entry_bytes.append(&mut bincode::serialize(img.file.as_ref().unwrap()).unwrap());
|
|
entry_bytes.append(&mut [0; 100].to_vec());
|
|
}
|
|
|
|
entry_bytes.append(&mut args);
|
|
}
|
|
|
|
bytes.append(&mut generate_header(ChunkType::DataSet, 84, entry_bytes.len()));
|
|
bytes.append(&mut bincode::serialize(&data_set.data).unwrap());
|
|
bytes.append(&mut [0; 80].to_vec());
|
|
|
|
bytes.append(&mut entry_bytes);
|
|
}
|
|
|
|
let sdb = bincode::serialize(&adb.data.as_ref().unwrap()).unwrap();
|
|
bytes = [generate_header(ChunkType::ArtworkDB, 56 + 64, bytes.len()), sdb, [0; 64].to_vec(), bytes].concat();
|
|
bytes
|
|
}
|
|
|
|
fn string_to_ipod16(str: &str) -> Vec<u8> {
|
|
str.encode_utf16().flat_map(|f| [f as u8, (f >> 8) as u8]).collect()
|
|
}
|
|
|
|
fn generate_header(ct: ChunkType, header_size: usize, data_len: usize) -> Vec<u8> {
|
|
let header_size = 12 + header_size as u32;
|
|
let header = ChunkHeader{ chunk_type: ct.into(), end_of_chunk: header_size, children_count: header_size + data_len as u32};
|
|
bincode::serialize(&header).unwrap()
|
|
}
|
|
|
|
fn generate_header_raw(ct: ChunkType, header_size: usize, child_cnt: usize) -> Vec<u8> {
|
|
let header_size = 12 + header_size as u32;
|
|
let header = ChunkHeader{ chunk_type: ct.into(), end_of_chunk: header_size, children_count: child_cnt as u32};
|
|
bincode::serialize(&header).unwrap()
|
|
}
|
|
}
|
|
|
|
pub mod aobjects {
|
|
use crate::artworkdb::objects::{ChunkHeader, DataSet, Database, ImageFile, ImageItem, ImageName, LocationTag};
|
|
|
|
#[derive(Debug, serde::Serialize)]
|
|
pub struct ADatabase {
|
|
pub header: Option<ChunkHeader>,
|
|
pub data: Option<Database>,
|
|
pub children: Vec<ADataSet>
|
|
}
|
|
|
|
impl ADatabase {
|
|
pub(crate) fn find_dataset(&mut self, p0: u32) -> &mut ADataSet {
|
|
self.children.iter_mut().find(|d| d.data.data_type == p0).unwrap()
|
|
}
|
|
|
|
fn get_last_cor_id(&mut self) -> u32{
|
|
let files = self.find_dataset(3);
|
|
let image_files = &mut files.child;
|
|
image_files.iter().map(|p| p.file.as_ref().unwrap().correlation_id).max().unwrap_or(1028)
|
|
}
|
|
|
|
pub fn add_images(&mut self, small_image_path: &str, large_image_path: &str, song_dbid: u64) {
|
|
let last_cor_id = self.get_last_cor_id();
|
|
let ds = self.find_dataset(1);
|
|
let image_items = &mut ds.child;
|
|
let image_item = crate::artworkdb::objects::ImageItem {
|
|
number_of_children: 3,
|
|
id: image_items.last().map_or(100, |p| p.data.as_ref().unwrap().id + 1),
|
|
song_dbid,
|
|
unknown4: 0,
|
|
rating: 0,
|
|
unknown6: 0,
|
|
original_date: 0,
|
|
digitized_date: 0,
|
|
source_image_size: 0
|
|
};
|
|
|
|
let small_tag = SecondTypeTag {
|
|
data: LocationTag {
|
|
tag_type: 2,
|
|
unk1: 0,
|
|
unk2: 0
|
|
},
|
|
str: None,
|
|
child: Some(
|
|
AImageName {
|
|
iname: ImageName {
|
|
number_of_children: 1,
|
|
correlation_id: last_cor_id + 1,
|
|
ithmb_offset: 0,
|
|
image_size: 20000,
|
|
vertical_padding: 0,
|
|
horizontal_padding: 0,
|
|
image_height: 100,
|
|
image_width: 100,
|
|
unknown: 0,
|
|
image_size_n: 20000,
|
|
},
|
|
tag: Some(
|
|
ThirdTypeTag {
|
|
data: LocationTag {
|
|
tag_type: 3,
|
|
unk1: 0,
|
|
unk2: 0,
|
|
},
|
|
str: Some(
|
|
[":", small_image_path].concat(),
|
|
),
|
|
}
|
|
),
|
|
}
|
|
),
|
|
};
|
|
|
|
let large_tag = SecondTypeTag {
|
|
data: LocationTag {
|
|
tag_type: 2,
|
|
unk1: 0,
|
|
unk2: 0
|
|
},
|
|
str: None,
|
|
child: Some(
|
|
AImageName {
|
|
iname: ImageName {
|
|
number_of_children: 1,
|
|
correlation_id: last_cor_id + 2,
|
|
ithmb_offset: 80000,
|
|
image_size: 80000,
|
|
vertical_padding: 0,
|
|
horizontal_padding: 0,
|
|
image_height: 200,
|
|
image_width: 200,
|
|
unknown: 0,
|
|
image_size_n: 80000,
|
|
},
|
|
tag: Some(
|
|
ThirdTypeTag {
|
|
data: LocationTag {
|
|
tag_type: 3,
|
|
unk1: 0,
|
|
unk2: 0,
|
|
},
|
|
str: Some(
|
|
[":", large_image_path].concat(),
|
|
),
|
|
}
|
|
),
|
|
}
|
|
),
|
|
};
|
|
|
|
|
|
image_items.push(AImageItem {
|
|
tag: vec![small_tag, large_tag],
|
|
data: Some(image_item),
|
|
file: None,
|
|
});
|
|
|
|
let files = self.find_dataset(3);
|
|
let image_files = &mut files.child;
|
|
|
|
image_files.append(&mut vec![
|
|
AImageItem {
|
|
tag: Vec::new(),
|
|
data: None,
|
|
file: Some(
|
|
ImageFile {
|
|
unknown1: 0,
|
|
correlation_id: last_cor_id + 1,
|
|
image_size: 20000,
|
|
},
|
|
),
|
|
},
|
|
AImageItem {
|
|
tag: Vec::new(),
|
|
data: None,
|
|
file: Some(
|
|
ImageFile {
|
|
unknown1: 0,
|
|
correlation_id: last_cor_id + 2,
|
|
image_size: 80000,
|
|
},
|
|
),
|
|
},
|
|
]);
|
|
}
|
|
}
|
|
|
|
#[derive(Debug, serde::Serialize)]
|
|
pub struct ADataSet {
|
|
pub header: ChunkHeader,
|
|
pub data: DataSet,
|
|
pub child: Vec<AImageItem>
|
|
}
|
|
|
|
#[derive(Debug, serde::Serialize)]
|
|
pub struct AImageItem {
|
|
pub tag: Vec<SecondTypeTag>,
|
|
pub data: Option<ImageItem>,
|
|
pub file: Option<ImageFile>,
|
|
}
|
|
|
|
#[derive(Debug, serde::Serialize)]
|
|
pub struct AImageName {
|
|
pub iname: ImageName,
|
|
pub tag: Option<ThirdTypeTag>,
|
|
}
|
|
|
|
#[derive(Debug, serde::Serialize)]
|
|
pub struct SecondTypeTag {
|
|
pub data: LocationTag,
|
|
pub str: Option<String>,
|
|
pub child: Option<AImageName>
|
|
}
|
|
|
|
#[derive(Debug, serde::Serialize)]
|
|
pub struct ThirdTypeTag {
|
|
pub data: LocationTag,
|
|
pub str: Option<String>
|
|
}
|
|
}
|
|
|
|
pub mod objects {
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
pub enum ChunkType {
|
|
ArtworkDB,
|
|
DataSet,
|
|
ImageList,
|
|
AlbumList,
|
|
FileList,
|
|
ImageItem,
|
|
LocationTag,
|
|
ImageName,
|
|
FileImage,
|
|
Unknown,
|
|
}
|
|
|
|
impl From<[u8; 4]> for ChunkType {
|
|
fn from(value: [u8; 4]) -> Self {
|
|
match value {
|
|
[0x6D, 0x68, 0x66, 0x64] => ChunkType::ArtworkDB,
|
|
[0x6D, 0x68, 0x73, 0x64] => ChunkType::DataSet,
|
|
[0x6D, 0x68, 0x6C, 0x69] => ChunkType::ImageList,
|
|
[0x6D, 0x68, 0x69, 0x69] => ChunkType::ImageItem,
|
|
[0x6D, 0x68, 0x6F, 0x64] => ChunkType::LocationTag,
|
|
[0x6D, 0x68, 0x6E, 0x69] => ChunkType::ImageName,
|
|
[0x6D, 0x68, 0x6C, 0x61] => ChunkType::AlbumList,
|
|
[0x6D, 0x68, 0x6C, 0x66] => ChunkType::FileList,
|
|
[0x6D, 0x68, 0x69, 0x66] => ChunkType::FileImage,
|
|
_ => ChunkType::Unknown,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl From<ChunkType> for [u8; 4] {
|
|
fn from(value: ChunkType) -> Self {
|
|
match value {
|
|
ChunkType::ArtworkDB => [0x6D, 0x68, 0x66, 0x64],
|
|
ChunkType::DataSet => [0x6D, 0x68, 0x73, 0x64],
|
|
ChunkType::Unknown => [0x00, 0x00, 0x00, 0x00],
|
|
ChunkType::ImageList => [0x6D, 0x68, 0x6C, 0x69],
|
|
ChunkType::ImageItem => [0x6D, 0x68, 0x69, 0x69],
|
|
ChunkType::LocationTag => [0x6D, 0x68, 0x6F, 0x64],
|
|
ChunkType::ImageName => [0x6D, 0x68, 0x6E, 0x69],
|
|
ChunkType::AlbumList => [0x6D, 0x68, 0x6C, 0x61],
|
|
ChunkType::FileList => [0x6D, 0x68, 0x6C, 0x66],
|
|
ChunkType::FileImage => [0x6D, 0x68, 0x69, 0x66],
|
|
}
|
|
}
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Debug, Clone, Copy)]
|
|
pub struct ChunkHeader {
|
|
pub chunk_type: [u8; 4],
|
|
pub end_of_chunk: u32,
|
|
pub children_count: u32,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Debug)]
|
|
pub struct Database {
|
|
unknown1: u32,
|
|
unknown2: u32,
|
|
number_of_children: u32,
|
|
unknown3: u32,
|
|
next_id_for_mhii: u32,
|
|
unknown5: u64,
|
|
unknown6: u64,
|
|
unknown7: u32,
|
|
unknown8: u32,
|
|
unknown9: u32,
|
|
unknown10: u64
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Debug, Clone, Copy)]
|
|
pub struct DataSet {
|
|
pub data_type: u32,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Debug, Clone, Copy)]
|
|
pub struct LocationTag {
|
|
pub tag_type: u32,
|
|
pub unk1: u32,
|
|
pub unk2: u32,
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Debug)]
|
|
pub struct ImageItem {
|
|
pub number_of_children: u32,
|
|
pub id: u32,
|
|
pub song_dbid: u64,
|
|
pub unknown4: u32,
|
|
pub rating: u32,
|
|
pub unknown6: u32,
|
|
pub original_date: u32,
|
|
pub digitized_date: u32,
|
|
pub source_image_size: u32
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Debug)]
|
|
pub struct ImageFile {
|
|
pub(crate) unknown1: u32,
|
|
pub(crate) correlation_id: u32,
|
|
pub(crate) image_size: u32
|
|
}
|
|
|
|
#[derive(Serialize, Deserialize, PartialEq, Debug)]
|
|
pub struct ImageName {
|
|
pub(crate) number_of_children: u32,
|
|
pub(crate) correlation_id: u32,
|
|
pub(crate) ithmb_offset: u32,
|
|
pub(crate) image_size: u32,
|
|
pub(crate) vertical_padding: u16,
|
|
pub(crate) horizontal_padding: u16,
|
|
pub(crate) image_height: u16,
|
|
pub(crate) image_width: u16,
|
|
pub(crate) unknown: u32,
|
|
pub(crate) image_size_n: u32,
|
|
}
|
|
} |