donaldp 02/02/14 02:36:30
Modified:
proposal/myrmidon/src/main/org/apache/tools/ant/taskdefs/condition
ConditionBase.java
Added: proposal/myrmidon/src/java/org/apache/antlib/build
Checksum.java
Removed: proposal/myrmidon/src/main/org/apache/tools/ant/taskdefs
Checksum.java
Log:
Move CheckSum into antlib.build package
Revision Changes Path
1.8 +2 -2
jakarta-ant/proposal/myrmidon/src/main/org/apache/tools/ant/taskdefs/condition/ConditionBase.java
Index: ConditionBase.java
===================================================================
RCS file:
/home/cvs/jakarta-ant/proposal/myrmidon/src/main/org/apache/tools/ant/taskdefs/condition/ConditionBase.java,v
retrieving revision 1.7
retrieving revision 1.8
diff -u -r1.7 -r1.8
--- ConditionBase.java 12 Jan 2002 04:55:34 -0000 1.7
+++ ConditionBase.java 14 Feb 2002 10:36:30 -0000 1.8
@@ -12,7 +12,7 @@
import java.util.NoSuchElementException;
import org.apache.tools.ant.ProjectComponent;
import org.apache.tools.ant.taskdefs.Available;
-import org.apache.tools.ant.taskdefs.Checksum;
+import org.apache.antlib.build.Checksum;
import org.apache.tools.ant.taskdefs.UpToDate;
/**
@@ -21,7 +21,7 @@
* conditions are in sync.
*
* @author <a href="mailto:[EMAIL PROTECTED]">Stefan Bodewig</a>
- * @version $Revision: 1.7 $
+ * @version $Revision: 1.8 $
*/
public abstract class ConditionBase
extends ProjectComponent
1.1
jakarta-ant/proposal/myrmidon/src/java/org/apache/antlib/build/Checksum.java
Index: Checksum.java
===================================================================
/*
* Copyright (C) The Apache Software Foundation. All rights reserved.
*
* This software is published under the terms of the Apache Software License
* version 1.1, a copy of which has been included with this distribution in
* the LICENSE.txt file.
*/
package org.apache.antlib.build;
import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.security.DigestInputStream;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.security.NoSuchProviderException;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.Hashtable;
import org.apache.avalon.excalibur.io.IOUtil;
import org.apache.myrmidon.api.TaskException;
import org.apache.tools.ant.taskdefs.condition.Condition;
import org.apache.tools.ant.taskdefs.MatchingTask;
import org.apache.tools.ant.types.DirectoryScanner;
import org.apache.tools.ant.types.FileSet;
import org.apache.tools.ant.types.ScannerUtil;
/**
* This task can be used to create checksums for files. It can also be used to
* verify checksums.
*
* @author <a href="mailto:[EMAIL PROTECTED]">Magesh Umasankar</a>
*/
public class Checksum
extends MatchingTask
implements Condition
{
/**
* File for which checksum is to be calculated.
*/
private File m_file;
/**
* MessageDigest algorithm to be used.
*/
private String m_algorithm = "MD5";
/**
* MessageDigest Algorithm provider
*/
private String m_provider;
/**
* ArrayList to hold source file sets.
*/
private ArrayList m_filesets = new ArrayList();
/**
* Stores SourceFile, DestFile pairs and SourceFile, Property String
pairs.
*/
private Hashtable m_includeFileMap = new Hashtable();
/**
* File Extension that is be to used to create or identify destination
file
*/
private String m_fileext;
/**
* Create new destination file? Defaults to false.
*/
private boolean m_forceOverwrite;
/**
* is this task being used as a nested condition element?
*/
private boolean m_isCondition;
/**
* Message Digest instance
*/
private MessageDigest m_messageDigest;
/**
* Holds generated checksum and gets set as a Project Property.
*/
private String m_property;
/**
* Contains the result of a checksum verification. ("true" or "false")
*/
private String m_verifyProperty;
/**
* Sets the MessageDigest algorithm to be used to calculate the checksum.
*
* @param algorithm The new Algorithm value
*/
public void setAlgorithm( final String algorithm )
{
m_algorithm = algorithm;
}
/**
* Sets the file for which the checksum is to be calculated.
*
* @param file The new File value
*/
public void setFile( final File file )
{
m_file = file;
}
/**
* Sets the File Extension that is be to used to create or identify
* destination file
*
* @param fileext The new Fileext value
*/
public void setFileext( final String fileext )
{
m_fileext = fileext;
}
/**
* Overwrite existing file irrespective of whether it is newer than the
* source file? Defaults to false.
*
* @param forceOverwrite The new ForceOverwrite value
*/
public void setForceOverwrite( boolean forceOverwrite )
{
this.m_forceOverwrite = forceOverwrite;
}
/**
* Sets the property to hold the generated checksum
*
* @param property The new Property value
*/
public void setProperty( String property )
{
this.m_property = property;
}
/**
* Sets the MessageDigest algorithm provider to be used to calculate the
* checksum.
*
* @param provider The new Provider value
*/
public void setProvider( final String provider )
{
m_provider = provider;
}
/**
* Sets verify property. This project property holds the result of a
* checksum verification - "true" or "false"
*
* @param verifyProperty The new Verifyproperty value
*/
public void setVerifyproperty( final String verifyProperty )
{
m_verifyProperty = verifyProperty;
}
/**
* Adds a set of files (nested fileset attribute).
*
* @param set The feature to be added to the Fileset attribute
*/
public void addFileset( final FileSet set )
{
m_filesets.add( set );
}
/**
* Calculate the checksum(s)
*
* @return Returns true if the checksum verification test passed, false
* otherwise.
*/
public boolean eval()
throws TaskException
{
m_isCondition = true;
return validateAndExecute();
}
/**
* Calculate the checksum(s).
*
* @exception TaskException Description of Exception
*/
public void execute()
throws TaskException
{
final boolean value = validateAndExecute();
if( m_verifyProperty != null )
{
final String name = m_verifyProperty;
final Object value1 = new Boolean( value ).toString();
getContext().setProperty( name, value1 );
}
}
/**
* Add key-value pair to the hashtable upon which to later operate upon.
*
* @param file The feature to be added to the ToIncludeFileMap attribute
* @exception TaskException Description of Exception
*/
private void addToIncludeFileMap( final File file )
throws TaskException
{
if( file != null )
{
if( file.exists() )
{
if( m_property == null )
{
final File dest = new File( file.getParent(),
file.getName() + m_fileext );
if( m_forceOverwrite || m_isCondition ||
( file.lastModified() > dest.lastModified() ) )
{
m_includeFileMap.put( file, dest );
}
else
{
final String message = file + " omitted as " + dest +
" is up to date.";
getLogger().debug( message );
}
}
else
{
m_includeFileMap.put( file, m_property );
}
}
else
{
final String message = "Could not find file " +
file.getAbsolutePath() +
" to generate checksum for.";
getLogger().info( message );
throw new TaskException( message );
}
}
}
/**
* Generate checksum(s) using the message digest created earlier.
*/
private boolean generateChecksums()
throws TaskException
{
boolean checksumMatches = true;
final Enumeration includes = m_includeFileMap.keys();
while( includes.hasMoreElements() )
{
final File src = (File)includes.nextElement();
if( !m_isCondition )
{
final String message = "Calculating " + m_algorithm + "
checksum for " + src;
getLogger().info( message );
}
checksumMatches = z( src, checksumMatches );
}
return checksumMatches;
}
private boolean z( final File src, boolean checksumMatches )
throws TaskException
{
FileInputStream fis = null;
FileOutputStream fos = null;
try
{
fis = new FileInputStream( src );
final byte[] fileDigest = buildDigest( fis );
IOUtil.shutdownStream( fis );
String checksum = "";
for( int i = 0; i < fileDigest.length; i++ )
{
final String hexStr = Integer.toHexString( 0x00ff &
fileDigest[ i ] );
if( hexStr.length() < 2 )
{
checksum += "0";
}
checksum += hexStr;
}
//can either be a property name string or a file
Object destination = m_includeFileMap.get( src );
if( destination instanceof String )
{
final String prop = (String)destination;
if( m_isCondition )
{
checksumMatches = checksum.equals( m_property );
}
else
{
final Object value = checksum;
getContext().setProperty( prop, value );
}
}
else if( destination instanceof File )
{
final File file = (File)destination;
if( m_isCondition )
{
if( file.exists() &&
file.length() == checksum.length() )
{
fis = new FileInputStream( file );
InputStreamReader isr = new InputStreamReader( fis );
BufferedReader br = new BufferedReader( isr );
String suppliedChecksum = br.readLine();
fis.close();
fis = null;
br.close();
isr.close();
checksumMatches =
checksum.equals( suppliedChecksum );
}
else
{
checksumMatches = false;
}
}
else
{
fos = new FileOutputStream( file );
fos.write( checksum.getBytes() );
fos.close();
fos = null;
}
}
}
catch( final Exception e )
{
throw new TaskException( e.getMessage(), e );
}
finally
{
IOUtil.shutdownStream( fis );
IOUtil.shutdownStream( fos );
}
return checksumMatches;
}
private byte[] buildDigest( final InputStream input )
throws IOException
{
m_messageDigest.reset();
final DigestInputStream digester =
new DigestInputStream( input, m_messageDigest );
while( digester.read() != -1 )
{
}
digester.close();
return m_messageDigest.digest();
}
/**
* Validate attributes and get down to business.
*/
private boolean validateAndExecute()
throws TaskException
{
if( null == m_file && 0 == m_filesets.size() )
{
final String message = "Specify at least one source - a file or a
fileset.";
throw new TaskException( message );
}
if( null != m_file && m_file.exists() && m_file.isDirectory() )
{
final String message = "Checksum cannot be generated for
directories";
throw new TaskException( message );
}
if( null != m_property && null != m_fileext )
{
final String message = "Property and FileExt cannot co-exist.";
throw new TaskException( message );
}
if( m_property != null )
{
if( m_forceOverwrite )
{
final String message =
"ForceOverwrite cannot be used when Property is
specified";
throw new TaskException( message );
}
if( m_file != null )
{
if( m_filesets.size() > 0 )
{
final String message =
"Multiple files cannot be used when Property is
specified";
throw new TaskException( message );
}
}
else
{
if( m_filesets.size() > 1 )
{
final String message =
"Multiple files cannot be used when Property is
specified";
throw new TaskException( message );
}
}
}
if( m_verifyProperty != null )
{
m_isCondition = true;
}
if( m_verifyProperty != null && m_forceOverwrite )
{
final String message = "VerifyProperty and ForceOverwrite cannot
co-exist.";
throw new TaskException( message );
}
if( m_isCondition && m_forceOverwrite )
{
final String message = "ForceOverwrite cannot be used when
conditions are being used.";
throw new TaskException( message );
}
if( m_fileext == null )
{
m_fileext = "." + m_algorithm;
}
else if( m_fileext.trim().length() == 0 )
{
final String message = "File extension when specified must not be
an empty string";
throw new TaskException( message );
}
setupMessageDigest();
if( m_messageDigest == null )
{
final String message = "Unable to create Message Digest";
throw new TaskException( message );
}
addIncludes();
return generateChecksums();
}
private void setupMessageDigest()
throws TaskException
{
m_messageDigest = null;
if( m_provider != null )
{
try
{
m_messageDigest = MessageDigest.getInstance( m_algorithm,
m_provider );
}
catch( final NoSuchAlgorithmException nsae )
{
throw new TaskException( nsae.toString(), nsae );
}
catch( final NoSuchProviderException nspe )
{
throw new TaskException( nspe.toString(), nspe );
}
}
else
{
try
{
m_messageDigest = MessageDigest.getInstance( m_algorithm );
}
catch( final NoSuchAlgorithmException nsae )
{
throw new TaskException( nsae.toString(), nsae );
}
}
}
private void addIncludes()
throws TaskException
{
addToIncludeFileMap( m_file );
final int size = m_filesets.size();
for( int i = 0; i < size; i++ )
{
final FileSet fileSet = (FileSet)m_filesets.get( i );
final DirectoryScanner scanner = ScannerUtil.getDirectoryScanner(
fileSet );
final String[] srcFiles = scanner.getIncludedFiles();
for( int j = 0; j < srcFiles.length; j++ )
{
final File src = new File( fileSet.getDir(), srcFiles[ j ] );
addToIncludeFileMap( src );
}
}
}
}
--
To unsubscribe, e-mail: <mailto:[EMAIL PROTECTED]>
For additional commands, e-mail: <mailto:[EMAIL PROTECTED]>
