Kriminalvården är en statlig myndighet med ansvar för häkten , fängelser och frivård .