001 /* 002 * The Apache Software License, Version 1.1 003 * 004 * Copyright (c) 2000-2002 The Apache Software Foundation. All rights 005 * reserved. 006 * 007 * Redistribution and use in source and binary forms, with or without 008 * modification, are permitted provided that the following conditions 009 * are met: 010 * 011 * 1. Redistributions of source code must retain the above copyright 012 * notice, this list of conditions and the following disclaimer. 013 * 014 * 2. Redistributions in binary form must reproduce the above copyright 015 * notice, this list of conditions and the following disclaimer in 016 * the documentation and/or other materials provided with the 017 * distribution. 018 * 019 * 3. The end-user documentation included with the redistribution, if 020 * any, must include the following acknowlegement: 021 * "This product includes software developed by the 022 * Apache Software Foundation (http://www.apache.org/)." 023 * Alternately, this acknowlegement may appear in the software itself, 024 * if and wherever such third-party acknowlegements normally appear. 025 * 026 * 4. The names "Ant" and "Apache Software 027 * Foundation" must not be used to endorse or promote products derived 028 * from this software without prior written permission. For written 029 * permission, please contact apache@apache.org. 030 * 031 * 5. Products derived from this software may not be called "Apache" 032 * nor may "Apache" appear in their names without prior written 033 * permission of the Apache Group. 034 * 035 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED 036 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 037 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 038 * DISCLAIMED. IN NO EVENT SHALL THE APACHE SOFTWARE FOUNDATION OR 039 * ITS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 040 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 041 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF 042 * USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 043 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 044 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT 045 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 046 * SUCH DAMAGE. 047 * ==================================================================== 048 * 049 * This software consists of voluntary contributions made by many 050 * individuals on behalf of the Apache Software Foundation. For more 051 * information on the Apache Software Foundation, please see 052 * <http://www.apache.org/>. 053 */ 054 package com.jcoverage.ant; 055 056 import com.jcoverage.coverage.Version; 057 058 import java.io.File; 059 import java.io.FileInputStream; 060 import java.io.FileOutputStream; 061 import java.io.InputStream; 062 import java.io.IOException; 063 import java.io.OutputStream; 064 065 import java.util.Arrays; 066 import java.util.HashSet; 067 import java.util.Iterator; 068 import java.util.LinkedList; 069 import java.util.List; 070 import java.util.Set; 071 072 import org.apache.tools.ant.AntClassLoader; 073 import org.apache.tools.ant.BuildException; 074 import org.apache.tools.ant.Project; 075 import org.apache.tools.ant.taskdefs.Copy; 076 import org.apache.tools.ant.taskdefs.Java; 077 import org.apache.tools.ant.taskdefs.MatchingTask; 078 import org.apache.tools.ant.types.FileSet; 079 import org.apache.tools.ant.types.Path; 080 import org.apache.tools.ant.types.Reference; 081 import org.apache.tools.ant.util.IdentityMapper; 082 import org.apache.tools.ant.util.SourceFileScanner; 083 084 public class InstrumentTask extends CommonMatchingTask { 085 Ignore ignoreRegex=null; 086 087 public InstrumentTask() { 088 super("com.jcoverage.coverage.Instrument"); 089 } 090 091 public Ignore createIgnore() { 092 ignoreRegex=new Ignore(); 093 return ignoreRegex; 094 } 095 096 void coverage(String[] filenames) { 097 if(filenames.length==0) { 098 return; 099 } 100 101 for(int i=0;i<filenames.length;i++) { 102 getProject().log("Adding "+filenames[i]+" to list",Project.MSG_VERBOSE); 103 getJava().createArg().setValue(filenames[i]); 104 } 105 106 System.out.println("jcoverage "+Version.VERSION_STRING+" copyright (c)2003 jcoverage ltd. http://jcoverage.com/"); 107 System.out.println("jcoverage is licensed under the GNU General Public License"); 108 System.out.println("jcoverage comes with ABSOLUTELY NO WARRANTY"); 109 System.out.println("instrumenting "+filenames.length+" "+(filenames.length==1?"class":"classes")+(toDir!=null?" to "+toDir:"")); 110 111 if(getJava().executeJava()!=0) { 112 throw new BuildException(); 113 } 114 } 115 116 public void execute() throws BuildException { 117 if(toDir!=null) { 118 getJava().createArg().setValue("-d"); 119 getJava().createArg().setValue(toDir.toString()); 120 } 121 122 if(ignoreRegex!=null) { 123 getJava().createArg().setValue("-ignore"); 124 getJava().createArg().setValue(ignoreRegex.getRegex()); 125 } 126 127 128 Set filenames=new HashSet(); 129 Iterator i=fileSets.iterator(); 130 131 while(i.hasNext()) { 132 FileSet fileSet=(FileSet)i.next(); 133 134 getJava().createArg().setValue("-basedir"); 135 getJava().createArg().setValue(baseDir(fileSet)); 136 137 filenames.addAll(Arrays.asList(getFilenames(fileSet))); 138 } 139 140 coverage((String[])filenames.toArray(new String[filenames.size()])); 141 } 142 }