Train travel from Hong kong to Beijing and Shanghai to Hong Kong

I am planning to be in Hong Kong in July and have a tour to China. we intend to travel by train to Beijing and our trip ends in Shanghai and want to come to Hong Kong by train. Kindly advise me about Customs & Immigration when on train and also availabilty of tickets. Will it be safe to book the tickets in Hong Kong or before hand.