unfortunately no, at least not in the betas. I believe the buttons on the headphones are controlled via hardware in the socket not software on the iPhone, so I don't believe this is possible for the 3G or original iPhone.
From what I've determined, no it can't be done. I don't think Apple was planning on releasing new headphones when the 3G came out, and because of that it doesn't have the hardware necessary to support them.