Energy security top of mind ahead of budget speech