libzypp 17.36.3
request.cc
Go to the documentation of this file.
1/*---------------------------------------------------------------------\
2| ____ _ __ __ ___ |
3| |__ / \ / / . \ . \ |
4| / / \ V /| _/ _/ |
5| / /__ | | | | | | |
6| /_____||_| |_| |_| |
7| |
8----------------------------------------------------------------------*/
13#include <zypp-core/zyppng/base/EventDispatcher>
15#include <zypp-core/zyppng/core/String>
18#include <zypp-curl/CurlConfig>
19#include <zypp-curl/auth/CurlAuthData>
20#include <zypp-media/MediaConfig>
24#include <zypp-core/Pathname.h>
25#include <curl/curl.h>
26#include <stdio.h>
27#include <fcntl.h>
28#include <utility>
29
30#include <iostream>
31#include <boost/variant.hpp>
32#include <boost/variant/polymorphic_get.hpp>
33
34
35namespace zyppng {
36
37 namespace {
38 static size_t nwr_headerCallback ( char *ptr, size_t size, size_t nmemb, void *userdata ) {
39 if ( !userdata )
40 return 0;
41
42 NetworkRequestPrivate *that = reinterpret_cast<NetworkRequestPrivate *>( userdata );
43 return that->headerfunction( ptr, size * nmemb );
44 }
45 static size_t nwr_writeCallback ( char *ptr, size_t size, size_t nmemb, void *userdata ) {
46 if ( !userdata )
47 return 0;
48
49 NetworkRequestPrivate *that = reinterpret_cast<NetworkRequestPrivate *>( userdata );
50 return that->writefunction( ptr, {}, size * nmemb );
51 }
52
53 //helper for std::visit
54 template<class T> struct always_false : std::false_type {};
55 }
56
62
66
68 : _outFile( std::move(prevState._outFile) )
69 , _partialHelper( std::move(prevState._partialHelper) )
70 , _downloaded( prevState._downloaded )
71 { }
72
74 : BasePrivate(p)
75 , _url ( std::move(url) )
76 , _targetFile ( std::move( targetFile) )
77 , _fMode ( std::move(fMode) )
78 , _headers( std::unique_ptr< curl_slist, decltype (&curl_slist_free_all) >( nullptr, &curl_slist_free_all ) )
79 { }
80
82 {
83 if ( _easyHandle ) {
84 //clean up for now, later we might reuse handles
85 curl_easy_cleanup( _easyHandle );
86 //reset in request but make sure the request was not enqueued again and got a new handle
87 _easyHandle = nullptr;
88 }
89 }
90
91 bool NetworkRequestPrivate::initialize( std::string &errBuf )
92 {
93 reset();
94
95 if ( _easyHandle )
96 //will reset to defaults but keep live connections, session ID and DNS caches
97 curl_easy_reset( _easyHandle );
98 else
99 _easyHandle = curl_easy_init();
100 return setupHandle ( errBuf );
101 }
102
103 bool NetworkRequestPrivate::setupHandle( std::string &errBuf )
104 {
106 curl_easy_setopt( _easyHandle, CURLOPT_ERRORBUFFER, this->_errorBuf.data() );
107
108 const std::string urlScheme = _url.getScheme();
109 if ( urlScheme == "http" || urlScheme == "https" )
111
112 try {
113
114 setCurlOption( CURLOPT_PRIVATE, this );
116 setCurlOption( CURLOPT_XFERINFODATA, this );
117 setCurlOption( CURLOPT_NOPROGRESS, 0L);
118 setCurlOption( CURLOPT_FAILONERROR, 1L);
119 setCurlOption( CURLOPT_NOSIGNAL, 1L);
120
121 std::string urlBuffer( _url.asString() );
122 setCurlOption( CURLOPT_URL, urlBuffer.c_str() );
123
124 setCurlOption( CURLOPT_WRITEFUNCTION, nwr_writeCallback );
125 setCurlOption( CURLOPT_WRITEDATA, this );
126
128 setCurlOption( CURLOPT_CONNECT_ONLY, 1L );
129 setCurlOption( CURLOPT_FRESH_CONNECT, 1L );
130 }
132 // instead of returning no data with NOBODY, we return
133 // little data, that works with broken servers, and
134 // works for ftp as well, because retrieving only headers
135 // ftp will return always OK code ?
136 // See http://curl.haxx.se/docs/knownbugs.html #58
137 if ( _protocolMode == ProtocolMode::HTTP && _settings.headRequestsAllowed() )
138 setCurlOption( CURLOPT_NOBODY, 1L );
139 else
140 setCurlOption( CURLOPT_RANGE, "0-1" );
141 }
142
143 //make a local copy of the settings, so headers are not added multiple times
145
146 if ( _dispatcher ) {
147 locSet.setUserAgentString( _dispatcher->agentString().c_str() );
148
149 // add custom headers as configured (bsc#955801)
150 const auto &cHeaders = _dispatcher->hostSpecificHeaders();
151 if ( auto i = cHeaders.find(_url.getHost()); i != cHeaders.end() ) {
152 for ( const auto &[key, value] : i->second ) {
154 "%s: %s", key.c_str(), value.c_str() )
155 ));
156 }
157 }
158 }
159
160 locSet.addHeader("Pragma:");
161
164 {
165 case 4: setCurlOption( CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V4 ); break;
166 case 6: setCurlOption( CURLOPT_IPRESOLVE, CURL_IPRESOLVE_V6 ); break;
167 default: break;
168 }
169
170 setCurlOption( CURLOPT_HEADERFUNCTION, &nwr_headerCallback );
171 setCurlOption( CURLOPT_HEADERDATA, this );
172
176 setCurlOption( CURLOPT_CONNECTTIMEOUT, locSet.connectTimeout() );
177 // If a transfer timeout is set, also set CURLOPT_TIMEOUT to an upper limit
178 // just in case curl does not trigger its progress callback frequently
179 // enough.
180 if ( locSet.timeout() )
181 {
182 setCurlOption( CURLOPT_TIMEOUT, 3600L );
183 }
184
185 if ( urlScheme == "https" )
186 {
187 if ( :: internal::setCurlRedirProtocols ( _easyHandle ) != CURLE_OK ) {
189 }
190
191 if( locSet.verifyPeerEnabled() ||
192 locSet.verifyHostEnabled() )
193 {
194 setCurlOption(CURLOPT_CAPATH, locSet.certificateAuthoritiesPath().c_str());
195 }
196
197 if( ! locSet.clientCertificatePath().empty() )
198 {
199 setCurlOption(CURLOPT_SSLCERT, locSet.clientCertificatePath().c_str());
200 }
201 if( ! locSet.clientKeyPath().empty() )
202 {
203 setCurlOption(CURLOPT_SSLKEY, locSet.clientKeyPath().c_str());
204 }
205
206#ifdef CURLSSLOPT_ALLOW_BEAST
207 // see bnc#779177
208 setCurlOption( CURLOPT_SSL_OPTIONS, CURLSSLOPT_ALLOW_BEAST );
209#endif
210 setCurlOption(CURLOPT_SSL_VERIFYPEER, locSet.verifyPeerEnabled() ? 1L : 0L);
211 setCurlOption(CURLOPT_SSL_VERIFYHOST, locSet.verifyHostEnabled() ? 2L : 0L);
212 // bnc#903405 - POODLE: libzypp should only talk TLS
213 setCurlOption(CURLOPT_SSLVERSION, CURL_SSLVERSION_TLSv1);
214 }
215
216 // follow any Location: header that the server sends as part of
217 // an HTTP header (#113275)
218 setCurlOption( CURLOPT_FOLLOWLOCATION, 1L);
219 // 3 redirects seem to be too few in some cases (bnc #465532)
220 setCurlOption( CURLOPT_MAXREDIRS, 6L );
221
222 //set the user agent
223 setCurlOption(CURLOPT_USERAGENT, locSet.userAgentString().c_str() );
224
225
226 /*---------------------------------------------------------------*
227 CURLOPT_USERPWD: [user name]:[password]
228 Url::username/password -> CURLOPT_USERPWD
229 If not provided, anonymous FTP identification
230 *---------------------------------------------------------------*/
231 if ( locSet.userPassword().size() )
232 {
233 setCurlOption(CURLOPT_USERPWD, locSet.userPassword().c_str());
234 std::string use_auth = _settings.authType();
235 if (use_auth.empty())
236 use_auth = "digest,basic"; // our default
238 if( auth != CURLAUTH_NONE)
239 {
240 DBG << _easyHandle << " " << "Enabling HTTP authentication methods: " << use_auth
241 << " (CURLOPT_HTTPAUTH=" << auth << ")" << std::endl;
242 setCurlOption(CURLOPT_HTTPAUTH, auth);
243 }
244 }
245
246 if ( locSet.proxyEnabled() && ! locSet.proxy().empty() )
247 {
248 DBG << _easyHandle << " " << "Proxy: '" << locSet.proxy() << "'" << std::endl;
249 setCurlOption(CURLOPT_PROXY, locSet.proxy().c_str());
250 setCurlOption(CURLOPT_PROXYAUTH, CURLAUTH_BASIC|CURLAUTH_DIGEST|CURLAUTH_NTLM );
251
252 /*---------------------------------------------------------------*
253 * CURLOPT_PROXYUSERPWD: [user name]:[password]
254 *
255 * Url::option(proxyuser and proxypassword) -> CURLOPT_PROXYUSERPWD
256 * If not provided, $HOME/.curlrc is evaluated
257 *---------------------------------------------------------------*/
258
259 std::string proxyuserpwd = locSet.proxyUserPassword();
260
261 if ( proxyuserpwd.empty() )
262 {
264 zypp::media::CurlConfig::parseConfig(curlconf); // parse ~/.curlrc
265 if ( curlconf.proxyuserpwd.empty() )
266 DBG << _easyHandle << " " << "Proxy: ~/.curlrc does not contain the proxy-user option" << std::endl;
267 else
268 {
269 proxyuserpwd = curlconf.proxyuserpwd;
270 DBG << _easyHandle << " " << "Proxy: using proxy-user from ~/.curlrc" << std::endl;
271 }
272 }
273 else
274 {
275 DBG << _easyHandle << " " << _easyHandle << " " << "Proxy: using provided proxy-user '" << _settings.proxyUsername() << "'" << std::endl;
276 }
277
278 if ( ! proxyuserpwd.empty() )
279 {
280 setCurlOption(CURLOPT_PROXYUSERPWD, ::internal::curlUnEscape( proxyuserpwd ).c_str());
281 }
282 }
283#if CURLVERSION_AT_LEAST(7,19,4)
284 else if ( locSet.proxy() == EXPLICITLY_NO_PROXY )
285 {
286 // Explicitly disabled in URL (see fillSettingsFromUrl()).
287 // This should also prevent libcurl from looking into the environment.
288 DBG << _easyHandle << " " << "Proxy: explicitly NOPROXY" << std::endl;
289 setCurlOption(CURLOPT_NOPROXY, "*");
290 }
291
292#endif
293 // else: Proxy: not explicitly set; libcurl may look into the environment
294
296 if ( locSet.minDownloadSpeed() != 0 )
297 {
298 setCurlOption(CURLOPT_LOW_SPEED_LIMIT, locSet.minDownloadSpeed());
299 // default to 10 seconds at low speed
300 setCurlOption(CURLOPT_LOW_SPEED_TIME, 60L);
301 }
302
303#if CURLVERSION_AT_LEAST(7,15,5)
304 if ( locSet.maxDownloadSpeed() != 0 )
305 setCurlOption(CURLOPT_MAX_RECV_SPEED_LARGE, locSet.maxDownloadSpeed());
306#endif
307
309 if ( locSet.cookieFileEnabled() )
310 setCurlOption( CURLOPT_COOKIEFILE, _currentCookieFile.c_str() );
311 else
312 MIL << _easyHandle << " " << "No cookies requested" << std::endl;
313 setCurlOption(CURLOPT_COOKIEJAR, _currentCookieFile.c_str() );
314
315#if CURLVERSION_AT_LEAST(7,18,0)
316 // bnc #306272
317 setCurlOption(CURLOPT_PROXY_TRANSFER_MODE, 1L );
318#endif
319
320 // Append settings custom headers to curl.
321 // TransferSettings assert strings are trimmed (HTTP/2 RFC 9113)
322 for ( const auto &header : locSet.headers() ) {
323 if ( !z_func()->addRequestHeader( header.c_str() ) )
325 }
326
327 if ( _headers )
328 setCurlOption( CURLOPT_HTTPHEADER, _headers.get() );
329
330 // set up ranges if required
332 if ( _requestedRanges.size() ) {
333
334 std::sort( _requestedRanges.begin(), _requestedRanges.end(), []( const auto &elem1, const auto &elem2 ){
335 return ( elem1._start < elem2._start );
336 });
337
338 CurlMultiPartHandler *helper = nullptr;
339 if ( auto initState = std::get_if<pending_t>(&_runningMode) ) {
340
342 initState->_partialHelper = std::make_unique<CurlMultiPartHandler>(
343 multiPartMode
346 , *this
347 );
348 helper = initState->_partialHelper.get();
349
350 } else if ( auto pendingState = std::get_if<prepareNextRangeBatch_t>(&_runningMode) ) {
351 helper = pendingState->_partialHelper.get();
352 } else {
353 errBuf = "Request is in invalid state to call setupHandle";
354 return false;
355 }
356
357 if ( !helper->prepare () ) {
358 errBuf = helper->lastErrorMessage ();
359 return false;
360 }
361 }
362 }
363
364 return true;
365
366 } catch ( const zypp::Exception &excp ) {
367 ZYPP_CAUGHT(excp);
368 errBuf = excp.asString();
369 }
370 return false;
371 }
372
374 {
375 auto rmode = std::get_if<NetworkRequestPrivate::running_t>( &_runningMode );
376 if ( !rmode ) {
377 DBG << _easyHandle << "Can only create output file in running mode" << std::endl;
378 return false;
379 }
380 // if we have no open file create or open it
381 if ( !rmode->_outFile ) {
382 std::string openMode = "w+b";
384 openMode = "r+b";
385
386 rmode->_outFile = fopen( _targetFile.asString().c_str() , openMode.c_str() );
387
388 //if the file does not exist create a new one
389 if ( !rmode->_outFile && _fMode == NetworkRequest::WriteShared ) {
390 rmode->_outFile = fopen( _targetFile.asString().c_str() , "w+b" );
391 }
392
393 if ( !rmode->_outFile ) {
395 ,zypp::str::Format("Unable to open target file (%1%). Errno: (%2%:%3%)") % _targetFile.asString() % errno % strerr_cxx() );
396 return false;
397 }
398 }
399
400 return true;
401 }
402
404 {
405 // We can recover from RangeFail errors if the helper indicates it
406 auto rmode = std::get_if<NetworkRequestPrivate::running_t>( &_runningMode );
407 if ( rmode->_partialHelper ) return rmode->_partialHelper->canRecover();
408 return false;
409 }
410
412 {
413 auto rmode = std::get_if<NetworkRequestPrivate::running_t>( &_runningMode );
414 if ( !rmode ) {
415 errBuf = "NetworkRequestPrivate::prepareToContinue called in invalid state";
416 return false;
417 }
418
419 if ( rmode->_partialHelper && rmode->_partialHelper->hasMoreWork() ) {
420
421 bool hadRangeFail = rmode->_partialHelper->lastError () == NetworkRequestError::RangeFail;
422
423 _runningMode = prepareNextRangeBatch_t( std::move(std::get<running_t>( _runningMode )) );
424
425 auto &prepMode = std::get<prepareNextRangeBatch_t>(_runningMode);
426 if ( !prepMode._partialHelper->prepareToContinue() ) {
427 errBuf = prepMode._partialHelper->lastErrorMessage();
428 return false;
429 }
430
431 if ( hadRangeFail ) {
432 // we reset the handle to default values. We do this to not run into
433 // "transfer closed with outstanding read data remaining" error CURL sometimes returns when
434 // we cancel a connection because of a range error to request a smaller batch.
435 // The error will still happen but much less frequently than without resetting the handle.
436 //
437 // Note: Even creating a new handle will NOT fix the issue
438 curl_easy_reset( _easyHandle );
439 }
440 if ( !setupHandle(errBuf))
441 return false;
442
443 return true;
444 }
445 errBuf = "Request has no more work";
446 return false;
447
448 }
449
451 {
452 // check if we have ranges that have never been requested
453 return std::any_of( _requestedRanges.begin(), _requestedRanges.end(), []( const auto &range ){ return range._rangeState == CurlMultiPartHandler::Pending; });
454 }
455
457 {
458 bool isRangeContinuation = std::holds_alternative<prepareNextRangeBatch_t>( _runningMode );
459 if ( isRangeContinuation ) {
460 MIL << _easyHandle << " " << "Continuing a previously started range batch." << std::endl;
461 _runningMode = running_t( std::move(std::get<prepareNextRangeBatch_t>( _runningMode )) );
462 } else {
463 _runningMode = running_t( std::move(std::get<pending_t>( _runningMode )) );
464 }
465
466 auto &m = std::get<running_t>( _runningMode );
467
468 if ( m._activityTimer ) {
469 DBG_MEDIA << _easyHandle << " Setting activity timeout to: " << _settings.timeout() << std::endl;
470 m._activityTimer->connect( &Timer::sigExpired, *this, &NetworkRequestPrivate::onActivityTimeout );
471 m._activityTimer->start( static_cast<uint64_t>( _settings.timeout() * 1000 ) );
472 }
473
474 if ( !isRangeContinuation )
475 _sigStarted.emit( *z_func() );
476 }
477
479 {
480 if ( std::holds_alternative<running_t>(_runningMode) ) {
481 auto &rmode = std::get<running_t>( _runningMode );
482 if ( rmode._partialHelper )
483 rmode._partialHelper->finalize();
484 }
485 }
486
488 {
489 finished_t resState;
490 resState._result = std::move(err);
491
492 if ( std::holds_alternative<running_t>(_runningMode) ) {
493
494 auto &rmode = std::get<running_t>( _runningMode );
495 resState._downloaded = rmode._downloaded;
496 resState._contentLenght = rmode._contentLenght;
497
499 if ( _requestedRanges.size( ) ) {
500 //we have a successful download lets see if we got everything we needed
501 if ( !rmode._partialHelper->verifyData() ){
502 NetworkRequestError::Type err = rmode._partialHelper->lastError();
503 resState._result = NetworkRequestErrorPrivate::customError( err, std::string(rmode._partialHelper->lastErrorMessage()) );
504 }
505
506 // if we have ranges we need to fill our digest from the full file
508 if ( fseek( rmode._outFile, 0, SEEK_SET ) != 0 ) {
509 resState._result = NetworkRequestErrorPrivate::customError( NetworkRequestError::InternalError, "Unable to set output file pointer." );
510 } else {
511 constexpr size_t bufSize = 4096;
512 char buf[bufSize];
513 while( auto cnt = fread(buf, 1, bufSize, rmode._outFile ) > 0 ) {
514 _fileVerification->_fileDigest.update(buf, cnt);
515 }
516 }
517 }
518 } // if ( _requestedRanges.size( ) )
519 }
520
521 // finally check the file digest if we have one
523 const UByteArray &calcSum = _fileVerification->_fileDigest.digestVector ();
524 const UByteArray &expSum = zypp::Digest::hexStringToUByteArray( _fileVerification->_fileChecksum.checksum () );
525 if ( calcSum != expSum ) {
528 , (zypp::str::Format("Invalid file checksum %1%, expected checksum %2%")
529 % _fileVerification->_fileDigest.digest()
530 % _fileVerification->_fileChecksum.checksum () ) );
531 }
532 }
533
534 rmode._outFile.reset();
535 }
536
537 _runningMode = std::move( resState );
538 _sigFinished.emit( *z_func(), std::get<finished_t>(_runningMode)._result );
539 }
540
542 {
544 _headers.reset( nullptr );
545 _errorBuf.fill( 0 );
547
548 if ( _fileVerification )
549 _fileVerification->_fileDigest.reset ();
550
551 std::for_each( _requestedRanges.begin (), _requestedRanges.end(), []( CurlMultiPartHandler::Range &range ) {
552 range.restart();
553 });
554 }
555
557 {
558 MIL_MEDIA << _easyHandle << " Request timeout interval: " << t.interval()<< " remaining: " << t.remaining() << std::endl;
559 std::map<std::string, boost::any> extraInfo;
560 extraInfo.insert( {"requestUrl", _url } );
561 extraInfo.insert( {"filepath", _targetFile } );
562 _dispatcher->cancel( *z_func(), NetworkRequestErrorPrivate::customError( NetworkRequestError::Timeout, "Download timed out", std::move(extraInfo) ) );
563 }
564
566 {
567 return std::string( _errorBuf.data() );
568 }
569
571 {
572 if ( std::holds_alternative<running_t>( _runningMode ) ){
573 auto &rmode = std::get<running_t>( _runningMode );
574 if ( rmode._activityTimer && rmode._activityTimer->isRunning() )
575 rmode._activityTimer->start();
576 }
577 }
578
579 int NetworkRequestPrivate::curlProgressCallback( void *clientp, curl_off_t dltotal, curl_off_t dlnow, curl_off_t ultotal, curl_off_t ulnow )
580 {
581 if ( !clientp )
582 return CURLE_OK;
583 NetworkRequestPrivate *that = reinterpret_cast<NetworkRequestPrivate *>( clientp );
584
585 if ( !std::holds_alternative<running_t>(that->_runningMode) ){
586 DBG << that->_easyHandle << " " << "Curl progress callback was called in invalid state "<< that->z_func()->state() << std::endl;
587 return -1;
588 }
589
590 auto &rmode = std::get<running_t>( that->_runningMode );
591
592 //reset the timer
593 that->resetActivityTimer();
594
595 rmode._isInCallback = true;
596 if ( rmode._lastProgressNow != dlnow ) {
597 rmode._lastProgressNow = dlnow;
598 that->_sigProgress.emit( *that->z_func(), dltotal, dlnow, ultotal, ulnow );
599 }
600 rmode._isInCallback = false;
601
602 return rmode._cachedResult ? CURLE_ABORTED_BY_CALLBACK : CURLE_OK;
603 }
604
605 size_t NetworkRequestPrivate::headerfunction( char *ptr, size_t bytes )
606 {
607 //it is valid to call this function with no data to write, just return OK
608 if ( bytes == 0)
609 return 0;
610
612
614
615 std::string_view hdr( ptr, bytes );
616
617 hdr.remove_prefix( std::min( hdr.find_first_not_of(" \t\r\n"), hdr.size() ) );
618 const auto lastNonWhitespace = hdr.find_last_not_of(" \t\r\n");
619 if ( lastNonWhitespace != hdr.npos )
620 hdr.remove_suffix( hdr.size() - (lastNonWhitespace + 1) );
621 else
622 hdr = std::string_view();
623
624 auto &rmode = std::get<running_t>( _runningMode );
625 if ( !hdr.size() ) {
626 return ( bytes );
627 }
628 if ( _expectedFileSize && rmode._partialHelper ) {
629 const auto &repSize = rmode._partialHelper->reportedFileSize ();
630 if ( repSize && repSize != _expectedFileSize ) {
631 rmode._cachedResult = NetworkRequestErrorPrivate::customError( NetworkRequestError::InternalError, "Reported downloaded data length is not what was expected." );
632 return 0;
633 }
634 }
635 if ( zypp::strv::hasPrefixCI( hdr, "HTTP/" ) ) {
636
637 long statuscode = 0;
638 (void)curl_easy_getinfo( _easyHandle, CURLINFO_RESPONSE_CODE, &statuscode);
639
640 // if we have a status 204 we need to create a empty file
641 if( statuscode == 204 && !( _options & NetworkRequest::ConnectionTest ) && !( _options & NetworkRequest::HeadRequest ) )
643
644 } else if ( zypp::strv::hasPrefixCI( hdr, "Location:" ) ) {
645 _lastRedirect = hdr.substr( 9 );
646 DBG << _easyHandle << " " << "redirecting to " << _lastRedirect << std::endl;
647
648 } else if ( zypp::strv::hasPrefixCI( hdr, "Content-Length:") ) {
649 auto lenStr = str::trim( hdr.substr( 15 ), zypp::str::TRIM );
650 auto str = std::string ( lenStr.data(), lenStr.length() );
652 if ( len > 0 ) {
653 DBG << _easyHandle << " " << "Got Content-Length Header: " << len << std::endl;
654 rmode._contentLenght = zypp::ByteCount(len, zypp::ByteCount::B);
655 }
656 }
657 }
658
659 return bytes;
660 }
661
662 size_t NetworkRequestPrivate::writefunction( char *data, std::optional<off_t> offset, size_t max )
663 {
665
666 //it is valid to call this function with no data to write, just return OK
667 if ( max == 0)
668 return 0;
669
670 //in case of a HEAD request, we do not write anything
672 return ( max );
673 }
674
675 auto &rmode = std::get<running_t>( _runningMode );
676
677 // if we have no open file create or open it
678 if ( !assertOutputFile() )
679 return 0;
680
681 if ( offset ) {
682 // seek to the given offset
683 if ( fseek( rmode._outFile, *offset, SEEK_SET ) != 0 ) {
685 "Unable to set output file pointer." );
686 return 0;
687 }
688 rmode._currentFileOffset = *offset;
689 }
690
691 if ( _expectedFileSize && rmode._partialHelper ) {
692 const auto &repSize = rmode._partialHelper->reportedFileSize ();
693 if ( repSize && repSize != _expectedFileSize ) {
694 rmode._cachedResult = NetworkRequestErrorPrivate::customError( NetworkRequestError::InternalError, "Reported downloaded data length is not what was expected." );
695 return 0;
696 }
697 }
698
699 //make sure we do not write after the expected file size
700 if ( _expectedFileSize && static_cast<zypp::ByteCount::SizeType>( rmode._currentFileOffset + max) > _expectedFileSize ) {
701 rmode._cachedResult = NetworkRequestErrorPrivate::customError( NetworkRequestError::ExceededMaxLen, "Downloaded data exceeds expected length." );
702 return 0;
703 }
704
705 auto written = fwrite( data, 1, max, rmode._outFile );
706 if ( written == 0 )
707 return 0;
708
709 // if we are not downloading in ranges, we can update the file digest on the fly if we have one
710 if ( !rmode._partialHelper && _fileVerification ) {
711 _fileVerification->_fileDigest.update( data, written );
712 }
713
714 rmode._currentFileOffset += written;
715
716 // count the number of real bytes we have downloaded so far
717 rmode._downloaded += written;
718 _sigBytesDownloaded.emit( *z_func(), rmode._downloaded );
719
720 return written;
721 }
722
724 {
725 auto rmode = std::get_if<NetworkRequestPrivate::running_t>( &_runningMode );
726 if ( !rmode || !rmode->_partialHelper || !rmode->_partialHelper->hasError() )
727 return;
728
729 // oldest cached result wins
730 if ( rmode->_cachedResult )
731 return;
732
733 auto lastErr = NetworkRequestErrorPrivate::customError( rmode->_partialHelper->lastError() , std::string(rmode->_partialHelper->lastErrorMessage()) );
734 MIL_MEDIA << _easyHandle << " Multipart handler announced error code " << lastErr.toString () << std::endl;
735 rmode->_cachedResult = lastErr;
736 }
737
739
741 : Base ( *new NetworkRequestPrivate( std::move(url), std::move(targetFile), std::move(fMode), *this ) )
742 {
743 }
744
746 {
747 Z_D();
748
749 if ( d->_dispatcher )
750 d->_dispatcher->cancel( *this, "Request destroyed while still running" );
751 }
752
754 {
755 d_func()->_expectedFileSize = std::move( expectedFileSize );
756 }
757
759 {
760 return d_func()->_expectedFileSize;
761 }
762
763 void NetworkRequest::setPriority( NetworkRequest::Priority prio, bool triggerReschedule )
764 {
765 Z_D();
766 d->_priority = prio;
767 if ( state() == Pending && triggerReschedule && d->_dispatcher )
768 d->_dispatcher->reschedule();
769 }
770
772 {
773 return d_func()->_priority;
774 }
775
776 void NetworkRequest::setOptions( Options opt )
777 {
778 d_func()->_options = opt;
779 }
780
781 NetworkRequest::Options NetworkRequest::options() const
782 {
783 return d_func()->_options;
784 }
785
786 void NetworkRequest::addRequestRange(size_t start, size_t len, std::optional<zypp::Digest> &&digest, CheckSumBytes expectedChkSum , std::any userData, std::optional<size_t> digestCompareLen, std::optional<size_t> chksumpad )
787 {
788 Z_D();
789 if ( state() == Running )
790 return;
791
792 d->_requestedRanges.push_back( Range::make( start, len, std::move(digest), std::move( expectedChkSum ), std::move( userData ), digestCompareLen, chksumpad ) );
793 }
794
796 {
797 Z_D();
798 if ( state() == Running )
799 return;
800
801 d->_requestedRanges.push_back( std::move(range) );
802 auto &rng = d->_requestedRanges.back();
803 rng._rangeState = CurlMultiPartHandler::Pending;
804 rng.bytesWritten = 0;
805 if ( rng._digest )
806 rng._digest->reset();
807 }
808
810 {
811 Z_D();
812 if ( state() == Running )
813 return false;
814
815 zypp::Digest fDig;
816 if ( !fDig.create( expected.type () ) )
817 return false;
818
819 d->_fileVerification = NetworkRequestPrivate::FileVerifyInfo{
820 ._fileDigest = std::move(fDig),
821 ._fileChecksum = expected
822 };
823 return true;
824 }
825
827 {
828 Z_D();
829 if ( state() == Running )
830 return;
831 d->_requestedRanges.clear();
832 }
833
834 std::vector<NetworkRequest::Range> NetworkRequest::failedRanges() const
835 {
836 const auto mystate = state();
837 if ( mystate != Finished && mystate != Error )
838 return {};
839
840 Z_D();
841
842 std::vector<Range> failed;
843 for ( auto &r : d->_requestedRanges ) {
844 if ( r._rangeState != CurlMultiPartHandler::Finished ) {
845 failed.push_back( r.clone() );
846 }
847 }
848 return failed;
849 }
850
851 const std::vector<NetworkRequest::Range> &NetworkRequest::requestedRanges() const
852 {
853 return d_func()->_requestedRanges;
854 }
855
856 const std::string &NetworkRequest::lastRedirectInfo() const
857 {
858 return d_func()->_lastRedirect;
859 }
860
862 {
863 return d_func()->_easyHandle;
864 }
865
866 std::optional<zyppng::NetworkRequest::Timings> NetworkRequest::timings() const
867 {
868 const auto myerr = error();
869 const auto mystate = state();
870 if ( mystate != Finished )
871 return {};
872
873 Timings t;
874
875 auto getMeasurement = [ this ]( const CURLINFO info, std::chrono::microseconds &target ){
876 using FPSeconds = std::chrono::duration<double, std::chrono::seconds::period>;
877 double val = 0;
878 const auto res = curl_easy_getinfo( d_func()->_easyHandle, info, &val );
879 if ( CURLE_OK == res ) {
880 target = std::chrono::duration_cast<std::chrono::microseconds>( FPSeconds(val) );
881 }
882 };
883
884 getMeasurement( CURLINFO_NAMELOOKUP_TIME, t.namelookup );
885 getMeasurement( CURLINFO_CONNECT_TIME, t.connect);
886 getMeasurement( CURLINFO_APPCONNECT_TIME, t.appconnect);
887 getMeasurement( CURLINFO_PRETRANSFER_TIME , t.pretransfer);
888 getMeasurement( CURLINFO_TOTAL_TIME, t.total);
889 getMeasurement( CURLINFO_REDIRECT_TIME, t.redirect);
890
891 return t;
892 }
893
894 std::vector<char> NetworkRequest::peekData( off_t offset, size_t count ) const
895 {
896 Z_D();
897
898 if ( !std::holds_alternative<NetworkRequestPrivate::running_t>( d->_runningMode) )
899 return {};
900
901 const auto &rmode = std::get<NetworkRequestPrivate::running_t>( d->_runningMode );
902 return zypp::io::peek_data_fd( rmode._outFile, offset, count );
903 }
904
906 {
907 return d_func()->_url;
908 }
909
911 {
912 Z_D();
914 return;
915
916 d->_url = url;
917 }
918
920 {
921 return d_func()->_targetFile;
922 }
923
925 {
926 Z_D();
928 return;
929 d->_targetFile = path;
930 }
931
933 {
934 return d_func()->_fMode;
935 }
936
938 {
939 Z_D();
941 return;
942 d->_fMode = std::move( mode );
943 }
944
945 std::string NetworkRequest::contentType() const
946 {
947 char *ptr = NULL;
948 if ( curl_easy_getinfo( d_func()->_easyHandle, CURLINFO_CONTENT_TYPE, &ptr ) == CURLE_OK && ptr )
949 return std::string(ptr);
950 return std::string();
951 }
952
954 {
955 return std::visit([](auto& arg) -> zypp::ByteCount {
956 using T = std::decay_t<decltype(arg)>;
957 if constexpr (std::is_same_v<T, NetworkRequestPrivate::pending_t> || std::is_same_v<T, NetworkRequestPrivate::prepareNextRangeBatch_t> )
958 return zypp::ByteCount(0);
959 else if constexpr (std::is_same_v<T, NetworkRequestPrivate::running_t>
960 || std::is_same_v<T, NetworkRequestPrivate::finished_t>)
961 return arg._contentLenght;
962 else
963 static_assert(always_false<T>::value, "Unhandled state type");
964 }, d_func()->_runningMode);
965 }
966
968 {
969 return std::visit([](auto& arg) -> zypp::ByteCount {
970 using T = std::decay_t<decltype(arg)>;
971 if constexpr (std::is_same_v<T, NetworkRequestPrivate::pending_t>)
972 return zypp::ByteCount();
973 else if constexpr (std::is_same_v<T, NetworkRequestPrivate::running_t>
974 || std::is_same_v<T, NetworkRequestPrivate::prepareNextRangeBatch_t>
975 || std::is_same_v<T, NetworkRequestPrivate::finished_t>)
976 return arg._downloaded;
977 else
978 static_assert(always_false<T>::value, "Unhandled state type");
979 }, d_func()->_runningMode);
980 }
981
983 {
984 return d_func()->_settings;
985 }
986
988 {
989 return std::visit([this](auto& arg) {
990 using T = std::decay_t<decltype(arg)>;
991 if constexpr (std::is_same_v<T, NetworkRequestPrivate::pending_t>)
992 return Pending;
993 else if constexpr (std::is_same_v<T, NetworkRequestPrivate::running_t> || std::is_same_v<T, NetworkRequestPrivate::prepareNextRangeBatch_t> )
994 return Running;
995 else if constexpr (std::is_same_v<T, NetworkRequestPrivate::finished_t>) {
996 if ( std::get<NetworkRequestPrivate::finished_t>( d_func()->_runningMode )._result.isError() )
997 return Error;
998 else
999 return Finished;
1000 }
1001 else
1002 static_assert(always_false<T>::value, "Unhandled state type");
1003 }, d_func()->_runningMode);
1004 }
1005
1007 {
1008 const auto s = state();
1009 if ( s != Error && s != Finished )
1010 return NetworkRequestError();
1011 return std::get<NetworkRequestPrivate::finished_t>( d_func()->_runningMode)._result;
1012 }
1013
1015 {
1016 if ( !hasError() )
1017 return std::string();
1018
1019 return error().nativeErrorString();
1020 }
1021
1023 {
1024 return error().isError();
1025 }
1026
1027 bool NetworkRequest::addRequestHeader( const std::string &header )
1028 {
1029 Z_D();
1030
1031 curl_slist *res = curl_slist_append( d->_headers ? d->_headers.get() : nullptr, header.c_str() );
1032 if ( !res )
1033 return false;
1034
1035 if ( !d->_headers )
1036 d->_headers = std::unique_ptr< curl_slist, decltype (&curl_slist_free_all) >( res, &curl_slist_free_all );
1037
1038 return true;
1039 }
1040
1042 {
1043 return d_func()->_currentCookieFile;
1044 }
1045
1047 {
1048 d_func()->_currentCookieFile = std::move(cookieFile);
1049 }
1050
1052 {
1053 return d_func()->_sigStarted;
1054 }
1055
1057 {
1058 return d_func()->_sigBytesDownloaded;
1059 }
1060
1061 SignalProxy<void (NetworkRequest &req, off_t dltotal, off_t dlnow, off_t ultotal, off_t ulnow)> NetworkRequest::sigProgress()
1062 {
1063 return d_func()->_sigProgress;
1064 }
1065
1067 {
1068 return d_func()->_sigFinished;
1069 }
1070
1071}
Store and operate with byte count.
Definition ByteCount.h:32
Unit::ValueType SizeType
Definition ByteCount.h:38
static const Unit B
1 Byte
Definition ByteCount.h:43
Compute Message Digests (MD5, SHA1 etc)
Definition Digest.h:38
bool create(const std::string &name)
initialize creation of a new message digest
Definition Digest.cc:197
Base class for Exception.
Definition Exception.h:147
std::string asString() const
Error message provided by dumpOn as string.
Definition Exception.cc:111
Url manipulation class.
Definition Url.h:93
const char * c_str() const
String representation.
Definition Pathname.h:112
bool empty() const
Test for an empty path.
Definition Pathname.h:116
static long auth_type_str2long(std::string &auth_type_str)
Converts a string of comma separated list of authetication type names into a long of ORed CURLAUTH_* ...
long maxDownloadSpeed() const
Maximum download speed (bytes per second)
long connectTimeout() const
connection timeout
long timeout() const
transfer timeout
const Pathname & clientCertificatePath() const
SSL client certificate file.
std::string userPassword() const
returns the user and password as a user:pass string
long minDownloadSpeed() const
Minimum download speed (bytes per second) until the connection is dropped.
const Headers & headers() const
returns a list of all added headers (trimmed)
const std::string & proxy() const
proxy host
const Pathname & clientKeyPath() const
SSL client key file.
void setUserAgentString(std::string &&val_r)
sets the user agent ie: "Mozilla v3" (trims)
void addHeader(std::string &&val_r)
add a header, on the form "Foo: Bar" (trims)
std::string proxyUserPassword() const
returns the proxy user and password as a user:pass string
bool verifyHostEnabled() const
Whether to verify host for ssl.
const std::string & userAgentString() const
user agent string (trimmed)
bool proxyEnabled() const
proxy is enabled
const Pathname & certificateAuthoritiesPath() const
SSL certificate authorities path ( default: /etc/ssl/certs )
bool verifyPeerEnabled() const
Whether to verify peer for ssl.
BasePrivate(Base &b)
Definition base_p.h:17
The CurlMultiPartHandler class.
const std::string & lastErrorMessage() const
static zyppng::NetworkRequestError customError(NetworkRequestError::Type t, std::string &&errorMsg="", std::map< std::string, boost::any > &&extraInfo={})
The NetworkRequestError class Represents a error that occured in.
Type type() const
type Returns the type of the error
std::string nativeErrorString() const
bool isError() const
isError Will return true if this is a actual error
size_t headerfunction(char *ptr, size_t bytes) override
Definition request.cc:605
std::optional< FileVerifyInfo > _fileVerification
The digest for the full file.
Definition request_p.h:117
enum zyppng::NetworkRequestPrivate::ProtocolMode _protocolMode
void notifyErrorCodeChanged() override
Definition request.cc:723
zypp::Pathname _currentCookieFile
Definition request_p.h:123
Signal< void(NetworkRequest &req, zypp::ByteCount count)> _sigBytesDownloaded
Definition request_p.h:130
NetworkRequestDispatcher * _dispatcher
Definition request_p.h:126
std::vector< NetworkRequest::Range > _requestedRanges
the requested ranges that need to be downloaded
Definition request_p.h:111
size_t writefunction(char *ptr, std::optional< off_t > offset, size_t bytes) override
Definition request.cc:662
static int curlProgressCallback(void *clientp, curl_off_t dltotal, curl_off_t dlnow, curl_off_t ultotal, curl_off_t ulnow)
Definition request.cc:579
std::string errorMessage() const
Definition request.cc:565
Signal< void(NetworkRequest &req)> _sigStarted
Definition request_p.h:129
NetworkRequest::FileMode _fMode
Definition request_p.h:119
std::variant< pending_t, running_t, prepareNextRangeBatch_t, finished_t > _runningMode
Definition request_p.h:188
bool initialize(std::string &errBuf)
Definition request.cc:91
void onActivityTimeout(Timer &)
Definition request.cc:556
Signal< void(NetworkRequest &req, off_t dltotal, off_t dlnow, off_t ultotal, off_t ulnow)> _sigProgress
Definition request_p.h:131
std::string _lastRedirect
to log/report redirections
Definition request_p.h:122
NetworkRequest::Options _options
Definition request_p.h:109
bool prepareToContinue(std::string &errBuf)
Definition request.cc:411
void setResult(NetworkRequestError &&err)
Definition request.cc:487
~NetworkRequestPrivate() override
Definition request.cc:81
std::array< char, CURL_ERROR_SIZE+1 > _errorBuf
Definition request_p.h:95
bool setupHandle(std::string &errBuf)
Definition request.cc:103
NetworkRequestPrivate(Url &&url, zypp::Pathname &&targetFile, NetworkRequest::FileMode fMode, NetworkRequest &p)
Definition request.cc:73
TransferSettings _settings
Definition request_p.h:108
void setCurlOption(CURLoption opt, T data)
Definition request_p.h:98
zypp::ByteCount _expectedFileSize
Definition request_p.h:110
Signal< void(NetworkRequest &req, const NetworkRequestError &err)> _sigFinished
Definition request_p.h:132
std::unique_ptr< curl_slist, decltype(&curl_slist_free_all) > _headers
Definition request_p.h:140
bool setExpectedFileChecksum(const zypp::CheckSum &expected)
Definition request.cc:809
zypp::ByteCount reportedByteCount() const
Returns the number of bytes that are reported from the backend as the full download size,...
Definition request.cc:953
const zypp::Pathname & targetFilePath() const
Returns the target filename path.
Definition request.cc:919
zypp::ByteCount downloadedByteCount() const
Returns the number of already downloaded bytes as reported by the backend.
Definition request.cc:967
void setUrl(const Url &url)
This will change the URL of the request.
Definition request.cc:910
void setExpectedFileSize(zypp::ByteCount expectedFileSize)
Definition request.cc:753
void setPriority(Priority prio, bool triggerReschedule=true)
Definition request.cc:763
std::vector< char > peekData(off_t offset, size_t count) const
Definition request.cc:894
std::string contentType() const
Returns the content type as reported from the server.
Definition request.cc:945
void setFileOpenMode(FileMode mode)
Sets the file open mode to mode.
Definition request.cc:937
bool addRequestHeader(const std::string &header)
Definition request.cc:1027
~NetworkRequest() override
Definition request.cc:745
void setOptions(Options opt)
Definition request.cc:776
FileMode fileOpenMode() const
Returns the currently configured file open mode.
Definition request.cc:932
zypp::ByteCount expectedFileSize() const
Definition request.cc:758
bool hasError() const
Checks if there was a error with the request.
Definition request.cc:1022
State state() const
Returns the current state the HttpDownloadRequest is in.
Definition request.cc:987
SignalProxy< void(NetworkRequest &req, const NetworkRequestError &err)> sigFinished()
Signals that the download finished.
Definition request.cc:1066
UByteArray CheckSumBytes
Definition request.h:49
Options options() const
Definition request.cc:781
SignalProxy< void(NetworkRequest &req, zypp::ByteCount count)> sigBytesDownloaded()
Signals that new data has been downloaded, this is only the payload and does not include control data...
Definition request.cc:1056
std::optional< Timings > timings() const
After the request is finished query the timings that were collected during download.
Definition request.cc:866
std::string extendedErrorString() const
In some cases, curl can provide extended error information collected at runtime.
Definition request.cc:1014
NetworkRequest(Url url, zypp::Pathname targetFile, FileMode fMode=WriteExclusive)
Definition request.cc:740
Priority priority() const
Definition request.cc:771
NetworkRequestError error() const
Returns the last set Error.
Definition request.cc:1006
void setTargetFilePath(const zypp::Pathname &path)
Changes the target file path of the download.
Definition request.cc:924
const zypp::Pathname & cookieFile() const
Definition request.cc:1041
void * nativeHandle() const
Definition request.cc:861
void setCookieFile(zypp::Pathname cookieFile)
Definition request.cc:1046
void addRequestRange(size_t start, size_t len=0, std::optional< zypp::Digest > &&digest={}, CheckSumBytes expectedChkSum=CheckSumBytes(), std::any userData=std::any(), std::optional< size_t > digestCompareLen={}, std::optional< size_t > chksumpad={})
Definition request.cc:786
std::vector< Range > failedRanges() const
Definition request.cc:834
const std::vector< Range > & requestedRanges() const
Definition request.cc:851
SignalProxy< void(NetworkRequest &req)> sigStarted()
Signals that the dispatcher dequeued the request and actually starts downloading data.
Definition request.cc:1051
SignalProxy< void(NetworkRequest &req, off_t dltotal, off_t dlnow, off_t ultotal, off_t ulnow)> sigProgress()
Signals if there was data read from the download.
Definition request.cc:1061
TransferSettings & transferSettings()
Definition request.cc:982
CurlMultiPartHandler::Range Range
Definition request.h:76
const std::string & lastRedirectInfo() const
Definition request.cc:856
The Timer class provides repetitive and single-shot timers.
Definition timer.h:45
SignalProxy< void(Timer &t)> sigExpired()
This signal is always emitted when the timer expires.
Definition timer.cc:120
uint64_t remaining() const
Definition timer.cc:99
uint64_t interval() const
Definition timer.cc:94
#define EXPLICITLY_NO_PROXY
#define MIL_MEDIA
#define DBG_MEDIA
std::string curlUnEscape(const std::string &text_r)
void setupZYPP_MEDIA_CURL_DEBUG(CURL *curl)
Setup CURLOPT_VERBOSE and CURLOPT_DEBUGFUNCTION according to env::ZYPP_MEDIA_CURL_DEBUG.
CURLcode setCurlRedirProtocols(CURL *curl)
Definition Arch.h:364
typename decay< T >::type decay_t
Definition TypeTraits.h:42
String related utilities and Regular expression matching.
int ZYPP_MEDIA_CURL_IPRESOLVE()
4/6 to force IPv4/v6
Definition curlhelper.cc:45
Types and functions for filesystem operations.
Definition Glob.cc:24
int assert_file_mode(const Pathname &path, unsigned mode)
Like assert_file but enforce mode even if the file already exists.
Definition PathInfo.cc:1210
std::vector< char > peek_data_fd(FILE *fd, off_t offset, size_t count)
Definition IOTools.cc:171
std::string form(const char *format,...) __attribute__((format(printf
Printf style construction of std::string.
Definition String.cc:37
TInt strtonum(const C_Str &str)
Parsing numbers from string.
std::string trim(const std::string &s, const Trim trim_r)
Definition String.cc:224
Url details namespace.
Definition UrlBase.cc:58
Easy-to use interface to the ZYPP dependency resolver.
T trim(StrType &&s, const Trim trim_r)
Definition string.h:35
zypp::media::TransferSettings TransferSettings
std::string strerr_cxx(const int err=-1)
Structure holding values of curlrc options.
Definition curlconfig.h:27
std::string proxyuserpwd
Definition curlconfig.h:49
static int parseConfig(CurlConfig &config, const std::string &filename="")
Parse a curlrc file and store the result in the config structure.
Definition curlconfig.cc:24
Convenient building of std::string with boost::format.
Definition String.h:253
static Range make(size_t start, size_t len=0, std::optional< zypp::Digest > &&digest={}, CheckSumBytes &&expectedChkSum=CheckSumBytes(), std::any &&userData=std::any(), std::optional< size_t > digestCompareLen={}, std::optional< size_t > _dataBlockPadding={})
std::unique_ptr< CurlMultiPartHandler > _partialHelper
Definition request_p.h:156
std::unique_ptr< CurlMultiPartHandler > _partialHelper
Definition request_p.h:166
running_t(pending_t &&prevState)
Definition request.cc:63
std::chrono::microseconds appconnect
Definition request.h:81
std::chrono::microseconds redirect
Definition request.h:84
std::chrono::microseconds pretransfer
Definition request.h:82
std::chrono::microseconds total
Definition request.h:83
std::chrono::microseconds namelookup
Definition request.h:79
std::chrono::microseconds connect
Definition request.h:80
#define ZYPP_CAUGHT(EXCPT)
Drops a logline telling the Exception was caught (in order to handle it).
Definition Exception.h:440
#define ZYPP_THROW(EXCPT)
Drops a logline and throws the Exception.
Definition Exception.h:424
#define DBG
Definition Logger.h:99
#define MIL
Definition Logger.h:100
#define ZYPP_IMPL_PRIVATE(Class)
Definition zyppglobal.h:92
#define Z_D()
Definition zyppglobal.h:105