JNUSTool/src/main/java/de/mas/wiiu/jnus/jnustool/util/Downloader.java

package de.mas.wiiu.jnus.jnustool.util;

import java.io.ByteArrayOutputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;

import de.mas.wiiu.jnus.jnustool.FEntry;
import de.mas.wiiu.jnus.jnustool.Logger;
import de.mas.wiiu.jnus.jnustool.Progress;

public class Downloader {
	private static Downloader instance;
	
	public static Downloader getInstance(){
		if(instance == null){
			instance = new Downloader();
		}
		return instance;		
	}
	private Downloader(){		
		
	}
	
	
	public byte[] downloadAndDecrypt(FEntry toDownload, Progress progressOfFile, boolean asByteArray) throws IOException{
		String URL = URL_BASE + "/" + String.format("%016X", toDownload.getTitleID()) +  "/" + String.format("%08X", toDownload.getNUScontentID());		
		URL url = new URL(URL);
		String [] path = toDownload.getFullPath().split("/");
		boolean decryptWithHash = false;
		if(/*!path[1].equals("code") &&*/ toDownload.isExtractWithHash()){
			decryptWithHash = true;
		}
		HttpURLConnection connection =(HttpURLConnection) url.openConnection();	
		long fileOffset = toDownload.getFileOffset();
		
		if(decryptWithHash){
			int BLOCKSIZE = 0x10000;
			int HASHBLOCKSIZE = 0xFC00;
			fileOffset = ((toDownload.getFileOffset() / HASHBLOCKSIZE) * BLOCKSIZE);
		}

		connection.setRequestProperty("Range", "bytes=" + fileOffset +"-");	
		try{
			connection.connect();
		}catch(Exception e){
			
		}
	    
	    Decryption decryption = new Decryption(toDownload.getTicket());
	    decryption.setProgressListener(progressOfFile);
	    InputStream input = connection.getInputStream();
	    OutputStream outputStream;
	    
	    if(!asByteArray){
	    	outputStream = new FileOutputStream(toDownload.getTargetPath() +"/" + toDownload.getFullPath().substring(1, toDownload.getFullPath().length()));
	    }else{
	    	outputStream = new ByteArrayOutputStream((int) toDownload.getFileLength());
	    }
	    
    	if(!decryptWithHash){
	        decryption.decryptFile(input, outputStream, toDownload);
	    }else{
	        byte[] h3 = toDownload.getH3();
	    	decryption.decryptFileHash(input, outputStream, toDownload,h3);
	    }    		
        connection.disconnect();
        if(asByteArray){
        	return ((ByteArrayOutputStream)outputStream).toByteArray();
        }else{
        	return null;
        }
	}	
	
	public static String URL_BASE = "";

	public void downloadTMD(long titleID,int version,String path) throws IOException {
		String version_suf = "";
		if(version > 0) version_suf = "." + version;
		String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/tmd" + version_suf;
		downloadFile(URL, "title.tmd",path,null);
		if(Settings.DL_ALL_VERSIONS) downloadFile(URL, "title.tmd"+version_suf,path,null);
	}
	
	public void downloadFile(String fileURL,String filename,String tmpPath, Progress progress) throws IOException{
		int BUFFER_SIZE = 0x800;
		URL url = new URL(fileURL);
	    HttpURLConnection httpConn = (HttpURLConnection) url.openConnection();
	
        InputStream inputStream = httpConn.getInputStream();
        if(tmpPath != null){
        	filename = tmpPath + "/" + filename;
        }
        
        FileOutputStream outputStream = new FileOutputStream(filename);
        if(progress != null){
        	progress.resetCurrent();
        }
        int bytesRead = -1;
        byte[] buffer = new byte[BUFFER_SIZE];
        while ((bytesRead = inputStream.read(buffer)) != -1) {
            outputStream.write(buffer, 0, bytesRead);
            if(progress != null){
            	progress.addCurrent(bytesRead);
            }
        }

        outputStream.close();
        inputStream.close();
	  
	    httpConn.disconnect();
	}
	
