That should work! Using the SHA3_sponge function along with SHA3_absorb and SHA3_squeeze from the assembly code can be sufficient, as long as the calling conventions align correctly between the C functions and assembly. Just make sure everything integrates smoothly, and you should be good to go.