Finding multiple of the same patterns in a binary

[LawrenceColombo]

Hey all, I wrote the below pattern to find uncompressed WAVE files embedded in a binary, and while I can search for the start offsets, I don't know how to highlight them in the hex editor view unless I do it manually? (and there are a few hundred) You can see below with my WIP that I'm search for the offset of RIFF but WAVEHeader header @ riff_offset; doesn't seem to highlight it, anyways fantastic tool so far

import type.magic;
import std.io;

struct WAVEHeader {
    u32 fileBlockId;
    u32 fileSize;
    u32 fileFormatId;
   
    
    u32 FormatBlocID;

    u32 BlocSize;       
    u16 AudioFormat;    
    u16 NbrChannels;    
    u32 Frequency;      
    u32 BytePerSec;     
    u16 BytePerBloc;    
    u16 BitsPerSample;  

    u32 DataBlocID;    
    le u32 DataSize;
    u8 Data[DataSize];       
};

// Placed headers at known offsets
//WAVEHeader waveheader @0x000BC5B8;
//WAVEHeader waveheader2 @0x000BD8A4;

u32 offset = std::mem::find_sequence(0, 0x52, 0x49, 0x46, 0x46);
std::print("First RIFF offset: {:#x}", offset);

// WIP
u32 sequenceOffset = 0x0;
while (true) {
    u32 riff_offset = std::mem::find_sequence(sequenceOffset, 0x52, 0x49, 0x46, 0x46);
    std::print("While loop RIFF offset: {:#x}", riff_offset);
    WAVEHeader header @ riff_offset;
    break;
};

[paxcut]

The reason it doesn't work is that you can't use loops to place variables. Instead what you need to do is to turn the loop into a struct which is indexed using a while loop. As a possible example something like this may work:

u32 sequenceOffset = 0x0;
struct Finder {
    u32 riff_offset = std::mem::find_sequence(sequenceOffset, 0x52, 0x49, 0x46, 0x46);
    if (riff_offset == -1)
        break;
    std::print("While loop RIFF offset: {:#x}", riff_offset);
    WAVEHeader header @ riff_offset;
};
Finder find[while(true)]@0;