From 826bbfc01c2e157ac35a53c41fc6caa8255e4773 Mon Sep 17 00:00:00 2001 From: Daniel Stenberg Date: Fri, 18 May 2007 10:40:43 +0000 Subject: better fix for the dl/ul counters --- lib/tftp.c | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) (limited to 'lib') diff --git a/lib/tftp.c b/lib/tftp.c index 9104d605c..d63ad773b 100644 --- a/lib/tftp.c +++ b/lib/tftp.c @@ -431,6 +431,7 @@ static CURLcode tftp_tx(tftp_state_data_t *state, tftp_event_t event) int sbytes; int rblock; CURLcode res = CURLE_OK; + struct Curl_transfer_keeper *k = &data->reqdata.keep; switch(event) { @@ -485,6 +486,9 @@ static CURLcode tftp_tx(tftp_state_data_t *state, tftp_event_t event) failf(data, "%s\n", Curl_strerror(state->conn, SOCKERRNO)); return CURLE_SEND_ERROR; } + /* Update the progress meter */ + k->writebytecount += state->sbytes; + Curl_pgrsSetUploadCounter(data, k->writebytecount); break; case TFTP_EVENT_TIMEOUT: @@ -508,6 +512,8 @@ static CURLcode tftp_tx(tftp_state_data_t *state, tftp_event_t event) failf(data, "%s\n", Curl_strerror(state->conn, SOCKERRNO)); return CURLE_SEND_ERROR; } + /* since this was a re-send, we remain at the still byte position */ + Curl_pgrsSetUploadCounter(data, k->writebytecount); } break; @@ -520,9 +526,6 @@ static CURLcode tftp_tx(tftp_state_data_t *state, tftp_event_t event) break; } - /* Update the progress meter */ - Curl_pgrsSetUploadCounter(data, (curl_off_t) state->block*TFTP_BLOCKSIZE); - return res; } @@ -667,6 +670,7 @@ CURLcode Curl_tftp(struct connectdata *conn, bool *done) struct Curl_sockaddr_storage fromaddr; socklen_t fromlen; int check_time = 0; + struct Curl_transfer_keeper *k = &data->reqdata.keep; *done = TRUE; @@ -739,8 +743,8 @@ CURLcode Curl_tftp(struct connectdata *conn, bool *done) state->rbytes-4); if(code) return code; - Curl_pgrsSetDownloadCounter(data, - (curl_off_t) state->rbytes-4); + k->bytecount += state->rbytes-4; + Curl_pgrsSetDownloadCounter(data, (curl_off_t) k->bytecount); } break; case TFTP_EVENT_ERROR: -- cgit v1.2.3