mm/rmap: extend rmap_walk_xxx() to cope with different cases