Tools/QtTestBrowser/urlloader.cpp

   1 /*
   2  * Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies)
   3  * Copyright (C) 2009 University of Szeged
   4  *
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
  17  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR
  20  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  21  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  22  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  23  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  24  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include "urlloader.h"
  30
  31 #include <QFile>
  32 #include <QDebug>
  33 #include <QWebPage>
  34
  35 UrlLoader::UrlLoader(QWebFrame* frame, const QString& inputFileName, int timeoutSeconds, int extraTimeSeconds)
  36     : m_frame(frame)
  37     , m_stdOut(stdout)
  38     , m_loaded(0)
  39     , m_numFramesLoading(0)
  40 {
  41     m_checkIfFinishedTimer.setInterval(200);
  42     m_checkIfFinishedTimer.setSingleShot(true);
  43     connect(&m_checkIfFinishedTimer, SIGNAL(timeout()), this, SLOT(checkIfFinished()));
  44     // loadStarted and loadFinished on QWebPage is emitted for each frame/sub-frame
  45     connect(m_frame->page(), SIGNAL(loadStarted()), this, SLOT(frameLoadStarted()));
  46     connect(m_frame->page(), SIGNAL(loadFinished(bool)), this, SLOT(frameLoadFinished()));
  47
  48     if (timeoutSeconds) {
  49         m_timeoutTimer.setInterval(timeoutSeconds * 1000);
  50         m_timeoutTimer.setSingleShot(true);
  51         connect(frame, SIGNAL(loadStarted()), &m_timeoutTimer, SLOT(start()));
  52         connect(&m_timeoutTimer, SIGNAL(timeout()), this, SLOT(loadNext()));
  53     }
  54     if (extraTimeSeconds) {
  55         m_extraTimeTimer.setInterval(extraTimeSeconds * 1000);
  56         m_extraTimeTimer.setSingleShot(true);
  57         connect(this, SIGNAL(pageLoadFinished()), &m_extraTimeTimer, SLOT(start()));
  58         connect(&m_extraTimeTimer, SIGNAL(timeout()), this, SLOT(loadNext()));
  59     } else
  60         connect(this, SIGNAL(pageLoadFinished()), this, SLOT(loadNext()));
  61     loadUrlList(inputFileName);
  62 }
  63
  64 void UrlLoader::loadNext()
  65 {
  66     m_timeoutTimer.stop();
  67     m_extraTimeTimer.stop();
  68     m_checkIfFinishedTimer.stop();
  69     m_numFramesLoading = 0;
  70     QString qstr;
  71     if (getUrl(qstr)) {
  72         QUrl url(qstr, QUrl::StrictMode);
  73         if (url.isValid()) {
  74             m_stdOut << "Loading " << qstr << " ......" << ++m_loaded << endl;
  75             m_frame->load(url);
  76         } else
  77             loadNext();
  78     } else
  79         disconnect(m_frame, 0, this, 0);
  80 }
  81
  82 void UrlLoader::checkIfFinished()
  83 {
  84     if (!m_numFramesLoading)
  85         emit pageLoadFinished();
  86 }
  87
  88 void UrlLoader::frameLoadStarted()
  89 {
  90     ++m_numFramesLoading;
  91     m_checkIfFinishedTimer.stop();
  92 }
  93
  94 void UrlLoader::frameLoadFinished()
  95 {
  96     Q_ASSERT(m_numFramesLoading > 0);
  97     --m_numFramesLoading;
  98     // Once our frame has finished loading, wait a moment to call loadNext for cases
  99     // where a sub-frame starts loading or another frame is loaded through JavaScript.
 100     m_checkIfFinishedTimer.start();
 101 }
 102
 103 void UrlLoader::loadUrlList(const QString& inputFileName)
 104 {
 105     QFile inputFile(inputFileName);
 106     if (inputFile.open(QIODevice::ReadOnly | QIODevice::Text)) {
 107         QTextStream stream(&inputFile);
 108         QString line;
 109         while (true) {
 110             line = stream.readLine();
 111             if (line.isNull())
 112                 break;
 113             m_urls.append(line);
 114         }
 115     } else {
 116         qDebug() << "Can't open list file";
 117         exit(0);
 118     }
 119     m_index = 0;
 120     inputFile.close();
 121 }
 122
 123 bool UrlLoader::getUrl(QString& qstr)
 124 {
 125     if (m_index == m_urls.size())
 126         return false;
 127
 128     qstr = m_urls[m_index++];
 129     return true;
 130 }