| /* |
| * Copyright 2000-2013 JetBrains s.r.o. |
| * |
| * 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.intellij.cvsSupport2.connections.ext; |
| |
| import com.intellij.cvsSupport2.config.ExtConfiguration; |
| import com.intellij.cvsSupport2.connections.ConnectionOnProcess; |
| import com.intellij.cvsSupport2.errorHandling.ErrorRegistry; |
| import com.intellij.cvsSupport2.javacvsImpl.io.InputStreamWrapper; |
| import com.intellij.cvsSupport2.javacvsImpl.io.ReadWriteStatistics; |
| import com.intellij.execution.configurations.GeneralCommandLine; |
| import com.intellij.CvsBundle; |
| import com.intellij.openapi.util.text.StringUtil; |
| import org.jetbrains.annotations.NonNls; |
| import org.netbeans.lib.cvsclient.ICvsCommandStopper; |
| import org.netbeans.lib.cvsclient.connection.AuthenticationException; |
| |
| import java.io.IOException; |
| import java.util.StringTokenizer; |
| |
| /** |
| * author: lesya |
| */ |
| public class ExtConnection extends ConnectionOnProcess { |
| |
| @NonNls public static final String DEFAULT_RSH = "ssh"; |
| |
| private final String myHost; |
| private final String myUserName; |
| private final ExtConfiguration myConfiguration; |
| |
| public ExtConnection(String host, String user, String repository, ExtConfiguration configuration, ErrorRegistry errorRegistry) { |
| super(repository, errorRegistry); |
| myHost = host; |
| myUserName = user; |
| myConfiguration = configuration; |
| } |
| |
| |
| public void open() throws AuthenticationException { |
| try { |
| //noinspection HardCodedStringLiteral |
| open(new String[]{"cvs", "server"}, null, null); |
| } |
| catch (IOException e) { |
| throw new AuthenticationException(CvsBundle.message("error.message.cannot.establish.external.connection"), e); |
| } |
| } |
| |
| private void open(String[] commands, String expectedResult, ICvsCommandStopper stopper) |
| throws AuthenticationException, IOException { |
| if (isOpen()) throw new RuntimeException(CvsBundle.message("error.message.connection.already.open")); |
| |
| GeneralCommandLine command = createRshCommand(myHost, myUserName, myConfiguration); |
| command.addParameters(commands); |
| execute(command); |
| |
| if (expectedResult != null) { |
| check(stopper, expectedResult); |
| } |
| } |
| |
| @SuppressWarnings({"IOResourceOpenedButNotSafelyClosed"}) |
| private void check(ICvsCommandStopper stopper, String expectedResult) throws IOException, AuthenticationException { |
| InputStreamWrapper streamWrapper = new InputStreamWrapper(myInputStream, stopper, new ReadWriteStatistics()); |
| try { |
| StringBuilder buffer = new StringBuilder(); |
| while (true) { |
| int i = streamWrapper.read(); |
| if (i == -1 || i == '\n' || i == ' ' || i == '\r') break; |
| buffer.append((char)i); |
| } |
| String read = buffer.toString().trim(); |
| if (!expectedResult.equals(read)) { |
| if (StringUtil.startsWithConcatenation(read, myUserName, "@", myHost)) { |
| throw new AuthenticationException(CvsBundle.message("exception.text.ext.server.rejected.access"), null); |
| } |
| else { |
| if (myErrorText.length() > 0) { |
| throw new AuthenticationException(myErrorText.toString(), null); |
| } |
| else { |
| throw new AuthenticationException(CvsBundle.message("exception.text.ext.cannot.establish.external.connection"), null); |
| } |
| } |
| } |
| } |
| finally { |
| streamWrapper.close(); |
| } |
| } |
| |
| private static GeneralCommandLine createRshCommand(String host, String userName, ExtConfiguration config) { |
| GeneralCommandLine command = new GeneralCommandLine(); |
| command.setExePath(config.CVS_RSH); |
| command.addParameter(host); |
| command.addParameter("-l"); |
| command.addParameter(userName); |
| |
| if (!config.PRIVATE_KEY_FILE.isEmpty()) { |
| command.addParameter("-i"); |
| command.addParameter(config.PRIVATE_KEY_FILE); |
| } |
| |
| if (!config.ADDITIONAL_PARAMETERS.isEmpty()) { |
| StringTokenizer parameters = new StringTokenizer(config.ADDITIONAL_PARAMETERS, " "); |
| while (parameters.hasMoreTokens()) command.addParameter(parameters.nextToken()); |
| } |
| |
| return command; |
| } |
| } |