I was actually looking for something more flexible which would allow me to measure the time and bits during communication.

It should be possible to do this in the current proposal of the communication hook. Could you elaborate a bit more on the limitations in the current proposal that might prevent us from doing these measurements?