Store custom extension callbacks according to their extension types instead of function signature

Store custom extension callbacks according to their extension types instead of function signature #2432