1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
/*= -*- c-basic-offset: 4; indent-tabs-mode: nil; -*-
*
* librsync -- the library for network deltas
*
* Copyright (C) 1999, 2000, 2001 by Martin Pool <mbp@sourcefrog.net>
* Copyright (C) 1999 by Andrew Tridgell <tridge@samba.org>
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public License
* as published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful, but
* WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this program; if not, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
/** \file readsums.c
* Load signatures from a file. */
#include "librsync.h"
#include "job.h"
#include "sumset.h"
#include "scoop.h"
#include "netint.h"
#include "trace.h"
#include "util.h"
static rs_result rs_loadsig_s_weak(rs_job_t *job);
static rs_result rs_loadsig_s_strong(rs_job_t *job);
/** Add a just-read-in checksum pair to the signature block. */
static rs_result rs_loadsig_add_sum(rs_job_t *job, rs_strong_sum_t *strong)
{
rs_signature_t *sig = job->signature;
if (rs_trace_enabled()) {
char hexbuf[RS_MAX_STRONG_SUM_LENGTH * 2 + 2];
rs_hexify(hexbuf, strong, sig->strong_sum_len);
rs_trace("got block: weak=" FMT_WEAKSUM ", strong=%s", job->weak_sig,
hexbuf);
}
rs_signature_add_block(job->signature, job->weak_sig, strong);
job->stats.sig_blocks++;
return RS_RUNNING;
}
static rs_result rs_loadsig_s_weak(rs_job_t *job)
{
int l;
rs_result result;
if ((result = rs_suck_n4(job, &l)) != RS_DONE) {
if (result == RS_INPUT_ENDED) /* ending here is OK */
return RS_DONE;
return result;
}
job->weak_sig = l;
job->statefn = rs_loadsig_s_strong;
return RS_RUNNING;
}
static rs_result rs_loadsig_s_strong(rs_job_t *job)
{
rs_result result;
rs_strong_sum_t *strongsum;
if ((result =
rs_scoop_read(job, job->signature->strong_sum_len,
(void **)&strongsum)) != RS_DONE)
return result;
job->statefn = rs_loadsig_s_weak;
return rs_loadsig_add_sum(job, strongsum);
}
static rs_result rs_loadsig_s_stronglen(rs_job_t *job)
{
int l;
rs_result result;
if ((result = rs_suck_n4(job, &l)) != RS_DONE)
return result;
if (l < 0 || l > RS_MAX_STRONG_SUM_LENGTH) {
rs_error("strong sum length %d is implausible", l);
return RS_CORRUPT;
}
rs_trace("got strong sum length %d", l);
job->sig_strong_len = l;
/* Initialize the signature. */
if ((result =
rs_signature_init(job->signature, job->sig_magic, job->sig_block_len,
job->sig_strong_len, job->sig_fsize)) != RS_DONE)
return result;
job->statefn = rs_loadsig_s_weak;
return RS_RUNNING;
}
static rs_result rs_loadsig_s_blocklen(rs_job_t *job)
{
int l;
rs_result result;
if ((result = rs_suck_n4(job, &l)) != RS_DONE)
return result;
if (l < 1) {
rs_error("block length of %d is bogus", l);
return RS_CORRUPT;
}
rs_trace("got block length %d", l);
job->sig_block_len = l;
job->stats.block_len = l;
job->statefn = rs_loadsig_s_stronglen;
return RS_RUNNING;
}
static rs_result rs_loadsig_s_magic(rs_job_t *job)
{
int l;
rs_result result;
if ((result = rs_suck_n4(job, &l)) != RS_DONE)
return result;
rs_trace("got signature magic %#x", l);
job->sig_magic = l;
job->statefn = rs_loadsig_s_blocklen;
return RS_RUNNING;
}
rs_job_t *rs_loadsig_begin(rs_signature_t **signature)
{
rs_job_t *job;
job = rs_job_new("loadsig", rs_loadsig_s_magic);
*signature = job->signature = rs_alloc_struct(rs_signature_t);
return job;
}
|
