Cleanup

diff --git a/mech_eap/util_name.c b/mech_eap/util_name.c
index 4c6c324..cc00c42 100644 (file)
--- a/mech_eap/util_name.c
+++ b/mech_eap/util_name.c
@@ -132,6 +132,12 @@ krbPrincipalToName(OM_uint32 *minor,
     name->krbPrincipal = *principal;
     *principal = NULL;
 
+    if (name->krbPrincipal->length == 1) {
+        name->flags |= NAME_FLAG_NAI;
+    } else {
+        name->flags |= NAME_FLAG_SERVICE;
+    }
+
     *minor = 0;
     return GSS_S_COMPLETE;
 }