/data/refman/pasoursint/CMSSW_6_1_2_SLHC4_patch1/src/EventFilter/Utilities/src/ParameterSetRetriever.cc

Go to the documentation of this file.

#include "EventFilter/Utilities/interface/ParameterSetRetriever.h"
#include "EventFilter/Utilities/interface/Exception.h"
#include "EventFilter/Utilities/interface/CurlUtils.h"
#include "EventFilter/Utilities/interface/SquidNet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "curl/curl.h"
#include <netdb.h>
#include <sys/socket.h>        /* for AF_INET */

#include <fstream>



namespace evf{
  //______________________________________________________________________________


  const std::string ParameterSetRetriever::fileheading="file:";
  const std::string ParameterSetRetriever::dbheading  ="db:";  
  const std::string ParameterSetRetriever::webheading ="http://";  

  ParameterSetRetriever::ParameterSetRetriever(const std::string& in)
  {
    using std::string;
    using std::ostringstream; 
    using std::ifstream; 



    if (fileheading==in.substr(0,fileheading.size()))
      { 
        string filename=in.substr(fileheading.size());
        edm::LogInfo("psetRetriever")<<"filename is --> "<<filename<<" <--";
      
        string   line;
        ifstream configFile(filename.c_str());
        while(std::getline(configFile,line)) {
          pset+=line;
          pset+="\n";
        }
      }
    else if (webheading==in.substr(0,webheading.size()))
      {
        string hostname = getHostString(in);
        SquidNet sn(3128,"http://localhost:8000/RELEASE-NOTES.txt");    
        edm::LogInfo("psetRetriever")<<"Using cfg from " << in;
        CURL* han = curl_easy_init();
        if(han==0)
          {
            XCEPT_RAISE(evf::Exception,"could not create handle for web ParameterSet");     
          }

        struct curl_slist *headers=NULL; /* init to NULL is important */
        headers = curl_slist_append(headers, "Pragma:");
        curl_easy_setopt(han, CURLOPT_HTTPHEADER, headers);
        char error[CURL_ERROR_SIZE];
        if(sn.check())
          curl_easy_setopt(han, CURLOPT_PROXY, "localhost:3128");

        curl_easy_setopt(han, CURLOPT_URL, hostname.c_str());
        curl_easy_setopt(han, CURLOPT_VERBOSE,"");
        curl_easy_setopt(han, CURLOPT_NOSIGNAL,"");
        //      curl_easy_setopt(han, CURLOPT_TIMEOUT, 60.0L);

        curl_easy_setopt(han, CURLOPT_WRITEFUNCTION, &write_data);
        curl_easy_setopt(han, CURLOPT_WRITEDATA, &pset);
        curl_easy_setopt(han, CURLOPT_ERRORBUFFER, error);
        int success = curl_easy_perform(han);
        curl_slist_free_all(headers); /* free the header list */
        curl_easy_cleanup(han);
        han = 0;
        if(success != 0)
          {
            ostringstream msg;
            msg <<  "could not get config from url " << in << " error #" 
                << success << " " << error;
            XCEPT_RAISE(evf::Exception,msg.str().c_str());
          }

        //now get path-index table      
        han = curl_easy_init();
        if(han==0)
          {
            XCEPT_RAISE(evf::Exception,"could not create handle for web ParameterSet");     
          }
        headers = NULL;
        headers = curl_slist_append(headers, "Pragma:");
        curl_easy_setopt(han, CURLOPT_HTTPHEADER, headers);
        if(sn.check())
          curl_easy_setopt(han, CURLOPT_PROXY, "localhost:3128");
        hostname = getHostString(in,"path");
        curl_easy_setopt(han, CURLOPT_URL, hostname.c_str());
        curl_easy_setopt(han, CURLOPT_VERBOSE,"");
        curl_easy_setopt(han, CURLOPT_NOSIGNAL,"");
        //      curl_easy_setopt(han, CURLOPT_TIMEOUT, 60.0L);

        curl_easy_setopt(han, CURLOPT_WRITEFUNCTION, &write_data);
        curl_easy_setopt(han, CURLOPT_WRITEDATA, &pathIndexTable);
        curl_easy_setopt(han, CURLOPT_ERRORBUFFER, error);
        success = curl_easy_perform(han);
        curl_slist_free_all(headers); /* free the header list */
        curl_easy_cleanup(han);

        if(success != 0)
          {
            ostringstream msg;
            msg <<  "could not get pathindex table from url " << in << " error #" 
                << success << " " << error;
            XCEPT_RAISE(evf::Exception,msg.str().c_str());
          }


      }
    else if (dbheading==in.substr(0,dbheading.size()))
      {
        XCEPT_RAISE(evf::Exception,"db access for ParameterSet not yet implemented");
      } 
    else
      {
        edm::LogInfo("psetRetriever")<<"Using string cfg from RunControl or XML";
        pset = in;
      }
  }


  //______________________________________________________________________________
  std::string ParameterSetRetriever::getAsString() const
  {
    return pset;
  }
  //______________________________________________________________________________
  std::string ParameterSetRetriever::getPathTableAsString() const
  {
    return pathIndexTable;
  }


  std::string ParameterSetRetriever::getHostString(const std::string &in, std::string modifier) const
  {
    using std::string;
    string innohttp = in.substr(webheading.size());
    string::size_type pos = innohttp.find(':',0);
    string host = innohttp.substr(0,pos);
    string path = innohttp.substr(pos);
    if(modifier != "")
      {
        pos = path.find_last_of('.');
        if(pos != string::npos)
          {
            string upath = path.substr(0,pos);
            path = upath + '.' + modifier;
          }
      }
    struct hostent *heb = gethostbyname(host.c_str());    
    struct hostent *he = gethostbyaddr(heb->h_addr_list[0], heb->h_length, heb->h_addrtype);
    string completehost = "http://";
    completehost += he->h_name;
    completehost += path;
    return completehost;
  }
} //end namespace evf