/*
* Created on Apr 16, 2004
* Created by Paul Gardner
* Copyright (C) 2004, 2005, 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 46,603.30 euros
* 8 Allee Lenotre, La Grille Royale, 78600 Le Mesnil le Roi, France.
*
*/
package org.gudy.azureus2.platform.win32.access.impl;
/**
* @author parg
*
*/
import java.io.File;
import java.net.InetAddress;
import java.net.UnknownHostException;
import java.util.*;
// don't use any core stuff in here as we need this access stub to be able to run in isolation
import org.gudy.azureus2.core3.config.COConfigurationManager;
import org.gudy.azureus2.core3.util.Constants;
import org.gudy.azureus2.core3.util.Debug;
import org.gudy.azureus2.platform.PlatformManagerPingCallback;
import org.gudy.azureus2.platform.win32.access.*;
import com.aelitis.azureus.util.MapUtils;
public class
AEWin32AccessImpl
implements AEWin32Access, AEWin32AccessCallback
{
protected static AEWin32AccessImpl singleton;
public static synchronized AEWin32Access
getSingleton(
boolean fully_initialise )
{
if ( singleton == null ){
singleton = new AEWin32AccessImpl(fully_initialise);
}
return( singleton );
}
private boolean fully_initialise;
private int trace_id_next = new Random().nextInt();
private List listeners = new ArrayList();
protected
AEWin32AccessImpl(
boolean _fully_initialise )
{
fully_initialise = _fully_initialise;
if ( isEnabled()){
AEWin32AccessInterface.load( this, fully_initialise );
}
}
public boolean
isEnabled()
{
return( AEWin32AccessInterface.isEnabled( fully_initialise ));
}
public long
windowsMessage(
int msg,
int param1,
long param2 )
{
int type = -1;
if ( msg == AEWin32AccessInterface.WM_ENDSESSION ){
type = AEWin32AccessListener.ET_SHUTDOWN;
}else if ( msg == AEWin32AccessInterface.WM_POWERBROADCAST ){
if ( param1 == AEWin32AccessInterface.PBT_APMQUERYSUSPEND ){
type = AEWin32AccessListener.ET_SUSPEND;
}else if ( param1 == AEWin32AccessInterface.PBT_APMRESUMESUSPEND ){
type = AEWin32AccessListener.ET_RESUME;
}
}
int result = -1;
if ( type != -1 ){
for (int i=0;i<listeners.size();i++){
try{
int temp = ((AEWin32AccessListener)listeners.get(i)).eventOccurred( type );
if ( temp != result ){
if ( result != -1 ){
Debug.out( "Incompatible results received: " + result + "/" + temp );
}
result = temp;
}
}catch( Throwable e ){
e.printStackTrace();
}
}
}
if ( result == AEWin32AccessListener.RT_SUSPEND_DENY ){
if (( param2 & 0x0001 ) != 0 ){
return( AEWin32AccessInterface.BROADCAST_QUERY_DENY );
}else{
Debug.out( "Ignoring suspend deny request as not permitted" );
}
}
return( result );
}
public long
generalMessage(
String str )
{
return( 0 );
}
public String
getVersion()
{
return( AEWin32AccessInterface.getVersion());
}
public String
readStringValue(
int type,
String subkey,
String value_name )
throws AEWin32AccessException
{
return( AEWin32AccessInterface.readStringValue( type, subkey, value_name ));
}
public void
writeStringValue(
int type,
String subkey,
String value_name,
String value_value )
throws AEWin32AccessException
{
AEWin32AccessInterface.writeStringValue( type, subkey, value_name, value_value );
}
public int
readWordValue(
int type,
String subkey,
String value_name )
throws AEWin32AccessException
{
return( AEWin32AccessInterface.readWordValue( type, subkey, value_name ));
}
public void
writeWordValue(
int type,
String subkey,
String value_name,
int value_value )
throws AEWin32AccessException
{
AEWin32AccessInterface.writeWordValue( type, subkey, value_name, value_value );
}
public void
deleteKey(
int type,
String subkey )
throws AEWin32AccessException
{
deleteKey( type, subkey, false );
}
public void
deleteKey(
int type,
String subkey,
boolean recursive )
throws AEWin32AccessException
{
AEWin32AccessInterface.deleteKey( type, subkey, recursive );
}
public void
deleteValue(
int type,
String subkey,
String value_name )
throws AEWin32AccessException
{
AEWin32AccessInterface.deleteValue( type, subkey, value_name );
}
public String
getUserAppData()
throws AEWin32AccessException
{
String app_data_key = "software\\microsoft\\windows\\currentversion\\explorer\\shell folders";
String app_data_name = "appdata";
return( readStringValue(
HKEY_CURRENT_USER,
app_data_key,
app_data_name ));
}
public String
getCommonAppData()
throws AEWin32AccessException
{
String app_data_key = "software\\microsoft\\windows\\currentversion\\explorer\\shell folders";
String app_data_name = "Common AppData";
return( readStringValue(
HKEY_LOCAL_MACHINE,
app_data_key,
app_data_name ));
}
public String
getLocalAppData()
throws AEWin32AccessException
{
String app_data_key = "software\\microsoft\\windows\\currentversion\\explorer\\shell folders";
String app_data_name = "Local AppData";
return( readStringValue(
HKEY_CURRENT_USER,
app_data_key,
app_data_name ));
}
public String
getUserDocumentsDir()
throws AEWin32AccessException
{
String app_data_key = "software\\microsoft\\windows\\currentversion\\explorer\\shell folders";
String app_data_name = "personal";
return( readStringValue(
HKEY_CURRENT_USER,
app_data_key,
app_data_name ));
}
public String
getUserMusicDir()
throws AEWin32AccessException
{
String app_data_key = "software\\microsoft\\windows\\currentversion\\explorer\\shell folders";
String app_data_name = "my music";
try {
return( readStringValue(
HKEY_CURRENT_USER,
app_data_key,
app_data_name ));
} catch (AEWin32AccessException e) {
// Win98 doesn't have it
String s = getUserDocumentsDir();
if (s != null) {
s += "\\My Music";
}
return s;
}
}
public String
getUserVideoDir()
throws AEWin32AccessException
{
String app_data_key = "software\\microsoft\\windows\\currentversion\\explorer\\shell folders";
String app_data_name = "my video";
try {
return( readStringValue(
HKEY_CURRENT_USER,
app_data_key,
app_data_name ));
} catch (AEWin32AccessException e) {
// Win98 doesn't have it
String s = getUserDocumentsDir();
if (s != null) {
s += "\\My Video";
}
return s;
}
}
public String
getProgramFilesDir()
throws AEWin32AccessException
{
String app_data_key = "software\\microsoft\\windows\\currentversion";
String app_data_name = "ProgramFilesDir";
return( readStringValue(
HKEY_LOCAL_MACHINE,
app_data_key,
app_data_name ));
}
public String
getApplicationInstallDir(
String app_name )
throws AEWin32AccessException
{
String res = "";
try{
res = readStringValue(
HKEY_CURRENT_USER,
"software\\" + app_name,
null );
}catch( AEWin32AccessException e ){
res = readStringValue(
HKEY_LOCAL_MACHINE,
"software\\" + app_name,
null );
}
return( res );
}
public void
createProcess(
String command_line,
boolean inherit_handles )
throws AEWin32AccessException
{
AEWin32AccessInterface.createProcess( command_line, inherit_handles );
}
public void
moveToRecycleBin(
String file_name )
throws AEWin32AccessException
{
AEWin32AccessInterface.moveToRecycleBin( file_name );
}
public void
copyFilePermissions(
String from_file_name,
String to_file_name )
throws AEWin32AccessException
{
AEWin32AccessInterface.copyPermission( from_file_name, to_file_name );
}
public boolean
testNativeAvailability(
String name )
throws AEWin32AccessException
{
return( AEWin32AccessInterface.testNativeAvailability( name ));
}
public int shellExecute(String operation, String file, String parameters,
String directory, int SW_const) throws AEWin32AccessException {
return AEWin32AccessInterface.shellExecute(operation, file, parameters,
directory, SW_const);
}
public int
shellExecuteAndWait(
String file,
String params )
throws AEWin32AccessException
{
return( AEWin32AccessInterface.shellExecuteAndWait(file, params ));
}
public void
traceRoute(
InetAddress source_address,
InetAddress target_address,
final PlatformManagerPingCallback callback )
throws AEWin32AccessException
{
traceRoute( source_address, target_address, false, callback );
}
public void
ping(
InetAddress source_address,
InetAddress target_address,
final PlatformManagerPingCallback callback )
throws AEWin32AccessException
{
if ( Constants.compareVersions( getVersion(), "1.15" ) < 0 ){
throw( new AEWin32AccessException( "Sorry, ping is broken in versions < 1.15" ));
}
traceRoute( source_address, target_address, true, callback );
}
protected void
traceRoute(
InetAddress source_address,
InetAddress target_address,
boolean ping_mode,
PlatformManagerPingCallback callback )
throws AEWin32AccessException
{
int trace_id;
synchronized( this ){
trace_id = trace_id_next++;
}
AEWin32AccessCallback cb = new traceRouteCallback( ping_mode, callback );
AEWin32AccessInterface.traceRoute(
trace_id,
addressToInt( source_address ),
addressToInt( target_address ),
ping_mode?1:0,
cb );
}
private int
addressToInt(
InetAddress address )
{
byte[] bytes = address.getAddress();
int resp = (bytes[0]<<24)&0xff000000 | (bytes[1] << 16)&0x00ff0000 | (bytes[2] << 8)&0x0000ff00 | bytes[3]&0x000000ff;
return( resp );
}
private InetAddress
intToAddress(
int address )
{
byte[] bytes = { (byte)(address>>24), (byte)(address>>16),(byte)(address>>8),(byte)address };
try{
InetAddress res = InetAddress.getByAddress(bytes);
return( res );
}catch( UnknownHostException e ){
return( null );
}
}
public void
addListener(
AEWin32AccessListener listener )
{
listeners.add( listener );
}
public void
removeListener(
AEWin32AccessListener listener )
{
listeners.remove( listener );
}
protected class
traceRouteCallback
implements AEWin32AccessCallback
{
private boolean ping_mode;
private PlatformManagerPingCallback cb;
protected
traceRouteCallback(
boolean _ping_mode,
PlatformManagerPingCallback _cb )
{
ping_mode = _ping_mode;
cb = _cb;
}
public long
windowsMessage(
int msg,
int param1,
long param2 )
{
return(0);
}
public long
generalMessage(
String msg )
{
StringTokenizer tok = new StringTokenizer( msg, "," );
int ttl = Integer.parseInt( tok.nextToken().trim());
int time = -1;
InetAddress address;
if ( tok.hasMoreTokens()){
int i_addr = Integer.parseInt( tok.nextToken().trim());
address = intToAddress( i_addr );
time = Integer.parseInt( tok.nextToken().trim());
//boolean is_udp = Integer.parseInt( tok.nextToken().trim()) == 1;
// System.out.println( "udp = " + is_udp );
}else{
address = null;
}
return( cb.reportNode( ping_mode?-1:ttl, address, time )?1:0 );
}
}
public Map<File, Map> getAllDrives() {
// gah, see http://pluscs.vuze.com/tickets/23804?col=1101604&page=1 we sometimes get crashes here
String state_key = "awein32.getalldrives.state.2";
int state = COConfigurationManager.getIntParameter( state_key, 0 );
if ( state == 1 ){
Debug.out( "Not enumerating system drives as it crashed last time we tried" );
return( new HashMap<File,Map>());
}
try{
COConfigurationManager.setParameter( state_key, 1 );
COConfigurationManager.save();
Map<File, Map> mapDrives = new HashMap<File, Map>();
try {
List availableDrives = AEWin32AccessInterface.getAvailableDrives();
if (availableDrives != null) {
for (Object object : availableDrives) {
File f = (File) object;
Map driveInfo = AEWin32AccessInterface.getDriveInfo(f.getPath().charAt(0));
boolean isWritableUSB = AEWin32Manager.getAccessor(false).isUSBDrive(driveInfo);
driveInfo.put("isWritableUSB", isWritableUSB);
mapDrives.put(f, driveInfo);
}
}
return mapDrives;
} catch (UnsatisfiedLinkError ue) {
Debug.outNoStack("Old aereg.dll");
} catch (Throwable e) {
Debug.out(e);
}
return Collections.emptyMap();
}finally{
COConfigurationManager.setParameter( state_key, 2 );
COConfigurationManager.setDirty();
}
}
public boolean isUSBDrive(Map driveInfo) {
if (driveInfo == null) {
return false;
}
boolean removeable = MapUtils.getMapBoolean(driveInfo, "Removable", false);
// values GetDriveType: {UNKNOWN, NO_ROOT_DIR, REMOVABLE, FIXED, REMOTE, CDROM, RAMDISK}
long driveType = MapUtils.getMapLong(driveInfo, "DriveType", 0);
// STORAGE_BUS_TYPE: http://msdn.microsoft.com/en-us/library/aa363465%28VS.85%29.aspx
long busType = MapUtils.getMapLong(driveInfo, "BusType", 0);
// MEDIA_TYPE: http://msdn.microsoft.com/en-us/library/aa365231%28VS.85%29.aspx
long mediaType = MapUtils.getMapLong(driveInfo, "MediaType", -1);
if (removeable && driveType == 2 && busType == 7
&& (mediaType == 11 || mediaType == -1)) {
return true;
}
return false;
}
public void
setThreadExecutionState(
int state )
{
AEWin32AccessInterface.setThreadExecutionState( state );
}
}