	public void downloadFile(String fileURL,String filename) throws IOException{
		downloadFile(fileURL, filename,null,null);
	}
	
	public void downloadTicket(long titleID,String path) throws IOException {
		String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/cetk";
		downloadFile(URL, "title.tik",path,null);
	}
	
	public void downloadContent(long titleID,int contentID,Progress progress) throws IOException {
		downloadContent(titleID,contentID, null,progress);
	}
	
	public byte[] downloadContentToByteArray(long titleID,int contentID) throws IOException {
		String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/" + String.format("%08X", contentID);
		return downloadFileToByteArray(URL);		
	}
	
	public byte[] downloadTMDToByteArray(long titleID, int version) throws IOException {
		String version_suf = "";
		if(version > 0) version_suf = "." + version;
		String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/tmd" +version_suf;
		return downloadFileToByteArray(URL);
	}
	
	public byte[] downloadFileToByteArray(String fileURL) throws IOException {
		
		int BUFFER_SIZE = 0x800;		
		URL url = new URL(fileURL);
	    HttpURLConnection httpConn = (HttpURLConnection) url.openConnection();
	    int responseCode = httpConn.getResponseCode();
	    
        // always check HTTP response code first
	    byte[] file = null;
	    
        if (responseCode == HttpURLConnection.HTTP_OK) {
		    int contentLength = httpConn.getContentLength();
		    
		    file = new byte[contentLength];
		    // always check HTTP response code first
	        
	        InputStream inputStream = httpConn.getInputStream();
	
	        int bytesRead = -1;
	        byte[] buffer = new byte[BUFFER_SIZE];
	        int filePostion = 0;
	        while ((bytesRead = inputStream.read(buffer)) != -1) {
	        	System.arraycopy(buffer, 0, file, filePostion,bytesRead);
	        	filePostion+=bytesRead;
	    		
	        }
	        inputStream.close();
        }else{
        	Logger.log("File not found: " + fileURL);
        }
	    httpConn.disconnect();
	    return file;
	}
	
public byte[] downloadHTTPSFileToByteArray(String fileURL) throws IOException {
   
        int BUFFER_SIZE = 0x800;        
        URL url = new URL(fileURL);
        HttpURLConnection httpConn = (HttpURLConnection) url.openConnection();
        int responseCode = httpConn.getResponseCode();
        
        // always check HTTP response code first
        byte[] file = null;
        
        if (responseCode == HttpURLConnection.HTTP_OK) {
            int contentLength = httpConn.getContentLength();
            
            file = new byte[contentLength];
            // always check HTTP response code first
            
            InputStream inputStream = httpConn.getInputStream();
    
            int bytesRead = -1;
            byte[] buffer = new byte[BUFFER_SIZE];
            int filePostion = 0;
            while ((bytesRead = inputStream.read(buffer)) != -1) {
                System.arraycopy(buffer, 0, file, filePostion,bytesRead);
                filePostion+=bytesRead;
                
            }
            inputStream.close();
        }else{
            Logger.log("File not found: " + fileURL);
        }
        httpConn.disconnect();
        return file;
    }
	
	
	
	public byte[] downloadTicketToByteArray(long titleID) throws IOException {
		String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/cetk";
		return downloadFileToByteArray(URL);
	}
	public void downloadContent(long titleID,int contentID, String tmpPath, Progress progress) throws IOException {
		String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/" + String.format("%08X", contentID);
		downloadFile(URL, String.format("%08x", contentID) +".app",tmpPath,progress);
		
	}
	public void downloadContentH3(long titleID, int contentID, String tmpPath, Progress progress) throws IOException {
		String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/" + String.format("%08X", contentID) + ".h3";
		downloadFile(URL, String.format("%08X", contentID) +".h3",tmpPath,progress);
	}
	
	public byte[] downloadContentH3AsByte(long titleID, int contentID) throws IOException {
        String URL = URL_BASE + "/" + String.format("%016X", titleID) +  "/" + String.format("%08X", contentID) + ".h3";
        return downloadFileToByteArray(URL);
    }
	
	
}