/*

  * This file is part of dependency-check-core.

  *

  * Licensed under the Apache License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may obtain a copy of the License at

  *

  *     http://www.apache.org/licenses/LICENSE-2.0

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  *

  * Copyright (c) 2013 Jeremy Long. All Rights Reserved.

  */

 package org.owasp.dependencycheck.data.cpe;

 import java.io.IOException;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.util.HashMap;

 import java.util.Map;

 import java.util.logging.Level;

 import java.util.logging.Logger;

 import org.apache.lucene.analysis.Analyzer;

 import org.apache.lucene.analysis.core.KeywordAnalyzer;

 import org.apache.lucene.analysis.miscellaneous.PerFieldAnalyzerWrapper;

 import org.apache.lucene.document.Document;

 import org.apache.lucene.document.Field;

 import org.apache.lucene.document.TextField;

 import org.apache.lucene.index.CorruptIndexException;

 import org.apache.lucene.index.DirectoryReader;

 import org.apache.lucene.index.IndexReader;

 import org.apache.lucene.index.IndexWriter;

 import org.apache.lucene.index.IndexWriterConfig;

 import org.apache.lucene.queryparser.classic.ParseException;

 import org.apache.lucene.queryparser.classic.QueryParser;

 import org.apache.lucene.search.IndexSearcher;

 import org.apache.lucene.search.Query;

 import org.apache.lucene.search.TopDocs;

 import org.apache.lucene.store.RAMDirectory;

 import org.owasp.dependencycheck.data.lucene.FieldAnalyzer;

 import org.owasp.dependencycheck.data.lucene.LuceneUtils;

 import org.owasp.dependencycheck.data.lucene.SearchFieldAnalyzer;

 import org.owasp.dependencycheck.data.nvdcve.CveDB;

 /**

  * An in memory lucene index that contains the vendor/product combinations from the CPE (application) identifiers within

  * the NVD CVE data.

  *

  * @author Jeremy Long <jeremy.long@owasp.org>

  */

 public final class CpeMemoryIndex {

     /**

      * singleton instance.

      */

     private static CpeMemoryIndex instance = new CpeMemoryIndex();

     /**

      * private constructor for singleton.

      */

     private CpeMemoryIndex() {

     }

     /**

      * Gets the singleton instance of the CpeMemoryIndex.

      *

      * @return the instance of the CpeMemoryIndex

      */

     public static CpeMemoryIndex getInstance() {

         return instance;

     }

     /**

      * The in memory Lucene index.

      */

     private RAMDirectory index;

     /**

      * The Lucene IndexReader.

      */

     private IndexReader indexReader;

     /**

      * The Lucene IndexSearcher.

      */

     private IndexSearcher indexSearcher;

     /**

      * The Lucene Analyzer used for Searching.

      */

     private Analyzer searchingAnalyzer;

     /**

      * The Lucene QueryParser used for Searching.

      */

     private QueryParser queryParser;

     /**

      * The search field analyzer for the product field.

      */

     private SearchFieldAnalyzer productSearchFieldAnalyzer;

     /**

      * The search field analyzer for the vendor field.

      */

     private SearchFieldAnalyzer vendorSearchFieldAnalyzer;

     /**

      * Creates and loads data into an in memory index.

      *

      * @param cve the data source to retrieve the cpe data

      * @throws IndexException thrown if there is an error creating the index

      */

     public void open(CveDB cve) throws IndexException {

         if (!openState) {

             index = new RAMDirectory();

             buildIndex(cve);

             try {

                 indexReader = DirectoryReader.open(index);

             } catch (IOException ex) {

                 throw new IndexException(ex);

             }

             indexSearcher = new IndexSearcher(indexReader);

             searchingAnalyzer = createSearchingAnalyzer();

             queryParser = new QueryParser(LuceneUtils.CURRENT_VERSION, Fields.DOCUMENT_KEY, searchingAnalyzer);

             openState = true;

         }

     }

     /**

      * A flag indicating whether or not the index is open.

      */

     private boolean openState = false;

     /**

      * returns whether or not the index is open.

      *

      * @return whether or not the index is open

      */

     public boolean isOpen() {

         return openState;

     }

     /**

      * Creates the indexing analyzer for the CPE Index.

      *

      * @return the CPE Analyzer.

      */

     @SuppressWarnings("unchecked")

     private Analyzer createIndexingAnalyzer() {

         final Map fieldAnalyzers = new HashMap();

         fieldAnalyzers.put(Fields.DOCUMENT_KEY, new KeywordAnalyzer());

         return new PerFieldAnalyzerWrapper(new FieldAnalyzer(LuceneUtils.CURRENT_VERSION), fieldAnalyzers);

     }

     /**

      * Creates an Analyzer for searching the CPE Index.

      *

      * @return the CPE Analyzer.

      */

     @SuppressWarnings("unchecked")

     private Analyzer createSearchingAnalyzer() {

         final Map fieldAnalyzers = new HashMap();

         fieldAnalyzers.put(Fields.DOCUMENT_KEY, new KeywordAnalyzer());

         productSearchFieldAnalyzer = new SearchFieldAnalyzer(LuceneUtils.CURRENT_VERSION);

         vendorSearchFieldAnalyzer = new SearchFieldAnalyzer(LuceneUtils.CURRENT_VERSION);

         fieldAnalyzers.put(Fields.PRODUCT, productSearchFieldAnalyzer);

         fieldAnalyzers.put(Fields.VENDOR, vendorSearchFieldAnalyzer);

         return new PerFieldAnalyzerWrapper(new FieldAnalyzer(LuceneUtils.CURRENT_VERSION), fieldAnalyzers);

     }

     /**

      * Saves a CPE IndexEntry into the Lucene index.

      *

      * @param vendor the vendor to index

      * @param product the product to index

      * @param indexWriter the index writer to write the entry into

      * @throws CorruptIndexException is thrown if the index is corrupt

      * @throws IOException is thrown if an IOException occurs

      */

     public void saveEntry(String vendor, String product, IndexWriter indexWriter) throws CorruptIndexException, IOException {

         final Document doc = new Document();

         final Field v = new TextField(Fields.VENDOR, vendor, Field.Store.YES);

         final Field p = new TextField(Fields.PRODUCT, product, Field.Store.YES);

         doc.add(v);

         doc.add(p);

         indexWriter.addDocument(doc);

     }

     /**

      * Closes the CPE Index.

      */

     public void close() {

         if (searchingAnalyzer != null) {

             searchingAnalyzer.close();

             searchingAnalyzer = null;

         }

         if (indexReader != null) {

             try {

                 indexReader.close();

             } catch (IOException ex) {

                 Logger.getLogger(CpeMemoryIndex.class.getName()).log(Level.FINEST, null, ex);

             }

             indexReader = null;

         }

         queryParser = null;

         indexSearcher = null;

         if (index != null) {

             index.close();

             index = null;

         }

         openState = false;

     }

     /**

      * Builds the lucene index based off of the data within the CveDB.

      *

      * @param cve the data base containing the CPE data

      * @throws IndexException thrown if there is an issue creating the index

      */

     private void buildIndex(CveDB cve) throws IndexException {

         Analyzer analyzer = null;

         IndexWriter indexWriter = null;

         try {

             analyzer = createIndexingAnalyzer();

             final IndexWriterConfig conf = new IndexWriterConfig(LuceneUtils.CURRENT_VERSION, analyzer);

             indexWriter = new IndexWriter(index, conf);

             final ResultSet rs = cve.getVendorProductList();

             if (rs == null) {

                 throw new IndexException("No data exists");

             }

             try {

                 while (rs.next()) {

                     saveEntry(rs.getString(1), rs.getString(2), indexWriter);

                 }

             } catch (SQLException ex) {

                 Logger.getLogger(CpeMemoryIndex.class.getName()).log(Level.FINE, null, ex);

                 throw new IndexException("Error reading CPE data", ex);

             }

         } catch (CorruptIndexException ex) {

             throw new IndexException("Unable to close an in-memory index", ex);

         } catch (IOException ex) {

             throw new IndexException("Unable to close an in-memory index", ex);

         } finally {

             if (indexWriter != null) {

                 try {

                     try {

                         indexWriter.commit();

                     } finally {

                         indexWriter.close(true);

                     }

                 } catch (CorruptIndexException ex) {

                     throw new IndexException("Unable to close an in-memory index", ex);

                 } catch (IOException ex) {

                     throw new IndexException("Unable to close an in-memory index", ex);

                 }

                 if (analyzer != null) {

                     analyzer.close();

                 }

             }

         }

     }

     /**

      * Resets the searching analyzers

      */

     private void resetSearchingAnalyzer() {

         if (productSearchFieldAnalyzer != null) {

             productSearchFieldAnalyzer.clear();

         }

         if (vendorSearchFieldAnalyzer != null) {

             vendorSearchFieldAnalyzer.clear();

         }

     }

     /**

      * Searches the index using the given search string.

      *

      * @param searchString the query text

      * @param maxQueryResults the maximum number of documents to return

      * @return the TopDocs found by the search

      * @throws ParseException thrown when the searchString is invalid

      * @throws IOException is thrown if there is an issue with the underlying Index

      */

     public TopDocs search(String searchString, int maxQueryResults) throws ParseException, IOException {

         if (searchString == null || searchString.trim().isEmpty()) {

             throw new ParseException("Query is null or empty");

         }

         final Query query = queryParser.parse(searchString);

         return indexSearcher.search(query, maxQueryResults);

     }

     /**

      * Searches the index using the given query.

      *

      * @param query the query used to search the index

      * @param maxQueryResults the max number of results to return

      * @return the TopDocs found be the query

      * @throws CorruptIndexException thrown if the Index is corrupt

      * @throws IOException thrown if there is an IOException

      */

     public TopDocs search(Query query, int maxQueryResults) throws CorruptIndexException, IOException {

         resetSearchingAnalyzer();

         return indexSearcher.search(query, maxQueryResults);

     }

     /**

      * Retrieves a document from the Index.

      *

      * @param documentId the id of the document to retrieve

      * @return the Document

      * @throws IOException thrown if there is an IOException

      */

     public Document getDocument(int documentId) throws IOException {

         return indexSearcher.doc(documentId);

     }

     /**

      * Returns the number of CPE entries stored in the index.

      *

      * @return the number of CPE entries stored in the index

      */

     public int numDocs() {

         if (indexReader == null) {

             return -1;

         }

         return indexReader.numDocs();

     }

 }