From f1720ca884bc3fa9da1288ad955e46f165aa4168 Mon Sep 17 00:00:00 2001
From: James Moger <james.moger@gitblit.com>
Date: Wed, 01 Jun 2011 08:09:46 -0400
Subject: [PATCH] Unit testing. Disable links on first commit. Initial stats page.
---
src/com/gitblit/JettyLoginService.java | 110 ++++++++++++++++++++++++++++++++++++++-----------------
1 files changed, 76 insertions(+), 34 deletions(-)
diff --git a/src/com/gitblit/JettyLoginService.java b/src/com/gitblit/JettyLoginService.java
index fb510ee..63a9861 100644
--- a/src/com/gitblit/JettyLoginService.java
+++ b/src/com/gitblit/JettyLoginService.java
@@ -1,3 +1,18 @@
+/*
+ * Copyright 2011 gitblit.com.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
package com.gitblit;
import java.io.File;
@@ -23,8 +38,7 @@
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-import com.gitblit.utils.StringUtils;
-import com.gitblit.wicket.models.UserModel;
+import com.gitblit.models.UserModel;
public class JettyLoginService extends MappedLoginService implements ILoginService {
@@ -45,8 +59,7 @@
return null;
}
UserModel user = new UserModel(username);
- user.setCookie(StringUtils.getSHA1((Constants.NAME + username + new String(password))));
- user.canAdmin(identity.isUserInRole(Constants.ADMIN_ROLE, null));
+ user.canAdmin = identity.isUserInRole(Constants.ADMIN_ROLE, null);
// Add repositories
for (Principal principal : identity.getSubject().getPrincipals()) {
@@ -62,15 +75,12 @@
}
@Override
- public UserModel authenticate(char[] cookie) {
- // TODO cookie login
- return null;
- }
-
- @Override
public UserModel getUserModel(String username) {
- UserModel model = new UserModel(username);
UserIdentity identity = _users.get(username);
+ if (identity == null) {
+ return null;
+ }
+ UserModel model = new UserModel(username);
Subject subject = identity.getSubject();
for (Principal principal : subject.getPrincipals()) {
if (principal instanceof RolePrincipal) {
@@ -80,7 +90,7 @@
case '#':
// Permissions
if (name.equalsIgnoreCase(Constants.ADMIN_ROLE)) {
- model.canAdmin(true);
+ model.canAdmin = true;
}
break;
default:
@@ -95,7 +105,7 @@
Properties allUsers = readRealmFile();
String value = allUsers.getProperty(username);
String password = value.split(",")[0];
- model.setPassword(password);
+ model.password = password;
} catch (Throwable t) {
logger.error(MessageFormat.format("Failed to read password for user {0}!", username), t);
}
@@ -104,17 +114,22 @@
@Override
public boolean updateUserModel(UserModel model) {
+ return updateUserModel(model.username, model);
+ }
+
+ @Override
+ public boolean updateUserModel(String username, UserModel model) {
try {
Properties allUsers = readRealmFile();
- ArrayList<String> roles = new ArrayList<String>(model.getRepositories());
+ ArrayList<String> roles = new ArrayList<String>(model.repositories);
// Permissions
- if (model.canAdmin()) {
+ if (model.canAdmin) {
roles.add(Constants.ADMIN_ROLE);
}
StringBuilder sb = new StringBuilder();
- sb.append(model.getPassword());
+ sb.append(model.password);
sb.append(',');
for (String role : roles) {
sb.append(role);
@@ -122,32 +137,41 @@
}
// trim trailing comma
sb.setLength(sb.length() - 1);
- allUsers.put(model.getUsername(), sb.toString());
+ allUsers.remove(username);
+ allUsers.put(model.username, sb.toString());
writeRealmFile(allUsers);
// Update login service
- putUser(model.getUsername(), Credential.getCredential(model.getPassword()), roles.toArray(new String[0]));
+ removeUser(username);
+ putUser(model.username, Credential.getCredential(model.password),
+ roles.toArray(new String[0]));
return true;
} catch (Throwable t) {
- logger.error(MessageFormat.format("Failed to update user model {0}!", model.getUsername()), t);
+ logger.error(MessageFormat.format("Failed to update user model {0}!", model.username),
+ t);
}
return false;
}
@Override
public boolean deleteUserModel(UserModel model) {
+ return deleteUser(model.username);
+ }
+
+ @Override
+ public boolean deleteUser(String username) {
try {
// Read realm file
Properties allUsers = readRealmFile();
- allUsers.remove(model.getUsername());
+ allUsers.remove(username);
writeRealmFile(allUsers);
// Drop user from map
- _users.remove(model.getUsername());
+ removeUser(username);
return true;
} catch (Throwable t) {
- logger.error(MessageFormat.format("Failed to delete user model {0}!", model.getUsername()), t);
+ logger.error(MessageFormat.format("Failed to delete user {0}!", username), t);
}
return false;
}
@@ -213,7 +237,7 @@
// add roles to users
for (String user : needsAddRole) {
String userValues = allUsers.getProperty(user);
- userValues += ("," + role);
+ userValues += "," + role;
allUsers.put(user, userValues);
String[] values = userValues.split(",");
String password = values[0];
@@ -245,7 +269,8 @@
allUsers.put(user, sb.toString());
// update memory
- putUser(user, Credential.getCredential(password), revisedRoles.toArray(new String[0]));
+ putUser(user, Credential.getCredential(password),
+ revisedRoles.toArray(new String[0]));
}
// persist changes
@@ -302,14 +327,16 @@
allUsers.put(user, sb.toString());
// update memory
- putUser(user, Credential.getCredential(password), revisedRoles.toArray(new String[0]));
+ putUser(user, Credential.getCredential(password),
+ revisedRoles.toArray(new String[0]));
}
// persist changes
writeRealmFile(allUsers);
return true;
} catch (Throwable t) {
- logger.error(MessageFormat.format("Failed to rename role {0} to {1}!", oldRole, newRole), t);
+ logger.error(
+ MessageFormat.format("Failed to rename role {0} to {1}!", oldRole, newRole), t);
}
return false;
}
@@ -358,11 +385,13 @@
allUsers.put(user, sb.toString());
// update memory
- putUser(user, Credential.getCredential(password), revisedRoles.toArray(new String[0]));
+ putUser(user, Credential.getCredential(password),
+ revisedRoles.toArray(new String[0]));
}
// persist changes
writeRealmFile(allUsers);
+ return true;
} catch (Throwable t) {
logger.error(MessageFormat.format("Failed to delete role {0}!", role), t);
}
@@ -381,24 +410,36 @@
// Update realm file
File realmFileCopy = new File(realmFile.getAbsolutePath() + ".tmp");
FileWriter writer = new FileWriter(realmFileCopy);
- properties.store(writer, "# Git:Blit realm file format: username=password,\\#permission,repository1,repository2...");
+ properties
+ .store(writer,
+ "# Git:Blit realm file format: username=password,\\#permission,repository1,repository2...");
writer.close();
if (realmFileCopy.exists() && realmFileCopy.length() > 0) {
- realmFile.delete();
- realmFileCopy.renameTo(realmFile);
+ if (realmFile.delete()) {
+ if (!realmFileCopy.renameTo(realmFile)) {
+ throw new IOException(MessageFormat.format("Failed to rename {0} to {1}!",
+ realmFileCopy.getAbsolutePath(), realmFile.getAbsolutePath()));
+ }
+ } else {
+ throw new IOException(MessageFormat.format("Failed to delete (0)!",
+ realmFile.getAbsolutePath()));
+ }
} else {
- throw new IOException("Failed to save realmfile!");
+ throw new IOException(MessageFormat.format("Failed to save {0}!",
+ realmFileCopy.getAbsolutePath()));
}
}
/* ------------------------------------------------------------ */
@Override
public void loadUsers() throws IOException {
- if (realmFile == null)
+ if (realmFile == null) {
return;
+ }
- if (Log.isDebugEnabled())
+ if (Log.isDebugEnabled()) {
Log.debug("Load " + this + " from " + realmFile);
+ }
Properties allUsers = readRealmFile();
// Map Users
@@ -412,7 +453,8 @@
credentials = credentials.substring(0, c).trim();
}
- if (username != null && username.length() > 0 && credentials != null && credentials.length() > 0) {
+ if (username != null && username.length() > 0 && credentials != null
+ && credentials.length() > 0) {
String[] roleArray = IdentityService.NO_ROLES;
if (roles != null && roles.length() > 0) {
roleArray = roles.split(",");
--
Gitblit v1.9.1