Fixed Change Password Bug

session.go

@@ -0,0 +1,181 @@
+/*
+Handles session login and lougout with HTTP stuff
+*/
+package main
+
+import (
+	"net/http"
+)
+
+func checkLogin(w http.ResponseWriter, r *http.Request) *LoginStatus {
+	var login_info *LoginInfo
+
+	session, err := store.Get(r, SESSION_NAME)
+	if err == nil {
+		username, ok := session.Values["login_username"]
+		password, ok2 := session.Values["login_password"]
+		user_dn, ok3 := session.Values["login_dn"]
+
+		if ok && ok2 && ok3 {
+			login_info = &LoginInfo{
+				DN:       user_dn.(string),
+				Username: username.(string),
+				Password: password.(string),
+			}
+		}
+	}
+
+	if login_info == nil {
+		login_info = handleLogin(w, r)
+		if login_info == nil {
+			return nil
+		}
+	}
+
+	l, err := ldapOpen(w)
+	if l == nil {
+		return nil
+	}
+
+	err = bind(User{
+		DN:       login_info.DN,
+		Password: login_info.Password,
+	}, config, l)
+
+	if err != nil {
+		delete(session.Values, "login_username")
+		delete(session.Values, "login_password")
+		delete(session.Values, "login_dn")
+
+		err = session.Save(r, w)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return nil
+		}
+		return checkLogin(w, r)
+	}
+
+	ldapUser, err := get(User{
+		DN: login_info.DN,
+		CN: login_info.Username,
+	}, config, l)
+
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return nil
+	}
+
+	userEntry := ldapUser.UserEntry
+
+	loginStatus := &LoginStatus{
+		Info:      login_info,
+		conn:      l,
+		UserEntry: userEntry,
+		CanAdmin:  ldapUser.CanAdmin,
+		CanInvite: ldapUser.CanInvite,
+	}
+
+	/*
+
+		requestKind := "(objectClass=organizationalPerson)"
+		if strings.EqualFold(login_info.DN, config.AdminAccount) {
+			requestKind = "(objectclass=*)"
+		}
+		searchRequest := ldap.NewSearchRequest(
+			login_info.DN,
+			ldap.ScopeBaseObject, ldap.NeverDerefAliases, 0, 0, false,
+			requestKind,
+			[]string{
+				"dn",
+				"displayname",
+				"givenname",
+				"sn",
+				"mail",
+				"cn",
+				"memberof",
+				"description",
+				"garage_s3_access_key",
+			},
+			nil)
+		//			FIELD_NAME_DIRECTORY_VISIBILITY,
+		//			FIELD_NAME_PROFILE_PICTURE,
+
+		sr, err := l.Search(searchRequest)
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return nil
+		}
+
+		if len(sr.Entries) != 1 {
+			http.Error(w, fmt.Sprintf("Unable to find entry for %s", login_info.DN), http.StatusInternalServerError)
+			return nil
+		}
+
+		loginStatus.UserEntry = sr.Entries[0]
+
+		loginStatus.CanAdmin = strings.EqualFold(loginStatus.Info.DN, config.AdminAccount)
+		loginStatus.CanInvite = false
+
+		groups := []EntryName{}
+		searchRequest = ldap.NewSearchRequest(
+			config.GroupBaseDN,
+			ldap.ScopeWholeSubtree, ldap.NeverDerefAliases, 0, 0, false,
+			fmt.Sprintf("(&(objectClass=groupOfNames)(member=%s))", login_info.DN),
+			[]string{"dn", "displayName", "cn", "description"},
+			nil)
+		// // log.Printf(fmt.Sprintf("708: %v",searchRequest))
+		sr, err = l.Search(searchRequest)
+		// if err != nil {
+		// 	http.Error(w, err.Error(), http.StatusInternalServerError)
+		// 	return
+		// }
+		//// log.Printf(fmt.Sprintf("303: %v",sr.Entries))
+		for _, ent := range sr.Entries {
+			// log.Printf(fmt.Sprintf("305: %v",ent.DN))
+			groups = append(groups, EntryName{
+				DN:   ent.DN,
+				Name: ent.GetAttributeValue("cn"),
+			})
+			// log.Printf(fmt.Sprintf("310: %v",config.GroupCanInvite))
+			if config.GroupCanInvite != "" && strings.EqualFold(ent.DN, config.GroupCanInvite) {
+				loginStatus.CanInvite = true
+			}
+			// log.Printf(fmt.Sprintf("314: %v",config.GroupCanAdmin))
+			if config.GroupCanAdmin != "" && strings.EqualFold(ent.DN, config.GroupCanAdmin) {
+				loginStatus.CanAdmin = true
+			}
+		}
+
+		// for _, attr := range loginStatus.UserEntry.Attributes {
+		// 	if strings.EqualFold(attr.Name, "memberof") {
+		// 		for _, group := range attr.Values {
+		// 			if config.GroupCanInvite != "" && strings.EqualFold(group, config.GroupCanInvite) {
+		// 				loginStatus.CanInvite = true
+		// 			}
+		// 			if config.GroupCanAdmin != "" && strings.EqualFold(group, config.GroupCanAdmin) {
+		// 				loginStatus.CanAdmin = true
+		// 			}
+		// 		}
+		// 	}
+		// }
+
+		return loginStatus
+	*/
+
+	return loginStatus
+}
+
+func logout(w http.ResponseWriter, r *http.Request) error {
+	session, err := store.Get(r, SESSION_NAME)
+	if err != nil {
+		session, _ = store.New(r, SESSION_NAME)
+		return err
+	}
+
+	delete(session.Values, "login_username")
+	delete(session.Values, "login_password")
+	delete(session.Values, "login_dn")
+
+	err = session.Save(r, w)
+	return err
+}