*исправление ошибок: двой перезаписи, исправление записи с BOM, добавлена проверка в insert_tags_into_audio

This commit is contained in:
K1ndWha1e committed 2026-10-07 23:24:01 +03:00
1 parent ffa8d80875
commit 078e68efa8
5 files changed
+181 -55

No files matched your search

+59 -21
View File
@@ -72,22 +72,16 @@ pub fn get_current_tag_size[T](mut tag T, mut audio os.File) {
tag.size = (u64(first_ten_bytes[6]) << 21) | (u64(first_ten_bytes[7]) << 14) | (u64(first_ten_bytes[8]) << 7) | u64(first_ten_bytes[9])
}
// Кодирование размера заголовка со всеми тэгами
pub fn encode_header_size(size u64, idv3_version u8) []u8 {
stride := if idv3_version == 3 {
[u64(24), u64(16), u64(8), u64(0xFF)]
} else {
[u64(21), u64(14), u64(7), u64(0x7F)]
}
// Кодирование размера заголовка со всеми тэгами.
// В заголовке размер всегда syncsafe (по 7 бит в байте), в том числе в v2.3
pub fn encode_header_size(size u64) []u8 {
mut encoded_size := []u8{len: 4}
encoded_size[0] = u8((size >> stride[0]) & stride[3])
encoded_size[1] = u8((size >> stride[1]) & stride[3])
encoded_size[2] = u8((size >> stride[2]) & stride[3])
encoded_size[3] = u8(size & stride[3])
encoded_size[0] = u8((size >> 21) & 0x7F)
encoded_size[1] = u8((size >> 14) & 0x7F)
encoded_size[2] = u8((size >> 7) & 0x7F)
encoded_size[3] = u8(size & 0x7F)
return encoded_size
}
// Кодирование первых 10 байт ID3
@@ -99,8 +93,7 @@ pub fn encode_header[T](mut id3v T) {
// Минорная версия и спец. флаги
id3v.encoded_header << []u8{len: 2}
// Размер всего тэга в заголовке всегда syncsafe (по 7 бит в байте), в том числе в v2.3
id3v.encoded_header << metadata.encode_header_size(id3v.size, 4)
id3v.encoded_header << metadata.encode_header_size(id3v.size)
}
// Количество нулей в конце тэга
@@ -116,16 +109,23 @@ pub fn encode_embed_tags[T](mut id3v2 T) {
$if field.typ is Frame {
// Читаются только заполненные тэги
if id3v2.$(field.name).cur_size > 0 {
if id3v2.$(field.name).name.to_lower() != "apic" {
// Кодируем значения полей
tag_bytes << id3v2.$(field.name).get_encoded_frame_tag(id3v2.version)
} else {
// Картинка не трогается
tag_bytes << id3v2.$(field.name).apic or { []u8{} }
// Фрейма не было в файле и init не вызывался: имя берётся из поля структуры
if id3v2.$(field.name).name == "" && field.name.len == 4 {
id3v2.$(field.name).init(field.name, id3v2.version)
}
// Без 4-символьного ID фрейм испортит весь тэг
if id3v2.$(field.name).name.len == 4 {
tag_bytes << id3v2.$(field.name).to_bytes(id3v2.version)
}
}
}
}
// Фреймы, для которых нет отдельного поля в структуре
for mut frame in id3v2.others {
tag_bytes << frame.to_bytes(id3v2.version)
}
// Запас нулей чтобы не пересчитывать весь тэг при маленьком редактировании
tag_bytes << []u8{len: tag_padding}
@@ -136,3 +136,41 @@ pub fn encode_embed_tags[T](mut id3v2 T) {
encode_header(mut id3v2)
id3v2.encoded_header << tag_bytes
}
// Добавляет аудио из файла после нового тэга в encoded_header, старый тэг пропускается
pub fn insert_tags_into_audio[T](mut id3v2 T, mut audio os.File) ! {
header := get_bytes(mut audio, 0, 10)
if header.len < 10 || header[..3].bytestr() != "ID3" {
return error("в файле нет тэга ID3v2")
}
// Размер старого тэга берётся из файла, id3v2.size уже хранит размер нового.
// В заголовке он syncsafe в любой версии, плюс 10 байт самого заголовка
mut audio_start := get_size_v24(header[6..10]) + 10
// В v2.4 после тэга может идти футер ещё на 10 байт
if header[3] == 4 && header[5] & 0x10 != 0 {
audio_start += 10
}
audio.seek(0, .end)!
file_size := u64(audio.tell()!)
if audio_start > file_size {
return error("размер тэга (${audio_start}) больше размера файла (${file_size})")
}
// Аудио читается сразу в конец encoded_header, без промежуточной копии всего файла
sound_len := int(file_size - audio_start)
header_len := id3v2.encoded_header.len
unsafe { id3v2.encoded_header.grow_len(sound_len) }
audio.seek(i64(audio_start), .start)!
mut read := 0
for read < sound_len {
n := unsafe { audio.read_into_ptr(&u8(id3v2.encoded_header.data) + header_len + read, sound_len - read)! }
if n == 0 {
return error("файл закончился раньше ожидаемого")
}
read += n
}
audio.seek(0, .start)!
}