/* Copyright (c) 2009, Nathan Freitas, Orbot / The Guardian Project - http://openideals.com/guardian */
/* See LICENSE for licensing information */
package org.torproject.android.service;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.StringTokenizer;

import org.sufficientlysecure.rootcommands.Shell;
import org.sufficientlysecure.rootcommands.command.SimpleCommand;
import org.torproject.android.OrbotApp;
import org.torproject.android.OrbotConstants;
import org.torproject.android.Prefs;

import android.annotation.SuppressLint;
import android.annotation.TargetApi;
import android.content.Context;
import android.content.SharedPreferences;
import android.os.Build;
import android.util.Log;

public class TorServiceUtils implements TorServiceConstants {

	
	
	public static int findProcessId(String command) throws IOException 
	{
		int procId = findProcessIdWithPS(command);
		return procId;
	}
	
	//use 'pidof' command
	/**
	public static int findProcessIdWithPidOf(String command) throws Exception
	{
		
		int procId = -1;
		
		Runtime r = Runtime.getRuntime();
		    	
		Process procPs = null;
		
		String baseName = new File(command).getName();
		//fix contributed my mikos on 2010.12.10
		procPs = r.exec(new String[] {SHELL_CMD_PIDOF, baseName});
        //procPs = r.exec(SHELL_CMD_PIDOF);
            
        BufferedReader reader = new BufferedReader(new InputStreamReader(procPs.getInputStream()));
        String line = null;

        while ((line = reader.readLine())!=null)
        {
        
        	try
        	{
        		//this line should just be the process id
        		procId = Integer.parseInt(line.trim());
        		break;
        	}
        	catch (NumberFormatException e)
        	{
        		Log.e("TorServiceUtils","unable to parse process pid: " + line,e);
        	}
        }
            
       
        return procId;

	}
	 * @throws IOException */
	
	//use 'ps' command
	public static int findProcessIdWithPS(String command) throws IOException 
	{
		
		int procId = -1;
		
		Runtime r = Runtime.getRuntime();
		    	
		Process procPs = null;
		
		String processKey = new File(command).getName();
		
        procPs = r.exec(SHELL_CMD_PS + ' ' + processKey); // this is the android ps <name> command
            
        BufferedReader reader = new BufferedReader(new InputStreamReader(procPs.getInputStream()));
        String line = null;
        
        while ((line = reader.readLine())!=null)
        {
        	if (line.contains("PID"))
        		continue;
        		
        	if (line.contains('/' + processKey))
        	{
        		
        		String[] lineParts = line.split("\\s+");
        		
        	    try { 
        	        
        	        procId = Integer.parseInt(lineParts[1]); //for most devices it is the second number
        	    } catch(NumberFormatException e) {
        	    	procId = Integer.parseInt(lineParts[0]); //but for samsungs it is the first
        	        
        	    }
        		
        		
        		break;
        	}
        }
        
        try { procPs.destroy(); } catch (Exception e) {} // try to destroy just to make sure we clean it up
       
        return procId;

	}
	
	public static SharedPreferences getSharedPrefs (Context context)
	{
		if (Build.VERSION.SDK_INT>=Build.VERSION_CODES.HONEYCOMB)
			return context.getSharedPreferences(OrbotConstants.PREF_TOR_SHARED_PREFS,0 | Context.MODE_MULTI_PROCESS);
		else
			return context.getSharedPreferences(OrbotConstants.PREF_TOR_SHARED_PREFS,Context.MODE_PRIVATE);
		
	}
	
	public static void killProcess(File fileProcBin) throws Exception {
        killProcess(fileProcBin, "-9"); // this is -KILL
    }

    public static void killProcess(File fileProcBin, String signal) throws Exception {
        int procId = -1;
        int killAttempts = 0;

        while ((procId = TorServiceUtils.findProcessId(fileProcBin.getCanonicalPath())) != -1) {
            killAttempts++;
            //logNotice("Found " + fileProcBin.getName() + " PID=" + procId + " - killing now...");
            String pidString = String.valueOf(procId);
            /*
             * first try as the normal app user to be safe, then if that fails,
             * try root since the process might be left over from
             * uninstall/reinstall with different UID.
             */
            Shell shell;
            if (Prefs.useRoot() && killAttempts > 2) {
                shell = Shell.startRootShell();
                Log.i(OrbotApp.TAG, "using a root shell");
            } else {
                shell = Shell.startShell();
            }
            shell.add(new SimpleCommand("busybox killall " + signal + " " + fileProcBin.getName()));
            shell.add(new SimpleCommand("toolbox kill " + signal + " " + pidString));
            shell.add(new SimpleCommand("busybox kill " + signal + " " + pidString));
            shell.add(new SimpleCommand("kill " + signal + " " + pidString));
            try {
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                // ignored
            }
            shell.close();
            if (killAttempts > 4)
                throw new Exception("Cannot kill: " + fileProcBin.getAbsolutePath());
        }
    }


}