How do I read the disk configuration with AHCI driver?

Question

Hi I'm learning OS and trying to follow a book to write a kernel.

The book using IDE to control disk, while I tried to use AHCI with MSI to make it easier and more efficient.

I successfully read/write information and issued MSI interrupt. But I do not know how to read the disk configuration, especially numbers of total sectors available.

Thanks a lot!