For the Steve Vai sounds, use a guitar with humbuckers, and crank everything. Make sure your sound has a lot of bottom, almost to the point where it is muddy. I cant really suggest anything else cause I dot know what kinda pedal you have or anything.
And God said, 'Let there be rock!'
-And it was good