/*
* Created on 4 Oct 2006
* Created by Paul Gardner
* Copyright (C) 2006 Aelitis, All Rights Reserved.
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
* This program 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 this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
*
* AELITIS, SAS au capital de 63.529,40 euros
* 8 Allee Lenotre, La Grille Royale, 78600 Le Mesnil le Roi, France.
*
*/
package com.aelitis.azureus.core.networkmanager.impl.http;
import java.io.IOException;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.StringTokenizer;
import org.gudy.azureus2.core3.disk.DiskManager;
import org.gudy.azureus2.core3.disk.DiskManagerFileInfo;
import org.gudy.azureus2.core3.internat.MessageText;
import org.gudy.azureus2.core3.peer.impl.PEPeerControl;
import org.gudy.azureus2.core3.peer.impl.PEPeerTransport;
import org.gudy.azureus2.core3.torrent.TOTorrent;
import org.gudy.azureus2.core3.torrent.TOTorrentFile;
import org.gudy.azureus2.core3.util.Debug;
import com.aelitis.azureus.core.networkmanager.NetworkConnection;
import com.aelitis.azureus.core.util.HTTPUtils;
public class
HTTPNetworkConnectionFile
extends HTTPNetworkConnection
{
private boolean switching;
protected
HTTPNetworkConnectionFile(
HTTPNetworkManager _manager,
NetworkConnection _connection,
PEPeerTransport _peer )
{
super( _manager, _connection, _peer );
}
protected void
decodeHeader(
HTTPMessageDecoder decoder,
final String header )
throws IOException
{
if ( switching ){
Debug.out( "new header received while paused" );
throw( new IOException( "Bork" ));
}
if ( !isSeed()){
return;
}
PEPeerControl control = getPeerControl();
DiskManager dm = control.getDiskManager();
if ( dm == null ){
Debug.out( "Disk manager is null" );
throw( new IOException( "Disk manager unavailable" ));
}
TOTorrent to_torrent = dm.getTorrent();
char[] chars = header.toCharArray();
int last_pos = 0;
int line_num = 0;
String target_str = null;
DiskManagerFileInfo target_file = null;
long file_offset = 0;
List<long[]> ranges = new ArrayList<long[]>();
boolean keep_alive = false;
for (int i=1;i<chars.length;i++){
if ( chars[i-1] == '\r' && chars[i] == '\n' ){
String line = new String( chars, last_pos, i - last_pos ).trim();
last_pos = i;
line_num++;
// System.out.println( "line " + line_num + " -> " + line );
if ( line_num == 1 ){
line = line.substring( line.indexOf( "files/" ) + 6 );
int hash_end = line.indexOf( "/" );
final byte[] old_hash = control.getHash();
final byte[] new_hash = URLDecoder.decode(line.substring(0, hash_end), "ISO-8859-1").getBytes( "ISO-8859-1" );
if ( !Arrays.equals( new_hash, old_hash )){
switching = true;
decoder.pauseInternally();
flushRequests(
new flushListener()
{
private boolean triggered;
public void
flushed()
{
synchronized( this ){
if ( triggered ){
return;
}
triggered = true;
}
getManager().reRoute(
HTTPNetworkConnectionFile.this,
old_hash, new_hash, header );
}
});
return;
}
line = line.substring( hash_end + 1 );
line = line.substring( 0, line.lastIndexOf( ' ' ));
String file = line;
if ( to_torrent.isSimpleTorrent()){
// optimise for simple torrents. also support the case where
// client has the hash but doesn't know the file name
target_file = dm.getFiles()[0];
}else{
target_str = file;
StringTokenizer tok = new StringTokenizer( file, "/" );
List<byte[]> bits = new ArrayList<byte[]>();
while( tok.hasMoreTokens()){
bits.add( URLDecoder.decode(tok.nextToken(), "ISO-8859-1").getBytes( "ISO-8859-1" ));
}
// latest spec has torrent file name encoded first for non-simple torrents
// remove it if we find it so we have some backward compat
if ( !to_torrent.isSimpleTorrent() && bits.size() > 1 ){
if ( Arrays.equals( to_torrent.getName(), (byte[])bits.get(0))){
bits.remove(0);
}
}
DiskManagerFileInfo[] files = dm.getFiles();
file_offset = 0;
for (int j=0;j<files.length;j++){
TOTorrentFile torrent_file = files[j].getTorrentFile();
byte[][] comps = torrent_file.getPathComponents();
if ( comps.length == bits.size()){
boolean match = true;
for (int k=0;k<comps.length;k++){
if ( !Arrays.equals( comps[k], (byte[])bits.get(k))){
match = false;
break;
}
}
if ( match ){
target_file = files[j];
break;
}
}
file_offset += torrent_file.getLength();
}
}
}else{
line = line.toLowerCase( MessageText.LOCALE_ENGLISH );
if ( line.startsWith( "range" ) && target_file != null ){
line = line.substring(5).trim();
if ( line.startsWith(":" )){
String range_str = line.substring(1).trim();
if ( range_str.startsWith( "bytes=" )){
long file_length = target_file.getLength();
StringTokenizer tok2 = new StringTokenizer( range_str.substring( 6 ), "," );
while( tok2.hasMoreTokens()){
String range = tok2.nextToken();
try{
int pos = range.indexOf('-');
if ( pos != -1 ){
String lhs = range.substring( 0,pos);
String rhs = range.substring(pos+1);
long start;
long end;
if ( lhs.length() == 0 ){
// -222 is last 222 bytes of file
end = file_length - 1;
start = file_length - Long.parseLong( rhs );
}else if ( rhs.length() == 0 ){
end = file_length - 1;
start = Long.parseLong( lhs );
}else{
start = Long.parseLong( lhs );
end = Long.parseLong( rhs );
}
ranges.add( new long[]{ start, end });
}
}catch( Throwable e ){
}
}
}
if ( ranges.size() == 0 ){
log( "Invalid range specification: '" + line + "'" );
sendAndClose( getManager().getRangeNotSatisfiable());
return;
}
}
}else if ( line.indexOf( "keep-alive" ) != -1 ){
keep_alive = true;
}
}
}
}
if ( target_file == null ){
log( "Failed to find file '" + target_str + "'" );
sendAndClose( getManager().getNotFound());
return;
}
try{
String name = target_file.getFile( true ).getName();
int pos = name.lastIndexOf( "." );
if ( pos != -1 ){
setContentType( HTTPUtils.guessContentTypeFromFileType( name.substring( pos+1 )));
}
}catch( Throwable e ){
}
long file_length = target_file.getLength();
boolean partial_content = ranges.size() > 0;
if ( !partial_content ){
ranges.add( new long[]{ 0, file_length - 1});
}
long[] offsets = new long[ranges.size()];
long[] lengths = new long[ranges.size()];
for (int i=0;i<ranges.size();i++){
long[] range = (long[])ranges.get(i);
long start = range[0];
long end = range[1];
if ( start < 0 || start >= file_length ||
end < 0 || end >= file_length ||
start > end ){
log( "Invalid range specification: '" + start + "-" + end + "'" );
sendAndClose( getManager().getRangeNotSatisfiable());
return;
}
offsets[i] = file_offset + start;
lengths[i] = ( end - start ) + 1;
}
addRequest( new httpRequest( offsets, lengths, file_length, partial_content, keep_alive ));
}
}