From 04a98505a4ab8f48aee22800fcac193d9367d0ae Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Fri, 29 Nov 2013 11:05:51 -0500 Subject: [PATCH] Refactor user services and separate authentication (issue-281) --- src/main/java/com/gitblit/auth/WindowsAuthProvider.java | 372 +++++++++++++++++++++++++--------------------------- 1 files changed, 177 insertions(+), 195 deletions(-) diff --git a/src/main/java/com/gitblit/WindowsUserService.java b/src/main/java/com/gitblit/auth/WindowsAuthProvider.java similarity index 81% rename from src/main/java/com/gitblit/WindowsUserService.java rename to src/main/java/com/gitblit/auth/WindowsAuthProvider.java index 99077c6..d455d58 100644 --- a/src/main/java/com/gitblit/WindowsUserService.java +++ b/src/main/java/com/gitblit/auth/WindowsAuthProvider.java @@ -1,195 +1,177 @@ -/* - * Copyright 2013 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; -import java.util.Set; -import java.util.TreeSet; - -import org.slf4j.Logger; -import org.slf4j.LoggerFactory; - -import waffle.windows.auth.IWindowsAccount; -import waffle.windows.auth.IWindowsAuthProvider; -import waffle.windows.auth.IWindowsComputer; -import waffle.windows.auth.IWindowsIdentity; -import waffle.windows.auth.impl.WindowsAuthProviderImpl; - -import com.gitblit.Constants.AccountType; -import com.gitblit.manager.IRuntimeManager; -import com.gitblit.models.UserModel; -import com.gitblit.utils.ArrayUtils; -import com.gitblit.utils.StringUtils; -import com.sun.jna.platform.win32.Win32Exception; - -/** - * Implementation of a Windows user service. - * - * @author James Moger - */ -public class WindowsUserService extends GitblitUserService { - - private final Logger logger = LoggerFactory.getLogger(WindowsUserService.class); - - private IStoredSettings settings; - - private IWindowsAuthProvider waffle; - - public WindowsUserService() { - super(); - } - - @Override - public void setup(IRuntimeManager runtimeManager) { - this.settings = runtimeManager.getSettings(); - - String file = settings.getString(Keys.realm.windows.backingUserService, "${baseFolder}/users.conf"); - File realmFile = runtimeManager.getFileOrFolder(file); - - serviceImpl = createUserService(realmFile); - logger.info("Windows User Service backed by " + serviceImpl.toString()); - - waffle = new WindowsAuthProviderImpl(); - IWindowsComputer computer = waffle.getCurrentComputer(); - logger.info(" name = " + computer.getComputerName()); - logger.info(" status = " + describeJoinStatus(computer.getJoinStatus())); - logger.info(" memberOf = " + computer.getMemberOf()); - //logger.info(" groups = " + Arrays.asList(computer.getGroups())); - } - - protected String describeJoinStatus(String value) { - if ("NetSetupUnknownStatus".equals(value)) { - return "unknown"; - } else if ("NetSetupUnjoined".equals(value)) { - return "not joined"; - } else if ("NetSetupWorkgroupName".equals(value)) { - return "joined to a workgroup"; - } else if ("NetSetupDomainName".equals(value)) { - return "joined to a domain"; - } - return value; - } - - @Override - public boolean supportsCredentialChanges() { - return false; - } - - @Override - public boolean supportsDisplayNameChanges() { - return false; - } - - @Override - public boolean supportsEmailAddressChanges() { - return true; - } - - @Override - public boolean supportsTeamMembershipChanges() { - return true; - } - - @Override - public AccountType getAccountType() { - return AccountType.WINDOWS; - } - - @Override - public UserModel authenticate(String username, char[] password) { - if (isLocalAccount(username)) { - // local account, bypass Windows authentication - return super.authenticate(username, password); - } - - String defaultDomain = settings.getString(Keys.realm.windows.defaultDomain, null); - if (StringUtils.isEmpty(defaultDomain)) { - // ensure that default domain is null - defaultDomain = null; - } - - if (defaultDomain != null) { - // sanitize username - if (username.startsWith(defaultDomain + "\\")) { - // strip default domain from domain\ username - username = username.substring(defaultDomain.length() + 1); - } else if (username.endsWith("@" + defaultDomain)) { - // strip default domain from username@domain - username = username.substring(0, username.lastIndexOf('@')); - } - } - - IWindowsIdentity identity = null; - try { - if (username.indexOf('@') > -1 || username.indexOf('\\') > -1) { - // manually specified domain - identity = waffle.logonUser(username, new String(password)); - } else { - // no domain specified, use default domain - identity = waffle.logonDomainUser(username, defaultDomain, new String(password)); - } - } catch (Win32Exception e) { - logger.error(e.getMessage()); - return null; - } - - if (identity.isGuest() && !settings.getBoolean(Keys.realm.windows.allowGuests, false)) { - logger.warn("Guest account access is disabled"); - identity.dispose(); - return null; - } - - UserModel user = getUserModel(username); - if (user == null) // create user object for new authenticated user - user = new UserModel(username.toLowerCase()); - - // create a user cookie - if (StringUtils.isEmpty(user.cookie) && !ArrayUtils.isEmpty(password)) { - user.cookie = StringUtils.getSHA1(user.username + new String(password)); - } - - // update user attributes from Windows identity - user.accountType = getAccountType(); - String fqn = identity.getFqn(); - if (fqn.indexOf('\\') > -1) { - user.displayName = fqn.substring(fqn.lastIndexOf('\\') + 1); - } else { - user.displayName = fqn; - } - user.password = Constants.EXTERNAL_ACCOUNT; - - Set<String> groupNames = new TreeSet<String>(); - for (IWindowsAccount group : identity.getGroups()) { - groupNames.add(group.getFqn()); - } - - if (groupNames.contains("BUILTIN\\Administrators")) { - // local administrator - user.canAdmin = true; - } - - // TODO consider mapping Windows groups to teams - - // push the changes to the backing user service - super.updateUserModel(user); - - - // cleanup resources - identity.dispose(); - - return user; - } -} +/* + * Copyright 2013 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.auth; + +import java.util.Set; +import java.util.TreeSet; + +import waffle.windows.auth.IWindowsAccount; +import waffle.windows.auth.IWindowsAuthProvider; +import waffle.windows.auth.IWindowsComputer; +import waffle.windows.auth.IWindowsIdentity; +import waffle.windows.auth.impl.WindowsAuthProviderImpl; + +import com.gitblit.Constants; +import com.gitblit.Constants.AccountType; +import com.gitblit.Keys; +import com.gitblit.auth.AuthenticationProvider.UsernamePasswordAuthenticationProvider; +import com.gitblit.models.UserModel; +import com.gitblit.utils.ArrayUtils; +import com.gitblit.utils.StringUtils; +import com.sun.jna.platform.win32.Win32Exception; + +/** + * Implementation of a Windows authentication provider. + * + * @author James Moger + */ +public class WindowsAuthProvider extends UsernamePasswordAuthenticationProvider { + + private IWindowsAuthProvider waffle; + + public WindowsAuthProvider() { + super("windows"); + } + + @Override + public void setup() { + + waffle = new WindowsAuthProviderImpl(); + IWindowsComputer computer = waffle.getCurrentComputer(); + logger.info("Windows Authentication Provider"); + logger.info(" name = " + computer.getComputerName()); + logger.info(" status = " + describeJoinStatus(computer.getJoinStatus())); + logger.info(" memberOf = " + computer.getMemberOf()); + //logger.info(" groups = " + Arrays.asList(computer.getGroups())); + } + + protected String describeJoinStatus(String value) { + if ("NetSetupUnknownStatus".equals(value)) { + return "unknown"; + } else if ("NetSetupUnjoined".equals(value)) { + return "not joined"; + } else if ("NetSetupWorkgroupName".equals(value)) { + return "joined to a workgroup"; + } else if ("NetSetupDomainName".equals(value)) { + return "joined to a domain"; + } + return value; + } + + @Override + public boolean supportsCredentialChanges() { + return false; + } + + @Override + public boolean supportsDisplayNameChanges() { + return false; + } + + @Override + public boolean supportsEmailAddressChanges() { + return true; + } + + @Override + public boolean supportsTeamMembershipChanges() { + return true; + } + + @Override + public AccountType getAccountType() { + return AccountType.WINDOWS; + } + + @Override + public UserModel authenticate(String username, char[] password) { + String defaultDomain = settings.getString(Keys.realm.windows.defaultDomain, null); + if (StringUtils.isEmpty(defaultDomain)) { + // ensure that default domain is null + defaultDomain = null; + } + + if (defaultDomain != null) { + // sanitize username + if (username.startsWith(defaultDomain + "\\")) { + // strip default domain from domain\ username + username = username.substring(defaultDomain.length() + 1); + } else if (username.endsWith("@" + defaultDomain)) { + // strip default domain from username@domain + username = username.substring(0, username.lastIndexOf('@')); + } + } + + IWindowsIdentity identity = null; + try { + if (username.indexOf('@') > -1 || username.indexOf('\\') > -1) { + // manually specified domain + identity = waffle.logonUser(username, new String(password)); + } else { + // no domain specified, use default domain + identity = waffle.logonDomainUser(username, defaultDomain, new String(password)); + } + } catch (Win32Exception e) { + logger.error(e.getMessage()); + return null; + } + + if (identity.isGuest() && !settings.getBoolean(Keys.realm.windows.allowGuests, false)) { + logger.warn("Guest account access is disabled"); + identity.dispose(); + return null; + } + + UserModel user = userManager.getUserModel(username); + if (user == null) // create user object for new authenticated user + user = new UserModel(username.toLowerCase()); + + // create a user cookie + if (StringUtils.isEmpty(user.cookie) && !ArrayUtils.isEmpty(password)) { + user.cookie = StringUtils.getSHA1(user.username + new String(password)); + } + + // update user attributes from Windows identity + user.accountType = getAccountType(); + String fqn = identity.getFqn(); + if (fqn.indexOf('\\') > -1) { + user.displayName = fqn.substring(fqn.lastIndexOf('\\') + 1); + } else { + user.displayName = fqn; + } + user.password = Constants.EXTERNAL_ACCOUNT; + + Set<String> groupNames = new TreeSet<String>(); + for (IWindowsAccount group : identity.getGroups()) { + groupNames.add(group.getFqn()); + } + + if (groupNames.contains("BUILTIN\\Administrators")) { + // local administrator + user.canAdmin = true; + } + + // TODO consider mapping Windows groups to teams + + // push the changes to the backing user service + updateUser(user); + + // cleanup resources + identity.dispose(); + + return user; + } +} -- Gitblit v1.9.1