Police are set to receive greater protection from terrorists, with a squad of security specialists and building upgrades tipped to be announced in the federal budget.