00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
#include "config.h"
00026
00027
#include <sys/types.h>
00028
#include <netinet/in.h>
00029
#include <limits.h>
00030
#include <unistd.h>
00031
00032
#ifdef HAVE_RES_INIT
00033
# include <sys/stat.h>
00034
# include <resolv.h>
00035
#endif
00036
00037
#include <qapplication.h>
00038
#include <qstring.h>
00039
#include <qcstring.h>
00040
#include <qptrlist.h>
00041
#include <qtimer.h>
00042
#include <qmutex.h>
00043
#include <qthread.h>
00044
#include <qwaitcondition.h>
00045
#include <qsemaphore.h>
00046
00047
#include "kresolver.h"
00048
#include "kresolver_p.h"
00049
#include "kresolverworkerbase.h"
00050
#include "kresolverstandardworkers_p.h"
00051
00052
using namespace KNetwork;
00053
using namespace KNetwork::Internal;
00054
00055
00056
00057
00058
00059
00060
00061
00062
00063
00064
00065
00066
00067
00068
00069
00070
00071
00072
00073
00074
00075
00076
00077
00078
00079
00080
00081
00082
00083
00084
00085
00086
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098
00099
00100
00101
namespace
00102
{
00103
00104
00105
00106
00107
00108
00109
00110
00111
00112
00113
00114
class ResInitUsage
00115 {
00116
#ifdef HAVE_RES_INIT
00117
time_t mTime;
00118
QWaitCondition cond;
00119
QMutex mutex;
00120
int useCount;
00121
00122
bool shouldResInit()
00123 {
00124
00125
struct stat st;
00126
if (
stat(
"/etc/resolv.conf", &st) != 0)
00127
return false;
00128
00129
if (mTime < st.st_mtime)
00130 {
00131
00132
return true;
00133 }
00134
return false;
00135 }
00136
00137
void reResInit()
00138 {
00139
00140 res_init();
00141
00142
struct stat st;
00143
if (
stat(
"/etc/resolv.conf", &st) == 0)
00144 mTime = st.st_mtime;
00145 }
00146
00147
public:
00148 ResInitUsage()
00149 : mTime(0), useCount(0)
00150 { }
00151
00152
00153
00154
00155
void operator--(
int)
00156 {
00157 mutex.lock();
00158
if (--useCount == 0)
00159
00160 cond.wakeAll();
00161 mutex.unlock();
00162 }
00163
00164
00165
00166
00167
void operator++(
int)
00168 {
00169 mutex.lock();
00170
00171
if (shouldResInit())
00172 {
00173
if (useCount)
00174 {
00175
00176
00177
00178 cond.wait(&mutex);
00179 }
00180 reResInit();
00181 }
00182 useCount++;
00183 mutex.unlock();
00184 }
00185
00186
#else
00187
public:
00188 ResInitUsage()
00189 { }
00190
00191
void operator--(
int)
00192 { }
00193
00194
void operator++(
int)
00195 { }
00196
#endif
00197
00198 } resInit;
00199
00200
00201
00202
00203
00204
00205
00206
static const int maxThreadWaitTime = 20000;
00207
static const int maxThreads = 5;
00208
00209
static pid_t pid;
00210
00211 KResolverThread::KResolverThread()
00212 : data(0L)
00213 {
00214 }
00215
00216
00217
void KResolverThread::run()
00218 {
00219
00220
00221
00222
00223 KResolverManager::manager()->registerThread(
this);
00224
while (
true)
00225 {
00226 data = KResolverManager::manager()->requestData(
this, ::maxThreadWaitTime);
00227
00228
00229
if (data)
00230 {
00231
00232
00233
00234
00235 ;
00236
00237
00238 data->worker->run();
00239
00240
00241 KResolverManager::manager()->releaseData(
this, data);
00242
00243
00244 }
00245
else
00246
break;
00247 }
00248
00249 KResolverManager::manager()->unregisterThread(
this);
00250
00251 }
00252
00253
static KResolverManager *globalManager;
00254
00255 KResolverManager* KResolverManager::manager()
00256 {
00257
if (globalManager == 0L)
00258
new KResolverManager();
00259
return globalManager;
00260 }
00261
00262 KResolverManager::KResolverManager()
00263 : runningThreads(0), availableThreads(0)
00264 {
00265 globalManager =
this;
00266 workers.setAutoDelete(
true);
00267 currentRequests.setAutoDelete(
true);
00268 initStandardWorkers();
00269
00270 pid = getpid();
00271 }
00272
00273 KResolverManager::~KResolverManager()
00274 {
00275
00276
00277
00278
for (workers.first(); workers.current(); workers.next())
00279 workers.current()->terminate();
00280 }
00281
00282
void KResolverManager::registerThread(KResolverThread* )
00283 {
00284 }
00285
00286
void KResolverManager::unregisterThread(KResolverThread*)
00287 {
00288 }
00289
00290
00291 RequestData* KResolverManager::requestData(KResolverThread *th,
int maxWaitTime)
00292 {
00294
00296
00297 resInit++;
00298
00299
00300
00301
QMutexLocker locker(&mutex);
00302 RequestData *data = findData(th);
00303
00304
if (data)
00305
00306
return data;
00307
00308
00309 availableThreads++;
00310 feedWorkers.wait(&mutex, maxWaitTime);
00311 availableThreads--;
00312
00313 data = findData(th);
00314
if (data == 0L)
00315 {
00316
00317 runningThreads--;
00318 resInit--;
00319 }
00320
return data;
00321 }
00322
00323 RequestData* KResolverManager::findData(KResolverThread* th)
00324 {
00326
00327
00329
00330
00331
for (RequestData *curr = newRequests.first(); curr; curr = newRequests.next())
00332
if (!curr->worker->m_finished)
00333 {
00334
00335
if (curr->obj)
00336 curr->obj->status = KResolver::InProgress;
00337 curr->worker->th = th;
00338
00339
00340 currentRequests.append(newRequests.take());
00341
00342
return curr;
00343 }
00344
00345
00346
return 0L;
00347 }
00348
00349
00350
void KResolverManager::releaseData(KResolverThread *, RequestData* data)
00351 {
00353
00355
00356 resInit--;
00357
00358
00359
00360
00361
if (data->obj)
00362 {
00363
if (data->nRequests > 0)
00364
00365
00366 data->obj->status = KResolver::PostProcessing;
00367
else
00368
00369 data->obj->status = data->worker->results.isEmpty() ? KResolver::Failed : KResolver::Success;
00370 }
00371
00372 data->worker->m_finished =
true;
00373 data->worker->th = 0L;
00374
00375
00376 handleFinished();
00377 }
00378
00379
00380
void KResolverManager::handleFinished()
00381 {
00382
bool redo =
false;
00383
QPtrQueue<RequestData> doneRequests;
00384
00385 mutex.lock();
00386
00387
00388
00389
00390 RequestData *curr = currentRequests.last();
00391
while (curr)
00392 {
00393
if (curr->worker->th == 0L)
00394 {
00395
if (handleFinishedItem(curr))
00396 {
00397 doneRequests.enqueue(currentRequests.take());
00398
if (curr->requestor &&
00399 curr->requestor->nRequests == 0 &&
00400 curr->requestor->worker->m_finished)
00401
00402
redo =
true;
00403 }
00404 }
00405
00406 curr = currentRequests.prev();
00407 }
00408
00409
00410
while (RequestData *d = doneRequests.dequeue())
00411 doNotifying(d);
00412
00413 mutex.unlock();
00414
00415
if (
redo)
00416 {
00417
00418
00419 handleFinished();
00420 }
00421 }
00422
00423
00424
bool KResolverManager::handleFinishedItem(RequestData* curr)
00425
00426 {
00427
00428
00429
00430
if (curr->worker->m_finished && curr->nRequests == 0)
00431 {
00432
00433
if (curr->obj)
00434 curr->obj->status = KResolver::Success;
00435
00436
if (curr->requestor)
00437 --curr->requestor->nRequests;
00438
00439
00440
00441
return true;
00442 }
00443
return false;
00444 }
00445
00446
00447
00448
void KResolverManager::registerNewWorker(KResolverWorkerFactoryBase *factory)
00449 {
00450 workerFactories.append(factory);
00451 }
00452
00453 KResolverWorkerBase* KResolverManager::findWorker(KResolverPrivate* p)
00454 {
00456
00458
00459
00460
00461
00462
00463
00464
00465
00466
00467
00468 KResolverWorkerBase *worker;
00469
for (KResolverWorkerFactoryBase *factory = workerFactories.first(); factory;
00470 factory = workerFactories.next())
00471 {
00472 worker = factory->create();
00473
00474
00475 worker->input = &p->input;
00476
00477
if (worker->preprocess())
00478 {
00479
00480
if (worker->m_finished)
00481 p->status = !worker->results.isEmpty() ?
00482 KResolver::Success : KResolver::Failed;
00483
else
00484 p->status = KResolver::Queued;
00485
return worker;
00486 }
00487
00488
00489
delete worker;
00490 }
00491
00492
00493
return 0L;
00494 }
00495
00496
void KResolverManager::doNotifying(RequestData *p)
00497 {
00499
00500
00502
00503
00504
00505
00506
00507
00508
00509
00510
00511
00512
00513
00514
00515
00516
00517
00518
00519
00520
00521
00522
00523
00524
if (p->obj)
00525 {
00526
00527 p->obj->mutex.lock();
00528
KResolver* parent = p->obj->parent;
00529
00530
00531
if (p->obj->status != KResolver::Canceled
00532 && p->worker)
00533 p->worker->postprocess();
00534
00535
00536
00537
KResolverResults& r = p->obj->results;
00538
if (p->worker)
00539 {
00540 r = p->worker->results;
00541
00542 r.
setAddress(p->input->node, p->input->service);
00543
00544
00545
00546
00547
if (p->obj->status != KResolver::Canceled)
00548 {
00549 p->obj->errorcode = r.
error();
00550 p->obj->syserror = r.
systemError();
00551 p->obj->status = !r.isEmpty() ?
00552 KResolver::Success : KResolver::Failed;
00553 }
00554
else
00555 {
00556 p->obj->status = KResolver::Canceled;
00557 p->obj->errorcode = KResolver::Canceled;
00558 p->obj->syserror = 0;
00559 }
00560 }
00561
else
00562 {
00563 r.empty();
00564 r.
setError(p->obj->errorcode, p->obj->syserror);
00565 }
00566
00567
00568
if (!p->obj->waiting && parent)
00569
00570
00571
00572 QApplication::postEvent(parent,
new QEvent((QEvent::Type)(ResolutionCompleted)));
00573
00574
00575 p->obj->mutex.unlock();
00576
00577
if (parent == 0L)
00578
00579
00580
delete p->obj;
00581 }
00582
else
00583 {
00584
00585
if (p->worker)
00586 p->worker->postprocess();
00587 }
00588
00589
delete p->worker;
00590
00591
00592
00593
00594
delete p;
00595
00596
00597 notifyWaiters.wakeAll();
00598 }
00599
00600
00601
00602
00603
void KResolverManager::enqueue(
KResolver *obj, RequestData *requestor)
00604 {
00605 RequestData *newrequest =
new RequestData;
00606 newrequest->nRequests = 0;
00607 newrequest->obj = obj->
d;
00608 newrequest->input = &obj->
d->input;
00609 newrequest->requestor = requestor;
00610
00611
00612
00613
if ((newrequest->worker = findWorker(obj->
d)) == 0L)
00614 {
00615
00616
00617 obj->
d->status = KResolver::Failed;
00618 obj->
d->errorcode = KResolver::UnsupportedFamily;
00619 obj->
d->syserror = 0;
00620
00621 doNotifying(newrequest);
00622
return;
00623 }
00624
00625
00626
00627
if (requestor)
00628 requestor->nRequests++;
00629
00630
if (!newrequest->worker->m_finished)
00631 dispatch(newrequest);
00632
else if (newrequest->nRequests > 0)
00633 {
00634 mutex.lock();
00635 currentRequests.append(newrequest);
00636 mutex.unlock();
00637 }
00638
else
00639
00640 doNotifying(newrequest);
00641 }
00642
00643
00644
00645
void KResolverManager::dispatch(RequestData *data)
00646 {
00647
00648
00649
00650
00651
QMutexLocker locker(&mutex);
00652
00653
00654 newRequests.append(data);
00655
00656
00657
00658
00659
00660
00661
00662
00663
00664
00665
00666
00667
00668
00669
00670
00671
00672
00673
00674
00675
00676
00677
00678
00679
00680
00681
00682
if (availableThreads == 0 && runningThreads < maxThreads)
00683 {
00684
00685
00686
00687 KResolverThread *th = workers.first();
00688
while (th && th->running())
00689 th = workers.next();
00690
00691
if (th == 0L)
00692
00693 th =
new KResolverThread;
00694
else
00695 workers.take();
00696
00697 th->start();
00698 workers.append(th);
00699 runningThreads++;
00700 }
00701
00702 feedWorkers.wakeAll();
00703
00704
00705 workers.first();
00706
while (workers.current())
00707 {
00708
if (!workers.current()->running())
00709 workers.remove();
00710
else
00711 workers.next();
00712 }
00713 }
00714
00715
00716
00717
bool KResolverManager::dequeueNew(
KResolver* obj)
00718 {
00719
00720
00721
00722
00723
00724
if (!obj->
isRunning())
00725 {
00726 KResolverPrivate *d = obj->
d;
00727
00728
00729 RequestData *curr = newRequests.first();
00730
while (curr)
00731
if (curr->obj == d)
00732 {
00733
00734
00735 d->status = KResolver::Canceled;
00736 newRequests.take();
00737 doNotifying(curr);
00738
00739
return true;
00740 }
00741
else
00742 curr = currentRequests.next();
00743
00744
00745
return true;
00746 }
00747
00748
00749
00750 obj->
d->mutex.lock();
00751 obj->
d->status = KResolver::Canceled;
00752 obj->
d->mutex.unlock();
00753
return false;
00754 }
00755
00756
00757
00758
void KResolverManager::dequeue(
KResolver *obj)
00759 {
00760
QMutexLocker locker(&mutex);
00761 dequeueNew(obj);
00762 }
00763
00764
00765
00766
void KResolverManager::aboutToBeDeleted(
KResolver *obj)
00767 {
00768
00769
00770
00771
00772
QMutexLocker locker(&mutex);
00773 KResolverPrivate* d = obj->
d;
00774
00775
if (!dequeueNew(obj))
00776 {
00777
00778
00779 RequestData *curr = currentRequests.first();
00780
while (curr)
00781
if (curr->obj == d)
00782 {
00783
00784
00785 d->parent = 0L;
00786 d->emitSignal =
false;
00787 d->status = KResolver::Canceled;
00788
00789
return;
00790 }
00791
else
00792 curr = currentRequests.next();
00793
00794
00795 qWarning(
"KResolverManager::aboutToBeDeleted: pid = %u, obj = %p: could not find obj in list",
00796 pid, (
void*)obj);
00797 }
00798
00799
00800
00801
delete d;
00802 }
00803
00804 }