fix flakiness of tst_qlocalsocket::processConnection
[qt:qtbase.git] / tests / auto / network / socket / qlocalsocket / socketprocess / main.cpp
1 /****************************************************************************
2 **
3 ** Copyright (C) 2012 Digia Plc and/or its subsidiary(-ies).
4 ** Contact: http://www.qt-project.org/legal
5 **
6 ** This file is part of the test suite of the Qt Toolkit.
7 **
8 ** $QT_BEGIN_LICENSE:LGPL$
9 ** Commercial License Usage
10 ** Licensees holding valid commercial Qt licenses may use this file in
11 ** accordance with the commercial license agreement provided with the
12 ** Software or, alternatively, in accordance with the terms contained in
13 ** a written agreement between you and Digia.  For licensing terms and
14 ** conditions see http://qt.digia.com/licensing.  For further information
15 ** use the contact form at http://qt.digia.com/contact-us.
16 **
17 ** GNU Lesser General Public License Usage
18 ** Alternatively, this file may be used under the terms of the GNU Lesser
19 ** General Public License version 2.1 as published by the Free Software
20 ** Foundation and appearing in the file LICENSE.LGPL included in the
21 ** packaging of this file.  Please review the following information to
22 ** ensure the GNU Lesser General Public License version 2.1 requirements
23 ** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
24 **
25 ** In addition, as a special exception, Digia gives you certain additional
26 ** rights.  These rights are described in the Digia Qt LGPL Exception
27 ** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.
28 **
29 ** GNU General Public License Usage
30 ** Alternatively, this file may be used under the terms of the GNU
31 ** General Public License version 3.0 as published by the Free Software
32 ** Foundation and appearing in the file LICENSE.GPL included in the
33 ** packaging of this file.  Please review the following information to
34 ** ensure the GNU General Public License version 3.0 requirements will be
35 ** met: http://www.gnu.org/copyleft/gpl.html.
36 **
37 **
38 ** $QT_END_LICENSE$
39 **
40 ****************************************************************************/
41
42
43 #include <qcoreapplication.h>
44 #include <qelapsedtimer.h>
45 #include <qeventloop.h>
46 #include <qlocalsocket.h>
47 #include <qlocalserver.h>
48 #include <qtimer.h>
49
50 const QString serverName = QStringLiteral("qlocalsocket_autotest");
51 const QByteArray testData("test");
52
53 bool runServer(int numberOfConnections)
54 {
55     QLocalServer *server = new QLocalServer(qApp);
56     if (!server->removeServer(serverName)) {
57         fprintf(stderr, "server: cannot remove server: %s\n", qPrintable(server->errorString()));
58         return false;
59     }
60     printf("server: listen on \"%s\"\n", qPrintable(serverName));
61     if (!server->listen(serverName)) {
62         fprintf(stderr, "server: listen failed: %s\n", qPrintable(server->errorString()));
63         return false;
64     }
65     for (int i = 1; i <= numberOfConnections; ++i) {
66         printf("server: wait for connection %d\n", i);
67         if (!server->waitForNewConnection(30000)) {
68             fprintf(stderr, "server: waitForNewConnection failed: %s\n",
69                     qPrintable(server->errorString()));
70             return false;
71         }
72         QLocalSocket *socket = server->nextPendingConnection();
73         printf("server: writing \"%s\"\n", testData.data());
74         socket->write(testData);
75         if (!socket->waitForBytesWritten()) {
76             fprintf(stderr, "server: waitForBytesWritten failed: %s\n",
77                     qPrintable(socket->errorString()));
78             return false;
79         }
80         printf("server: data written\n");
81         if (socket->error() != QLocalSocket::UnknownSocketError) {
82             fprintf(stderr, "server: socket error %d\n", socket->error());
83             return false;
84         }
85     }
86     return true;
87 }
88
89 bool runClient()
90 {
91     QLocalSocket socket;
92     printf("client: connecting to \"%s\"\n", qPrintable(serverName));
93     QElapsedTimer connectTimer;
94     connectTimer.start();
95     forever {
96         socket.connectToServer(serverName, QLocalSocket::ReadWrite);
97         if (socket.waitForConnected())
98             break;
99         if (socket.error() == QLocalSocket::ServerNotFoundError) {
100             if (connectTimer.elapsed() > 5000) {
101                 fprintf(stderr, "client: server not found. Giving up.\n");
102                 return false;
103             }
104             printf("client: server not found. Trying again...\n");
105             QEventLoop eventLoop;
106             QTimer::singleShot(500, &eventLoop, SLOT(quit()));
107             eventLoop.exec();
108             continue;
109         }
110         fprintf(stderr, "client: waitForConnected failed: %s\n",
111                 qPrintable(socket.errorString()));
112         return false;
113     }
114     printf("client: connected\n");
115     if (!socket.waitForReadyRead()) {
116         fprintf(stderr, "client: waitForReadyRead failed: %s\n",
117                 qPrintable(socket.errorString()));
118         return false;
119     }
120     printf("client: data is available for reading\n");
121     const QByteArray data = socket.readLine();
122     printf("client: received \"%s\"\n", data.data());
123     if (data != testData) {
124         fprintf(stderr, "client: received unexpected data\n");
125         return false;
126     }
127     return true;
128 }
129
130 int main(int argc, char **argv)
131 {
132     QCoreApplication app(argc, argv);
133     if (argc < 2)
134         return EXIT_FAILURE;
135     if (strcmp(argv[1], "--server") == 0) {
136         if (argc < 3) {
137             fprintf(stderr, "--server needs the number of incoming connections\n");
138             return EXIT_FAILURE;
139         }
140         bool ok;
141         int n = QByteArray(argv[2]).toInt(&ok);
142         if (!ok) {
143             fprintf(stderr, "Cannot convert %s to a number.\n", argv[2]);
144             return EXIT_FAILURE;
145         }
146         if (!runServer(n))
147             return EXIT_FAILURE;
148     } else if (strcmp(argv[1], "--client") == 0) {
149         if (!runClient())
150             return EXIT_FAILURE;
151     } else {
152         fprintf(stderr, "unknown command line option: %s\n", argv[1]);
153         return EXIT_FAILURE;
154     }
155     return EXIT_SUCCESS;
156 }