Maybe a bit of both. Tom Hagen was a legal advisor and defends Michael Corleone during the Senate hearings on the mob.
That part is def based on Sidney Korshak.