HttpResponse.java

/**
 * Copyright 2003-2016 SSHTOOLS Limited. All Rights Reserved.
 *
 * For product documentation visit https://www.sshtools.com/
 *
 * This file is part of J2SSH Maverick.
 *
 * J2SSH Maverick is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * J2SSH Maverick is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with J2SSH Maverick.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.sshtools.net;

import java.io.IOException;
import java.io.InputStream;
import java.util.NoSuchElementException;
import java.util.StringTokenizer;

/**
 * Utility class to process HTTP responses.
 * 
 * @author Lee David Painter
 * 
 */
public class HttpResponse extends HttpHeader {
	private String version;
	private int status;
	private String reason;

	public HttpResponse(InputStream input) throws IOException {
		begin = readLine(input);

		while (begin.trim().length() == 0) {
			begin = readLine(input);
		}

		processResponse();
		processHeaderFields(input);
	}

	public String getVersion() {
		return version;
	}

	public int getStatus() {
		return status;
	}

	public String getReason() {
		return reason;
	}

	private void processResponse() throws IOException {
		StringTokenizer tokens = new StringTokenizer(begin, white_SPACE, false);

		try {
			version = tokens.nextToken();
			status = Integer.parseInt(tokens.nextToken());
			reason = tokens.nextToken();
		} catch (NoSuchElementException e) {
			throw new IOException("Failed to read HTTP repsonse header");
		} catch (NumberFormatException e) {
			throw new IOException("Failed to read HTTP resposne header");
		}
	}

	public String getAuthenticationMethod() {
		String auth = getHeaderField("Proxy-Authenticate");
		String method = null;

		if (auth != null) {
			int n = auth.indexOf(' ');
			method = auth.substring(0, n);
		}

		return method;
	}

	public String getAuthenticationRealm() {
		String auth = getHeaderField("Proxy-Authenticate");
		String realm = "";

		if (auth != null) {
			int l;
			int r = auth.indexOf('=');

			while (r >= 0) {
				l = auth.lastIndexOf(' ', r);
				if (l > -1) {
					String val = auth.substring(l + 1, r);

					if (val.equalsIgnoreCase("realm")) {
						l = r + 2;
						r = auth.indexOf('"', l);
						realm = auth.substring(l, r);

						break;
					}

					r = auth.indexOf('=', r + 1);
				}
			}
		}

		return realm;
	}

}