agp: Switch mask_memory() method to take address argument again, not page