runtime/kioslave/cgi/cgi.cpp

   1 /*
   2    Copyright (C) 2002 Cornelius Schumacher <schumacher@kde.org>
   3
   4    This program is free software; you can redistribute it and/or modify
   5    it under the terms of the GNU General Public License as published by
   6    the Free Software Foundation; either version 2 of the License, or
   7    (at your option) any later version.
   8
   9    This program is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12    GNU General Public License for more details.
  13
  14    You should have received a copy of the GNU General Public License
  15    along with this program; if not, write to the Free Software
  16    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  17 */
  18
  19 #include "cgi.h"
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <unistd.h> // getpid()
  24
  25 #include <QDir>
  26 #include <QRegExp>
  27
  28 #include <kdebug.h>
  29 #include <kshell.h>
  30 #include <kstandarddirs.h>
  31 #include <kcomponentdata.h>
  32 #include <klocale.h>
  33 #include <kconfig.h>
  34 #include <kconfiggroup.h>
  35
  36
  37 using namespace KIO;
  38
  39 CgiProtocol::CgiProtocol( const QByteArray &pool, const QByteArray &app )
  40     : SlaveBase( "cgi", pool, app )
  41 {
  42   kDebug(7124) << "CgiProtocol::CgiProtocol";
  43
  44   KConfig _cfg( "kcmcgirc" );
  45   KConfigGroup cfg(&_cfg, "General" );
  46   mCgiPaths = cfg.readEntry( "Paths" , QStringList() );
  47 }
  48
  49 CgiProtocol::~CgiProtocol()
  50 {
  51   kDebug(7124) << "CgiProtocol::~CgiProtocol";
  52 }
  53
  54 void CgiProtocol::get( const KUrl& url )
  55 {
  56   kDebug(7124) << "CgiProtocol::get()";
  57   kDebug(7124) << " URL: " << url.url();
  58 #if 0
  59   kDebug(7124) << " Path: " << url.path();
  60   kDebug(7124) << " Query: " << url.query();
  61   kDebug(7124) << " Protocol: " << url.protocol();
  62   kDebug(7124) << " Filename: " << url.filename();
  63 #endif
  64   QByteArray protocol = "SERVER_PROTOCOL=HTTP";
  65   putenv( protocol.data() );
  66
  67   QByteArray requestMethod = "REQUEST_METHOD=GET";
  68   putenv( requestMethod.data() );
  69
  70   QByteArray query = url.query().mid( 1 ).toLocal8Bit();
  71   query.prepend( "QUERY_STRING=" );
  72   putenv( query.data() );
  73
  74   QString path = url.path();
  75
  76   QString file;
  77
  78   int pos = path.lastIndexOf('/');
  79   if ( pos >= 0 ) file = path.mid( pos + 1 );
  80   else file = path;
  81
  82   QString cmd;
  83
  84   bool stripHeader = false;
  85   bool forwardFile = true;
  86
  87   QStringList::ConstIterator it;
  88   for( it = mCgiPaths.constBegin(); it != mCgiPaths.constEnd(); ++it ) {
  89     cmd = *it;
  90     if ( !(*it).endsWith('/') )
  91         cmd += '/';
  92     cmd += file;
  93     if ( KStandardDirs::exists( cmd ) ) {
  94       forwardFile = false;
  95       stripHeader = true;
  96       break;
  97     }
  98   }
  99
 100   FILE *fd;
 101
 102   if ( forwardFile ) {
 103     kDebug(7124) << "Forwarding to '" << path << "'";
 104
 105     QByteArray filepath = QFile::encodeName( path );
 106
 107     fd = fopen( filepath.data(), "r" );
 108
 109     if ( !fd ) {
 110       kDebug(7124) << "Error opening '" << filepath << "'";
 111       error( KIO::ERR_CANNOT_OPEN_FOR_READING, filepath );
 112       return;
 113     }
 114   } else {
 115     kDebug(7124) << "Cmd: " << cmd;
 116
 117     fd = popen( QFile::encodeName(KShell::quoteArg( cmd )).data(), "r" );
 118
 119     if ( !fd ) {
 120       kDebug(7124) << "Error running '" << cmd << "'";
 121       error( KIO::ERR_CANNOT_OPEN_FOR_READING, cmd );
 122       return;
 123     }
 124   }
 125
 126   char buffer[ 4090 ];
 127
 128   while ( !feof( fd ) )
 129   {
 130     int n = fread( buffer, 1, 2048, fd );
 131
 132     if ( n == -1 )
 133     {
 134       // ERROR
 135       if ( forwardFile ) {
 136         fclose( fd );
 137       } else {
 138         pclose( fd );
 139       }
 140       return;
 141     }
 142
 143     buffer[n] = 0;
 144
 145     if ( stripHeader ) {
 146       QByteArray output = buffer; // this assumes buffer is text and not binary
 147       int colon = output.indexOf( ':' );
 148       int newline = output.indexOf( '\n' );
 149       int semicolon = output.lastIndexOf( ';', newline );
 150       int end;
 151       if ( semicolon < 0 ) end = newline;
 152       else end = semicolon;
 153
 154 #if 0
 155       kDebug(7124) << "  colon: " << colon;
 156       kDebug(7124) << "  newline: " << newline;
 157       kDebug(7124) << "  semicolon: " << semicolon;
 158       kDebug(7124) << "  end: " << end;
 159 #endif
 160
 161       QByteArray contentType = output.mid( colon + 1, end - colon - 1 );
 162
 163       contentType = contentType.trimmed();
 164
 165       kDebug(7124) << "ContentType: '" << contentType << "'";
 166
 167       mimeType( contentType );
 168
 169       int start = output.indexOf( "\r\n\r\n" );
 170       if ( start >= 0 ) start += 4;
 171       else {
 172         start = output.indexOf( "\n\n" );
 173         if ( start >= 0 ) start += 2;
 174       }
 175
 176       if ( start >= 0 ) output = output.mid( start );
 177
 178       stripHeader = false;
 179       data( output );
 180     } else {
 181       data( QByteArray::fromRawData( buffer, n ) );
 182     }
 183   }
 184
 185   if ( forwardFile ) {
 186     fclose( fd );
 187   } else {
 188     pclose( fd );
 189   }
 190
 191   finished();
 192
 193   kDebug(7124) << "CgiProtocol::get - done";
 194 }
 195
 196 extern "C" { int KDE_EXPORT kdemain( int argc, char **argv ); }
 197
 198 /*! The kdemain function generates an instance of the ioslave and starts its
 199  * dispatch loop. */
 200
 201 int kdemain( int argc, char **argv )
 202 {
 203   KComponentData componentData( "kio_cgi" );
 204
 205   kDebug(7124) << "kio_cgi starting " << getpid();
 206
 207   if (argc != 4)
 208   {
 209      fprintf(stderr, "Usage: kio_cgi protocol domain-socket1 domain-socket2\n");
 210      exit(-1);
 211   }
 212
 213   CgiProtocol slave( argv[2], argv[3] );
 214   slave.dispatchLoop();
 215
 216   return 0;
 217 }