Home | History | Annotate | Download | only in ijar
      1 // Copyright 2001,2007 Alan Donovan. All rights reserved.
      2 //
      3 // Author: Alan Donovan <adonovan (at) google.com>
      4 //
      5 // Licensed under the Apache License, Version 2.0 (the "License");
      6 // you may not use this file except in compliance with the License.
      7 // You may obtain a copy of the License at
      8 //
      9 //    http://www.apache.org/licenses/LICENSE-2.0
     10 //
     11 // Unless required by applicable law or agreed to in writing, software
     12 // distributed under the License is distributed on an "AS IS" BASIS,
     13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     14 // See the License for the specific language governing permissions and
     15 // limitations under the License.
     16 //
     17 // ijar.cpp -- .jar -> _interface.jar tool.
     18 //
     19 
     20 #include <stdio.h>
     21 #include <string.h>
     22 #include <stdlib.h>
     23 #include <limits.h>
     24 #include <errno.h>
     25 #include <memory>
     26 
     27 #include "zip.h"
     28 
     29 namespace devtools_ijar {
     30 
     31 bool verbose = false;
     32 
     33 // Reads a JVM class from classdata_in (of the specified length), and
     34 // writes out a simplified class to classdata_out, advancing the
     35 // pointer.
     36 void StripClass(u1 *&classdata_out, const u1 *classdata_in, size_t in_length);
     37 
     38 const char* CLASS_EXTENSION = ".class";
     39 const size_t CLASS_EXTENSION_LENGTH = strlen(CLASS_EXTENSION);
     40 
     41 // ZipExtractorProcessor that select only .class file and use
     42 // StripClass to generate an interface class, storing as a new file
     43 // in the specified ZipBuilder.
     44 class JarStripperProcessor : public ZipExtractorProcessor {
     45  public:
     46   JarStripperProcessor() {}
     47   virtual ~JarStripperProcessor() {}
     48 
     49   virtual void Process(const char* filename, const u4 attr,
     50                        const u1* data, const size_t size);
     51   virtual bool Accept(const char* filename, const u4 attr);
     52 
     53  private:
     54   // Not owned by JarStripperProcessor, see SetZipBuilder().
     55   ZipBuilder* builder;
     56 
     57  public:
     58   // Set the ZipBuilder to add the ijar class to the output zip file.
     59   // This pointer should not be deleted while this class is still in use and
     60   // it should be set before any call to the Process() method.
     61   void SetZipBuilder(ZipBuilder* builder) {
     62     this->builder = builder;
     63   }
     64 };
     65 
     66 bool JarStripperProcessor::Accept(const char* filename, const u4) {
     67   ssize_t offset = strlen(filename) - CLASS_EXTENSION_LENGTH;
     68   if (offset >= 0) {
     69     return strcmp(filename + offset, CLASS_EXTENSION) == 0;
     70   }
     71   return false;
     72 }
     73 
     74 void JarStripperProcessor::Process(const char* filename, const u4,
     75                                    const u1* data, const size_t size) {
     76   if (verbose) {
     77     fprintf(stderr, "INFO: StripClass: %s\n", filename);
     78   }
     79   u1 *q = builder->NewFile(filename, 0);
     80   u1 *classdata_out = q;
     81   StripClass(q, data, size);  // actually process it
     82   size_t out_length = q - classdata_out;
     83   builder->FinishFile(out_length);
     84 }
     85 
     86 // Opens "file_in" (a .jar file) for reading, and writes an interface
     87 // .jar to "file_out".
     88 void OpenFilesAndProcessJar(const char *file_out, const char *file_in) {
     89   JarStripperProcessor processor;
     90   std::unique_ptr<ZipExtractor> in(ZipExtractor::Create(file_in, &processor));
     91   if (in.get() == NULL) {
     92     fprintf(stderr, "Unable to open Zip file %s: %s\n", file_in,
     93             strerror(errno));
     94     abort();
     95   }
     96   u8 output_length = in->CalculateOutputLength();
     97   std::unique_ptr<ZipBuilder> out(ZipBuilder::Create(file_out, output_length));
     98   if (out.get() == NULL) {
     99     fprintf(stderr, "Unable to open output file %s: %s\n", file_out,
    100             strerror(errno));
    101     abort();
    102   }
    103   processor.SetZipBuilder(out.get());
    104 
    105   // Process all files in the zip
    106   if (in->ProcessAll() < 0) {
    107     fprintf(stderr, "%s\n", in->GetError());
    108     abort();
    109   }
    110 
    111   // Add dummy file, since javac doesn't like truly empty jars.
    112   if (out->GetNumberFiles() == 0) {
    113     out->WriteEmptyFile("dummy");
    114   }
    115   // Finish writing the output file
    116   if (out->Finish() < 0) {
    117     fprintf(stderr, "%s\n", out->GetError());
    118     abort();
    119   }
    120   // Get all file size
    121   size_t in_length = in->GetSize();
    122   size_t out_length = out->GetSize();
    123   if (verbose) {
    124     fprintf(stderr, "INFO: produced interface jar: %s -> %s (%d%%).\n",
    125             file_in, file_out,
    126             static_cast<int>(100.0 * out_length / in_length));
    127   }
    128 }
    129 
    130 }  // namespace devtools_ijar
    131 
    132 //
    133 // main method
    134 //
    135 static void usage() {
    136   fprintf(stderr, "Usage: ijar [-v] x.jar [x_interface.jar>]\n");
    137   fprintf(stderr, "Creates an interface jar from the specified jar file.\n");
    138   exit(1);
    139 }
    140 
    141 int main(int argc, char **argv) {
    142   const char *filename_in = NULL;
    143   const char *filename_out = NULL;
    144 
    145   for (int ii = 1; ii < argc; ++ii) {
    146     if (strcmp(argv[ii], "-v") == 0) {
    147       devtools_ijar::verbose = true;
    148     } else if (filename_in == NULL) {
    149       filename_in = argv[ii];
    150     } else if (filename_out == NULL) {
    151       filename_out = argv[ii];
    152     } else {
    153       usage();
    154     }
    155   }
    156 
    157   if (filename_in == NULL) {
    158     usage();
    159   }
    160 
    161   // Guess output filename from input:
    162   char filename_out_buf[PATH_MAX];
    163   if (filename_out == NULL) {
    164     size_t len = strlen(filename_in);
    165     if (len > 4 && strncmp(filename_in + len - 4, ".jar", 4) == 0) {
    166       strcpy(filename_out_buf, filename_in);
    167       strcpy(filename_out_buf + len - 4, "-interface.jar");
    168       filename_out = filename_out_buf;
    169     } else {
    170       fprintf(stderr, "Can't determine output filename since input filename "
    171               "doesn't end with '.jar'.\n");
    172       return 1;
    173     }
    174   }
    175 
    176   if (devtools_ijar::verbose) {
    177     fprintf(stderr, "INFO: writing to '%s'.\n", filename_out);
    178   }
    179 
    180   devtools_ijar::OpenFilesAndProcessJar(filename_out, filename_in);
    181   return 0;
    182 }
    183