pflocal: add payload-aware intrans functions

* pflocal/mig-mutate.h: Add mutators.
* pflocal/mig-decls.c (begin_using_sock_user_payload): New function.
(begin_using_addr_payload): Likewise.

2 files changed, 15 insertions, 0 deletions

diff --git a/pflocal/mig-decls.h b/pflocal/mig-decls.h
index 983de9d1..b1da797d 100644
--- a/pflocal/mig-decls.h
+++ b/pflocal/mig-decls.h
@@ -36,6 +36,12 @@ begin_using_sock_user_port(mach_port_t port)
   return (sock_user_t)ports_lookup_port (0, port, sock_user_port_class);
 }
 
+static inline sock_user_t __attribute__ ((unused))
+begin_using_sock_user_payload (unsigned long payload)
+{
+  return ports_lookup_payload (NULL, payload, sock_user_port_class);
+}
+
 static inline void __attribute__ ((unused))
 end_using_sock_user_port (sock_user_t sock_user)
 {
@@ -49,6 +55,12 @@ begin_using_addr_port(mach_port_t port)
   return (addr_t)ports_lookup_port (0, port, addr_port_class);
 }
 
+static inline addr_t __attribute__ ((unused))
+begin_using_addr_payload (unsigned long payload)
+{
+  return ports_lookup_port (NULL, payload, addr_port_class);
+}
+
 static inline void __attribute__ ((unused))
 end_using_addr_port (addr_t addr)
 {
diff --git a/pflocal/mig-mutate.h b/pflocal/mig-mutate.h
index b1494730..238c806b 100644
--- a/pflocal/mig-mutate.h
+++ b/pflocal/mig-mutate.h
@@ -21,13 +21,16 @@
 #define IO_SELECT_REPLY_PORT
 
 #define IO_INTRAN sock_user_t begin_using_sock_user_port (io_t)
+#define IO_INTRAN_PAYLOAD sock_user_t begin_using_sock_user_payload
 #define IO_DESTRUCTOR end_using_sock_user_port (sock_user_t)
 
 #define IO_IMPORTS import "mig-decls.h";
 
 #define SOCKET_INTRAN sock_user_t begin_using_sock_user_port (socket_t)
+#define SOCKET_INTRAN_PAYLOAD sock_user_t begin_using_sock_user_payload
 #define SOCKET_DESTRUCTOR end_using_sock_user_port (sock_user_t)
 #define ADDRPORT_INTRAN addr_t begin_using_addr_port (addr_port_t)
+#define ADDRPORT_INTRAN_PAYLOAD addr_t begin_using_addr_payload
 #define ADDRPORT_DESTRUCTOR end_using_addr_port (addr_t)
 
 #define SOCKET_IMPORTS import "mig-decls.h";