Sort shuffle read improvements

Hi @sryza , I've done some improvements in current SortSHuffleRead, would you mind help reviewing this and give me some comments, thanks a lot!

1. copy memory shuffle block from off-heap to on-heap if memory is allocated with direct memory to control the off-heap memory size.
2. Some code style modification and better failure processing logic.
I didn't try this, I think this can be worked from my understanding. Meantime I think `while` is just the style choice, also very simple and straightforward here :)