diff options
Diffstat (limited to 'libshouldbeinlibc/exec-reauth.c')
| -rw-r--r-- | libshouldbeinlibc/exec-reauth.c | 32 |
1 files changed, 17 insertions, 15 deletions
diff --git a/libshouldbeinlibc/exec-reauth.c b/libshouldbeinlibc/exec-reauth.c index cd217e02..71e41b8b 100644 --- a/libshouldbeinlibc/exec-reauth.c +++ b/libshouldbeinlibc/exec-reauth.c @@ -60,22 +60,24 @@ exec_reauth (auth_t auth, int secure, int must_reauth, err = auth_user_authenticate (auth, ref, MACH_MSG_TYPE_MAKE_SEND, &newport); mach_port_mod_refs (mach_task_self (), ref, MACH_PORT_RIGHT_RECEIVE, -1); + /* If they gave us nothing, make sure not to proceed. */ + if (!err && newport == MACH_PORT_NULL) + err = KERN_INVALID_ARGUMENT; if (err) - { - if (must_reauth) - return err; - /* Nothing Happens. */ - } - else - { - if (isproc) - mach_port_deallocate (mach_task_self (), newport); - else - { - mach_port_deallocate (mach_task_self (), *port); - *port = newport; - } - } + return must_reauth ? err : 0; + + if (isproc) + { + err = proc_complete_reauthentication (newport); + if (err) + { + mach_port_deallocate (mach_task_self (), newport); + return must_reauth ? err : 0; + } + } + + mach_port_deallocate (mach_task_self (), *port); + *port = newport; } return 0; } |