1 | The following changes since commit 248b23735645f7cbb503d9be6f5bf825f2a603ab: | 1 | The following changes since commit 136c67e07869227b21b3f627316e03679ce7b738: |
---|---|---|---|
2 | 2 | ||
3 | Update version for v2.10.0-rc4 release (2017-08-24 17:34:26 +0100) | 3 | Merge remote-tracking branch 'remotes/bkoppelmann/tags/pull-tricore-2018-03-02' into staging (2018-03-02 16:56:20 +0000) |
4 | 4 | ||
5 | are available in the git repository at: | 5 | are available in the Git repository at: |
6 | 6 | ||
7 | git://github.com/stefanha/qemu.git tags/block-pull-request | 7 | git://github.com/stefanha/qemu.git tags/block-pull-request |
8 | 8 | ||
9 | for you to fetch changes up to 3e4c705212abfe8c9882a00beb2d1466a8a53cec: | 9 | for you to fetch changes up to 23500c6a9409efc80d696aede0629bfbe7556a90: |
10 | 10 | ||
11 | qcow2: allocate cluster_cache/cluster_data on demand (2017-08-30 18:02:10 +0100) | 11 | README: Document 'git-publish' workflow (2018-03-05 09:03:17 +0000) |
12 | |||
13 | ---------------------------------------------------------------- | ||
14 | Pull request | ||
15 | |||
16 | Mostly patches that are only indirectly related to the block layer, but I've | ||
17 | reviewed them and there is no maintainer. | ||
12 | 18 | ||
13 | ---------------------------------------------------------------- | 19 | ---------------------------------------------------------------- |
14 | 20 | ||
15 | ---------------------------------------------------------------- | 21 | Fam Zheng (2): |
22 | Add a git-publish configuration file | ||
23 | README: Document 'git-publish' workflow | ||
16 | 24 | ||
17 | Alberto Garcia (8): | 25 | Su Hang (3): |
18 | throttle: Fix wrong variable name in the header documentation | 26 | util/uri.c: Coding style check, Only whitespace involved |
19 | throttle: Update the throttle_fix_bucket() documentation | 27 | util/uri.c: remove brackets that wrap `return` statement's content. |
20 | throttle: Make throttle_is_valid() a bit less verbose | 28 | util/uri.c: wrap single statement blocks with braces {} |
21 | throttle: Remove throttle_fix_bucket() / throttle_unfix_bucket() | ||
22 | throttle: Make LeakyBucket.avg and LeakyBucket.max integer types | ||
23 | throttle: Make burst_length 64bit and add range checks | ||
24 | throttle: Test the valid range of config values | ||
25 | misc: Remove unused Error variables | ||
26 | 29 | ||
27 | Dan Aloni (1): | 30 | Thomas Huth (1): |
28 | nvme: Fix get/set number of queues feature, again | 31 | tests/libqos: Check for valid dev pointer when looking for PCI devices |
29 | 32 | ||
30 | Eduardo Habkost (1): | 33 | tests/libqos/virtio-pci.c | 4 +- |
31 | oslib-posix: Print errors before aborting on qemu_alloc_stack() | 34 | util/uri.c | 1733 ++++++++++++++++++++++++--------------------- |
32 | 35 | .gitpublish | 51 ++ | |
33 | Fred Rolland (1): | 36 | README | 31 +- |
34 | qemu-doc: Add UUID support in initiator name | 37 | 4 files changed, 1014 insertions(+), 805 deletions(-) |
35 | 38 | create mode 100644 .gitpublish | |
36 | Stefan Hajnoczi (4): | ||
37 | scripts: add argparse module for Python 2.6 compatibility | ||
38 | docker.py: Python 2.6 argparse compatibility | ||
39 | tests: migration/guestperf Python 2.6 argparse compatibility | ||
40 | qcow2: allocate cluster_cache/cluster_data on demand | ||
41 | |||
42 | include/qemu/throttle.h | 8 +- | ||
43 | block/qcow.c | 12 +- | ||
44 | block/qcow2-cluster.c | 17 + | ||
45 | block/qcow2.c | 20 +- | ||
46 | dump.c | 4 +- | ||
47 | hw/block/nvme.c | 4 +- | ||
48 | tests/test-throttle.c | 80 +- | ||
49 | util/oslib-posix.c | 2 + | ||
50 | util/throttle.c | 86 +- | ||
51 | COPYING.PYTHON | 270 ++++ | ||
52 | qemu-doc.texi | 5 +- | ||
53 | scripts/argparse.py | 2406 ++++++++++++++++++++++++++++++++++++ | ||
54 | tests/docker/docker.py | 4 +- | ||
55 | tests/migration/guestperf/shell.py | 8 +- | ||
56 | 14 files changed, 2831 insertions(+), 95 deletions(-) | ||
57 | create mode 100644 COPYING.PYTHON | ||
58 | create mode 100644 scripts/argparse.py | ||
59 | 39 | ||
60 | -- | 40 | -- |
61 | 2.13.5 | 41 | 2.14.3 |
62 | 42 | ||
63 | 43 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Dan Aloni <dan@kernelim.com> | ||
2 | 1 | ||
3 | The number of queues that should be return by the admin command should: | ||
4 | |||
5 | 1) Only mention the number of non-admin queues. | ||
6 | 2) It is zero-based, meaning that '0 == one non-admin queue', | ||
7 | '1 == two non-admin queues', and so forth. | ||
8 | |||
9 | Because our `num_queues` means the number of queues _plus_ the admin | ||
10 | queue, then the right calculation for the number returned from the admin | ||
11 | command is `num_queues - 2`, combining the two requirements mentioned. | ||
12 | |||
13 | The issue was discovered by reducing num_queues from 64 to 8 and running | ||
14 | a Linux VM with an SMP parameter larger than that (e.g. 22). It tries to | ||
15 | utilize all queues, and therefore fails with an invalid queue number | ||
16 | when trying to queue I/Os on the last queue. | ||
17 | |||
18 | Signed-off-by: Dan Aloni <dan@kernelim.com> | ||
19 | CC: Alex Friedman <alex@e8storage.com> | ||
20 | CC: Keith Busch <keith.busch@intel.com> | ||
21 | CC: Stefan Hajnoczi <stefanha@redhat.com> | ||
22 | Reviewed-by: Keith Busch <keith.busch@intel.com> | ||
23 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
24 | --- | ||
25 | hw/block/nvme.c | 4 ++-- | ||
26 | 1 file changed, 2 insertions(+), 2 deletions(-) | ||
27 | |||
28 | diff --git a/hw/block/nvme.c b/hw/block/nvme.c | ||
29 | index XXXXXXX..XXXXXXX 100644 | ||
30 | --- a/hw/block/nvme.c | ||
31 | +++ b/hw/block/nvme.c | ||
32 | @@ -XXX,XX +XXX,XX @@ static uint16_t nvme_get_feature(NvmeCtrl *n, NvmeCmd *cmd, NvmeRequest *req) | ||
33 | result = blk_enable_write_cache(n->conf.blk); | ||
34 | break; | ||
35 | case NVME_NUMBER_OF_QUEUES: | ||
36 | - result = cpu_to_le32((n->num_queues - 1) | ((n->num_queues - 1) << 16)); | ||
37 | + result = cpu_to_le32((n->num_queues - 2) | ((n->num_queues - 2) << 16)); | ||
38 | break; | ||
39 | default: | ||
40 | return NVME_INVALID_FIELD | NVME_DNR; | ||
41 | @@ -XXX,XX +XXX,XX @@ static uint16_t nvme_set_feature(NvmeCtrl *n, NvmeCmd *cmd, NvmeRequest *req) | ||
42 | break; | ||
43 | case NVME_NUMBER_OF_QUEUES: | ||
44 | req->cqe.result = | ||
45 | - cpu_to_le32((n->num_queues - 1) | ((n->num_queues - 1) << 16)); | ||
46 | + cpu_to_le32((n->num_queues - 2) | ((n->num_queues - 2) << 16)); | ||
47 | break; | ||
48 | default: | ||
49 | return NVME_INVALID_FIELD | NVME_DNR; | ||
50 | -- | ||
51 | 2.13.5 | ||
52 | |||
53 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Alberto Garcia <berto@igalia.com> | ||
2 | 1 | ||
3 | The level of the burst bucket is stored in bkt.burst_level, not | ||
4 | bkt.burst_length. | ||
5 | |||
6 | Signed-off-by: Alberto Garcia <berto@igalia.com> | ||
7 | Reviewed-by: Manos Pitsidianakis <el13635@mail.ntua.gr> | ||
8 | Message-id: 49aab2711d02f285567f3b3b13a113847af33812.1503580370.git.berto@igalia.com | ||
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | ||
11 | include/qemu/throttle.h | 2 +- | ||
12 | 1 file changed, 1 insertion(+), 1 deletion(-) | ||
13 | |||
14 | diff --git a/include/qemu/throttle.h b/include/qemu/throttle.h | ||
15 | index XXXXXXX..XXXXXXX 100644 | ||
16 | --- a/include/qemu/throttle.h | ||
17 | +++ b/include/qemu/throttle.h | ||
18 | @@ -XXX,XX +XXX,XX @@ typedef enum { | ||
19 | * - The bkt.avg rate does not apply until the bucket is full, | ||
20 | * allowing the user to do bursts until then. The I/O limit during | ||
21 | * bursts is bkt.max. To enforce this limit we keep an additional | ||
22 | - * bucket in bkt.burst_length that leaks at a rate of bkt.max units | ||
23 | + * bucket in bkt.burst_level that leaks at a rate of bkt.max units | ||
24 | * per second. | ||
25 | * | ||
26 | * - Because of all of the above, the user can perform I/O at a | ||
27 | -- | ||
28 | 2.13.5 | ||
29 | |||
30 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Alberto Garcia <berto@igalia.com> | ||
2 | 1 | ||
3 | The way the throttling algorithm works is that requests start being | ||
4 | throttled once the bucket level exceeds the burst limit. When we get | ||
5 | there the bucket leaks at the level set by the user (bkt->avg), and | ||
6 | that leak rate is what prevents guest I/O from exceeding the desired | ||
7 | limit. | ||
8 | |||
9 | If we don't allow bursts (i.e. bkt->max == 0) then we can start | ||
10 | throttling requests immediately. The problem with keeping the | ||
11 | threshold at 0 is that it only allows one request at a time, and as | ||
12 | soon as there's a bit of I/O from the guest every other request will | ||
13 | be throttled and performance will suffer considerably. That can even | ||
14 | make the guest unable to reach the throttle limit if that limit is | ||
15 | high enough, and that happens regardless of the block scheduler used | ||
16 | by the guest. | ||
17 | |||
18 | Increasing that threshold gives flexibility to the guest, allowing it | ||
19 | to perform short bursts of I/O before being throttled. Increasing the | ||
20 | threshold too much does not make a difference in the long run (because | ||
21 | it's the leak rate what defines the actual throughput) but it does | ||
22 | allow the guest to perform longer initial bursts and exceed the | ||
23 | throttle limit for a short while. | ||
24 | |||
25 | A burst value of bkt->avg / 10 allows the guest to perform 100ms' | ||
26 | worth of I/O at the target rate without being throttled. | ||
27 | |||
28 | Signed-off-by: Alberto Garcia <berto@igalia.com> | ||
29 | Message-id: 31aae6645f0d1fbf3860fb2b528b757236f0c0a7.1503580370.git.berto@igalia.com | ||
30 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
31 | --- | ||
32 | util/throttle.c | 11 +++-------- | ||
33 | 1 file changed, 3 insertions(+), 8 deletions(-) | ||
34 | |||
35 | diff --git a/util/throttle.c b/util/throttle.c | ||
36 | index XXXXXXX..XXXXXXX 100644 | ||
37 | --- a/util/throttle.c | ||
38 | +++ b/util/throttle.c | ||
39 | @@ -XXX,XX +XXX,XX @@ static void throttle_fix_bucket(LeakyBucket *bkt) | ||
40 | /* zero bucket level */ | ||
41 | bkt->level = bkt->burst_level = 0; | ||
42 | |||
43 | - /* The following is done to cope with the Linux CFQ block scheduler | ||
44 | - * which regroup reads and writes by block of 100ms in the guest. | ||
45 | - * When they are two process one making reads and one making writes cfq | ||
46 | - * make a pattern looking like the following: | ||
47 | - * WWWWWWWWWWWRRRRRRRRRRRRRRWWWWWWWWWWWWWwRRRRRRRRRRRRRRRRR | ||
48 | - * Having a max burst value of 100ms of the average will help smooth the | ||
49 | - * throttling | ||
50 | - */ | ||
51 | + /* If bkt->max is 0 we still want to allow short bursts of I/O | ||
52 | + * from the guest, otherwise every other request will be throttled | ||
53 | + * and performance will suffer considerably. */ | ||
54 | min = bkt->avg / 10; | ||
55 | if (bkt->avg && !bkt->max) { | ||
56 | bkt->max = min; | ||
57 | -- | ||
58 | 2.13.5 | ||
59 | |||
60 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Alberto Garcia <berto@igalia.com> | ||
2 | 1 | ||
3 | Use a pointer to the bucket instead of repeating cfg->buckets[i] all | ||
4 | the time. This makes the code more concise and will help us expand the | ||
5 | checks later and save a few line breaks. | ||
6 | |||
7 | Signed-off-by: Alberto Garcia <berto@igalia.com> | ||
8 | Message-id: 763ffc40a26b17d54cf93f5a999e4656049fcf0c.1503580370.git.berto@igalia.com | ||
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | ||
11 | util/throttle.c | 15 +++++++-------- | ||
12 | 1 file changed, 7 insertions(+), 8 deletions(-) | ||
13 | |||
14 | diff --git a/util/throttle.c b/util/throttle.c | ||
15 | index XXXXXXX..XXXXXXX 100644 | ||
16 | --- a/util/throttle.c | ||
17 | +++ b/util/throttle.c | ||
18 | @@ -XXX,XX +XXX,XX @@ bool throttle_is_valid(ThrottleConfig *cfg, Error **errp) | ||
19 | } | ||
20 | |||
21 | for (i = 0; i < BUCKETS_COUNT; i++) { | ||
22 | - if (cfg->buckets[i].avg < 0 || | ||
23 | - cfg->buckets[i].max < 0 || | ||
24 | - cfg->buckets[i].avg > THROTTLE_VALUE_MAX || | ||
25 | - cfg->buckets[i].max > THROTTLE_VALUE_MAX) { | ||
26 | + LeakyBucket *bkt = &cfg->buckets[i]; | ||
27 | + if (bkt->avg < 0 || bkt->max < 0 || | ||
28 | + bkt->avg > THROTTLE_VALUE_MAX || bkt->max > THROTTLE_VALUE_MAX) { | ||
29 | error_setg(errp, "bps/iops/max values must be within [0, %lld]", | ||
30 | THROTTLE_VALUE_MAX); | ||
31 | return false; | ||
32 | } | ||
33 | |||
34 | - if (!cfg->buckets[i].burst_length) { | ||
35 | + if (!bkt->burst_length) { | ||
36 | error_setg(errp, "the burst length cannot be 0"); | ||
37 | return false; | ||
38 | } | ||
39 | |||
40 | - if (cfg->buckets[i].burst_length > 1 && !cfg->buckets[i].max) { | ||
41 | + if (bkt->burst_length > 1 && !bkt->max) { | ||
42 | error_setg(errp, "burst length set without burst rate"); | ||
43 | return false; | ||
44 | } | ||
45 | |||
46 | - if (cfg->buckets[i].max && !cfg->buckets[i].avg) { | ||
47 | + if (bkt->max && !bkt->avg) { | ||
48 | error_setg(errp, "bps_max/iops_max require corresponding" | ||
49 | " bps/iops values"); | ||
50 | return false; | ||
51 | } | ||
52 | |||
53 | - if (cfg->buckets[i].max && cfg->buckets[i].max < cfg->buckets[i].avg) { | ||
54 | + if (bkt->max && bkt->max < bkt->avg) { | ||
55 | error_setg(errp, "bps_max/iops_max cannot be lower than bps/iops"); | ||
56 | return false; | ||
57 | } | ||
58 | -- | ||
59 | 2.13.5 | ||
60 | |||
61 | diff view generated by jsdifflib |
1 | From: Alberto Garcia <berto@igalia.com> | 1 | From: Su Hang <suhang16@mails.ucas.ac.cn> |
---|---|---|---|
2 | 2 | ||
3 | Signed-off-by: Alberto Garcia <berto@igalia.com> | 3 | Using `clang-format -i util/uri.c` first, then change back few code |
4 | Message-id: a57dd6274e1b6dc9c28769fec4c7ea543be5c5e3.1503580370.git.berto@igalia.com | 4 | manually, to make sure only whitespace involved. |
5 | |||
6 | Signed-off-by: Su Hang <suhang16@mails.ucas.ac.cn> | ||
7 | Reviewed-by: Thomas Huth <thuth@redhat.com> | ||
8 | Message-id: 1519533358-13759-2-git-send-email-suhang16@mails.ucas.ac.cn | ||
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
6 | --- | 10 | --- |
7 | tests/test-throttle.c | 77 +++++++++++++++++++++++++++++++++++++++++++++++++++ | 11 | util/uri.c | 1450 ++++++++++++++++++++++++++++++------------------------------ |
8 | 1 file changed, 77 insertions(+) | 12 | 1 file changed, 726 insertions(+), 724 deletions(-) |
9 | 13 | ||
10 | diff --git a/tests/test-throttle.c b/tests/test-throttle.c | 14 | diff --git a/util/uri.c b/util/uri.c |
11 | index XXXXXXX..XXXXXXX 100644 | 15 | index XXXXXXX..XXXXXXX 100644 |
12 | --- a/tests/test-throttle.c | 16 | --- a/util/uri.c |
13 | +++ b/tests/test-throttle.c | 17 | +++ b/util/uri.c |
14 | @@ -XXX,XX +XXX,XX @@ static void test_is_valid(void) | 18 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); |
15 | test_is_valid_for_value(1, true); | 19 | */ |
20 | #define IS_ALPHA(x) (IS_LOWALPHA(x) || IS_UPALPHA(x)) | ||
21 | |||
22 | - | ||
23 | /* | ||
24 | * lowalpha = "a" | "b" | "c" | "d" | "e" | "f" | "g" | "h" | "i" | "j" | | ||
25 | * "k" | "l" | "m" | "n" | "o" | "p" | "q" | "r" | "s" | "t" | | ||
26 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
27 | * mark = "-" | "_" | "." | "!" | "~" | "*" | "'" | "(" | ")" | ||
28 | */ | ||
29 | |||
30 | -#define IS_MARK(x) (((x) == '-') || ((x) == '_') || ((x) == '.') || \ | ||
31 | - ((x) == '!') || ((x) == '~') || ((x) == '*') || ((x) == '\'') || \ | ||
32 | +#define IS_MARK(x) (((x) == '-') || ((x) == '_') || ((x) == '.') || \ | ||
33 | + ((x) == '!') || ((x) == '~') || ((x) == '*') || ((x) == '\'') || \ | ||
34 | ((x) == '(') || ((x) == ')')) | ||
35 | |||
36 | /* | ||
37 | * unwise = "{" | "}" | "|" | "\" | "^" | "`" | ||
38 | */ | ||
39 | |||
40 | -#define IS_UNWISE(p) \ | ||
41 | - (((*(p) == '{')) || ((*(p) == '}')) || ((*(p) == '|')) || \ | ||
42 | - ((*(p) == '\\')) || ((*(p) == '^')) || ((*(p) == '[')) || \ | ||
43 | - ((*(p) == ']')) || ((*(p) == '`'))) | ||
44 | +#define IS_UNWISE(p) \ | ||
45 | + (((*(p) == '{')) || ((*(p) == '}')) || ((*(p) == '|')) || \ | ||
46 | + ((*(p) == '\\')) || ((*(p) == '^')) || ((*(p) == '[')) || \ | ||
47 | + ((*(p) == ']')) || ((*(p) == '`'))) | ||
48 | /* | ||
49 | * reserved = ";" | "/" | "?" | ":" | "@" | "&" | "=" | "+" | "$" | "," | | ||
50 | * "[" | "]" | ||
51 | */ | ||
52 | |||
53 | -#define IS_RESERVED(x) (((x) == ';') || ((x) == '/') || ((x) == '?') || \ | ||
54 | - ((x) == ':') || ((x) == '@') || ((x) == '&') || ((x) == '=') || \ | ||
55 | - ((x) == '+') || ((x) == '$') || ((x) == ',') || ((x) == '[') || \ | ||
56 | - ((x) == ']')) | ||
57 | +#define IS_RESERVED(x) (((x) == ';') || ((x) == '/') || ((x) == '?') || \ | ||
58 | + ((x) == ':') || ((x) == '@') || ((x) == '&') || ((x) == '=') || \ | ||
59 | + ((x) == '+') || ((x) == '$') || ((x) == ',') || ((x) == '[') || \ | ||
60 | + ((x) == ']')) | ||
61 | |||
62 | /* | ||
63 | * unreserved = alphanum | mark | ||
64 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
65 | * Skip to next pointer char, handle escaped sequences | ||
66 | */ | ||
67 | |||
68 | -#define NEXT(p) ((*p == '%')? p += 3 : p++) | ||
69 | +#define NEXT(p) ((*p == '%') ? p += 3 : p++) | ||
70 | |||
71 | /* | ||
72 | * Productions from the spec. | ||
73 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
74 | * path = [ abs_path | opaque_part ] | ||
75 | */ | ||
76 | |||
77 | - | ||
78 | /************************************************************************ | ||
79 | - * * | ||
80 | - * RFC 3986 parser * | ||
81 | - * * | ||
82 | + * * | ||
83 | + * RFC 3986 parser * | ||
84 | + * * | ||
85 | ************************************************************************/ | ||
86 | |||
87 | #define ISA_DIGIT(p) ((*(p) >= '0') && (*(p) <= '9')) | ||
88 | -#define ISA_ALPHA(p) (((*(p) >= 'a') && (*(p) <= 'z')) || \ | ||
89 | +#define ISA_ALPHA(p) (((*(p) >= 'a') && (*(p) <= 'z')) || \ | ||
90 | ((*(p) >= 'A') && (*(p) <= 'Z'))) | ||
91 | -#define ISA_HEXDIG(p) \ | ||
92 | - (ISA_DIGIT(p) || ((*(p) >= 'a') && (*(p) <= 'f')) || \ | ||
93 | - ((*(p) >= 'A') && (*(p) <= 'F'))) | ||
94 | +#define ISA_HEXDIG(p) \ | ||
95 | + (ISA_DIGIT(p) || ((*(p) >= 'a') && (*(p) <= 'f')) || \ | ||
96 | + ((*(p) >= 'A') && (*(p) <= 'F'))) | ||
97 | |||
98 | /* | ||
99 | * sub-delims = "!" / "$" / "&" / "'" / "(" / ")" | ||
100 | * / "*" / "+" / "," / ";" / "=" | ||
101 | */ | ||
102 | -#define ISA_SUB_DELIM(p) \ | ||
103 | - (((*(p) == '!')) || ((*(p) == '$')) || ((*(p) == '&')) || \ | ||
104 | - ((*(p) == '(')) || ((*(p) == ')')) || ((*(p) == '*')) || \ | ||
105 | - ((*(p) == '+')) || ((*(p) == ',')) || ((*(p) == ';')) || \ | ||
106 | - ((*(p) == '=')) || ((*(p) == '\''))) | ||
107 | +#define ISA_SUB_DELIM(p) \ | ||
108 | + (((*(p) == '!')) || ((*(p) == '$')) || ((*(p) == '&')) || \ | ||
109 | + ((*(p) == '(')) || ((*(p) == ')')) || ((*(p) == '*')) || \ | ||
110 | + ((*(p) == '+')) || ((*(p) == ',')) || ((*(p) == ';')) || \ | ||
111 | + ((*(p) == '=')) || ((*(p) == '\''))) | ||
112 | |||
113 | /* | ||
114 | * gen-delims = ":" / "/" / "?" / "#" / "[" / "]" / "@" | ||
115 | */ | ||
116 | -#define ISA_GEN_DELIM(p) \ | ||
117 | - (((*(p) == ':')) || ((*(p) == '/')) || ((*(p) == '?')) || \ | ||
118 | - ((*(p) == '#')) || ((*(p) == '[')) || ((*(p) == ']')) || \ | ||
119 | - ((*(p) == '@'))) | ||
120 | +#define ISA_GEN_DELIM(p) \ | ||
121 | + (((*(p) == ':')) || ((*(p) == '/')) || ((*(p) == '?')) || \ | ||
122 | + ((*(p) == '#')) || ((*(p) == '[')) || ((*(p) == ']')) || \ | ||
123 | + ((*(p) == '@'))) | ||
124 | |||
125 | /* | ||
126 | * reserved = gen-delims / sub-delims | ||
127 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
128 | /* | ||
129 | * unreserved = ALPHA / DIGIT / "-" / "." / "_" / "~" | ||
130 | */ | ||
131 | -#define ISA_UNRESERVED(p) \ | ||
132 | - ((ISA_ALPHA(p)) || (ISA_DIGIT(p)) || ((*(p) == '-')) || \ | ||
133 | - ((*(p) == '.')) || ((*(p) == '_')) || ((*(p) == '~'))) | ||
134 | +#define ISA_UNRESERVED(p) \ | ||
135 | + ((ISA_ALPHA(p)) || (ISA_DIGIT(p)) || ((*(p) == '-')) || \ | ||
136 | + ((*(p) == '.')) || ((*(p) == '_')) || ((*(p) == '~'))) | ||
137 | |||
138 | /* | ||
139 | * pct-encoded = "%" HEXDIG HEXDIG | ||
140 | */ | ||
141 | -#define ISA_PCT_ENCODED(p) \ | ||
142 | - ((*(p) == '%') && (ISA_HEXDIG(p + 1)) && (ISA_HEXDIG(p + 2))) | ||
143 | +#define ISA_PCT_ENCODED(p) \ | ||
144 | + ((*(p) == '%') && (ISA_HEXDIG(p + 1)) && (ISA_HEXDIG(p + 2))) | ||
145 | |||
146 | /* | ||
147 | * pchar = unreserved / pct-encoded / sub-delims / ":" / "@" | ||
148 | */ | ||
149 | -#define ISA_PCHAR(p) \ | ||
150 | - (ISA_UNRESERVED(p) || ISA_PCT_ENCODED(p) || ISA_SUB_DELIM(p) || \ | ||
151 | - ((*(p) == ':')) || ((*(p) == '@'))) | ||
152 | +#define ISA_PCHAR(p) \ | ||
153 | + (ISA_UNRESERVED(p) || ISA_PCT_ENCODED(p) || ISA_SUB_DELIM(p) || \ | ||
154 | + ((*(p) == ':')) || ((*(p) == '@'))) | ||
155 | |||
156 | /** | ||
157 | * rfc3986_parse_scheme: | ||
158 | @@ -XXX,XX +XXX,XX @@ static void uri_clean(URI *uri); | ||
159 | * | ||
160 | * Returns 0 or the error code | ||
161 | */ | ||
162 | -static int | ||
163 | -rfc3986_parse_scheme(URI *uri, const char **str) { | ||
164 | +static int rfc3986_parse_scheme(URI *uri, const char **str) | ||
165 | +{ | ||
166 | const char *cur; | ||
167 | |||
168 | if (str == NULL) | ||
169 | - return(-1); | ||
170 | + return (-1); | ||
171 | |||
172 | cur = *str; | ||
173 | if (!ISA_ALPHA(cur)) | ||
174 | - return(2); | ||
175 | + return (2); | ||
176 | cur++; | ||
177 | - while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || | ||
178 | - (*cur == '+') || (*cur == '-') || (*cur == '.')) cur++; | ||
179 | + while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || (*cur == '+') || (*cur == '-') || | ||
180 | + (*cur == '.')) | ||
181 | + cur++; | ||
182 | if (uri != NULL) { | ||
183 | g_free(uri->scheme); | ||
184 | - uri->scheme = g_strndup(*str, cur - *str); | ||
185 | + uri->scheme = g_strndup(*str, cur - *str); | ||
186 | } | ||
187 | *str = cur; | ||
188 | - return(0); | ||
189 | + return (0); | ||
16 | } | 190 | } |
17 | 191 | ||
18 | +static void test_ranges(void) | 192 | /** |
193 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_scheme(URI *uri, const char **str) { | ||
194 | * | ||
195 | * Returns 0 or the error code | ||
196 | */ | ||
197 | -static int | ||
198 | -rfc3986_parse_fragment(URI *uri, const char **str) | ||
199 | +static int rfc3986_parse_fragment(URI *uri, const char **str) | ||
200 | { | ||
201 | const char *cur; | ||
202 | |||
203 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_fragment(URI *uri, const char **str) | ||
204 | NEXT(cur); | ||
205 | if (uri != NULL) { | ||
206 | g_free(uri->fragment); | ||
207 | - if (uri->cleanup & 2) | ||
208 | - uri->fragment = g_strndup(*str, cur - *str); | ||
209 | - else | ||
210 | - uri->fragment = uri_string_unescape(*str, cur - *str, NULL); | ||
211 | + if (uri->cleanup & 2) | ||
212 | + uri->fragment = g_strndup(*str, cur - *str); | ||
213 | + else | ||
214 | + uri->fragment = uri_string_unescape(*str, cur - *str, NULL); | ||
215 | } | ||
216 | *str = cur; | ||
217 | return (0); | ||
218 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_fragment(URI *uri, const char **str) | ||
219 | * | ||
220 | * Returns 0 or the error code | ||
221 | */ | ||
222 | -static int | ||
223 | -rfc3986_parse_query(URI *uri, const char **str) | ||
224 | +static int rfc3986_parse_query(URI *uri, const char **str) | ||
225 | { | ||
226 | const char *cur; | ||
227 | |||
228 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_query(URI *uri, const char **str) | ||
229 | NEXT(cur); | ||
230 | if (uri != NULL) { | ||
231 | g_free(uri->query); | ||
232 | - uri->query = g_strndup (*str, cur - *str); | ||
233 | + uri->query = g_strndup(*str, cur - *str); | ||
234 | } | ||
235 | *str = cur; | ||
236 | return (0); | ||
237 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_query(URI *uri, const char **str) | ||
238 | * | ||
239 | * Returns 0 or the error code | ||
240 | */ | ||
241 | -static int | ||
242 | -rfc3986_parse_port(URI *uri, const char **str) | ||
243 | +static int rfc3986_parse_port(URI *uri, const char **str) | ||
244 | { | ||
245 | const char *cur = *str; | ||
246 | int port = 0; | ||
247 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_port(URI *uri, const char **str) | ||
248 | * | ||
249 | * Returns 0 or the error code | ||
250 | */ | ||
251 | -static int | ||
252 | -rfc3986_parse_user_info(URI *uri, const char **str) | ||
253 | +static int rfc3986_parse_user_info(URI *uri, const char **str) | ||
254 | { | ||
255 | const char *cur; | ||
256 | |||
257 | cur = *str; | ||
258 | - while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || | ||
259 | - ISA_SUB_DELIM(cur) || (*cur == ':')) | ||
260 | - NEXT(cur); | ||
261 | + while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur) || | ||
262 | + (*cur == ':')) | ||
263 | + NEXT(cur); | ||
264 | if (*cur == '@') { | ||
265 | - if (uri != NULL) { | ||
266 | + if (uri != NULL) { | ||
267 | g_free(uri->user); | ||
268 | - if (uri->cleanup & 2) | ||
269 | - uri->user = g_strndup(*str, cur - *str); | ||
270 | - else | ||
271 | - uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
272 | - } | ||
273 | - *str = cur; | ||
274 | - return(0); | ||
275 | + if (uri->cleanup & 2) | ||
276 | + uri->user = g_strndup(*str, cur - *str); | ||
277 | + else | ||
278 | + uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
279 | + } | ||
280 | + *str = cur; | ||
281 | + return (0); | ||
282 | } | ||
283 | - return(1); | ||
284 | + return (1); | ||
285 | } | ||
286 | |||
287 | /** | ||
288 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_user_info(URI *uri, const char **str) | ||
289 | * | ||
290 | * Returns 0 if found and skipped, 1 otherwise | ||
291 | */ | ||
292 | -static int | ||
293 | -rfc3986_parse_dec_octet(const char **str) { | ||
294 | +static int rfc3986_parse_dec_octet(const char **str) | ||
19 | +{ | 295 | +{ |
20 | + int i; | 296 | const char *cur = *str; |
21 | + | 297 | |
22 | + for (i = 0; i < BUCKETS_COUNT; i++) { | 298 | if (!(ISA_DIGIT(cur))) |
23 | + LeakyBucket *b = &cfg.buckets[i]; | 299 | - return(1); |
24 | + throttle_config_init(&cfg); | 300 | - if (!ISA_DIGIT(cur+1)) |
25 | + | 301 | - cur++; |
26 | + /* avg = 0 means throttling is disabled, but the config is valid */ | 302 | - else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur+2))) |
27 | + b->avg = 0; | 303 | - cur += 2; |
28 | + g_assert(throttle_is_valid(&cfg, NULL)); | 304 | + return (1); |
29 | + g_assert(!throttle_enabled(&cfg)); | 305 | + if (!ISA_DIGIT(cur + 1)) |
30 | + | 306 | + cur++; |
31 | + /* These are valid configurations (values <= THROTTLE_VALUE_MAX) */ | 307 | + else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) |
32 | + b->avg = 1; | 308 | + cur += 2; |
33 | + g_assert(throttle_is_valid(&cfg, NULL)); | 309 | else if ((*cur == '1') && (ISA_DIGIT(cur + 1)) && (ISA_DIGIT(cur + 2))) |
34 | + | 310 | - cur += 3; |
35 | + b->avg = THROTTLE_VALUE_MAX; | 311 | - else if ((*cur == '2') && (*(cur + 1) >= '0') && |
36 | + g_assert(throttle_is_valid(&cfg, NULL)); | 312 | - (*(cur + 1) <= '4') && (ISA_DIGIT(cur + 2))) |
37 | + | 313 | - cur += 3; |
38 | + b->avg = THROTTLE_VALUE_MAX; | 314 | - else if ((*cur == '2') && (*(cur + 1) == '5') && |
39 | + b->max = THROTTLE_VALUE_MAX; | 315 | - (*(cur + 2) >= '0') && (*(cur + 1) <= '5')) |
40 | + g_assert(throttle_is_valid(&cfg, NULL)); | 316 | - cur += 3; |
41 | + | 317 | + cur += 3; |
42 | + /* Values over THROTTLE_VALUE_MAX are not allowed */ | 318 | + else if ((*cur == '2') && (*(cur + 1) >= '0') && (*(cur + 1) <= '4') && |
43 | + b->avg = THROTTLE_VALUE_MAX + 1; | 319 | + (ISA_DIGIT(cur + 2))) |
44 | + g_assert(!throttle_is_valid(&cfg, NULL)); | 320 | + cur += 3; |
45 | + | 321 | + else if ((*cur == '2') && (*(cur + 1) == '5') && (*(cur + 2) >= '0') && |
46 | + b->avg = THROTTLE_VALUE_MAX; | 322 | + (*(cur + 1) <= '5')) |
47 | + b->max = THROTTLE_VALUE_MAX + 1; | 323 | + cur += 3; |
48 | + g_assert(!throttle_is_valid(&cfg, NULL)); | 324 | else |
49 | + | 325 | - return(1); |
50 | + /* burst_length must be between 1 and THROTTLE_VALUE_MAX */ | 326 | + return (1); |
51 | + b->avg = 1; | 327 | *str = cur; |
52 | + b->max = 1; | 328 | - return(0); |
53 | + b->burst_length = 0; | 329 | + return (0); |
54 | + g_assert(!throttle_is_valid(&cfg, NULL)); | 330 | } |
55 | + | 331 | /** |
56 | + b->avg = 1; | 332 | * rfc3986_parse_host: |
57 | + b->max = 1; | 333 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_dec_octet(const char **str) { |
58 | + b->burst_length = 1; | 334 | * |
59 | + g_assert(throttle_is_valid(&cfg, NULL)); | 335 | * Returns 0 or the error code |
60 | + | 336 | */ |
61 | + b->avg = 1; | 337 | -static int |
62 | + b->max = 1; | 338 | -rfc3986_parse_host(URI *uri, const char **str) |
63 | + b->burst_length = THROTTLE_VALUE_MAX; | 339 | +static int rfc3986_parse_host(URI *uri, const char **str) |
64 | + g_assert(throttle_is_valid(&cfg, NULL)); | 340 | { |
65 | + | 341 | const char *cur = *str; |
66 | + b->avg = 1; | 342 | const char *host; |
67 | + b->max = 1; | 343 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_host(URI *uri, const char **str) |
68 | + b->burst_length = THROTTLE_VALUE_MAX + 1; | 344 | */ |
69 | + g_assert(!throttle_is_valid(&cfg, NULL)); | 345 | if (*cur == '[') { |
70 | + | 346 | cur++; |
71 | + /* burst_length * max cannot exceed THROTTLE_VALUE_MAX */ | 347 | - while ((*cur != ']') && (*cur != 0)) |
72 | + b->avg = 1; | 348 | - cur++; |
73 | + b->max = 2; | 349 | - if (*cur != ']') |
74 | + b->burst_length = THROTTLE_VALUE_MAX / 2; | 350 | - return(1); |
75 | + g_assert(throttle_is_valid(&cfg, NULL)); | 351 | - cur++; |
76 | + | 352 | - goto found; |
77 | + b->avg = 1; | 353 | + while ((*cur != ']') && (*cur != 0)) |
78 | + b->max = 3; | 354 | + cur++; |
79 | + b->burst_length = THROTTLE_VALUE_MAX / 2; | 355 | + if (*cur != ']') |
80 | + g_assert(!throttle_is_valid(&cfg, NULL)); | 356 | + return (1); |
81 | + | 357 | + cur++; |
82 | + b->avg = 1; | 358 | + goto found; |
83 | + b->max = THROTTLE_VALUE_MAX; | 359 | } |
84 | + b->burst_length = 1; | 360 | /* |
85 | + g_assert(throttle_is_valid(&cfg, NULL)); | 361 | * try to parse an IPv4 |
86 | + | 362 | */ |
87 | + b->avg = 1; | 363 | if (ISA_DIGIT(cur)) { |
88 | + b->max = THROTTLE_VALUE_MAX; | 364 | if (rfc3986_parse_dec_octet(&cur) != 0) |
89 | + b->burst_length = 2; | 365 | - goto not_ipv4; |
90 | + g_assert(!throttle_is_valid(&cfg, NULL)); | 366 | - if (*cur != '.') |
91 | + } | 367 | - goto not_ipv4; |
92 | +} | 368 | - cur++; |
93 | + | 369 | + goto not_ipv4; |
94 | static void test_max_is_missing_limit(void) | 370 | + if (*cur != '.') |
371 | + goto not_ipv4; | ||
372 | + cur++; | ||
373 | if (rfc3986_parse_dec_octet(&cur) != 0) | ||
374 | - goto not_ipv4; | ||
375 | - if (*cur != '.') | ||
376 | - goto not_ipv4; | ||
377 | + goto not_ipv4; | ||
378 | + if (*cur != '.') | ||
379 | + goto not_ipv4; | ||
380 | if (rfc3986_parse_dec_octet(&cur) != 0) | ||
381 | - goto not_ipv4; | ||
382 | - if (*cur != '.') | ||
383 | - goto not_ipv4; | ||
384 | + goto not_ipv4; | ||
385 | + if (*cur != '.') | ||
386 | + goto not_ipv4; | ||
387 | if (rfc3986_parse_dec_octet(&cur) != 0) | ||
388 | - goto not_ipv4; | ||
389 | - goto found; | ||
390 | -not_ipv4: | ||
391 | + goto not_ipv4; | ||
392 | + goto found; | ||
393 | + not_ipv4: | ||
394 | cur = *str; | ||
395 | } | ||
396 | /* | ||
397 | @@ -XXX,XX +XXX,XX @@ not_ipv4: | ||
398 | found: | ||
399 | if (uri != NULL) { | ||
400 | g_free(uri->authority); | ||
401 | - uri->authority = NULL; | ||
402 | + uri->authority = NULL; | ||
403 | g_free(uri->server); | ||
404 | - if (cur != host) { | ||
405 | - if (uri->cleanup & 2) | ||
406 | - uri->server = g_strndup(host, cur - host); | ||
407 | - else | ||
408 | - uri->server = uri_string_unescape(host, cur - host, NULL); | ||
409 | - } else | ||
410 | - uri->server = NULL; | ||
411 | + if (cur != host) { | ||
412 | + if (uri->cleanup & 2) | ||
413 | + uri->server = g_strndup(host, cur - host); | ||
414 | + else | ||
415 | + uri->server = uri_string_unescape(host, cur - host, NULL); | ||
416 | + } else | ||
417 | + uri->server = NULL; | ||
418 | } | ||
419 | *str = cur; | ||
420 | - return(0); | ||
421 | + return (0); | ||
422 | } | ||
423 | |||
424 | /** | ||
425 | @@ -XXX,XX +XXX,XX @@ found: | ||
426 | * | ||
427 | * Returns 0 or the error code | ||
428 | */ | ||
429 | -static int | ||
430 | -rfc3986_parse_authority(URI *uri, const char **str) | ||
431 | +static int rfc3986_parse_authority(URI *uri, const char **str) | ||
432 | { | ||
433 | const char *cur; | ||
434 | int ret; | ||
435 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_authority(URI *uri, const char **str) | ||
436 | else | ||
437 | cur++; | ||
438 | ret = rfc3986_parse_host(uri, &cur); | ||
439 | - if (ret != 0) return(ret); | ||
440 | + if (ret != 0) | ||
441 | + return (ret); | ||
442 | if (*cur == ':') { | ||
443 | cur++; | ||
444 | ret = rfc3986_parse_port(uri, &cur); | ||
445 | - if (ret != 0) return(ret); | ||
446 | + if (ret != 0) | ||
447 | + return (ret); | ||
448 | } | ||
449 | *str = cur; | ||
450 | - return(0); | ||
451 | + return (0); | ||
452 | } | ||
453 | |||
454 | /** | ||
455 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_authority(URI *uri, const char **str) | ||
456 | * | ||
457 | * Returns 0 or the error code | ||
458 | */ | ||
459 | -static int | ||
460 | -rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
461 | +static int rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
462 | { | ||
463 | const char *cur; | ||
464 | |||
465 | cur = *str; | ||
466 | if (!ISA_PCHAR(cur)) { | ||
467 | if (empty) | ||
468 | - return(0); | ||
469 | - return(1); | ||
470 | + return (0); | ||
471 | + return (1); | ||
472 | } | ||
473 | while (ISA_PCHAR(cur) && (*cur != forbid)) | ||
474 | NEXT(cur); | ||
475 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
476 | * | ||
477 | * Returns 0 or the error code | ||
478 | */ | ||
479 | -static int | ||
480 | -rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
481 | +static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
482 | { | ||
483 | const char *cur; | ||
484 | int ret; | ||
485 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
486 | |||
487 | while (*cur == '/') { | ||
488 | cur++; | ||
489 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
490 | - if (ret != 0) return(ret); | ||
491 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
492 | + if (ret != 0) | ||
493 | + return (ret); | ||
494 | } | ||
495 | if (uri != NULL) { | ||
496 | g_free(uri->path); | ||
497 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
498 | * | ||
499 | * Returns 0 or the error code | ||
500 | */ | ||
501 | -static int | ||
502 | -rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
503 | +static int rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
504 | { | ||
505 | const char *cur; | ||
506 | int ret; | ||
507 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
508 | cur = *str; | ||
509 | |||
510 | if (*cur != '/') | ||
511 | - return(1); | ||
512 | + return (1); | ||
513 | cur++; | ||
514 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
515 | if (ret == 0) { | ||
516 | - while (*cur == '/') { | ||
517 | - cur++; | ||
518 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
519 | - if (ret != 0) return(ret); | ||
520 | - } | ||
521 | + while (*cur == '/') { | ||
522 | + cur++; | ||
523 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
524 | + if (ret != 0) | ||
525 | + return (ret); | ||
526 | + } | ||
527 | } | ||
528 | if (uri != NULL) { | ||
529 | g_free(uri->path); | ||
530 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
531 | * | ||
532 | * Returns 0 or the error code | ||
533 | */ | ||
534 | -static int | ||
535 | -rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
536 | +static int rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
537 | { | ||
538 | const char *cur; | ||
539 | int ret; | ||
540 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
541 | cur = *str; | ||
542 | |||
543 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
544 | - if (ret != 0) return(ret); | ||
545 | + if (ret != 0) | ||
546 | + return (ret); | ||
547 | while (*cur == '/') { | ||
548 | cur++; | ||
549 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
550 | - if (ret != 0) return(ret); | ||
551 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
552 | + if (ret != 0) | ||
553 | + return (ret); | ||
554 | } | ||
555 | if (uri != NULL) { | ||
556 | g_free(uri->path); | ||
557 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
558 | * | ||
559 | * Returns 0 or the error code | ||
560 | */ | ||
561 | -static int | ||
562 | -rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
563 | +static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
564 | { | ||
565 | const char *cur; | ||
566 | int ret; | ||
567 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
568 | cur = *str; | ||
569 | |||
570 | ret = rfc3986_parse_segment(&cur, ':', 0); | ||
571 | - if (ret != 0) return(ret); | ||
572 | + if (ret != 0) | ||
573 | + return (ret); | ||
574 | while (*cur == '/') { | ||
575 | cur++; | ||
576 | - ret = rfc3986_parse_segment(&cur, 0, 1); | ||
577 | - if (ret != 0) return(ret); | ||
578 | + ret = rfc3986_parse_segment(&cur, 0, 1); | ||
579 | + if (ret != 0) | ||
580 | + return (ret); | ||
581 | } | ||
582 | if (uri != NULL) { | ||
583 | g_free(uri->path); | ||
584 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
585 | * | ||
586 | * Returns 0 or the error code | ||
587 | */ | ||
588 | -static int | ||
589 | -rfc3986_parse_hier_part(URI *uri, const char **str) | ||
590 | +static int rfc3986_parse_hier_part(URI *uri, const char **str) | ||
591 | { | ||
592 | const char *cur; | ||
593 | int ret; | ||
594 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_hier_part(URI *uri, const char **str) | ||
595 | |||
596 | if ((*cur == '/') && (*(cur + 1) == '/')) { | ||
597 | cur += 2; | ||
598 | - ret = rfc3986_parse_authority(uri, &cur); | ||
599 | - if (ret != 0) return(ret); | ||
600 | - ret = rfc3986_parse_path_ab_empty(uri, &cur); | ||
601 | - if (ret != 0) return(ret); | ||
602 | - *str = cur; | ||
603 | - return(0); | ||
604 | + ret = rfc3986_parse_authority(uri, &cur); | ||
605 | + if (ret != 0) | ||
606 | + return (ret); | ||
607 | + ret = rfc3986_parse_path_ab_empty(uri, &cur); | ||
608 | + if (ret != 0) | ||
609 | + return (ret); | ||
610 | + *str = cur; | ||
611 | + return (0); | ||
612 | } else if (*cur == '/') { | ||
613 | ret = rfc3986_parse_path_absolute(uri, &cur); | ||
614 | - if (ret != 0) return(ret); | ||
615 | + if (ret != 0) | ||
616 | + return (ret); | ||
617 | } else if (ISA_PCHAR(cur)) { | ||
618 | ret = rfc3986_parse_path_rootless(uri, &cur); | ||
619 | - if (ret != 0) return(ret); | ||
620 | + if (ret != 0) | ||
621 | + return (ret); | ||
622 | } else { | ||
623 | - /* path-empty is effectively empty */ | ||
624 | - if (uri != NULL) { | ||
625 | + /* path-empty is effectively empty */ | ||
626 | + if (uri != NULL) { | ||
627 | g_free(uri->path); | ||
628 | - uri->path = NULL; | ||
629 | - } | ||
630 | + uri->path = NULL; | ||
631 | + } | ||
632 | } | ||
633 | *str = cur; | ||
634 | return (0); | ||
635 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_hier_part(URI *uri, const char **str) | ||
636 | * | ||
637 | * Returns 0 or the error code | ||
638 | */ | ||
639 | -static int | ||
640 | -rfc3986_parse_relative_ref(URI *uri, const char *str) { | ||
641 | +static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
642 | +{ | ||
643 | int ret; | ||
644 | |||
645 | if ((*str == '/') && (*(str + 1) == '/')) { | ||
646 | str += 2; | ||
647 | - ret = rfc3986_parse_authority(uri, &str); | ||
648 | - if (ret != 0) return(ret); | ||
649 | - ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
650 | - if (ret != 0) return(ret); | ||
651 | + ret = rfc3986_parse_authority(uri, &str); | ||
652 | + if (ret != 0) | ||
653 | + return (ret); | ||
654 | + ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
655 | + if (ret != 0) | ||
656 | + return (ret); | ||
657 | } else if (*str == '/') { | ||
658 | - ret = rfc3986_parse_path_absolute(uri, &str); | ||
659 | - if (ret != 0) return(ret); | ||
660 | + ret = rfc3986_parse_path_absolute(uri, &str); | ||
661 | + if (ret != 0) | ||
662 | + return (ret); | ||
663 | } else if (ISA_PCHAR(str)) { | ||
664 | ret = rfc3986_parse_path_no_scheme(uri, &str); | ||
665 | - if (ret != 0) return(ret); | ||
666 | + if (ret != 0) | ||
667 | + return (ret); | ||
668 | } else { | ||
669 | - /* path-empty is effectively empty */ | ||
670 | - if (uri != NULL) { | ||
671 | + /* path-empty is effectively empty */ | ||
672 | + if (uri != NULL) { | ||
673 | g_free(uri->path); | ||
674 | - uri->path = NULL; | ||
675 | - } | ||
676 | + uri->path = NULL; | ||
677 | + } | ||
678 | } | ||
679 | |||
680 | if (*str == '?') { | ||
681 | - str++; | ||
682 | - ret = rfc3986_parse_query(uri, &str); | ||
683 | - if (ret != 0) return(ret); | ||
684 | + str++; | ||
685 | + ret = rfc3986_parse_query(uri, &str); | ||
686 | + if (ret != 0) | ||
687 | + return (ret); | ||
688 | } | ||
689 | if (*str == '#') { | ||
690 | - str++; | ||
691 | - ret = rfc3986_parse_fragment(uri, &str); | ||
692 | - if (ret != 0) return(ret); | ||
693 | + str++; | ||
694 | + ret = rfc3986_parse_fragment(uri, &str); | ||
695 | + if (ret != 0) | ||
696 | + return (ret); | ||
697 | } | ||
698 | if (*str != 0) { | ||
699 | - uri_clean(uri); | ||
700 | - return(1); | ||
701 | + uri_clean(uri); | ||
702 | + return (1); | ||
703 | } | ||
704 | - return(0); | ||
705 | + return (0); | ||
706 | } | ||
707 | |||
708 | - | ||
709 | /** | ||
710 | * rfc3986_parse: | ||
711 | * @uri: pointer to an URI structure | ||
712 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_relative_ref(URI *uri, const char *str) { | ||
713 | * | ||
714 | * Returns 0 or the error code | ||
715 | */ | ||
716 | -static int | ||
717 | -rfc3986_parse(URI *uri, const char *str) { | ||
718 | +static int rfc3986_parse(URI *uri, const char *str) | ||
719 | +{ | ||
720 | int ret; | ||
721 | |||
722 | ret = rfc3986_parse_scheme(uri, &str); | ||
723 | - if (ret != 0) return(ret); | ||
724 | + if (ret != 0) | ||
725 | + return (ret); | ||
726 | if (*str != ':') { | ||
727 | - return(1); | ||
728 | + return (1); | ||
729 | } | ||
730 | str++; | ||
731 | ret = rfc3986_parse_hier_part(uri, &str); | ||
732 | - if (ret != 0) return(ret); | ||
733 | + if (ret != 0) | ||
734 | + return (ret); | ||
735 | if (*str == '?') { | ||
736 | - str++; | ||
737 | - ret = rfc3986_parse_query(uri, &str); | ||
738 | - if (ret != 0) return(ret); | ||
739 | + str++; | ||
740 | + ret = rfc3986_parse_query(uri, &str); | ||
741 | + if (ret != 0) | ||
742 | + return (ret); | ||
743 | } | ||
744 | if (*str == '#') { | ||
745 | - str++; | ||
746 | - ret = rfc3986_parse_fragment(uri, &str); | ||
747 | - if (ret != 0) return(ret); | ||
748 | + str++; | ||
749 | + ret = rfc3986_parse_fragment(uri, &str); | ||
750 | + if (ret != 0) | ||
751 | + return (ret); | ||
752 | } | ||
753 | if (*str != 0) { | ||
754 | - uri_clean(uri); | ||
755 | - return(1); | ||
756 | + uri_clean(uri); | ||
757 | + return (1); | ||
758 | } | ||
759 | - return(0); | ||
760 | + return (0); | ||
761 | } | ||
762 | |||
763 | /** | ||
764 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse(URI *uri, const char *str) { | ||
765 | * | ||
766 | * Returns 0 or the error code | ||
767 | */ | ||
768 | -static int | ||
769 | -rfc3986_parse_uri_reference(URI *uri, const char *str) { | ||
770 | +static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
771 | +{ | ||
772 | int ret; | ||
773 | |||
774 | if (str == NULL) | ||
775 | - return(-1); | ||
776 | + return (-1); | ||
777 | uri_clean(uri); | ||
778 | |||
779 | /* | ||
780 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_uri_reference(URI *uri, const char *str) { | ||
781 | */ | ||
782 | ret = rfc3986_parse(uri, str); | ||
783 | if (ret != 0) { | ||
784 | - uri_clean(uri); | ||
785 | + uri_clean(uri); | ||
786 | ret = rfc3986_parse_relative_ref(uri, str); | ||
787 | - if (ret != 0) { | ||
788 | - uri_clean(uri); | ||
789 | - return(ret); | ||
790 | - } | ||
791 | + if (ret != 0) { | ||
792 | + uri_clean(uri); | ||
793 | + return (ret); | ||
794 | + } | ||
795 | } | ||
796 | - return(0); | ||
797 | + return (0); | ||
798 | } | ||
799 | |||
800 | /** | ||
801 | @@ -XXX,XX +XXX,XX @@ rfc3986_parse_uri_reference(URI *uri, const char *str) { | ||
802 | * | ||
803 | * Returns a newly built URI or NULL in case of error | ||
804 | */ | ||
805 | -URI * | ||
806 | -uri_parse(const char *str) { | ||
807 | +URI *uri_parse(const char *str) | ||
808 | +{ | ||
809 | URI *uri; | ||
810 | int ret; | ||
811 | |||
812 | if (str == NULL) | ||
813 | - return(NULL); | ||
814 | + return (NULL); | ||
815 | uri = uri_new(); | ||
816 | ret = rfc3986_parse_uri_reference(uri, str); | ||
817 | if (ret) { | ||
818 | uri_free(uri); | ||
819 | - return(NULL); | ||
820 | + return (NULL); | ||
821 | } | ||
822 | - return(uri); | ||
823 | + return (uri); | ||
824 | } | ||
825 | |||
826 | /** | ||
827 | @@ -XXX,XX +XXX,XX @@ uri_parse(const char *str) { | ||
828 | * | ||
829 | * Returns 0 or the error code | ||
830 | */ | ||
831 | -int | ||
832 | -uri_parse_into(URI *uri, const char *str) { | ||
833 | - return(rfc3986_parse_uri_reference(uri, str)); | ||
834 | +int uri_parse_into(URI *uri, const char *str) | ||
835 | +{ | ||
836 | + return (rfc3986_parse_uri_reference(uri, str)); | ||
837 | } | ||
838 | |||
839 | /** | ||
840 | @@ -XXX,XX +XXX,XX @@ uri_parse_into(URI *uri, const char *str) { | ||
841 | * | ||
842 | * Returns a newly built URI or NULL in case of error | ||
843 | */ | ||
844 | -URI * | ||
845 | -uri_parse_raw(const char *str, int raw) { | ||
846 | +URI *uri_parse_raw(const char *str, int raw) | ||
847 | +{ | ||
848 | URI *uri; | ||
849 | int ret; | ||
850 | |||
851 | if (str == NULL) | ||
852 | - return(NULL); | ||
853 | + return (NULL); | ||
854 | uri = uri_new(); | ||
855 | if (raw) { | ||
856 | uri->cleanup |= 2; | ||
857 | @@ -XXX,XX +XXX,XX @@ uri_parse_raw(const char *str, int raw) { | ||
858 | ret = uri_parse_into(uri, str); | ||
859 | if (ret) { | ||
860 | uri_free(uri); | ||
861 | - return(NULL); | ||
862 | + return (NULL); | ||
863 | } | ||
864 | - return(uri); | ||
865 | + return (uri); | ||
866 | } | ||
867 | |||
868 | /************************************************************************ | ||
869 | - * * | ||
870 | - * Generic URI structure functions * | ||
871 | - * * | ||
872 | + * * | ||
873 | + * Generic URI structure functions * | ||
874 | + * * | ||
875 | ************************************************************************/ | ||
876 | |||
877 | /** | ||
878 | @@ -XXX,XX +XXX,XX @@ uri_parse_raw(const char *str, int raw) { | ||
879 | * | ||
880 | * Returns the new structure or NULL in case of error | ||
881 | */ | ||
882 | -URI * | ||
883 | -uri_new(void) { | ||
884 | +URI *uri_new(void) | ||
885 | +{ | ||
886 | URI *ret; | ||
887 | |||
888 | ret = g_new0(URI, 1); | ||
889 | - return(ret); | ||
890 | + return (ret); | ||
891 | } | ||
892 | |||
893 | /** | ||
894 | @@ -XXX,XX +XXX,XX @@ uri_new(void) { | ||
895 | * Function to handle properly a reallocation when saving an URI | ||
896 | * Also imposes some limit on the length of an URI string output | ||
897 | */ | ||
898 | -static char * | ||
899 | -realloc2n(char *ret, int *max) { | ||
900 | +static char *realloc2n(char *ret, int *max) | ||
901 | +{ | ||
902 | char *temp; | ||
903 | int tmp; | ||
904 | |||
905 | tmp = *max * 2; | ||
906 | temp = g_realloc(ret, (tmp + 1)); | ||
907 | *max = tmp; | ||
908 | - return(temp); | ||
909 | + return (temp); | ||
910 | } | ||
911 | |||
912 | /** | ||
913 | @@ -XXX,XX +XXX,XX @@ realloc2n(char *ret, int *max) { | ||
914 | * | ||
915 | * Returns a new string (to be deallocated by caller) | ||
916 | */ | ||
917 | -char * | ||
918 | -uri_to_string(URI *uri) { | ||
919 | +char *uri_to_string(URI *uri) | ||
920 | +{ | ||
921 | char *ret = NULL; | ||
922 | char *temp; | ||
923 | const char *p; | ||
924 | int len; | ||
925 | int max; | ||
926 | |||
927 | - if (uri == NULL) return(NULL); | ||
928 | - | ||
929 | + if (uri == NULL) | ||
930 | + return (NULL); | ||
931 | |||
932 | max = 80; | ||
933 | ret = g_malloc(max + 1); | ||
934 | len = 0; | ||
935 | |||
936 | if (uri->scheme != NULL) { | ||
937 | - p = uri->scheme; | ||
938 | - while (*p != 0) { | ||
939 | - if (len >= max) { | ||
940 | + p = uri->scheme; | ||
941 | + while (*p != 0) { | ||
942 | + if (len >= max) { | ||
943 | temp = realloc2n(ret, &max); | ||
944 | - ret = temp; | ||
945 | - } | ||
946 | - ret[len++] = *p++; | ||
947 | - } | ||
948 | - if (len >= max) { | ||
949 | + ret = temp; | ||
950 | + } | ||
951 | + ret[len++] = *p++; | ||
952 | + } | ||
953 | + if (len >= max) { | ||
954 | temp = realloc2n(ret, &max); | ||
955 | ret = temp; | ||
956 | - } | ||
957 | - ret[len++] = ':'; | ||
958 | + } | ||
959 | + ret[len++] = ':'; | ||
960 | } | ||
961 | if (uri->opaque != NULL) { | ||
962 | - p = uri->opaque; | ||
963 | - while (*p != 0) { | ||
964 | - if (len + 3 >= max) { | ||
965 | + p = uri->opaque; | ||
966 | + while (*p != 0) { | ||
967 | + if (len + 3 >= max) { | ||
968 | temp = realloc2n(ret, &max); | ||
969 | ret = temp; | ||
970 | - } | ||
971 | - if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) | ||
972 | - ret[len++] = *p++; | ||
973 | - else { | ||
974 | - int val = *(unsigned char *)p++; | ||
975 | - int hi = val / 0x10, lo = val % 0x10; | ||
976 | - ret[len++] = '%'; | ||
977 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
978 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
979 | - } | ||
980 | - } | ||
981 | + } | ||
982 | + if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) | ||
983 | + ret[len++] = *p++; | ||
984 | + else { | ||
985 | + int val = *(unsigned char *)p++; | ||
986 | + int hi = val / 0x10, lo = val % 0x10; | ||
987 | + ret[len++] = '%'; | ||
988 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
989 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
990 | + } | ||
991 | + } | ||
992 | } else { | ||
993 | - if (uri->server != NULL) { | ||
994 | - if (len + 3 >= max) { | ||
995 | + if (uri->server != NULL) { | ||
996 | + if (len + 3 >= max) { | ||
997 | temp = realloc2n(ret, &max); | ||
998 | ret = temp; | ||
999 | - } | ||
1000 | - ret[len++] = '/'; | ||
1001 | - ret[len++] = '/'; | ||
1002 | - if (uri->user != NULL) { | ||
1003 | - p = uri->user; | ||
1004 | - while (*p != 0) { | ||
1005 | - if (len + 3 >= max) { | ||
1006 | + } | ||
1007 | + ret[len++] = '/'; | ||
1008 | + ret[len++] = '/'; | ||
1009 | + if (uri->user != NULL) { | ||
1010 | + p = uri->user; | ||
1011 | + while (*p != 0) { | ||
1012 | + if (len + 3 >= max) { | ||
1013 | temp = realloc2n(ret, &max); | ||
1014 | ret = temp; | ||
1015 | - } | ||
1016 | - if ((IS_UNRESERVED(*(p))) || | ||
1017 | - ((*(p) == ';')) || ((*(p) == ':')) || | ||
1018 | - ((*(p) == '&')) || ((*(p) == '=')) || | ||
1019 | - ((*(p) == '+')) || ((*(p) == '$')) || | ||
1020 | - ((*(p) == ','))) | ||
1021 | - ret[len++] = *p++; | ||
1022 | - else { | ||
1023 | - int val = *(unsigned char *)p++; | ||
1024 | - int hi = val / 0x10, lo = val % 0x10; | ||
1025 | - ret[len++] = '%'; | ||
1026 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1027 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1028 | - } | ||
1029 | - } | ||
1030 | - if (len + 3 >= max) { | ||
1031 | + } | ||
1032 | + if ((IS_UNRESERVED(*(p))) || ((*(p) == ';')) || | ||
1033 | + ((*(p) == ':')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
1034 | + ((*(p) == '+')) || ((*(p) == '$')) || ((*(p) == ','))) | ||
1035 | + ret[len++] = *p++; | ||
1036 | + else { | ||
1037 | + int val = *(unsigned char *)p++; | ||
1038 | + int hi = val / 0x10, lo = val % 0x10; | ||
1039 | + ret[len++] = '%'; | ||
1040 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1041 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1042 | + } | ||
1043 | + } | ||
1044 | + if (len + 3 >= max) { | ||
1045 | temp = realloc2n(ret, &max); | ||
1046 | ret = temp; | ||
1047 | - } | ||
1048 | - ret[len++] = '@'; | ||
1049 | - } | ||
1050 | - p = uri->server; | ||
1051 | - while (*p != 0) { | ||
1052 | - if (len >= max) { | ||
1053 | + } | ||
1054 | + ret[len++] = '@'; | ||
1055 | + } | ||
1056 | + p = uri->server; | ||
1057 | + while (*p != 0) { | ||
1058 | + if (len >= max) { | ||
1059 | temp = realloc2n(ret, &max); | ||
1060 | ret = temp; | ||
1061 | - } | ||
1062 | - ret[len++] = *p++; | ||
1063 | - } | ||
1064 | - if (uri->port > 0) { | ||
1065 | - if (len + 10 >= max) { | ||
1066 | + } | ||
1067 | + ret[len++] = *p++; | ||
1068 | + } | ||
1069 | + if (uri->port > 0) { | ||
1070 | + if (len + 10 >= max) { | ||
1071 | temp = realloc2n(ret, &max); | ||
1072 | ret = temp; | ||
1073 | - } | ||
1074 | - len += snprintf(&ret[len], max - len, ":%d", uri->port); | ||
1075 | - } | ||
1076 | - } else if (uri->authority != NULL) { | ||
1077 | - if (len + 3 >= max) { | ||
1078 | + } | ||
1079 | + len += snprintf(&ret[len], max - len, ":%d", uri->port); | ||
1080 | + } | ||
1081 | + } else if (uri->authority != NULL) { | ||
1082 | + if (len + 3 >= max) { | ||
1083 | temp = realloc2n(ret, &max); | ||
1084 | ret = temp; | ||
1085 | - } | ||
1086 | - ret[len++] = '/'; | ||
1087 | - ret[len++] = '/'; | ||
1088 | - p = uri->authority; | ||
1089 | - while (*p != 0) { | ||
1090 | - if (len + 3 >= max) { | ||
1091 | + } | ||
1092 | + ret[len++] = '/'; | ||
1093 | + ret[len++] = '/'; | ||
1094 | + p = uri->authority; | ||
1095 | + while (*p != 0) { | ||
1096 | + if (len + 3 >= max) { | ||
1097 | temp = realloc2n(ret, &max); | ||
1098 | ret = temp; | ||
1099 | - } | ||
1100 | - if ((IS_UNRESERVED(*(p))) || | ||
1101 | - ((*(p) == '$')) || ((*(p) == ',')) || ((*(p) == ';')) || | ||
1102 | - ((*(p) == ':')) || ((*(p) == '@')) || ((*(p) == '&')) || | ||
1103 | - ((*(p) == '=')) || ((*(p) == '+'))) | ||
1104 | - ret[len++] = *p++; | ||
1105 | - else { | ||
1106 | - int val = *(unsigned char *)p++; | ||
1107 | - int hi = val / 0x10, lo = val % 0x10; | ||
1108 | - ret[len++] = '%'; | ||
1109 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1110 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1111 | - } | ||
1112 | - } | ||
1113 | - } else if (uri->scheme != NULL) { | ||
1114 | - if (len + 3 >= max) { | ||
1115 | + } | ||
1116 | + if ((IS_UNRESERVED(*(p))) || ((*(p) == '$')) || | ||
1117 | + ((*(p) == ',')) || ((*(p) == ';')) || ((*(p) == ':')) || | ||
1118 | + ((*(p) == '@')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
1119 | + ((*(p) == '+'))) | ||
1120 | + ret[len++] = *p++; | ||
1121 | + else { | ||
1122 | + int val = *(unsigned char *)p++; | ||
1123 | + int hi = val / 0x10, lo = val % 0x10; | ||
1124 | + ret[len++] = '%'; | ||
1125 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1126 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1127 | + } | ||
1128 | + } | ||
1129 | + } else if (uri->scheme != NULL) { | ||
1130 | + if (len + 3 >= max) { | ||
1131 | temp = realloc2n(ret, &max); | ||
1132 | ret = temp; | ||
1133 | - } | ||
1134 | - ret[len++] = '/'; | ||
1135 | - ret[len++] = '/'; | ||
1136 | - } | ||
1137 | - if (uri->path != NULL) { | ||
1138 | - p = uri->path; | ||
1139 | - /* | ||
1140 | - * the colon in file:///d: should not be escaped or | ||
1141 | - * Windows accesses fail later. | ||
1142 | - */ | ||
1143 | - if ((uri->scheme != NULL) && | ||
1144 | - (p[0] == '/') && | ||
1145 | - (((p[1] >= 'a') && (p[1] <= 'z')) || | ||
1146 | - ((p[1] >= 'A') && (p[1] <= 'Z'))) && | ||
1147 | - (p[2] == ':') && | ||
1148 | - (!strcmp(uri->scheme, "file"))) { | ||
1149 | - if (len + 3 >= max) { | ||
1150 | + } | ||
1151 | + ret[len++] = '/'; | ||
1152 | + ret[len++] = '/'; | ||
1153 | + } | ||
1154 | + if (uri->path != NULL) { | ||
1155 | + p = uri->path; | ||
1156 | + /* | ||
1157 | + * the colon in file:///d: should not be escaped or | ||
1158 | + * Windows accesses fail later. | ||
1159 | + */ | ||
1160 | + if ((uri->scheme != NULL) && (p[0] == '/') && | ||
1161 | + (((p[1] >= 'a') && (p[1] <= 'z')) || | ||
1162 | + ((p[1] >= 'A') && (p[1] <= 'Z'))) && | ||
1163 | + (p[2] == ':') && (!strcmp(uri->scheme, "file"))) { | ||
1164 | + if (len + 3 >= max) { | ||
1165 | temp = realloc2n(ret, &max); | ||
1166 | ret = temp; | ||
1167 | - } | ||
1168 | - ret[len++] = *p++; | ||
1169 | - ret[len++] = *p++; | ||
1170 | - ret[len++] = *p++; | ||
1171 | - } | ||
1172 | - while (*p != 0) { | ||
1173 | - if (len + 3 >= max) { | ||
1174 | + } | ||
1175 | + ret[len++] = *p++; | ||
1176 | + ret[len++] = *p++; | ||
1177 | + ret[len++] = *p++; | ||
1178 | + } | ||
1179 | + while (*p != 0) { | ||
1180 | + if (len + 3 >= max) { | ||
1181 | temp = realloc2n(ret, &max); | ||
1182 | ret = temp; | ||
1183 | - } | ||
1184 | - if ((IS_UNRESERVED(*(p))) || ((*(p) == '/')) || | ||
1185 | + } | ||
1186 | + if ((IS_UNRESERVED(*(p))) || ((*(p) == '/')) || | ||
1187 | ((*(p) == ';')) || ((*(p) == '@')) || ((*(p) == '&')) || | ||
1188 | - ((*(p) == '=')) || ((*(p) == '+')) || ((*(p) == '$')) || | ||
1189 | - ((*(p) == ','))) | ||
1190 | - ret[len++] = *p++; | ||
1191 | - else { | ||
1192 | - int val = *(unsigned char *)p++; | ||
1193 | - int hi = val / 0x10, lo = val % 0x10; | ||
1194 | - ret[len++] = '%'; | ||
1195 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1196 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1197 | - } | ||
1198 | - } | ||
1199 | - } | ||
1200 | - if (uri->query != NULL) { | ||
1201 | - if (len + 1 >= max) { | ||
1202 | + ((*(p) == '=')) || ((*(p) == '+')) || ((*(p) == '$')) || | ||
1203 | + ((*(p) == ','))) | ||
1204 | + ret[len++] = *p++; | ||
1205 | + else { | ||
1206 | + int val = *(unsigned char *)p++; | ||
1207 | + int hi = val / 0x10, lo = val % 0x10; | ||
1208 | + ret[len++] = '%'; | ||
1209 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1210 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1211 | + } | ||
1212 | + } | ||
1213 | + } | ||
1214 | + if (uri->query != NULL) { | ||
1215 | + if (len + 1 >= max) { | ||
1216 | temp = realloc2n(ret, &max); | ||
1217 | ret = temp; | ||
1218 | - } | ||
1219 | - ret[len++] = '?'; | ||
1220 | - p = uri->query; | ||
1221 | - while (*p != 0) { | ||
1222 | - if (len + 1 >= max) { | ||
1223 | + } | ||
1224 | + ret[len++] = '?'; | ||
1225 | + p = uri->query; | ||
1226 | + while (*p != 0) { | ||
1227 | + if (len + 1 >= max) { | ||
1228 | temp = realloc2n(ret, &max); | ||
1229 | ret = temp; | ||
1230 | - } | ||
1231 | - ret[len++] = *p++; | ||
1232 | - } | ||
1233 | - } | ||
1234 | + } | ||
1235 | + ret[len++] = *p++; | ||
1236 | + } | ||
1237 | + } | ||
1238 | } | ||
1239 | if (uri->fragment != NULL) { | ||
1240 | - if (len + 3 >= max) { | ||
1241 | + if (len + 3 >= max) { | ||
1242 | temp = realloc2n(ret, &max); | ||
1243 | ret = temp; | ||
1244 | - } | ||
1245 | - ret[len++] = '#'; | ||
1246 | - p = uri->fragment; | ||
1247 | - while (*p != 0) { | ||
1248 | - if (len + 3 >= max) { | ||
1249 | + } | ||
1250 | + ret[len++] = '#'; | ||
1251 | + p = uri->fragment; | ||
1252 | + while (*p != 0) { | ||
1253 | + if (len + 3 >= max) { | ||
1254 | temp = realloc2n(ret, &max); | ||
1255 | ret = temp; | ||
1256 | - } | ||
1257 | - if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) | ||
1258 | - ret[len++] = *p++; | ||
1259 | - else { | ||
1260 | - int val = *(unsigned char *)p++; | ||
1261 | - int hi = val / 0x10, lo = val % 0x10; | ||
1262 | - ret[len++] = '%'; | ||
1263 | - ret[len++] = hi + (hi > 9? 'A'-10 : '0'); | ||
1264 | - ret[len++] = lo + (lo > 9? 'A'-10 : '0'); | ||
1265 | - } | ||
1266 | - } | ||
1267 | + } | ||
1268 | + if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) | ||
1269 | + ret[len++] = *p++; | ||
1270 | + else { | ||
1271 | + int val = *(unsigned char *)p++; | ||
1272 | + int hi = val / 0x10, lo = val % 0x10; | ||
1273 | + ret[len++] = '%'; | ||
1274 | + ret[len++] = hi + (hi > 9 ? 'A' - 10 : '0'); | ||
1275 | + ret[len++] = lo + (lo > 9 ? 'A' - 10 : '0'); | ||
1276 | + } | ||
1277 | + } | ||
1278 | } | ||
1279 | if (len >= max) { | ||
1280 | temp = realloc2n(ret, &max); | ||
1281 | ret = temp; | ||
1282 | } | ||
1283 | ret[len] = 0; | ||
1284 | - return(ret); | ||
1285 | + return (ret); | ||
1286 | } | ||
1287 | |||
1288 | /** | ||
1289 | @@ -XXX,XX +XXX,XX @@ uri_to_string(URI *uri) { | ||
1290 | * | ||
1291 | * Make sure the URI struct is free of content | ||
1292 | */ | ||
1293 | -static void | ||
1294 | -uri_clean(URI *uri) { | ||
1295 | - if (uri == NULL) return; | ||
1296 | +static void uri_clean(URI *uri) | ||
1297 | +{ | ||
1298 | + if (uri == NULL) | ||
1299 | + return; | ||
1300 | |||
1301 | g_free(uri->scheme); | ||
1302 | uri->scheme = NULL; | ||
1303 | @@ -XXX,XX +XXX,XX @@ uri_clean(URI *uri) { | ||
1304 | * | ||
1305 | * Free up the URI struct | ||
1306 | */ | ||
1307 | -void | ||
1308 | -uri_free(URI *uri) { | ||
1309 | +void uri_free(URI *uri) | ||
1310 | +{ | ||
1311 | uri_clean(uri); | ||
1312 | g_free(uri); | ||
1313 | } | ||
1314 | |||
1315 | /************************************************************************ | ||
1316 | - * * | ||
1317 | - * Helper functions * | ||
1318 | - * * | ||
1319 | + * * | ||
1320 | + * Helper functions * | ||
1321 | + * * | ||
1322 | ************************************************************************/ | ||
1323 | |||
1324 | /** | ||
1325 | @@ -XXX,XX +XXX,XX @@ uri_free(URI *uri) { | ||
1326 | * | ||
1327 | * Returns 0 or an error code | ||
1328 | */ | ||
1329 | -static int | ||
1330 | -normalize_uri_path(char *path) { | ||
1331 | +static int normalize_uri_path(char *path) | ||
1332 | +{ | ||
1333 | char *cur, *out; | ||
1334 | |||
1335 | if (path == NULL) | ||
1336 | - return(-1); | ||
1337 | + return (-1); | ||
1338 | |||
1339 | /* Skip all initial "/" chars. We want to get to the beginning of the | ||
1340 | * first non-empty segment. | ||
1341 | */ | ||
1342 | cur = path; | ||
1343 | while (cur[0] == '/') | ||
1344 | - ++cur; | ||
1345 | + ++cur; | ||
1346 | if (cur[0] == '\0') | ||
1347 | - return(0); | ||
1348 | + return (0); | ||
1349 | |||
1350 | /* Keep everything we've seen so far. */ | ||
1351 | out = cur; | ||
1352 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1353 | * Analyze each segment in sequence for cases (c) and (d). | ||
1354 | */ | ||
1355 | while (cur[0] != '\0') { | ||
1356 | - /* | ||
1357 | - * c) All occurrences of "./", where "." is a complete path segment, | ||
1358 | - * are removed from the buffer string. | ||
1359 | - */ | ||
1360 | - if ((cur[0] == '.') && (cur[1] == '/')) { | ||
1361 | - cur += 2; | ||
1362 | - /* '//' normalization should be done at this point too */ | ||
1363 | - while (cur[0] == '/') | ||
1364 | - cur++; | ||
1365 | - continue; | ||
1366 | - } | ||
1367 | + /* | ||
1368 | + * c) All occurrences of "./", where "." is a complete path segment, | ||
1369 | + * are removed from the buffer string. | ||
1370 | + */ | ||
1371 | + if ((cur[0] == '.') && (cur[1] == '/')) { | ||
1372 | + cur += 2; | ||
1373 | + /* '//' normalization should be done at this point too */ | ||
1374 | + while (cur[0] == '/') | ||
1375 | + cur++; | ||
1376 | + continue; | ||
1377 | + } | ||
1378 | |||
1379 | - /* | ||
1380 | - * d) If the buffer string ends with "." as a complete path segment, | ||
1381 | - * that "." is removed. | ||
1382 | - */ | ||
1383 | - if ((cur[0] == '.') && (cur[1] == '\0')) | ||
1384 | - break; | ||
1385 | + /* | ||
1386 | + * d) If the buffer string ends with "." as a complete path segment, | ||
1387 | + * that "." is removed. | ||
1388 | + */ | ||
1389 | + if ((cur[0] == '.') && (cur[1] == '\0')) | ||
1390 | + break; | ||
1391 | |||
1392 | - /* Otherwise keep the segment. */ | ||
1393 | - while (cur[0] != '/') { | ||
1394 | + /* Otherwise keep the segment. */ | ||
1395 | + while (cur[0] != '/') { | ||
1396 | if (cur[0] == '\0') | ||
1397 | - goto done_cd; | ||
1398 | - (out++)[0] = (cur++)[0]; | ||
1399 | - } | ||
1400 | - /* nomalize // */ | ||
1401 | - while ((cur[0] == '/') && (cur[1] == '/')) | ||
1402 | - cur++; | ||
1403 | + goto done_cd; | ||
1404 | + (out++)[0] = (cur++)[0]; | ||
1405 | + } | ||
1406 | + /* nomalize // */ | ||
1407 | + while ((cur[0] == '/') && (cur[1] == '/')) | ||
1408 | + cur++; | ||
1409 | |||
1410 | (out++)[0] = (cur++)[0]; | ||
1411 | } | ||
1412 | - done_cd: | ||
1413 | +done_cd: | ||
1414 | out[0] = '\0'; | ||
1415 | |||
1416 | /* Reset to the beginning of the first segment for the next sequence. */ | ||
1417 | cur = path; | ||
1418 | while (cur[0] == '/') | ||
1419 | - ++cur; | ||
1420 | + ++cur; | ||
1421 | if (cur[0] == '\0') | ||
1422 | - return(0); | ||
1423 | + return (0); | ||
1424 | |||
1425 | /* | ||
1426 | * Analyze each segment in sequence for cases (e) and (f). | ||
1427 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1428 | /* Find the end of the current segment. */ | ||
1429 | segp = cur; | ||
1430 | while ((segp[0] != '/') && (segp[0] != '\0')) | ||
1431 | - ++segp; | ||
1432 | + ++segp; | ||
1433 | |||
1434 | /* If this is the last segment, we're done (we need at least two | ||
1435 | * segments to meet the criteria for the (e) and (f) cases). | ||
1436 | */ | ||
1437 | if (segp[0] == '\0') | ||
1438 | - break; | ||
1439 | + break; | ||
1440 | |||
1441 | /* If the first segment is "..", or if the next segment _isn't_ "..", | ||
1442 | * keep this segment and try the next one. | ||
1443 | */ | ||
1444 | ++segp; | ||
1445 | - if (((cur[0] == '.') && (cur[1] == '.') && (segp == cur+3)) | ||
1446 | - || ((segp[0] != '.') || (segp[1] != '.') | ||
1447 | - || ((segp[2] != '/') && (segp[2] != '\0')))) { | ||
1448 | - cur = segp; | ||
1449 | - continue; | ||
1450 | + if (((cur[0] == '.') && (cur[1] == '.') && (segp == cur + 3)) || | ||
1451 | + ((segp[0] != '.') || (segp[1] != '.') || | ||
1452 | + ((segp[2] != '/') && (segp[2] != '\0')))) { | ||
1453 | + cur = segp; | ||
1454 | + continue; | ||
1455 | } | ||
1456 | |||
1457 | /* If we get here, remove this segment and the next one and back up | ||
1458 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1459 | |||
1460 | /* If this is the end of the buffer, we're done. */ | ||
1461 | if (segp[2] == '\0') { | ||
1462 | - cur[0] = '\0'; | ||
1463 | - break; | ||
1464 | + cur[0] = '\0'; | ||
1465 | + break; | ||
1466 | } | ||
1467 | /* Valgrind complained, strcpy(cur, segp + 3); */ | ||
1468 | /* string will overlap, do not use strcpy */ | ||
1469 | tmp = cur; | ||
1470 | segp += 3; | ||
1471 | while ((*tmp++ = *segp++) != 0) | ||
1472 | - ; | ||
1473 | + ; | ||
1474 | |||
1475 | /* If there are no previous segments, then keep going from here. */ | ||
1476 | segp = cur; | ||
1477 | while ((segp > path) && ((--segp)[0] == '/')) | ||
1478 | - ; | ||
1479 | + ; | ||
1480 | if (segp == path) | ||
1481 | - continue; | ||
1482 | + continue; | ||
1483 | |||
1484 | /* "segp" is pointing to the end of a previous segment; find it's | ||
1485 | * start. We need to back up to the previous segment and start | ||
1486 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1487 | */ | ||
1488 | cur = segp; | ||
1489 | while ((cur > path) && (cur[-1] != '/')) | ||
1490 | - --cur; | ||
1491 | + --cur; | ||
1492 | } | ||
1493 | out[0] = '\0'; | ||
1494 | |||
1495 | @@ -XXX,XX +XXX,XX @@ normalize_uri_path(char *path) { | ||
1496 | * We discard them from the final path. | ||
1497 | */ | ||
1498 | if (path[0] == '/') { | ||
1499 | - cur = path; | ||
1500 | - while ((cur[0] == '/') && (cur[1] == '.') && (cur[2] == '.') | ||
1501 | - && ((cur[3] == '/') || (cur[3] == '\0'))) | ||
1502 | - cur += 3; | ||
1503 | + cur = path; | ||
1504 | + while ((cur[0] == '/') && (cur[1] == '.') && (cur[2] == '.') && | ||
1505 | + ((cur[3] == '/') || (cur[3] == '\0'))) | ||
1506 | + cur += 3; | ||
1507 | |||
1508 | - if (cur != path) { | ||
1509 | - out = path; | ||
1510 | - while (cur[0] != '\0') | ||
1511 | - (out++)[0] = (cur++)[0]; | ||
1512 | - out[0] = 0; | ||
1513 | - } | ||
1514 | + if (cur != path) { | ||
1515 | + out = path; | ||
1516 | + while (cur[0] != '\0') | ||
1517 | + (out++)[0] = (cur++)[0]; | ||
1518 | + out[0] = 0; | ||
1519 | + } | ||
1520 | } | ||
1521 | |||
1522 | - return(0); | ||
1523 | + return (0); | ||
1524 | } | ||
1525 | |||
1526 | -static int is_hex(char c) { | ||
1527 | - if (((c >= '0') && (c <= '9')) || | ||
1528 | - ((c >= 'a') && (c <= 'f')) || | ||
1529 | +static int is_hex(char c) | ||
1530 | +{ | ||
1531 | + if (((c >= '0') && (c <= '9')) || ((c >= 'a') && (c <= 'f')) || | ||
1532 | ((c >= 'A') && (c <= 'F'))) | ||
1533 | - return(1); | ||
1534 | - return(0); | ||
1535 | + return (1); | ||
1536 | + return (0); | ||
1537 | } | ||
1538 | |||
1539 | - | ||
1540 | /** | ||
1541 | * uri_string_unescape: | ||
1542 | * @str: the string to unescape | ||
1543 | @@ -XXX,XX +XXX,XX @@ static int is_hex(char c) { | ||
1544 | * Returns a copy of the string, but unescaped, will return NULL only in case | ||
1545 | * of error | ||
1546 | */ | ||
1547 | -char * | ||
1548 | -uri_string_unescape(const char *str, int len, char *target) { | ||
1549 | +char *uri_string_unescape(const char *str, int len, char *target) | ||
1550 | +{ | ||
1551 | char *ret, *out; | ||
1552 | const char *in; | ||
1553 | |||
1554 | if (str == NULL) | ||
1555 | - return(NULL); | ||
1556 | - if (len <= 0) len = strlen(str); | ||
1557 | - if (len < 0) return(NULL); | ||
1558 | + return (NULL); | ||
1559 | + if (len <= 0) | ||
1560 | + len = strlen(str); | ||
1561 | + if (len < 0) | ||
1562 | + return (NULL); | ||
1563 | |||
1564 | if (target == NULL) { | ||
1565 | - ret = g_malloc(len + 1); | ||
1566 | + ret = g_malloc(len + 1); | ||
1567 | } else | ||
1568 | - ret = target; | ||
1569 | + ret = target; | ||
1570 | in = str; | ||
1571 | out = ret; | ||
1572 | - while(len > 0) { | ||
1573 | - if ((len > 2) && (*in == '%') && (is_hex(in[1])) && (is_hex(in[2]))) { | ||
1574 | - in++; | ||
1575 | - if ((*in >= '0') && (*in <= '9')) | ||
1576 | - *out = (*in - '0'); | ||
1577 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
1578 | - *out = (*in - 'a') + 10; | ||
1579 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
1580 | - *out = (*in - 'A') + 10; | ||
1581 | - in++; | ||
1582 | - if ((*in >= '0') && (*in <= '9')) | ||
1583 | - *out = *out * 16 + (*in - '0'); | ||
1584 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
1585 | - *out = *out * 16 + (*in - 'a') + 10; | ||
1586 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
1587 | - *out = *out * 16 + (*in - 'A') + 10; | ||
1588 | - in++; | ||
1589 | - len -= 3; | ||
1590 | - out++; | ||
1591 | - } else { | ||
1592 | - *out++ = *in++; | ||
1593 | - len--; | ||
1594 | - } | ||
1595 | + while (len > 0) { | ||
1596 | + if ((len > 2) && (*in == '%') && (is_hex(in[1])) && (is_hex(in[2]))) { | ||
1597 | + in++; | ||
1598 | + if ((*in >= '0') && (*in <= '9')) | ||
1599 | + *out = (*in - '0'); | ||
1600 | + else if ((*in >= 'a') && (*in <= 'f')) | ||
1601 | + *out = (*in - 'a') + 10; | ||
1602 | + else if ((*in >= 'A') && (*in <= 'F')) | ||
1603 | + *out = (*in - 'A') + 10; | ||
1604 | + in++; | ||
1605 | + if ((*in >= '0') && (*in <= '9')) | ||
1606 | + *out = *out * 16 + (*in - '0'); | ||
1607 | + else if ((*in >= 'a') && (*in <= 'f')) | ||
1608 | + *out = *out * 16 + (*in - 'a') + 10; | ||
1609 | + else if ((*in >= 'A') && (*in <= 'F')) | ||
1610 | + *out = *out * 16 + (*in - 'A') + 10; | ||
1611 | + in++; | ||
1612 | + len -= 3; | ||
1613 | + out++; | ||
1614 | + } else { | ||
1615 | + *out++ = *in++; | ||
1616 | + len--; | ||
1617 | + } | ||
1618 | } | ||
1619 | *out = 0; | ||
1620 | - return(ret); | ||
1621 | + return (ret); | ||
1622 | } | ||
1623 | |||
1624 | /** | ||
1625 | @@ -XXX,XX +XXX,XX @@ uri_string_unescape(const char *str, int len, char *target) { | ||
1626 | * | ||
1627 | * Returns a new escaped string or NULL in case of error. | ||
1628 | */ | ||
1629 | -char * | ||
1630 | -uri_string_escape(const char *str, const char *list) { | ||
1631 | +char *uri_string_escape(const char *str, const char *list) | ||
1632 | +{ | ||
1633 | char *ret, ch; | ||
1634 | char *temp; | ||
1635 | const char *in; | ||
1636 | int len, out; | ||
1637 | |||
1638 | if (str == NULL) | ||
1639 | - return(NULL); | ||
1640 | + return (NULL); | ||
1641 | if (str[0] == 0) | ||
1642 | - return(g_strdup(str)); | ||
1643 | + return (g_strdup(str)); | ||
1644 | len = strlen(str); | ||
1645 | - if (!(len > 0)) return(NULL); | ||
1646 | + if (!(len > 0)) | ||
1647 | + return (NULL); | ||
1648 | |||
1649 | len += 20; | ||
1650 | ret = g_malloc(len); | ||
1651 | in = str; | ||
1652 | out = 0; | ||
1653 | - while(*in != 0) { | ||
1654 | - if (len - out <= 3) { | ||
1655 | + while (*in != 0) { | ||
1656 | + if (len - out <= 3) { | ||
1657 | temp = realloc2n(ret, &len); | ||
1658 | - ret = temp; | ||
1659 | - } | ||
1660 | + ret = temp; | ||
1661 | + } | ||
1662 | |||
1663 | - ch = *in; | ||
1664 | - | ||
1665 | - if ((ch != '@') && (!IS_UNRESERVED(ch)) && (!strchr(list, ch))) { | ||
1666 | - unsigned char val; | ||
1667 | - ret[out++] = '%'; | ||
1668 | - val = ch >> 4; | ||
1669 | - if (val <= 9) | ||
1670 | - ret[out++] = '0' + val; | ||
1671 | - else | ||
1672 | - ret[out++] = 'A' + val - 0xA; | ||
1673 | - val = ch & 0xF; | ||
1674 | - if (val <= 9) | ||
1675 | - ret[out++] = '0' + val; | ||
1676 | - else | ||
1677 | - ret[out++] = 'A' + val - 0xA; | ||
1678 | - in++; | ||
1679 | - } else { | ||
1680 | - ret[out++] = *in++; | ||
1681 | - } | ||
1682 | + ch = *in; | ||
1683 | |||
1684 | + if ((ch != '@') && (!IS_UNRESERVED(ch)) && (!strchr(list, ch))) { | ||
1685 | + unsigned char val; | ||
1686 | + ret[out++] = '%'; | ||
1687 | + val = ch >> 4; | ||
1688 | + if (val <= 9) | ||
1689 | + ret[out++] = '0' + val; | ||
1690 | + else | ||
1691 | + ret[out++] = 'A' + val - 0xA; | ||
1692 | + val = ch & 0xF; | ||
1693 | + if (val <= 9) | ||
1694 | + ret[out++] = '0' + val; | ||
1695 | + else | ||
1696 | + ret[out++] = 'A' + val - 0xA; | ||
1697 | + in++; | ||
1698 | + } else { | ||
1699 | + ret[out++] = *in++; | ||
1700 | + } | ||
1701 | } | ||
1702 | ret[out] = 0; | ||
1703 | - return(ret); | ||
1704 | + return (ret); | ||
1705 | } | ||
1706 | |||
1707 | /************************************************************************ | ||
1708 | - * * | ||
1709 | - * Public functions * | ||
1710 | - * * | ||
1711 | + * * | ||
1712 | + * Public functions * | ||
1713 | + * * | ||
1714 | ************************************************************************/ | ||
1715 | |||
1716 | /** | ||
1717 | @@ -XXX,XX +XXX,XX @@ uri_string_escape(const char *str, const char *list) { | ||
1718 | * Returns a new URI string (to be freed by the caller) or NULL in case | ||
1719 | * of error. | ||
1720 | */ | ||
1721 | -char * | ||
1722 | -uri_resolve(const char *uri, const char *base) { | ||
1723 | +char *uri_resolve(const char *uri, const char *base) | ||
1724 | +{ | ||
1725 | char *val = NULL; | ||
1726 | int ret, len, indx, cur, out; | ||
1727 | URI *ref = NULL; | ||
1728 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1729 | * URI. Should we do that here? | ||
1730 | */ | ||
1731 | if (uri == NULL) | ||
1732 | - ret = -1; | ||
1733 | + ret = -1; | ||
1734 | else { | ||
1735 | - if (*uri) { | ||
1736 | - ref = uri_new(); | ||
1737 | - ret = uri_parse_into(ref, uri); | ||
1738 | - } | ||
1739 | - else | ||
1740 | - ret = 0; | ||
1741 | + if (*uri) { | ||
1742 | + ref = uri_new(); | ||
1743 | + ret = uri_parse_into(ref, uri); | ||
1744 | + } else | ||
1745 | + ret = 0; | ||
1746 | } | ||
1747 | if (ret != 0) | ||
1748 | - goto done; | ||
1749 | + goto done; | ||
1750 | if ((ref != NULL) && (ref->scheme != NULL)) { | ||
1751 | - /* | ||
1752 | - * The URI is absolute don't modify. | ||
1753 | - */ | ||
1754 | - val = g_strdup(uri); | ||
1755 | - goto done; | ||
1756 | + /* | ||
1757 | + * The URI is absolute don't modify. | ||
1758 | + */ | ||
1759 | + val = g_strdup(uri); | ||
1760 | + goto done; | ||
1761 | } | ||
1762 | if (base == NULL) | ||
1763 | - ret = -1; | ||
1764 | + ret = -1; | ||
1765 | else { | ||
1766 | - bas = uri_new(); | ||
1767 | - ret = uri_parse_into(bas, base); | ||
1768 | + bas = uri_new(); | ||
1769 | + ret = uri_parse_into(bas, base); | ||
1770 | } | ||
1771 | if (ret != 0) { | ||
1772 | - if (ref) | ||
1773 | - val = uri_to_string(ref); | ||
1774 | - goto done; | ||
1775 | + if (ref) | ||
1776 | + val = uri_to_string(ref); | ||
1777 | + goto done; | ||
1778 | } | ||
1779 | if (ref == NULL) { | ||
1780 | - /* | ||
1781 | - * the base fragment must be ignored | ||
1782 | - */ | ||
1783 | + /* | ||
1784 | + * the base fragment must be ignored | ||
1785 | + */ | ||
1786 | g_free(bas->fragment); | ||
1787 | bas->fragment = NULL; | ||
1788 | - val = uri_to_string(bas); | ||
1789 | - goto done; | ||
1790 | + val = uri_to_string(bas); | ||
1791 | + goto done; | ||
1792 | } | ||
1793 | |||
1794 | /* | ||
1795 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1796 | */ | ||
1797 | res = uri_new(); | ||
1798 | if ((ref->scheme == NULL) && (ref->path == NULL) && | ||
1799 | - ((ref->authority == NULL) && (ref->server == NULL))) { | ||
1800 | + ((ref->authority == NULL) && (ref->server == NULL))) { | ||
1801 | res->scheme = g_strdup(bas->scheme); | ||
1802 | - if (bas->authority != NULL) | ||
1803 | - res->authority = g_strdup(bas->authority); | ||
1804 | - else if (bas->server != NULL) { | ||
1805 | + if (bas->authority != NULL) | ||
1806 | + res->authority = g_strdup(bas->authority); | ||
1807 | + else if (bas->server != NULL) { | ||
1808 | res->server = g_strdup(bas->server); | ||
1809 | res->user = g_strdup(bas->user); | ||
1810 | res->port = bas->port; | ||
1811 | - } | ||
1812 | + } | ||
1813 | res->path = g_strdup(bas->path); | ||
1814 | if (ref->query != NULL) { | ||
1815 | - res->query = g_strdup (ref->query); | ||
1816 | + res->query = g_strdup(ref->query); | ||
1817 | } else { | ||
1818 | res->query = g_strdup(bas->query); | ||
1819 | } | ||
1820 | res->fragment = g_strdup(ref->fragment); | ||
1821 | - goto step_7; | ||
1822 | + goto step_7; | ||
1823 | } | ||
1824 | |||
1825 | /* | ||
1826 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1827 | * scheme is inherited from the base URI's scheme component. | ||
1828 | */ | ||
1829 | if (ref->scheme != NULL) { | ||
1830 | - val = uri_to_string(ref); | ||
1831 | - goto done; | ||
1832 | + val = uri_to_string(ref); | ||
1833 | + goto done; | ||
1834 | } | ||
1835 | res->scheme = g_strdup(bas->scheme); | ||
1836 | |||
1837 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1838 | * use an authority component. | ||
1839 | */ | ||
1840 | if ((ref->authority != NULL) || (ref->server != NULL)) { | ||
1841 | - if (ref->authority != NULL) | ||
1842 | - res->authority = g_strdup(ref->authority); | ||
1843 | - else { | ||
1844 | - res->server = g_strdup(ref->server); | ||
1845 | + if (ref->authority != NULL) | ||
1846 | + res->authority = g_strdup(ref->authority); | ||
1847 | + else { | ||
1848 | + res->server = g_strdup(ref->server); | ||
1849 | res->user = g_strdup(ref->user); | ||
1850 | res->port = ref->port; | ||
1851 | - } | ||
1852 | + } | ||
1853 | res->path = g_strdup(ref->path); | ||
1854 | - goto step_7; | ||
1855 | + goto step_7; | ||
1856 | } | ||
1857 | if (bas->authority != NULL) | ||
1858 | - res->authority = g_strdup(bas->authority); | ||
1859 | + res->authority = g_strdup(bas->authority); | ||
1860 | else if (bas->server != NULL) { | ||
1861 | res->server = g_strdup(bas->server); | ||
1862 | res->user = g_strdup(bas->user); | ||
1863 | - res->port = bas->port; | ||
1864 | + res->port = bas->port; | ||
1865 | } | ||
1866 | |||
1867 | /* | ||
1868 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1869 | * the reference is an absolute-path and we skip to step 7. | ||
1870 | */ | ||
1871 | if ((ref->path != NULL) && (ref->path[0] == '/')) { | ||
1872 | - res->path = g_strdup(ref->path); | ||
1873 | - goto step_7; | ||
1874 | + res->path = g_strdup(ref->path); | ||
1875 | + goto step_7; | ||
1876 | } | ||
1877 | |||
1878 | - | ||
1879 | /* | ||
1880 | * 6) If this step is reached, then we are resolving a relative-path | ||
1881 | * reference. The relative path needs to be merged with the base | ||
1882 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1883 | */ | ||
1884 | len = 2; /* extra / and 0 */ | ||
1885 | if (ref->path != NULL) | ||
1886 | - len += strlen(ref->path); | ||
1887 | + len += strlen(ref->path); | ||
1888 | if (bas->path != NULL) | ||
1889 | - len += strlen(bas->path); | ||
1890 | + len += strlen(bas->path); | ||
1891 | res->path = g_malloc(len); | ||
1892 | res->path[0] = 0; | ||
1893 | |||
1894 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1895 | cur = 0; | ||
1896 | out = 0; | ||
1897 | if (bas->path != NULL) { | ||
1898 | - while (bas->path[cur] != 0) { | ||
1899 | - while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) | ||
1900 | - cur++; | ||
1901 | - if (bas->path[cur] == 0) | ||
1902 | - break; | ||
1903 | + while (bas->path[cur] != 0) { | ||
1904 | + while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) | ||
1905 | + cur++; | ||
1906 | + if (bas->path[cur] == 0) | ||
1907 | + break; | ||
1908 | |||
1909 | - cur++; | ||
1910 | - while (out < cur) { | ||
1911 | - res->path[out] = bas->path[out]; | ||
1912 | - out++; | ||
1913 | - } | ||
1914 | - } | ||
1915 | + cur++; | ||
1916 | + while (out < cur) { | ||
1917 | + res->path[out] = bas->path[out]; | ||
1918 | + out++; | ||
1919 | + } | ||
1920 | + } | ||
1921 | } | ||
1922 | res->path[out] = 0; | ||
1923 | |||
1924 | @@ -XXX,XX +XXX,XX @@ uri_resolve(const char *uri, const char *base) { | ||
1925 | * string. | ||
1926 | */ | ||
1927 | if (ref->path != NULL && ref->path[0] != 0) { | ||
1928 | - indx = 0; | ||
1929 | - /* | ||
1930 | - * Ensure the path includes a '/' | ||
1931 | - */ | ||
1932 | - if ((out == 0) && (bas->server != NULL)) | ||
1933 | - res->path[out++] = '/'; | ||
1934 | - while (ref->path[indx] != 0) { | ||
1935 | - res->path[out++] = ref->path[indx++]; | ||
1936 | - } | ||
1937 | + indx = 0; | ||
1938 | + /* | ||
1939 | + * Ensure the path includes a '/' | ||
1940 | + */ | ||
1941 | + if ((out == 0) && (bas->server != NULL)) | ||
1942 | + res->path[out++] = '/'; | ||
1943 | + while (ref->path[indx] != 0) { | ||
1944 | + res->path[out++] = ref->path[indx++]; | ||
1945 | + } | ||
1946 | } | ||
1947 | res->path[out] = 0; | ||
1948 | |||
1949 | @@ -XXX,XX +XXX,XX @@ step_7: | ||
1950 | |||
1951 | done: | ||
1952 | if (ref != NULL) | ||
1953 | - uri_free(ref); | ||
1954 | + uri_free(ref); | ||
1955 | if (bas != NULL) | ||
1956 | - uri_free(bas); | ||
1957 | + uri_free(bas); | ||
1958 | if (res != NULL) | ||
1959 | - uri_free(res); | ||
1960 | - return(val); | ||
1961 | + uri_free(res); | ||
1962 | + return (val); | ||
1963 | } | ||
1964 | |||
1965 | /** | ||
1966 | @@ -XXX,XX +XXX,XX @@ done: | ||
1967 | * Returns a new URI string (to be freed by the caller) or NULL in case | ||
1968 | * error. | ||
1969 | */ | ||
1970 | -char * | ||
1971 | -uri_resolve_relative (const char *uri, const char * base) | ||
1972 | +char *uri_resolve_relative(const char *uri, const char *base) | ||
1973 | { | ||
1974 | char *val = NULL; | ||
1975 | int ret; | ||
1976 | @@ -XXX,XX +XXX,XX @@ uri_resolve_relative (const char *uri, const char * base) | ||
1977 | int remove_path = 0; | ||
1978 | |||
1979 | if ((uri == NULL) || (*uri == 0)) | ||
1980 | - return NULL; | ||
1981 | + return NULL; | ||
1982 | |||
1983 | /* | ||
1984 | * First parse URI into a standard form | ||
1985 | */ | ||
1986 | - ref = uri_new (); | ||
1987 | + ref = uri_new(); | ||
1988 | /* If URI not already in "relative" form */ | ||
1989 | if (uri[0] != '.') { | ||
1990 | - ret = uri_parse_into (ref, uri); | ||
1991 | - if (ret != 0) | ||
1992 | - goto done; /* Error in URI, return NULL */ | ||
1993 | + ret = uri_parse_into(ref, uri); | ||
1994 | + if (ret != 0) | ||
1995 | + goto done; /* Error in URI, return NULL */ | ||
1996 | } else | ||
1997 | - ref->path = g_strdup(uri); | ||
1998 | + ref->path = g_strdup(uri); | ||
1999 | |||
2000 | /* | ||
2001 | * Next parse base into the same standard form | ||
2002 | */ | ||
2003 | if ((base == NULL) || (*base == 0)) { | ||
2004 | - val = g_strdup (uri); | ||
2005 | - goto done; | ||
2006 | + val = g_strdup(uri); | ||
2007 | + goto done; | ||
2008 | } | ||
2009 | - bas = uri_new (); | ||
2010 | + bas = uri_new(); | ||
2011 | if (base[0] != '.') { | ||
2012 | - ret = uri_parse_into (bas, base); | ||
2013 | - if (ret != 0) | ||
2014 | - goto done; /* Error in base, return NULL */ | ||
2015 | + ret = uri_parse_into(bas, base); | ||
2016 | + if (ret != 0) | ||
2017 | + goto done; /* Error in base, return NULL */ | ||
2018 | } else | ||
2019 | - bas->path = g_strdup(base); | ||
2020 | + bas->path = g_strdup(base); | ||
2021 | |||
2022 | /* | ||
2023 | * If the scheme / server on the URI differs from the base, | ||
2024 | * just return the URI | ||
2025 | */ | ||
2026 | if ((ref->scheme != NULL) && | ||
2027 | - ((bas->scheme == NULL) || | ||
2028 | - (strcmp (bas->scheme, ref->scheme)) || | ||
2029 | - (strcmp (bas->server, ref->server)))) { | ||
2030 | - val = g_strdup (uri); | ||
2031 | - goto done; | ||
2032 | + ((bas->scheme == NULL) || (strcmp(bas->scheme, ref->scheme)) || | ||
2033 | + (strcmp(bas->server, ref->server)))) { | ||
2034 | + val = g_strdup(uri); | ||
2035 | + goto done; | ||
2036 | } | ||
2037 | if (bas->path == ref->path || | ||
2038 | (bas->path && ref->path && !strcmp(bas->path, ref->path))) { | ||
2039 | - val = g_strdup(""); | ||
2040 | - goto done; | ||
2041 | + val = g_strdup(""); | ||
2042 | + goto done; | ||
2043 | } | ||
2044 | if (bas->path == NULL) { | ||
2045 | - val = g_strdup(ref->path); | ||
2046 | - goto done; | ||
2047 | + val = g_strdup(ref->path); | ||
2048 | + goto done; | ||
2049 | } | ||
2050 | if (ref->path == NULL) { | ||
2051 | - ref->path = (char *) "/"; | ||
2052 | - remove_path = 1; | ||
2053 | + ref->path = (char *)"/"; | ||
2054 | + remove_path = 1; | ||
2055 | } | ||
2056 | |||
2057 | /* | ||
2058 | @@ -XXX,XX +XXX,XX @@ uri_resolve_relative (const char *uri, const char * base) | ||
2059 | * two path components may be missing (bug 316224) | ||
2060 | */ | ||
2061 | if (bas->path == NULL) { | ||
2062 | - if (ref->path != NULL) { | ||
2063 | - uptr = ref->path; | ||
2064 | - if (*uptr == '/') | ||
2065 | - uptr++; | ||
2066 | - /* exception characters from uri_to_string */ | ||
2067 | - val = uri_string_escape(uptr, "/;&=+$,"); | ||
2068 | - } | ||
2069 | - goto done; | ||
2070 | + if (ref->path != NULL) { | ||
2071 | + uptr = ref->path; | ||
2072 | + if (*uptr == '/') | ||
2073 | + uptr++; | ||
2074 | + /* exception characters from uri_to_string */ | ||
2075 | + val = uri_string_escape(uptr, "/;&=+$,"); | ||
2076 | + } | ||
2077 | + goto done; | ||
2078 | } | ||
2079 | bptr = bas->path; | ||
2080 | if (ref->path == NULL) { | ||
2081 | - for (ix = 0; bptr[ix] != 0; ix++) { | ||
2082 | - if (bptr[ix] == '/') | ||
2083 | - nbslash++; | ||
2084 | - } | ||
2085 | - uptr = NULL; | ||
2086 | - len = 1; /* this is for a string terminator only */ | ||
2087 | + for (ix = 0; bptr[ix] != 0; ix++) { | ||
2088 | + if (bptr[ix] == '/') | ||
2089 | + nbslash++; | ||
2090 | + } | ||
2091 | + uptr = NULL; | ||
2092 | + len = 1; /* this is for a string terminator only */ | ||
2093 | } else { | ||
2094 | - /* | ||
2095 | - * Next we compare the two strings and find where they first differ | ||
2096 | - */ | ||
2097 | - if ((ref->path[pos] == '.') && (ref->path[pos+1] == '/')) | ||
2098 | + /* | ||
2099 | + * Next we compare the two strings and find where they first differ | ||
2100 | + */ | ||
2101 | + if ((ref->path[pos] == '.') && (ref->path[pos + 1] == '/')) | ||
2102 | pos += 2; | ||
2103 | - if ((*bptr == '.') && (bptr[1] == '/')) | ||
2104 | + if ((*bptr == '.') && (bptr[1] == '/')) | ||
2105 | bptr += 2; | ||
2106 | - else if ((*bptr == '/') && (ref->path[pos] != '/')) | ||
2107 | - bptr++; | ||
2108 | - while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) | ||
2109 | - pos++; | ||
2110 | + else if ((*bptr == '/') && (ref->path[pos] != '/')) | ||
2111 | + bptr++; | ||
2112 | + while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) | ||
2113 | + pos++; | ||
2114 | |||
2115 | - if (bptr[pos] == ref->path[pos]) { | ||
2116 | - val = g_strdup(""); | ||
2117 | - goto done; /* (I can't imagine why anyone would do this) */ | ||
2118 | - } | ||
2119 | + if (bptr[pos] == ref->path[pos]) { | ||
2120 | + val = g_strdup(""); | ||
2121 | + goto done; /* (I can't imagine why anyone would do this) */ | ||
2122 | + } | ||
2123 | |||
2124 | - /* | ||
2125 | - * In URI, "back up" to the last '/' encountered. This will be the | ||
2126 | - * beginning of the "unique" suffix of URI | ||
2127 | - */ | ||
2128 | - ix = pos; | ||
2129 | - if ((ref->path[ix] == '/') && (ix > 0)) | ||
2130 | - ix--; | ||
2131 | - else if ((ref->path[ix] == 0) && (ix > 1) && (ref->path[ix - 1] == '/')) | ||
2132 | - ix -= 2; | ||
2133 | - for (; ix > 0; ix--) { | ||
2134 | - if (ref->path[ix] == '/') | ||
2135 | - break; | ||
2136 | - } | ||
2137 | - if (ix == 0) { | ||
2138 | - uptr = ref->path; | ||
2139 | - } else { | ||
2140 | - ix++; | ||
2141 | - uptr = &ref->path[ix]; | ||
2142 | - } | ||
2143 | + /* | ||
2144 | + * In URI, "back up" to the last '/' encountered. This will be the | ||
2145 | + * beginning of the "unique" suffix of URI | ||
2146 | + */ | ||
2147 | + ix = pos; | ||
2148 | + if ((ref->path[ix] == '/') && (ix > 0)) | ||
2149 | + ix--; | ||
2150 | + else if ((ref->path[ix] == 0) && (ix > 1) && (ref->path[ix - 1] == '/')) | ||
2151 | + ix -= 2; | ||
2152 | + for (; ix > 0; ix--) { | ||
2153 | + if (ref->path[ix] == '/') | ||
2154 | + break; | ||
2155 | + } | ||
2156 | + if (ix == 0) { | ||
2157 | + uptr = ref->path; | ||
2158 | + } else { | ||
2159 | + ix++; | ||
2160 | + uptr = &ref->path[ix]; | ||
2161 | + } | ||
2162 | |||
2163 | - /* | ||
2164 | - * In base, count the number of '/' from the differing point | ||
2165 | - */ | ||
2166 | - if (bptr[pos] != ref->path[pos]) {/* check for trivial URI == base */ | ||
2167 | - for (; bptr[ix] != 0; ix++) { | ||
2168 | - if (bptr[ix] == '/') | ||
2169 | - nbslash++; | ||
2170 | - } | ||
2171 | - } | ||
2172 | - len = strlen (uptr) + 1; | ||
2173 | + /* | ||
2174 | + * In base, count the number of '/' from the differing point | ||
2175 | + */ | ||
2176 | + if (bptr[pos] != ref->path[pos]) { /* check for trivial URI == base */ | ||
2177 | + for (; bptr[ix] != 0; ix++) { | ||
2178 | + if (bptr[ix] == '/') | ||
2179 | + nbslash++; | ||
2180 | + } | ||
2181 | + } | ||
2182 | + len = strlen(uptr) + 1; | ||
2183 | } | ||
2184 | |||
2185 | if (nbslash == 0) { | ||
2186 | - if (uptr != NULL) | ||
2187 | - /* exception characters from uri_to_string */ | ||
2188 | - val = uri_string_escape(uptr, "/;&=+$,"); | ||
2189 | - goto done; | ||
2190 | + if (uptr != NULL) | ||
2191 | + /* exception characters from uri_to_string */ | ||
2192 | + val = uri_string_escape(uptr, "/;&=+$,"); | ||
2193 | + goto done; | ||
2194 | } | ||
2195 | |||
2196 | /* | ||
2197 | @@ -XXX,XX +XXX,XX @@ uri_resolve_relative (const char *uri, const char * base) | ||
2198 | * length of the remainder of the URI, plus enough space | ||
2199 | * for the "../" groups, plus one for the terminator | ||
2200 | */ | ||
2201 | - val = g_malloc (len + 3 * nbslash); | ||
2202 | + val = g_malloc(len + 3 * nbslash); | ||
2203 | vptr = val; | ||
2204 | /* | ||
2205 | * Put in as many "../" as needed | ||
2206 | */ | ||
2207 | - for (; nbslash>0; nbslash--) { | ||
2208 | - *vptr++ = '.'; | ||
2209 | - *vptr++ = '.'; | ||
2210 | - *vptr++ = '/'; | ||
2211 | + for (; nbslash > 0; nbslash--) { | ||
2212 | + *vptr++ = '.'; | ||
2213 | + *vptr++ = '.'; | ||
2214 | + *vptr++ = '/'; | ||
2215 | } | ||
2216 | /* | ||
2217 | * Finish up with the end of the URI | ||
2218 | */ | ||
2219 | if (uptr != NULL) { | ||
2220 | - if ((vptr > val) && (len > 0) && | ||
2221 | - (uptr[0] == '/') && (vptr[-1] == '/')) { | ||
2222 | - memcpy (vptr, uptr + 1, len - 1); | ||
2223 | - vptr[len - 2] = 0; | ||
2224 | - } else { | ||
2225 | - memcpy (vptr, uptr, len); | ||
2226 | - vptr[len - 1] = 0; | ||
2227 | - } | ||
2228 | + if ((vptr > val) && (len > 0) && (uptr[0] == '/') && | ||
2229 | + (vptr[-1] == '/')) { | ||
2230 | + memcpy(vptr, uptr + 1, len - 1); | ||
2231 | + vptr[len - 2] = 0; | ||
2232 | + } else { | ||
2233 | + memcpy(vptr, uptr, len); | ||
2234 | + vptr[len - 1] = 0; | ||
2235 | + } | ||
2236 | } else { | ||
2237 | - vptr[len - 1] = 0; | ||
2238 | + vptr[len - 1] = 0; | ||
2239 | } | ||
2240 | |||
2241 | /* escape the freshly-built path */ | ||
2242 | vptr = val; | ||
2243 | - /* exception characters from uri_to_string */ | ||
2244 | + /* exception characters from uri_to_string */ | ||
2245 | val = uri_string_escape(vptr, "/;&=+$,"); | ||
2246 | g_free(vptr); | ||
2247 | |||
2248 | @@ -XXX,XX +XXX,XX @@ done: | ||
2249 | if (remove_path != 0) | ||
2250 | ref->path = NULL; | ||
2251 | if (ref != NULL) | ||
2252 | - uri_free (ref); | ||
2253 | + uri_free(ref); | ||
2254 | if (bas != NULL) | ||
2255 | - uri_free (bas); | ||
2256 | + uri_free(bas); | ||
2257 | |||
2258 | return val; | ||
2259 | } | ||
2260 | @@ -XXX,XX +XXX,XX @@ done: | ||
2261 | * Utility functions to help parse and assemble query strings. | ||
2262 | */ | ||
2263 | |||
2264 | -struct QueryParams * | ||
2265 | -query_params_new (int init_alloc) | ||
2266 | +struct QueryParams *query_params_new(int init_alloc) | ||
2267 | { | ||
2268 | struct QueryParams *ps; | ||
2269 | |||
2270 | - if (init_alloc <= 0) init_alloc = 1; | ||
2271 | + if (init_alloc <= 0) | ||
2272 | + init_alloc = 1; | ||
2273 | |||
2274 | ps = g_new(QueryParams, 1); | ||
2275 | ps->n = 0; | ||
2276 | @@ -XXX,XX +XXX,XX @@ query_params_new (int init_alloc) | ||
2277 | /* Ensure there is space to store at least one more parameter | ||
2278 | * at the end of the set. | ||
2279 | */ | ||
2280 | -static int | ||
2281 | -query_params_append (struct QueryParams *ps, | ||
2282 | - const char *name, const char *value) | ||
2283 | +static int query_params_append(struct QueryParams *ps, const char *name, | ||
2284 | + const char *value) | ||
2285 | { | ||
2286 | if (ps->n >= ps->alloc) { | ||
2287 | ps->p = g_renew(QueryParam, ps->p, ps->alloc * 2); | ||
2288 | @@ -XXX,XX +XXX,XX @@ query_params_append (struct QueryParams *ps, | ||
2289 | return 0; | ||
2290 | } | ||
2291 | |||
2292 | -void | ||
2293 | -query_params_free (struct QueryParams *ps) | ||
2294 | +void query_params_free(struct QueryParams *ps) | ||
95 | { | 2295 | { |
96 | int i; | 2296 | int i; |
97 | @@ -XXX,XX +XXX,XX @@ int main(int argc, char **argv) | 2297 | |
98 | g_test_add_func("/throttle/config/enabled", test_enabled); | 2298 | for (i = 0; i < ps->n; ++i) { |
99 | g_test_add_func("/throttle/config/conflicting", test_conflicting_config); | 2299 | - g_free (ps->p[i].name); |
100 | g_test_add_func("/throttle/config/is_valid", test_is_valid); | 2300 | - g_free (ps->p[i].value); |
101 | + g_test_add_func("/throttle/config/ranges", test_ranges); | 2301 | + g_free(ps->p[i].name); |
102 | g_test_add_func("/throttle/config/max", test_max_is_missing_limit); | 2302 | + g_free(ps->p[i].value); |
103 | g_test_add_func("/throttle/config/iops_size", | 2303 | } |
104 | test_iops_size_is_missing_limit); | 2304 | - g_free (ps->p); |
2305 | - g_free (ps); | ||
2306 | + g_free(ps->p); | ||
2307 | + g_free(ps); | ||
2308 | } | ||
2309 | |||
2310 | -struct QueryParams * | ||
2311 | -query_params_parse (const char *query) | ||
2312 | +struct QueryParams *query_params_parse(const char *query) | ||
2313 | { | ||
2314 | struct QueryParams *ps; | ||
2315 | const char *end, *eq; | ||
2316 | |||
2317 | - ps = query_params_new (0); | ||
2318 | - if (!query || query[0] == '\0') return ps; | ||
2319 | + ps = query_params_new(0); | ||
2320 | + if (!query || query[0] == '\0') | ||
2321 | + return ps; | ||
2322 | |||
2323 | while (*query) { | ||
2324 | char *name = NULL, *value = NULL; | ||
2325 | |||
2326 | /* Find the next separator, or end of the string. */ | ||
2327 | - end = strchr (query, '&'); | ||
2328 | + end = strchr(query, '&'); | ||
2329 | if (!end) | ||
2330 | - end = strchr (query, ';'); | ||
2331 | + end = strchr(query, ';'); | ||
2332 | if (!end) | ||
2333 | - end = query + strlen (query); | ||
2334 | + end = query + strlen(query); | ||
2335 | |||
2336 | /* Find the first '=' character between here and end. */ | ||
2337 | - eq = strchr (query, '='); | ||
2338 | - if (eq && eq >= end) eq = NULL; | ||
2339 | + eq = strchr(query, '='); | ||
2340 | + if (eq && eq >= end) | ||
2341 | + eq = NULL; | ||
2342 | |||
2343 | /* Empty section (eg. "&&"). */ | ||
2344 | if (end == query) | ||
2345 | @@ -XXX,XX +XXX,XX @@ query_params_parse (const char *query) | ||
2346 | * and consistent with CGI.pm we assume value is "". | ||
2347 | */ | ||
2348 | else if (!eq) { | ||
2349 | - name = uri_string_unescape (query, end - query, NULL); | ||
2350 | + name = uri_string_unescape(query, end - query, NULL); | ||
2351 | value = NULL; | ||
2352 | } | ||
2353 | /* Or if we have "name=" here (works around annoying | ||
2354 | * problem when calling uri_string_unescape with len = 0). | ||
2355 | */ | ||
2356 | - else if (eq+1 == end) { | ||
2357 | - name = uri_string_unescape (query, eq - query, NULL); | ||
2358 | + else if (eq + 1 == end) { | ||
2359 | + name = uri_string_unescape(query, eq - query, NULL); | ||
2360 | value = g_new0(char, 1); | ||
2361 | } | ||
2362 | /* If the '=' character is at the beginning then we have | ||
2363 | @@ -XXX,XX +XXX,XX @@ query_params_parse (const char *query) | ||
2364 | |||
2365 | /* Otherwise it's "name=value". */ | ||
2366 | else { | ||
2367 | - name = uri_string_unescape (query, eq - query, NULL); | ||
2368 | - value = uri_string_unescape (eq+1, end - (eq+1), NULL); | ||
2369 | + name = uri_string_unescape(query, eq - query, NULL); | ||
2370 | + value = uri_string_unescape(eq + 1, end - (eq + 1), NULL); | ||
2371 | } | ||
2372 | |||
2373 | /* Append to the parameter set. */ | ||
2374 | - query_params_append (ps, name, value); | ||
2375 | + query_params_append(ps, name, value); | ||
2376 | g_free(name); | ||
2377 | g_free(value); | ||
2378 | |||
2379 | next: | ||
2380 | query = end; | ||
2381 | - if (*query) query ++; /* skip '&' separator */ | ||
2382 | + if (*query) | ||
2383 | + query++; /* skip '&' separator */ | ||
2384 | } | ||
2385 | |||
2386 | return ps; | ||
105 | -- | 2387 | -- |
106 | 2.13.5 | 2388 | 2.14.3 |
107 | 2389 | ||
108 | 2390 | diff view generated by jsdifflib |
1 | Most qcow2 files are uncompressed so it is wasteful to allocate (32 + 1) | 1 | From: Su Hang <suhang16@mails.ucas.ac.cn> |
---|---|---|---|
2 | * cluster_size + 512 bytes upfront. Allocate s->cluster_cache and | ||
3 | s->cluster_data when the first read operation is performance on a | ||
4 | compressed cluster. | ||
5 | 2 | ||
6 | The buffers are freed in .bdrv_close(). .bdrv_open() no longer has any | 3 | only remove brackets that wrap `return` statements' content. |
7 | code paths that can allocate these buffers, so remove the free functions | ||
8 | in the error code path. | ||
9 | 4 | ||
10 | This patch can result in significant memory savings when many qcow2 | 5 | use `perl -pi -e "s/return \((.*?)\);/return \1;/g" util/uri.c` |
11 | disks are attached or backing file chains are long: | 6 | to remove pattern like this: "return (1);" |
12 | 7 | ||
13 | Before 12.81% (1,023,193,088B) | 8 | Signed-off-by: Su Hang <suhang16@mails.ucas.ac.cn> |
14 | After 5.36% (393,893,888B) | 9 | Reviewed-by: Thomas Huth <thuth@redhat.com> |
15 | 10 | Message-id: 1519533358-13759-3-git-send-email-suhang16@mails.ucas.ac.cn | |
16 | Reported-by: Alexey Kardashevskiy <aik@ozlabs.ru> | ||
17 | Tested-by: Alexey Kardashevskiy <aik@ozlabs.ru> | ||
18 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
19 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
20 | Message-id: 20170821135530.32344-1-stefanha@redhat.com | ||
21 | Cc: Kevin Wolf <kwolf@redhat.com> | ||
22 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
23 | --- | 12 | --- |
24 | block/qcow2-cluster.c | 17 +++++++++++++++++ | 13 | util/uri.c | 160 ++++++++++++++++++++++++++++++------------------------------- |
25 | block/qcow2.c | 12 ------------ | 14 | 1 file changed, 80 insertions(+), 80 deletions(-) |
26 | 2 files changed, 17 insertions(+), 12 deletions(-) | ||
27 | 15 | ||
28 | diff --git a/block/qcow2-cluster.c b/block/qcow2-cluster.c | 16 | diff --git a/util/uri.c b/util/uri.c |
29 | index XXXXXXX..XXXXXXX 100644 | 17 | index XXXXXXX..XXXXXXX 100644 |
30 | --- a/block/qcow2-cluster.c | 18 | --- a/util/uri.c |
31 | +++ b/block/qcow2-cluster.c | 19 | +++ b/util/uri.c |
32 | @@ -XXX,XX +XXX,XX @@ int qcow2_decompress_cluster(BlockDriverState *bs, uint64_t cluster_offset) | 20 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_scheme(URI *uri, const char **str) |
33 | nb_csectors = ((cluster_offset >> s->csize_shift) & s->csize_mask) + 1; | 21 | const char *cur; |
34 | sector_offset = coffset & 511; | 22 | |
35 | csize = nb_csectors * 512 - sector_offset; | 23 | if (str == NULL) |
36 | + | 24 | - return (-1); |
37 | + /* Allocate buffers on first decompress operation, most images are | 25 | + return -1; |
38 | + * uncompressed and the memory overhead can be avoided. The buffers | 26 | |
39 | + * are freed in .bdrv_close(). | 27 | cur = *str; |
40 | + */ | 28 | if (!ISA_ALPHA(cur)) |
41 | + if (!s->cluster_data) { | 29 | - return (2); |
42 | + /* one more sector for decompressed data alignment */ | 30 | + return 2; |
43 | + s->cluster_data = qemu_try_blockalign(bs->file->bs, | 31 | cur++; |
44 | + QCOW_MAX_CRYPT_CLUSTERS * s->cluster_size + 512); | 32 | while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || (*cur == '+') || (*cur == '-') || |
45 | + if (!s->cluster_data) { | 33 | (*cur == '.')) |
46 | + return -ENOMEM; | 34 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_scheme(URI *uri, const char **str) |
47 | + } | 35 | uri->scheme = g_strndup(*str, cur - *str); |
48 | + } | 36 | } |
49 | + if (!s->cluster_cache) { | 37 | *str = cur; |
50 | + s->cluster_cache = g_malloc(s->cluster_size); | 38 | - return (0); |
51 | + } | 39 | + return 0; |
52 | + | 40 | } |
53 | BLKDBG_EVENT(bs->file, BLKDBG_READ_COMPRESSED); | 41 | |
54 | ret = bdrv_read(bs->file, coffset >> 9, s->cluster_data, | 42 | /** |
55 | nb_csectors); | 43 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_fragment(URI *uri, const char **str) |
56 | diff --git a/block/qcow2.c b/block/qcow2.c | 44 | const char *cur; |
57 | index XXXXXXX..XXXXXXX 100644 | 45 | |
58 | --- a/block/qcow2.c | 46 | if (str == NULL) |
59 | +++ b/block/qcow2.c | 47 | - return (-1); |
60 | @@ -XXX,XX +XXX,XX @@ static int qcow2_do_open(BlockDriverState *bs, QDict *options, int flags, | 48 | + return -1; |
61 | goto fail; | 49 | |
62 | } | 50 | cur = *str; |
63 | 51 | ||
64 | - s->cluster_cache = g_malloc(s->cluster_size); | 52 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_fragment(URI *uri, const char **str) |
65 | - /* one more sector for decompressed data alignment */ | 53 | uri->fragment = uri_string_unescape(*str, cur - *str, NULL); |
66 | - s->cluster_data = qemu_try_blockalign(bs->file->bs, QCOW_MAX_CRYPT_CLUSTERS | 54 | } |
67 | - * s->cluster_size + 512); | 55 | *str = cur; |
68 | - if (s->cluster_data == NULL) { | 56 | - return (0); |
69 | - error_setg(errp, "Could not allocate temporary cluster buffer"); | 57 | + return 0; |
70 | - ret = -ENOMEM; | 58 | } |
71 | - goto fail; | 59 | |
72 | - } | 60 | /** |
73 | - | 61 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_query(URI *uri, const char **str) |
74 | s->cluster_cache_offset = -1; | 62 | const char *cur; |
75 | s->flags = flags; | 63 | |
76 | 64 | if (str == NULL) | |
77 | @@ -XXX,XX +XXX,XX @@ static int qcow2_do_open(BlockDriverState *bs, QDict *options, int flags, | 65 | - return (-1); |
78 | if (s->refcount_block_cache) { | 66 | + return -1; |
79 | qcow2_cache_destroy(bs, s->refcount_block_cache); | 67 | |
80 | } | 68 | cur = *str; |
81 | - g_free(s->cluster_cache); | 69 | |
82 | - qemu_vfree(s->cluster_data); | 70 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_query(URI *uri, const char **str) |
83 | qcrypto_block_free(s->crypto); | 71 | uri->query = g_strndup(*str, cur - *str); |
84 | qapi_free_QCryptoBlockOpenOptions(s->crypto_opts); | 72 | } |
85 | return ret; | 73 | *str = cur; |
74 | - return (0); | ||
75 | + return 0; | ||
76 | } | ||
77 | |||
78 | /** | ||
79 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_user_info(URI *uri, const char **str) | ||
80 | uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
81 | } | ||
82 | *str = cur; | ||
83 | - return (0); | ||
84 | + return 0; | ||
85 | } | ||
86 | - return (1); | ||
87 | + return 1; | ||
88 | } | ||
89 | |||
90 | /** | ||
91 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_dec_octet(const char **str) | ||
92 | const char *cur = *str; | ||
93 | |||
94 | if (!(ISA_DIGIT(cur))) | ||
95 | - return (1); | ||
96 | + return 1; | ||
97 | if (!ISA_DIGIT(cur + 1)) | ||
98 | cur++; | ||
99 | else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) | ||
100 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_dec_octet(const char **str) | ||
101 | (*(cur + 1) <= '5')) | ||
102 | cur += 3; | ||
103 | else | ||
104 | - return (1); | ||
105 | + return 1; | ||
106 | *str = cur; | ||
107 | - return (0); | ||
108 | + return 0; | ||
109 | } | ||
110 | /** | ||
111 | * rfc3986_parse_host: | ||
112 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) | ||
113 | while ((*cur != ']') && (*cur != 0)) | ||
114 | cur++; | ||
115 | if (*cur != ']') | ||
116 | - return (1); | ||
117 | + return 1; | ||
118 | cur++; | ||
119 | goto found; | ||
120 | } | ||
121 | @@ -XXX,XX +XXX,XX @@ found: | ||
122 | uri->server = NULL; | ||
123 | } | ||
124 | *str = cur; | ||
125 | - return (0); | ||
126 | + return 0; | ||
127 | } | ||
128 | |||
129 | /** | ||
130 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_authority(URI *uri, const char **str) | ||
131 | cur++; | ||
132 | ret = rfc3986_parse_host(uri, &cur); | ||
133 | if (ret != 0) | ||
134 | - return (ret); | ||
135 | + return ret; | ||
136 | if (*cur == ':') { | ||
137 | cur++; | ||
138 | ret = rfc3986_parse_port(uri, &cur); | ||
139 | if (ret != 0) | ||
140 | - return (ret); | ||
141 | + return ret; | ||
142 | } | ||
143 | *str = cur; | ||
144 | - return (0); | ||
145 | + return 0; | ||
146 | } | ||
147 | |||
148 | /** | ||
149 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
150 | cur = *str; | ||
151 | if (!ISA_PCHAR(cur)) { | ||
152 | if (empty) | ||
153 | - return (0); | ||
154 | - return (1); | ||
155 | + return 0; | ||
156 | + return 1; | ||
157 | } | ||
158 | while (ISA_PCHAR(cur) && (*cur != forbid)) | ||
159 | NEXT(cur); | ||
160 | *str = cur; | ||
161 | - return (0); | ||
162 | + return 0; | ||
163 | } | ||
164 | |||
165 | /** | ||
166 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
167 | cur++; | ||
168 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
169 | if (ret != 0) | ||
170 | - return (ret); | ||
171 | + return ret; | ||
172 | } | ||
173 | if (uri != NULL) { | ||
174 | g_free(uri->path); | ||
175 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
176 | } | ||
177 | } | ||
178 | *str = cur; | ||
179 | - return (0); | ||
180 | + return 0; | ||
181 | } | ||
182 | |||
183 | /** | ||
184 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
185 | cur = *str; | ||
186 | |||
187 | if (*cur != '/') | ||
188 | - return (1); | ||
189 | + return 1; | ||
190 | cur++; | ||
191 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
192 | if (ret == 0) { | ||
193 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
194 | cur++; | ||
195 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
196 | if (ret != 0) | ||
197 | - return (ret); | ||
198 | + return ret; | ||
199 | } | ||
200 | } | ||
201 | if (uri != NULL) { | ||
202 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
203 | } | ||
204 | } | ||
205 | *str = cur; | ||
206 | - return (0); | ||
207 | + return 0; | ||
208 | } | ||
209 | |||
210 | /** | ||
211 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
212 | |||
213 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
214 | if (ret != 0) | ||
215 | - return (ret); | ||
216 | + return ret; | ||
217 | while (*cur == '/') { | ||
218 | cur++; | ||
219 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
220 | if (ret != 0) | ||
221 | - return (ret); | ||
222 | + return ret; | ||
223 | } | ||
224 | if (uri != NULL) { | ||
225 | g_free(uri->path); | ||
226 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
227 | } | ||
228 | } | ||
229 | *str = cur; | ||
230 | - return (0); | ||
231 | + return 0; | ||
232 | } | ||
233 | |||
234 | /** | ||
235 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
236 | |||
237 | ret = rfc3986_parse_segment(&cur, ':', 0); | ||
238 | if (ret != 0) | ||
239 | - return (ret); | ||
240 | + return ret; | ||
241 | while (*cur == '/') { | ||
242 | cur++; | ||
243 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
244 | if (ret != 0) | ||
245 | - return (ret); | ||
246 | + return ret; | ||
247 | } | ||
248 | if (uri != NULL) { | ||
249 | g_free(uri->path); | ||
250 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
251 | } | ||
252 | } | ||
253 | *str = cur; | ||
254 | - return (0); | ||
255 | + return 0; | ||
256 | } | ||
257 | |||
258 | /** | ||
259 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_hier_part(URI *uri, const char **str) | ||
260 | cur += 2; | ||
261 | ret = rfc3986_parse_authority(uri, &cur); | ||
262 | if (ret != 0) | ||
263 | - return (ret); | ||
264 | + return ret; | ||
265 | ret = rfc3986_parse_path_ab_empty(uri, &cur); | ||
266 | if (ret != 0) | ||
267 | - return (ret); | ||
268 | + return ret; | ||
269 | *str = cur; | ||
270 | - return (0); | ||
271 | + return 0; | ||
272 | } else if (*cur == '/') { | ||
273 | ret = rfc3986_parse_path_absolute(uri, &cur); | ||
274 | if (ret != 0) | ||
275 | - return (ret); | ||
276 | + return ret; | ||
277 | } else if (ISA_PCHAR(cur)) { | ||
278 | ret = rfc3986_parse_path_rootless(uri, &cur); | ||
279 | if (ret != 0) | ||
280 | - return (ret); | ||
281 | + return ret; | ||
282 | } else { | ||
283 | /* path-empty is effectively empty */ | ||
284 | if (uri != NULL) { | ||
285 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_hier_part(URI *uri, const char **str) | ||
286 | } | ||
287 | } | ||
288 | *str = cur; | ||
289 | - return (0); | ||
290 | + return 0; | ||
291 | } | ||
292 | |||
293 | /** | ||
294 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
295 | str += 2; | ||
296 | ret = rfc3986_parse_authority(uri, &str); | ||
297 | if (ret != 0) | ||
298 | - return (ret); | ||
299 | + return ret; | ||
300 | ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
301 | if (ret != 0) | ||
302 | - return (ret); | ||
303 | + return ret; | ||
304 | } else if (*str == '/') { | ||
305 | ret = rfc3986_parse_path_absolute(uri, &str); | ||
306 | if (ret != 0) | ||
307 | - return (ret); | ||
308 | + return ret; | ||
309 | } else if (ISA_PCHAR(str)) { | ||
310 | ret = rfc3986_parse_path_no_scheme(uri, &str); | ||
311 | if (ret != 0) | ||
312 | - return (ret); | ||
313 | + return ret; | ||
314 | } else { | ||
315 | /* path-empty is effectively empty */ | ||
316 | if (uri != NULL) { | ||
317 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
318 | str++; | ||
319 | ret = rfc3986_parse_query(uri, &str); | ||
320 | if (ret != 0) | ||
321 | - return (ret); | ||
322 | + return ret; | ||
323 | } | ||
324 | if (*str == '#') { | ||
325 | str++; | ||
326 | ret = rfc3986_parse_fragment(uri, &str); | ||
327 | if (ret != 0) | ||
328 | - return (ret); | ||
329 | + return ret; | ||
330 | } | ||
331 | if (*str != 0) { | ||
332 | uri_clean(uri); | ||
333 | - return (1); | ||
334 | + return 1; | ||
335 | } | ||
336 | - return (0); | ||
337 | + return 0; | ||
338 | } | ||
339 | |||
340 | /** | ||
341 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse(URI *uri, const char *str) | ||
342 | |||
343 | ret = rfc3986_parse_scheme(uri, &str); | ||
344 | if (ret != 0) | ||
345 | - return (ret); | ||
346 | + return ret; | ||
347 | if (*str != ':') { | ||
348 | - return (1); | ||
349 | + return 1; | ||
350 | } | ||
351 | str++; | ||
352 | ret = rfc3986_parse_hier_part(uri, &str); | ||
353 | if (ret != 0) | ||
354 | - return (ret); | ||
355 | + return ret; | ||
356 | if (*str == '?') { | ||
357 | str++; | ||
358 | ret = rfc3986_parse_query(uri, &str); | ||
359 | if (ret != 0) | ||
360 | - return (ret); | ||
361 | + return ret; | ||
362 | } | ||
363 | if (*str == '#') { | ||
364 | str++; | ||
365 | ret = rfc3986_parse_fragment(uri, &str); | ||
366 | if (ret != 0) | ||
367 | - return (ret); | ||
368 | + return ret; | ||
369 | } | ||
370 | if (*str != 0) { | ||
371 | uri_clean(uri); | ||
372 | - return (1); | ||
373 | + return 1; | ||
374 | } | ||
375 | - return (0); | ||
376 | + return 0; | ||
377 | } | ||
378 | |||
379 | /** | ||
380 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
381 | int ret; | ||
382 | |||
383 | if (str == NULL) | ||
384 | - return (-1); | ||
385 | + return -1; | ||
386 | uri_clean(uri); | ||
387 | |||
388 | /* | ||
389 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
390 | ret = rfc3986_parse_relative_ref(uri, str); | ||
391 | if (ret != 0) { | ||
392 | uri_clean(uri); | ||
393 | - return (ret); | ||
394 | + return ret; | ||
395 | } | ||
396 | } | ||
397 | - return (0); | ||
398 | + return 0; | ||
399 | } | ||
400 | |||
401 | /** | ||
402 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse(const char *str) | ||
403 | int ret; | ||
404 | |||
405 | if (str == NULL) | ||
406 | - return (NULL); | ||
407 | + return NULL; | ||
408 | uri = uri_new(); | ||
409 | ret = rfc3986_parse_uri_reference(uri, str); | ||
410 | if (ret) { | ||
411 | uri_free(uri); | ||
412 | - return (NULL); | ||
413 | + return NULL; | ||
414 | } | ||
415 | - return (uri); | ||
416 | + return uri; | ||
417 | } | ||
418 | |||
419 | /** | ||
420 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse(const char *str) | ||
421 | */ | ||
422 | int uri_parse_into(URI *uri, const char *str) | ||
423 | { | ||
424 | - return (rfc3986_parse_uri_reference(uri, str)); | ||
425 | + return rfc3986_parse_uri_reference(uri, str); | ||
426 | } | ||
427 | |||
428 | /** | ||
429 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse_raw(const char *str, int raw) | ||
430 | int ret; | ||
431 | |||
432 | if (str == NULL) | ||
433 | - return (NULL); | ||
434 | + return NULL; | ||
435 | uri = uri_new(); | ||
436 | if (raw) { | ||
437 | uri->cleanup |= 2; | ||
438 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse_raw(const char *str, int raw) | ||
439 | ret = uri_parse_into(uri, str); | ||
440 | if (ret) { | ||
441 | uri_free(uri); | ||
442 | - return (NULL); | ||
443 | + return NULL; | ||
444 | } | ||
445 | - return (uri); | ||
446 | + return uri; | ||
447 | } | ||
448 | |||
449 | /************************************************************************ | ||
450 | @@ -XXX,XX +XXX,XX @@ URI *uri_new(void) | ||
451 | URI *ret; | ||
452 | |||
453 | ret = g_new0(URI, 1); | ||
454 | - return (ret); | ||
455 | + return ret; | ||
456 | } | ||
457 | |||
458 | /** | ||
459 | @@ -XXX,XX +XXX,XX @@ static char *realloc2n(char *ret, int *max) | ||
460 | tmp = *max * 2; | ||
461 | temp = g_realloc(ret, (tmp + 1)); | ||
462 | *max = tmp; | ||
463 | - return (temp); | ||
464 | + return temp; | ||
465 | } | ||
466 | |||
467 | /** | ||
468 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
469 | int max; | ||
470 | |||
471 | if (uri == NULL) | ||
472 | - return (NULL); | ||
473 | + return NULL; | ||
474 | |||
475 | max = 80; | ||
476 | ret = g_malloc(max + 1); | ||
477 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
478 | ret = temp; | ||
479 | } | ||
480 | ret[len] = 0; | ||
481 | - return (ret); | ||
482 | + return ret; | ||
483 | } | ||
484 | |||
485 | /** | ||
486 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
487 | char *cur, *out; | ||
488 | |||
489 | if (path == NULL) | ||
490 | - return (-1); | ||
491 | + return -1; | ||
492 | |||
493 | /* Skip all initial "/" chars. We want to get to the beginning of the | ||
494 | * first non-empty segment. | ||
495 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
496 | while (cur[0] == '/') | ||
497 | ++cur; | ||
498 | if (cur[0] == '\0') | ||
499 | - return (0); | ||
500 | + return 0; | ||
501 | |||
502 | /* Keep everything we've seen so far. */ | ||
503 | out = cur; | ||
504 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
505 | while (cur[0] == '/') | ||
506 | ++cur; | ||
507 | if (cur[0] == '\0') | ||
508 | - return (0); | ||
509 | + return 0; | ||
510 | |||
511 | /* | ||
512 | * Analyze each segment in sequence for cases (e) and (f). | ||
513 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
514 | } | ||
515 | } | ||
516 | |||
517 | - return (0); | ||
518 | + return 0; | ||
519 | } | ||
520 | |||
521 | static int is_hex(char c) | ||
522 | { | ||
523 | if (((c >= '0') && (c <= '9')) || ((c >= 'a') && (c <= 'f')) || | ||
524 | ((c >= 'A') && (c <= 'F'))) | ||
525 | - return (1); | ||
526 | - return (0); | ||
527 | + return 1; | ||
528 | + return 0; | ||
529 | } | ||
530 | |||
531 | /** | ||
532 | @@ -XXX,XX +XXX,XX @@ char *uri_string_unescape(const char *str, int len, char *target) | ||
533 | const char *in; | ||
534 | |||
535 | if (str == NULL) | ||
536 | - return (NULL); | ||
537 | + return NULL; | ||
538 | if (len <= 0) | ||
539 | len = strlen(str); | ||
540 | if (len < 0) | ||
541 | - return (NULL); | ||
542 | + return NULL; | ||
543 | |||
544 | if (target == NULL) { | ||
545 | ret = g_malloc(len + 1); | ||
546 | @@ -XXX,XX +XXX,XX @@ char *uri_string_unescape(const char *str, int len, char *target) | ||
547 | } | ||
548 | } | ||
549 | *out = 0; | ||
550 | - return (ret); | ||
551 | + return ret; | ||
552 | } | ||
553 | |||
554 | /** | ||
555 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
556 | int len, out; | ||
557 | |||
558 | if (str == NULL) | ||
559 | - return (NULL); | ||
560 | + return NULL; | ||
561 | if (str[0] == 0) | ||
562 | - return (g_strdup(str)); | ||
563 | + return g_strdup(str); | ||
564 | len = strlen(str); | ||
565 | if (!(len > 0)) | ||
566 | - return (NULL); | ||
567 | + return NULL; | ||
568 | |||
569 | len += 20; | ||
570 | ret = g_malloc(len); | ||
571 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
572 | } | ||
573 | } | ||
574 | ret[out] = 0; | ||
575 | - return (ret); | ||
576 | + return ret; | ||
577 | } | ||
578 | |||
579 | /************************************************************************ | ||
580 | @@ -XXX,XX +XXX,XX @@ done: | ||
581 | uri_free(bas); | ||
582 | if (res != NULL) | ||
583 | uri_free(res); | ||
584 | - return (val); | ||
585 | + return val; | ||
586 | } | ||
587 | |||
588 | /** | ||
86 | -- | 589 | -- |
87 | 2.13.5 | 590 | 2.14.3 |
88 | 591 | ||
89 | 592 | diff view generated by jsdifflib |
1 | From: Alberto Garcia <berto@igalia.com> | 1 | From: Su Hang <suhang16@mails.ucas.ac.cn> |
---|---|---|---|
2 | 2 | ||
3 | The throttling code can change internally the value of bkt->max if it | 3 | For this patch, using curly braces to wrap `if` `while` `else` statements, |
4 | hasn't been set by the user. The problem with this is that if we want | 4 | which only hold single statement. For example: |
5 | to retrieve the original value we have to undo this change first. This | 5 | ''' |
6 | is ugly and unnecessary: this patch removes the throttle_fix_bucket() | 6 | if (cond) |
7 | and throttle_unfix_bucket() functions completely and moves the logic | 7 | statement; |
8 | to throttle_compute_wait(). | 8 | ''' |
9 | to | ||
10 | ''' | ||
11 | if (cond) { | ||
12 | statement; | ||
13 | } | ||
14 | ''' | ||
9 | 15 | ||
10 | Signed-off-by: Alberto Garcia <berto@igalia.com> | 16 | And using tricks that compare the disassemblies before and after |
11 | Reviewed-by: Manos Pitsidianakis <el13635@mail.ntua.gr> | 17 | code changes, to make sure code logic isn't changed: |
12 | Message-id: 5b0b9e1ac6eb208d709eddc7b09e7669a523bff3.1503580370.git.berto@igalia.com | 18 | ''' |
19 | git checkout master | ||
20 | make util/uri.o | ||
21 | strip util/uri.o | ||
22 | objdump -Drx util/uri.o > /tmp/uri-master.txt | ||
23 | git checkout cleanupbranch | ||
24 | make util/uri.o | ||
25 | strip util/uri.o | ||
26 | objdump -Drx util/uri.o > /tmp/uri-cleanup.txt | ||
27 | |||
13 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 28 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
14 | --- | 29 | --- |
15 | util/throttle.c | 62 +++++++++++++++++++++------------------------------------ | 30 | util/uri.c | 463 +++++++++++++++++++++++++++++++++++++++---------------------- |
16 | 1 file changed, 23 insertions(+), 39 deletions(-) | 31 | 1 file changed, 294 insertions(+), 169 deletions(-) |
17 | 32 | ||
18 | diff --git a/util/throttle.c b/util/throttle.c | 33 | diff --git a/util/uri.c b/util/uri.c |
19 | index XXXXXXX..XXXXXXX 100644 | 34 | index XXXXXXX..XXXXXXX 100644 |
20 | --- a/util/throttle.c | 35 | --- a/util/uri.c |
21 | +++ b/util/throttle.c | 36 | +++ b/util/uri.c |
22 | @@ -XXX,XX +XXX,XX @@ static int64_t throttle_do_compute_wait(double limit, double extra) | 37 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_scheme(URI *uri, const char **str) |
23 | int64_t throttle_compute_wait(LeakyBucket *bkt) | ||
24 | { | 38 | { |
25 | double extra; /* the number of extra units blocking the io */ | 39 | const char *cur; |
26 | + double bucket_size; /* I/O before throttling to bkt->avg */ | 40 | |
27 | + double burst_bucket_size; /* Before throttling to bkt->max */ | 41 | - if (str == NULL) |
28 | 42 | + if (str == NULL) { | |
29 | if (!bkt->avg) { | 43 | return -1; |
44 | + } | ||
45 | |||
46 | cur = *str; | ||
47 | - if (!ISA_ALPHA(cur)) | ||
48 | + if (!ISA_ALPHA(cur)) { | ||
49 | return 2; | ||
50 | + } | ||
51 | cur++; | ||
52 | while (ISA_ALPHA(cur) || ISA_DIGIT(cur) || (*cur == '+') || (*cur == '-') || | ||
53 | - (*cur == '.')) | ||
54 | + (*cur == '.')) { | ||
55 | cur++; | ||
56 | + } | ||
57 | if (uri != NULL) { | ||
58 | g_free(uri->scheme); | ||
59 | uri->scheme = g_strndup(*str, cur - *str); | ||
60 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_fragment(URI *uri, const char **str) | ||
61 | { | ||
62 | const char *cur; | ||
63 | |||
64 | - if (str == NULL) | ||
65 | + if (str == NULL) { | ||
66 | return -1; | ||
67 | + } | ||
68 | |||
69 | cur = *str; | ||
70 | |||
71 | while ((ISA_PCHAR(cur)) || (*cur == '/') || (*cur == '?') || | ||
72 | (*cur == '[') || (*cur == ']') || | ||
73 | - ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) | ||
74 | + ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) { | ||
75 | NEXT(cur); | ||
76 | + } | ||
77 | if (uri != NULL) { | ||
78 | g_free(uri->fragment); | ||
79 | - if (uri->cleanup & 2) | ||
80 | + if (uri->cleanup & 2) { | ||
81 | uri->fragment = g_strndup(*str, cur - *str); | ||
82 | - else | ||
83 | + } else { | ||
84 | uri->fragment = uri_string_unescape(*str, cur - *str, NULL); | ||
85 | + } | ||
86 | } | ||
87 | *str = cur; | ||
88 | return 0; | ||
89 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_query(URI *uri, const char **str) | ||
90 | { | ||
91 | const char *cur; | ||
92 | |||
93 | - if (str == NULL) | ||
94 | + if (str == NULL) { | ||
95 | return -1; | ||
96 | + } | ||
97 | |||
98 | cur = *str; | ||
99 | |||
100 | while ((ISA_PCHAR(cur)) || (*cur == '/') || (*cur == '?') || | ||
101 | - ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) | ||
102 | + ((uri != NULL) && (uri->cleanup & 1) && (IS_UNWISE(cur)))) { | ||
103 | NEXT(cur); | ||
104 | + } | ||
105 | if (uri != NULL) { | ||
106 | g_free(uri->query); | ||
107 | uri->query = g_strndup(*str, cur - *str); | ||
108 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_user_info(URI *uri, const char **str) | ||
109 | |||
110 | cur = *str; | ||
111 | while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur) || | ||
112 | - (*cur == ':')) | ||
113 | + (*cur == ':')) { | ||
114 | NEXT(cur); | ||
115 | + } | ||
116 | if (*cur == '@') { | ||
117 | if (uri != NULL) { | ||
118 | g_free(uri->user); | ||
119 | - if (uri->cleanup & 2) | ||
120 | + if (uri->cleanup & 2) { | ||
121 | uri->user = g_strndup(*str, cur - *str); | ||
122 | - else | ||
123 | + } else { | ||
124 | uri->user = uri_string_unescape(*str, cur - *str, NULL); | ||
125 | + } | ||
126 | } | ||
127 | *str = cur; | ||
30 | return 0; | 128 | return 0; |
31 | } | 129 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_dec_octet(const char **str) |
32 | 130 | { | |
33 | - /* If the bucket is full then we have to wait */ | 131 | const char *cur = *str; |
34 | - extra = bkt->level - bkt->max * bkt->burst_length; | 132 | |
35 | + if (!bkt->max) { | 133 | - if (!(ISA_DIGIT(cur))) |
36 | + /* If bkt->max is 0 we still want to allow short bursts of I/O | 134 | + if (!(ISA_DIGIT(cur))) { |
37 | + * from the guest, otherwise every other request will be throttled | 135 | return 1; |
38 | + * and performance will suffer considerably. */ | 136 | - if (!ISA_DIGIT(cur + 1)) |
39 | + bucket_size = bkt->avg / 10; | 137 | + } |
40 | + burst_bucket_size = 0; | 138 | + if (!ISA_DIGIT(cur + 1)) { |
139 | cur++; | ||
140 | - else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) | ||
141 | + } else if ((*cur != '0') && (ISA_DIGIT(cur + 1)) && (!ISA_DIGIT(cur + 2))) { | ||
142 | cur += 2; | ||
143 | - else if ((*cur == '1') && (ISA_DIGIT(cur + 1)) && (ISA_DIGIT(cur + 2))) | ||
144 | + } else if ((*cur == '1') && (ISA_DIGIT(cur + 1)) && (ISA_DIGIT(cur + 2))) { | ||
145 | cur += 3; | ||
146 | - else if ((*cur == '2') && (*(cur + 1) >= '0') && (*(cur + 1) <= '4') && | ||
147 | - (ISA_DIGIT(cur + 2))) | ||
148 | + } else if ((*cur == '2') && (*(cur + 1) >= '0') && (*(cur + 1) <= '4') && | ||
149 | + (ISA_DIGIT(cur + 2))) { | ||
150 | cur += 3; | ||
151 | - else if ((*cur == '2') && (*(cur + 1) == '5') && (*(cur + 2) >= '0') && | ||
152 | - (*(cur + 1) <= '5')) | ||
153 | + } else if ((*cur == '2') && (*(cur + 1) == '5') && (*(cur + 2) >= '0') && | ||
154 | + (*(cur + 1) <= '5')) { | ||
155 | cur += 3; | ||
156 | - else | ||
41 | + } else { | 157 | + } else { |
42 | + /* If we have a burst limit then we have to wait until all I/O | 158 | return 1; |
43 | + * at burst rate has finished before throttling to bkt->avg */ | 159 | + } |
44 | + bucket_size = bkt->max * bkt->burst_length; | 160 | *str = cur; |
45 | + burst_bucket_size = bkt->max / 10; | 161 | return 0; |
46 | + } | ||
47 | + | ||
48 | + /* If the main bucket is full then we have to wait */ | ||
49 | + extra = bkt->level - bucket_size; | ||
50 | if (extra > 0) { | ||
51 | return throttle_do_compute_wait(bkt->avg, extra); | ||
52 | } | ||
53 | |||
54 | - /* If the bucket is not full yet we have to make sure that we | ||
55 | - * fulfill the goal of bkt->max units per second. */ | ||
56 | + /* If the main bucket is not full yet we still have to check the | ||
57 | + * burst bucket in order to enforce the burst limit */ | ||
58 | if (bkt->burst_length > 1) { | ||
59 | - /* We use 1/10 of the max value to smooth the throttling. | ||
60 | - * See throttle_fix_bucket() for more details. */ | ||
61 | - extra = bkt->burst_level - bkt->max / 10; | ||
62 | + extra = bkt->burst_level - burst_bucket_size; | ||
63 | if (extra > 0) { | ||
64 | return throttle_do_compute_wait(bkt->max, extra); | ||
65 | } | ||
66 | @@ -XXX,XX +XXX,XX @@ bool throttle_is_valid(ThrottleConfig *cfg, Error **errp) | ||
67 | return true; | ||
68 | } | 162 | } |
69 | 163 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) | |
70 | -/* fix bucket parameters */ | 164 | */ |
71 | -static void throttle_fix_bucket(LeakyBucket *bkt) | 165 | if (*cur == '[') { |
72 | -{ | 166 | cur++; |
73 | - double min; | 167 | - while ((*cur != ']') && (*cur != 0)) |
74 | - | 168 | + while ((*cur != ']') && (*cur != 0)) { |
75 | - /* zero bucket level */ | 169 | cur++; |
76 | - bkt->level = bkt->burst_level = 0; | 170 | - if (*cur != ']') |
77 | - | 171 | + } |
78 | - /* If bkt->max is 0 we still want to allow short bursts of I/O | 172 | + if (*cur != ']') { |
79 | - * from the guest, otherwise every other request will be throttled | 173 | return 1; |
80 | - * and performance will suffer considerably. */ | 174 | + } |
81 | - min = bkt->avg / 10; | 175 | cur++; |
82 | - if (bkt->avg && !bkt->max) { | 176 | goto found; |
83 | - bkt->max = min; | 177 | } |
84 | - } | 178 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) |
85 | -} | 179 | * try to parse an IPv4 |
86 | - | 180 | */ |
87 | -/* undo internal bucket parameter changes (see throttle_fix_bucket()) */ | 181 | if (ISA_DIGIT(cur)) { |
88 | -static void throttle_unfix_bucket(LeakyBucket *bkt) | 182 | - if (rfc3986_parse_dec_octet(&cur) != 0) |
89 | -{ | 183 | + if (rfc3986_parse_dec_octet(&cur) != 0) { |
90 | - if (bkt->max < bkt->avg) { | 184 | goto not_ipv4; |
91 | - bkt->max = 0; | 185 | - if (*cur != '.') |
92 | - } | 186 | + } |
93 | -} | 187 | + if (*cur != '.') { |
94 | - | 188 | goto not_ipv4; |
95 | /* Used to configure the throttle | 189 | + } |
96 | * | 190 | cur++; |
97 | * @ts: the throttle state we are working on | 191 | - if (rfc3986_parse_dec_octet(&cur) != 0) |
98 | @@ -XXX,XX +XXX,XX @@ void throttle_config(ThrottleState *ts, | 192 | + if (rfc3986_parse_dec_octet(&cur) != 0) { |
99 | 193 | goto not_ipv4; | |
100 | ts->cfg = *cfg; | 194 | - if (*cur != '.') |
101 | 195 | + } | |
102 | + /* Zero bucket level */ | 196 | + if (*cur != '.') { |
103 | for (i = 0; i < BUCKETS_COUNT; i++) { | 197 | goto not_ipv4; |
104 | - throttle_fix_bucket(&ts->cfg.buckets[i]); | 198 | - if (rfc3986_parse_dec_octet(&cur) != 0) |
105 | + ts->cfg.buckets[i].level = 0; | 199 | + } |
106 | + ts->cfg.buckets[i].burst_level = 0; | 200 | + if (rfc3986_parse_dec_octet(&cur) != 0) { |
107 | } | 201 | goto not_ipv4; |
108 | 202 | - if (*cur != '.') | |
109 | ts->previous_leak = qemu_clock_get_ns(clock_type); | 203 | + } |
110 | @@ -XXX,XX +XXX,XX @@ void throttle_config(ThrottleState *ts, | 204 | + if (*cur != '.') { |
205 | goto not_ipv4; | ||
206 | - if (rfc3986_parse_dec_octet(&cur) != 0) | ||
207 | + } | ||
208 | + if (rfc3986_parse_dec_octet(&cur) != 0) { | ||
209 | goto not_ipv4; | ||
210 | + } | ||
211 | goto found; | ||
212 | not_ipv4: | ||
213 | cur = *str; | ||
214 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_host(URI *uri, const char **str) | ||
215 | /* | ||
216 | * then this should be a hostname which can be empty | ||
217 | */ | ||
218 | - while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur)) | ||
219 | + while (ISA_UNRESERVED(cur) || ISA_PCT_ENCODED(cur) || ISA_SUB_DELIM(cur)) { | ||
220 | NEXT(cur); | ||
221 | + } | ||
222 | found: | ||
223 | if (uri != NULL) { | ||
224 | g_free(uri->authority); | ||
225 | uri->authority = NULL; | ||
226 | g_free(uri->server); | ||
227 | if (cur != host) { | ||
228 | - if (uri->cleanup & 2) | ||
229 | + if (uri->cleanup & 2) { | ||
230 | uri->server = g_strndup(host, cur - host); | ||
231 | - else | ||
232 | + } else { | ||
233 | uri->server = uri_string_unescape(host, cur - host, NULL); | ||
234 | - } else | ||
235 | + } | ||
236 | + } else { | ||
237 | uri->server = NULL; | ||
238 | + } | ||
239 | } | ||
240 | *str = cur; | ||
241 | return 0; | ||
242 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_authority(URI *uri, const char **str) | ||
243 | * try to parse a userinfo and check for the trailing @ | ||
244 | */ | ||
245 | ret = rfc3986_parse_user_info(uri, &cur); | ||
246 | - if ((ret != 0) || (*cur != '@')) | ||
247 | + if ((ret != 0) || (*cur != '@')) { | ||
248 | cur = *str; | ||
249 | - else | ||
250 | + } else { | ||
251 | cur++; | ||
252 | + } | ||
253 | ret = rfc3986_parse_host(uri, &cur); | ||
254 | - if (ret != 0) | ||
255 | + if (ret != 0) { | ||
256 | return ret; | ||
257 | + } | ||
258 | if (*cur == ':') { | ||
259 | cur++; | ||
260 | ret = rfc3986_parse_port(uri, &cur); | ||
261 | - if (ret != 0) | ||
262 | + if (ret != 0) { | ||
263 | return ret; | ||
264 | + } | ||
265 | } | ||
266 | *str = cur; | ||
267 | return 0; | ||
268 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_segment(const char **str, char forbid, int empty) | ||
269 | |||
270 | cur = *str; | ||
271 | if (!ISA_PCHAR(cur)) { | ||
272 | - if (empty) | ||
273 | + if (empty) { | ||
274 | return 0; | ||
275 | + } | ||
276 | return 1; | ||
277 | } | ||
278 | - while (ISA_PCHAR(cur) && (*cur != forbid)) | ||
279 | + while (ISA_PCHAR(cur) && (*cur != forbid)) { | ||
280 | NEXT(cur); | ||
281 | + } | ||
282 | *str = cur; | ||
283 | return 0; | ||
284 | } | ||
285 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_ab_empty(URI *uri, const char **str) | ||
286 | while (*cur == '/') { | ||
287 | cur++; | ||
288 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
289 | - if (ret != 0) | ||
290 | + if (ret != 0) { | ||
291 | return ret; | ||
292 | + } | ||
293 | } | ||
294 | if (uri != NULL) { | ||
295 | g_free(uri->path); | ||
296 | if (*str != cur) { | ||
297 | - if (uri->cleanup & 2) | ||
298 | + if (uri->cleanup & 2) { | ||
299 | uri->path = g_strndup(*str, cur - *str); | ||
300 | - else | ||
301 | + } else { | ||
302 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
303 | + } | ||
304 | } else { | ||
305 | uri->path = NULL; | ||
306 | } | ||
307 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_absolute(URI *uri, const char **str) | ||
308 | |||
309 | cur = *str; | ||
310 | |||
311 | - if (*cur != '/') | ||
312 | + if (*cur != '/') { | ||
313 | return 1; | ||
314 | + } | ||
315 | cur++; | ||
316 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
317 | if (ret == 0) { | ||
318 | while (*cur == '/') { | ||
319 | cur++; | ||
320 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
321 | - if (ret != 0) | ||
322 | + if (ret != 0) { | ||
323 | return ret; | ||
324 | + } | ||
325 | } | ||
326 | } | ||
327 | if (uri != NULL) { | ||
328 | g_free(uri->path); | ||
329 | if (cur != *str) { | ||
330 | - if (uri->cleanup & 2) | ||
331 | + if (uri->cleanup & 2) { | ||
332 | uri->path = g_strndup(*str, cur - *str); | ||
333 | - else | ||
334 | + } else { | ||
335 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
336 | + } | ||
337 | } else { | ||
338 | uri->path = NULL; | ||
339 | } | ||
340 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_rootless(URI *uri, const char **str) | ||
341 | cur = *str; | ||
342 | |||
343 | ret = rfc3986_parse_segment(&cur, 0, 0); | ||
344 | - if (ret != 0) | ||
345 | + if (ret != 0) { | ||
346 | return ret; | ||
347 | + } | ||
348 | while (*cur == '/') { | ||
349 | cur++; | ||
350 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
351 | - if (ret != 0) | ||
352 | + if (ret != 0) { | ||
353 | return ret; | ||
354 | + } | ||
355 | } | ||
356 | if (uri != NULL) { | ||
357 | g_free(uri->path); | ||
358 | if (cur != *str) { | ||
359 | - if (uri->cleanup & 2) | ||
360 | + if (uri->cleanup & 2) { | ||
361 | uri->path = g_strndup(*str, cur - *str); | ||
362 | - else | ||
363 | + } else { | ||
364 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
365 | + } | ||
366 | } else { | ||
367 | uri->path = NULL; | ||
368 | } | ||
369 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_path_no_scheme(URI *uri, const char **str) | ||
370 | cur = *str; | ||
371 | |||
372 | ret = rfc3986_parse_segment(&cur, ':', 0); | ||
373 | - if (ret != 0) | ||
374 | + if (ret != 0) { | ||
375 | return ret; | ||
376 | + } | ||
377 | while (*cur == '/') { | ||
378 | cur++; | ||
379 | ret = rfc3986_parse_segment(&cur, 0, 1); | ||
380 | - if (ret != 0) | ||
381 | + if (ret != 0) { | ||
382 | return ret; | ||
383 | + } | ||
384 | } | ||
385 | if (uri != NULL) { | ||
386 | g_free(uri->path); | ||
387 | if (cur != *str) { | ||
388 | - if (uri->cleanup & 2) | ||
389 | + if (uri->cleanup & 2) { | ||
390 | uri->path = g_strndup(*str, cur - *str); | ||
391 | - else | ||
392 | + } else { | ||
393 | uri->path = uri_string_unescape(*str, cur - *str, NULL); | ||
394 | + } | ||
395 | } else { | ||
396 | uri->path = NULL; | ||
397 | } | ||
398 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_hier_part(URI *uri, const char **str) | ||
399 | if ((*cur == '/') && (*(cur + 1) == '/')) { | ||
400 | cur += 2; | ||
401 | ret = rfc3986_parse_authority(uri, &cur); | ||
402 | - if (ret != 0) | ||
403 | + if (ret != 0) { | ||
404 | return ret; | ||
405 | + } | ||
406 | ret = rfc3986_parse_path_ab_empty(uri, &cur); | ||
407 | - if (ret != 0) | ||
408 | + if (ret != 0) { | ||
409 | return ret; | ||
410 | + } | ||
411 | *str = cur; | ||
412 | return 0; | ||
413 | } else if (*cur == '/') { | ||
414 | ret = rfc3986_parse_path_absolute(uri, &cur); | ||
415 | - if (ret != 0) | ||
416 | + if (ret != 0) { | ||
417 | return ret; | ||
418 | + } | ||
419 | } else if (ISA_PCHAR(cur)) { | ||
420 | ret = rfc3986_parse_path_rootless(uri, &cur); | ||
421 | - if (ret != 0) | ||
422 | + if (ret != 0) { | ||
423 | return ret; | ||
424 | + } | ||
425 | } else { | ||
426 | /* path-empty is effectively empty */ | ||
427 | if (uri != NULL) { | ||
428 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
429 | if ((*str == '/') && (*(str + 1) == '/')) { | ||
430 | str += 2; | ||
431 | ret = rfc3986_parse_authority(uri, &str); | ||
432 | - if (ret != 0) | ||
433 | + if (ret != 0) { | ||
434 | return ret; | ||
435 | + } | ||
436 | ret = rfc3986_parse_path_ab_empty(uri, &str); | ||
437 | - if (ret != 0) | ||
438 | + if (ret != 0) { | ||
439 | return ret; | ||
440 | + } | ||
441 | } else if (*str == '/') { | ||
442 | ret = rfc3986_parse_path_absolute(uri, &str); | ||
443 | - if (ret != 0) | ||
444 | + if (ret != 0) { | ||
445 | return ret; | ||
446 | + } | ||
447 | } else if (ISA_PCHAR(str)) { | ||
448 | ret = rfc3986_parse_path_no_scheme(uri, &str); | ||
449 | - if (ret != 0) | ||
450 | + if (ret != 0) { | ||
451 | return ret; | ||
452 | + } | ||
453 | } else { | ||
454 | /* path-empty is effectively empty */ | ||
455 | if (uri != NULL) { | ||
456 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_relative_ref(URI *uri, const char *str) | ||
457 | if (*str == '?') { | ||
458 | str++; | ||
459 | ret = rfc3986_parse_query(uri, &str); | ||
460 | - if (ret != 0) | ||
461 | + if (ret != 0) { | ||
462 | return ret; | ||
463 | + } | ||
464 | } | ||
465 | if (*str == '#') { | ||
466 | str++; | ||
467 | ret = rfc3986_parse_fragment(uri, &str); | ||
468 | - if (ret != 0) | ||
469 | + if (ret != 0) { | ||
470 | return ret; | ||
471 | + } | ||
472 | } | ||
473 | if (*str != 0) { | ||
474 | uri_clean(uri); | ||
475 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse(URI *uri, const char *str) | ||
476 | int ret; | ||
477 | |||
478 | ret = rfc3986_parse_scheme(uri, &str); | ||
479 | - if (ret != 0) | ||
480 | + if (ret != 0) { | ||
481 | return ret; | ||
482 | + } | ||
483 | if (*str != ':') { | ||
484 | return 1; | ||
485 | } | ||
486 | str++; | ||
487 | ret = rfc3986_parse_hier_part(uri, &str); | ||
488 | - if (ret != 0) | ||
489 | + if (ret != 0) { | ||
490 | return ret; | ||
491 | + } | ||
492 | if (*str == '?') { | ||
493 | str++; | ||
494 | ret = rfc3986_parse_query(uri, &str); | ||
495 | - if (ret != 0) | ||
496 | + if (ret != 0) { | ||
497 | return ret; | ||
498 | + } | ||
499 | } | ||
500 | if (*str == '#') { | ||
501 | str++; | ||
502 | ret = rfc3986_parse_fragment(uri, &str); | ||
503 | - if (ret != 0) | ||
504 | + if (ret != 0) { | ||
505 | return ret; | ||
506 | + } | ||
507 | } | ||
508 | if (*str != 0) { | ||
509 | uri_clean(uri); | ||
510 | @@ -XXX,XX +XXX,XX @@ static int rfc3986_parse_uri_reference(URI *uri, const char *str) | ||
511 | { | ||
512 | int ret; | ||
513 | |||
514 | - if (str == NULL) | ||
515 | + if (str == NULL) { | ||
516 | return -1; | ||
517 | + } | ||
518 | uri_clean(uri); | ||
519 | |||
520 | /* | ||
521 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse(const char *str) | ||
522 | URI *uri; | ||
523 | int ret; | ||
524 | |||
525 | - if (str == NULL) | ||
526 | + if (str == NULL) { | ||
527 | return NULL; | ||
528 | + } | ||
529 | uri = uri_new(); | ||
530 | ret = rfc3986_parse_uri_reference(uri, str); | ||
531 | if (ret) { | ||
532 | @@ -XXX,XX +XXX,XX @@ URI *uri_parse_raw(const char *str, int raw) | ||
533 | URI *uri; | ||
534 | int ret; | ||
535 | |||
536 | - if (str == NULL) | ||
537 | + if (str == NULL) { | ||
538 | return NULL; | ||
539 | + } | ||
540 | uri = uri_new(); | ||
541 | if (raw) { | ||
542 | uri->cleanup |= 2; | ||
543 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
544 | int len; | ||
545 | int max; | ||
546 | |||
547 | - if (uri == NULL) | ||
548 | + if (uri == NULL) { | ||
549 | return NULL; | ||
550 | + } | ||
551 | |||
552 | max = 80; | ||
553 | ret = g_malloc(max + 1); | ||
554 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
555 | temp = realloc2n(ret, &max); | ||
556 | ret = temp; | ||
557 | } | ||
558 | - if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) | ||
559 | + if (IS_RESERVED(*(p)) || IS_UNRESERVED(*(p))) { | ||
560 | ret[len++] = *p++; | ||
561 | - else { | ||
562 | + } else { | ||
563 | int val = *(unsigned char *)p++; | ||
564 | int hi = val / 0x10, lo = val % 0x10; | ||
565 | ret[len++] = '%'; | ||
566 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
567 | } | ||
568 | if ((IS_UNRESERVED(*(p))) || ((*(p) == ';')) || | ||
569 | ((*(p) == ':')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
570 | - ((*(p) == '+')) || ((*(p) == '$')) || ((*(p) == ','))) | ||
571 | + ((*(p) == '+')) || ((*(p) == '$')) || ((*(p) == ','))) { | ||
572 | ret[len++] = *p++; | ||
573 | - else { | ||
574 | + } else { | ||
575 | int val = *(unsigned char *)p++; | ||
576 | int hi = val / 0x10, lo = val % 0x10; | ||
577 | ret[len++] = '%'; | ||
578 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
579 | if ((IS_UNRESERVED(*(p))) || ((*(p) == '$')) || | ||
580 | ((*(p) == ',')) || ((*(p) == ';')) || ((*(p) == ':')) || | ||
581 | ((*(p) == '@')) || ((*(p) == '&')) || ((*(p) == '=')) || | ||
582 | - ((*(p) == '+'))) | ||
583 | + ((*(p) == '+'))) { | ||
584 | ret[len++] = *p++; | ||
585 | - else { | ||
586 | + } else { | ||
587 | int val = *(unsigned char *)p++; | ||
588 | int hi = val / 0x10, lo = val % 0x10; | ||
589 | ret[len++] = '%'; | ||
590 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
591 | if ((IS_UNRESERVED(*(p))) || ((*(p) == '/')) || | ||
592 | ((*(p) == ';')) || ((*(p) == '@')) || ((*(p) == '&')) || | ||
593 | ((*(p) == '=')) || ((*(p) == '+')) || ((*(p) == '$')) || | ||
594 | - ((*(p) == ','))) | ||
595 | + ((*(p) == ','))) { | ||
596 | ret[len++] = *p++; | ||
597 | - else { | ||
598 | + } else { | ||
599 | int val = *(unsigned char *)p++; | ||
600 | int hi = val / 0x10, lo = val % 0x10; | ||
601 | ret[len++] = '%'; | ||
602 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
603 | temp = realloc2n(ret, &max); | ||
604 | ret = temp; | ||
605 | } | ||
606 | - if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) | ||
607 | + if ((IS_UNRESERVED(*(p))) || (IS_RESERVED(*(p)))) { | ||
608 | ret[len++] = *p++; | ||
609 | - else { | ||
610 | + } else { | ||
611 | int val = *(unsigned char *)p++; | ||
612 | int hi = val / 0x10, lo = val % 0x10; | ||
613 | ret[len++] = '%'; | ||
614 | @@ -XXX,XX +XXX,XX @@ char *uri_to_string(URI *uri) | ||
111 | */ | 615 | */ |
112 | void throttle_get_config(ThrottleState *ts, ThrottleConfig *cfg) | 616 | static void uri_clean(URI *uri) |
113 | { | 617 | { |
114 | - int i; | 618 | - if (uri == NULL) |
115 | - | 619 | + if (uri == NULL) { |
116 | *cfg = ts->cfg; | 620 | return; |
117 | - | 621 | + } |
118 | - for (i = 0; i < BUCKETS_COUNT; i++) { | 622 | |
119 | - throttle_unfix_bucket(&cfg->buckets[i]); | 623 | g_free(uri->scheme); |
120 | - } | 624 | uri->scheme = NULL; |
625 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
626 | { | ||
627 | char *cur, *out; | ||
628 | |||
629 | - if (path == NULL) | ||
630 | + if (path == NULL) { | ||
631 | return -1; | ||
632 | + } | ||
633 | |||
634 | /* Skip all initial "/" chars. We want to get to the beginning of the | ||
635 | * first non-empty segment. | ||
636 | */ | ||
637 | cur = path; | ||
638 | - while (cur[0] == '/') | ||
639 | + while (cur[0] == '/') { | ||
640 | ++cur; | ||
641 | - if (cur[0] == '\0') | ||
642 | + } | ||
643 | + if (cur[0] == '\0') { | ||
644 | return 0; | ||
645 | + } | ||
646 | |||
647 | /* Keep everything we've seen so far. */ | ||
648 | out = cur; | ||
649 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
650 | if ((cur[0] == '.') && (cur[1] == '/')) { | ||
651 | cur += 2; | ||
652 | /* '//' normalization should be done at this point too */ | ||
653 | - while (cur[0] == '/') | ||
654 | + while (cur[0] == '/') { | ||
655 | cur++; | ||
656 | + } | ||
657 | continue; | ||
658 | } | ||
659 | |||
660 | @@ -XXX,XX +XXX,XX @@ static int normalize_uri_path(char *path) | ||
661 | * d) If the buffer string ends with "." as a complete path segment, | ||
662 | * that "." is removed. | ||
663 | */ | ||
664 | - if ((cur[0] == '.') && (cur[1] == '\0')) | ||
665 | + if ((cur[0] == '.') && (cur[1] == '\0')) { | ||
666 | break; | ||
667 | + } | ||
668 | |||
669 | /* Otherwise keep the segment. */ | ||
670 | while (cur[0] != '/') { | ||
671 | - if (cur[0] == '\0') | ||
672 | + if (cur[0] == '\0') { | ||
673 | goto done_cd; | ||
674 | + } | ||
675 | (out++)[0] = (cur++)[0]; | ||
676 | } | ||
677 | /* nomalize // */ | ||
678 | - while ((cur[0] == '/') && (cur[1] == '/')) | ||
679 | + while ((cur[0] == '/') && (cur[1] == '/')) { | ||
680 | cur++; | ||
681 | + } | ||
682 | |||
683 | (out++)[0] = (cur++)[0]; | ||
684 | } | ||
685 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
686 | |||
687 | /* Reset to the beginning of the first segment for the next sequence. */ | ||
688 | cur = path; | ||
689 | - while (cur[0] == '/') | ||
690 | + while (cur[0] == '/') { | ||
691 | ++cur; | ||
692 | - if (cur[0] == '\0') | ||
693 | + } | ||
694 | + if (cur[0] == '\0') { | ||
695 | return 0; | ||
696 | + } | ||
697 | |||
698 | /* | ||
699 | * Analyze each segment in sequence for cases (e) and (f). | ||
700 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
701 | |||
702 | /* Find the end of the current segment. */ | ||
703 | segp = cur; | ||
704 | - while ((segp[0] != '/') && (segp[0] != '\0')) | ||
705 | + while ((segp[0] != '/') && (segp[0] != '\0')) { | ||
706 | ++segp; | ||
707 | + } | ||
708 | |||
709 | /* If this is the last segment, we're done (we need at least two | ||
710 | * segments to meet the criteria for the (e) and (f) cases). | ||
711 | */ | ||
712 | - if (segp[0] == '\0') | ||
713 | + if (segp[0] == '\0') { | ||
714 | break; | ||
715 | + } | ||
716 | |||
717 | /* If the first segment is "..", or if the next segment _isn't_ "..", | ||
718 | * keep this segment and try the next one. | ||
719 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
720 | /* string will overlap, do not use strcpy */ | ||
721 | tmp = cur; | ||
722 | segp += 3; | ||
723 | - while ((*tmp++ = *segp++) != 0) | ||
724 | - ; | ||
725 | + while ((*tmp++ = *segp++) != 0) { | ||
726 | + /* No further work */ | ||
727 | + } | ||
728 | |||
729 | /* If there are no previous segments, then keep going from here. */ | ||
730 | segp = cur; | ||
731 | - while ((segp > path) && ((--segp)[0] == '/')) | ||
732 | - ; | ||
733 | - if (segp == path) | ||
734 | + while ((segp > path) && ((--segp)[0] == '/')) { | ||
735 | + /* No further work */ | ||
736 | + } | ||
737 | + if (segp == path) { | ||
738 | continue; | ||
739 | + } | ||
740 | |||
741 | /* "segp" is pointing to the end of a previous segment; find it's | ||
742 | * start. We need to back up to the previous segment and start | ||
743 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
744 | * remove the "foo/..". | ||
745 | */ | ||
746 | cur = segp; | ||
747 | - while ((cur > path) && (cur[-1] != '/')) | ||
748 | + while ((cur > path) && (cur[-1] != '/')) { | ||
749 | --cur; | ||
750 | + } | ||
751 | } | ||
752 | out[0] = '\0'; | ||
753 | |||
754 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
755 | if (path[0] == '/') { | ||
756 | cur = path; | ||
757 | while ((cur[0] == '/') && (cur[1] == '.') && (cur[2] == '.') && | ||
758 | - ((cur[3] == '/') || (cur[3] == '\0'))) | ||
759 | + ((cur[3] == '/') || (cur[3] == '\0'))) { | ||
760 | cur += 3; | ||
761 | + } | ||
762 | |||
763 | if (cur != path) { | ||
764 | out = path; | ||
765 | - while (cur[0] != '\0') | ||
766 | + while (cur[0] != '\0') { | ||
767 | (out++)[0] = (cur++)[0]; | ||
768 | + } | ||
769 | out[0] = 0; | ||
770 | } | ||
771 | } | ||
772 | @@ -XXX,XX +XXX,XX @@ done_cd: | ||
773 | static int is_hex(char c) | ||
774 | { | ||
775 | if (((c >= '0') && (c <= '9')) || ((c >= 'a') && (c <= 'f')) || | ||
776 | - ((c >= 'A') && (c <= 'F'))) | ||
777 | + ((c >= 'A') && (c <= 'F'))) { | ||
778 | return 1; | ||
779 | + } | ||
780 | return 0; | ||
121 | } | 781 | } |
122 | 782 | ||
123 | 783 | @@ -XXX,XX +XXX,XX @@ char *uri_string_unescape(const char *str, int len, char *target) | |
784 | char *ret, *out; | ||
785 | const char *in; | ||
786 | |||
787 | - if (str == NULL) | ||
788 | + if (str == NULL) { | ||
789 | return NULL; | ||
790 | - if (len <= 0) | ||
791 | + } | ||
792 | + if (len <= 0) { | ||
793 | len = strlen(str); | ||
794 | - if (len < 0) | ||
795 | + } | ||
796 | + if (len < 0) { | ||
797 | return NULL; | ||
798 | + } | ||
799 | |||
800 | if (target == NULL) { | ||
801 | ret = g_malloc(len + 1); | ||
802 | - } else | ||
803 | + } else { | ||
804 | ret = target; | ||
805 | + } | ||
806 | in = str; | ||
807 | out = ret; | ||
808 | while (len > 0) { | ||
809 | if ((len > 2) && (*in == '%') && (is_hex(in[1])) && (is_hex(in[2]))) { | ||
810 | in++; | ||
811 | - if ((*in >= '0') && (*in <= '9')) | ||
812 | + if ((*in >= '0') && (*in <= '9')) { | ||
813 | *out = (*in - '0'); | ||
814 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
815 | + } else if ((*in >= 'a') && (*in <= 'f')) { | ||
816 | *out = (*in - 'a') + 10; | ||
817 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
818 | + } else if ((*in >= 'A') && (*in <= 'F')) { | ||
819 | *out = (*in - 'A') + 10; | ||
820 | + } | ||
821 | in++; | ||
822 | - if ((*in >= '0') && (*in <= '9')) | ||
823 | + if ((*in >= '0') && (*in <= '9')) { | ||
824 | *out = *out * 16 + (*in - '0'); | ||
825 | - else if ((*in >= 'a') && (*in <= 'f')) | ||
826 | + } else if ((*in >= 'a') && (*in <= 'f')) { | ||
827 | *out = *out * 16 + (*in - 'a') + 10; | ||
828 | - else if ((*in >= 'A') && (*in <= 'F')) | ||
829 | + } else if ((*in >= 'A') && (*in <= 'F')) { | ||
830 | *out = *out * 16 + (*in - 'A') + 10; | ||
831 | + } | ||
832 | in++; | ||
833 | len -= 3; | ||
834 | out++; | ||
835 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
836 | const char *in; | ||
837 | int len, out; | ||
838 | |||
839 | - if (str == NULL) | ||
840 | + if (str == NULL) { | ||
841 | return NULL; | ||
842 | - if (str[0] == 0) | ||
843 | + } | ||
844 | + if (str[0] == 0) { | ||
845 | return g_strdup(str); | ||
846 | + } | ||
847 | len = strlen(str); | ||
848 | - if (!(len > 0)) | ||
849 | + if (!(len > 0)) { | ||
850 | return NULL; | ||
851 | + } | ||
852 | |||
853 | len += 20; | ||
854 | ret = g_malloc(len); | ||
855 | @@ -XXX,XX +XXX,XX @@ char *uri_string_escape(const char *str, const char *list) | ||
856 | unsigned char val; | ||
857 | ret[out++] = '%'; | ||
858 | val = ch >> 4; | ||
859 | - if (val <= 9) | ||
860 | + if (val <= 9) { | ||
861 | ret[out++] = '0' + val; | ||
862 | - else | ||
863 | + } else { | ||
864 | ret[out++] = 'A' + val - 0xA; | ||
865 | + } | ||
866 | val = ch & 0xF; | ||
867 | - if (val <= 9) | ||
868 | + if (val <= 9) { | ||
869 | ret[out++] = '0' + val; | ||
870 | - else | ||
871 | + } else { | ||
872 | ret[out++] = 'A' + val - 0xA; | ||
873 | + } | ||
874 | in++; | ||
875 | } else { | ||
876 | ret[out++] = *in++; | ||
877 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
878 | * as a reference to "." rather than as a synonym for the current | ||
879 | * URI. Should we do that here? | ||
880 | */ | ||
881 | - if (uri == NULL) | ||
882 | + if (uri == NULL) { | ||
883 | ret = -1; | ||
884 | - else { | ||
885 | + } else { | ||
886 | if (*uri) { | ||
887 | ref = uri_new(); | ||
888 | ret = uri_parse_into(ref, uri); | ||
889 | - } else | ||
890 | + } else { | ||
891 | ret = 0; | ||
892 | + } | ||
893 | } | ||
894 | - if (ret != 0) | ||
895 | + if (ret != 0) { | ||
896 | goto done; | ||
897 | + } | ||
898 | if ((ref != NULL) && (ref->scheme != NULL)) { | ||
899 | /* | ||
900 | * The URI is absolute don't modify. | ||
901 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
902 | val = g_strdup(uri); | ||
903 | goto done; | ||
904 | } | ||
905 | - if (base == NULL) | ||
906 | + if (base == NULL) { | ||
907 | ret = -1; | ||
908 | - else { | ||
909 | + } else { | ||
910 | bas = uri_new(); | ||
911 | ret = uri_parse_into(bas, base); | ||
912 | } | ||
913 | if (ret != 0) { | ||
914 | - if (ref) | ||
915 | + if (ref) { | ||
916 | val = uri_to_string(ref); | ||
917 | + } | ||
918 | goto done; | ||
919 | } | ||
920 | if (ref == NULL) { | ||
921 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
922 | if ((ref->scheme == NULL) && (ref->path == NULL) && | ||
923 | ((ref->authority == NULL) && (ref->server == NULL))) { | ||
924 | res->scheme = g_strdup(bas->scheme); | ||
925 | - if (bas->authority != NULL) | ||
926 | + if (bas->authority != NULL) { | ||
927 | res->authority = g_strdup(bas->authority); | ||
928 | - else if (bas->server != NULL) { | ||
929 | + } else if (bas->server != NULL) { | ||
930 | res->server = g_strdup(bas->server); | ||
931 | res->user = g_strdup(bas->user); | ||
932 | res->port = bas->port; | ||
933 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
934 | * use an authority component. | ||
935 | */ | ||
936 | if ((ref->authority != NULL) || (ref->server != NULL)) { | ||
937 | - if (ref->authority != NULL) | ||
938 | + if (ref->authority != NULL) { | ||
939 | res->authority = g_strdup(ref->authority); | ||
940 | - else { | ||
941 | + } else { | ||
942 | res->server = g_strdup(ref->server); | ||
943 | res->user = g_strdup(ref->user); | ||
944 | res->port = ref->port; | ||
945 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
946 | res->path = g_strdup(ref->path); | ||
947 | goto step_7; | ||
948 | } | ||
949 | - if (bas->authority != NULL) | ||
950 | + if (bas->authority != NULL) { | ||
951 | res->authority = g_strdup(bas->authority); | ||
952 | - else if (bas->server != NULL) { | ||
953 | + } else if (bas->server != NULL) { | ||
954 | res->server = g_strdup(bas->server); | ||
955 | res->user = g_strdup(bas->user); | ||
956 | res->port = bas->port; | ||
957 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
958 | * Allocate a buffer large enough for the result string. | ||
959 | */ | ||
960 | len = 2; /* extra / and 0 */ | ||
961 | - if (ref->path != NULL) | ||
962 | + if (ref->path != NULL) { | ||
963 | len += strlen(ref->path); | ||
964 | - if (bas->path != NULL) | ||
965 | + } | ||
966 | + if (bas->path != NULL) { | ||
967 | len += strlen(bas->path); | ||
968 | + } | ||
969 | res->path = g_malloc(len); | ||
970 | res->path[0] = 0; | ||
971 | |||
972 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
973 | out = 0; | ||
974 | if (bas->path != NULL) { | ||
975 | while (bas->path[cur] != 0) { | ||
976 | - while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) | ||
977 | + while ((bas->path[cur] != 0) && (bas->path[cur] != '/')) { | ||
978 | cur++; | ||
979 | - if (bas->path[cur] == 0) | ||
980 | + } | ||
981 | + if (bas->path[cur] == 0) { | ||
982 | break; | ||
983 | + } | ||
984 | |||
985 | cur++; | ||
986 | while (out < cur) { | ||
987 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve(const char *uri, const char *base) | ||
988 | /* | ||
989 | * Ensure the path includes a '/' | ||
990 | */ | ||
991 | - if ((out == 0) && (bas->server != NULL)) | ||
992 | + if ((out == 0) && (bas->server != NULL)) { | ||
993 | res->path[out++] = '/'; | ||
994 | + } | ||
995 | while (ref->path[indx] != 0) { | ||
996 | res->path[out++] = ref->path[indx++]; | ||
997 | } | ||
998 | @@ -XXX,XX +XXX,XX @@ step_7: | ||
999 | val = uri_to_string(res); | ||
1000 | |||
1001 | done: | ||
1002 | - if (ref != NULL) | ||
1003 | + if (ref != NULL) { | ||
1004 | uri_free(ref); | ||
1005 | - if (bas != NULL) | ||
1006 | + } | ||
1007 | + if (bas != NULL) { | ||
1008 | uri_free(bas); | ||
1009 | - if (res != NULL) | ||
1010 | + } | ||
1011 | + if (res != NULL) { | ||
1012 | uri_free(res); | ||
1013 | + } | ||
1014 | return val; | ||
1015 | } | ||
1016 | |||
1017 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1018 | char *bptr, *uptr, *vptr; | ||
1019 | int remove_path = 0; | ||
1020 | |||
1021 | - if ((uri == NULL) || (*uri == 0)) | ||
1022 | + if ((uri == NULL) || (*uri == 0)) { | ||
1023 | return NULL; | ||
1024 | + } | ||
1025 | |||
1026 | /* | ||
1027 | * First parse URI into a standard form | ||
1028 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1029 | /* If URI not already in "relative" form */ | ||
1030 | if (uri[0] != '.') { | ||
1031 | ret = uri_parse_into(ref, uri); | ||
1032 | - if (ret != 0) | ||
1033 | + if (ret != 0) { | ||
1034 | goto done; /* Error in URI, return NULL */ | ||
1035 | - } else | ||
1036 | + } | ||
1037 | + } else { | ||
1038 | ref->path = g_strdup(uri); | ||
1039 | + } | ||
1040 | |||
1041 | /* | ||
1042 | * Next parse base into the same standard form | ||
1043 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1044 | bas = uri_new(); | ||
1045 | if (base[0] != '.') { | ||
1046 | ret = uri_parse_into(bas, base); | ||
1047 | - if (ret != 0) | ||
1048 | + if (ret != 0) { | ||
1049 | goto done; /* Error in base, return NULL */ | ||
1050 | - } else | ||
1051 | + } | ||
1052 | + } else { | ||
1053 | bas->path = g_strdup(base); | ||
1054 | + } | ||
1055 | |||
1056 | /* | ||
1057 | * If the scheme / server on the URI differs from the base, | ||
1058 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1059 | if (bas->path == NULL) { | ||
1060 | if (ref->path != NULL) { | ||
1061 | uptr = ref->path; | ||
1062 | - if (*uptr == '/') | ||
1063 | + if (*uptr == '/') { | ||
1064 | uptr++; | ||
1065 | + } | ||
1066 | /* exception characters from uri_to_string */ | ||
1067 | val = uri_string_escape(uptr, "/;&=+$,"); | ||
1068 | } | ||
1069 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1070 | bptr = bas->path; | ||
1071 | if (ref->path == NULL) { | ||
1072 | for (ix = 0; bptr[ix] != 0; ix++) { | ||
1073 | - if (bptr[ix] == '/') | ||
1074 | + if (bptr[ix] == '/') { | ||
1075 | nbslash++; | ||
1076 | + } | ||
1077 | } | ||
1078 | uptr = NULL; | ||
1079 | len = 1; /* this is for a string terminator only */ | ||
1080 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1081 | /* | ||
1082 | * Next we compare the two strings and find where they first differ | ||
1083 | */ | ||
1084 | - if ((ref->path[pos] == '.') && (ref->path[pos + 1] == '/')) | ||
1085 | + if ((ref->path[pos] == '.') && (ref->path[pos + 1] == '/')) { | ||
1086 | pos += 2; | ||
1087 | - if ((*bptr == '.') && (bptr[1] == '/')) | ||
1088 | + } | ||
1089 | + if ((*bptr == '.') && (bptr[1] == '/')) { | ||
1090 | bptr += 2; | ||
1091 | - else if ((*bptr == '/') && (ref->path[pos] != '/')) | ||
1092 | + } else if ((*bptr == '/') && (ref->path[pos] != '/')) { | ||
1093 | bptr++; | ||
1094 | - while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) | ||
1095 | + } | ||
1096 | + while ((bptr[pos] == ref->path[pos]) && (bptr[pos] != 0)) { | ||
1097 | pos++; | ||
1098 | + } | ||
1099 | |||
1100 | if (bptr[pos] == ref->path[pos]) { | ||
1101 | val = g_strdup(""); | ||
1102 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1103 | * beginning of the "unique" suffix of URI | ||
1104 | */ | ||
1105 | ix = pos; | ||
1106 | - if ((ref->path[ix] == '/') && (ix > 0)) | ||
1107 | + if ((ref->path[ix] == '/') && (ix > 0)) { | ||
1108 | ix--; | ||
1109 | - else if ((ref->path[ix] == 0) && (ix > 1) && (ref->path[ix - 1] == '/')) | ||
1110 | + } else if ((ref->path[ix] == 0) && (ix > 1) | ||
1111 | + && (ref->path[ix - 1] == '/')) { | ||
1112 | ix -= 2; | ||
1113 | + } | ||
1114 | for (; ix > 0; ix--) { | ||
1115 | - if (ref->path[ix] == '/') | ||
1116 | + if (ref->path[ix] == '/') { | ||
1117 | break; | ||
1118 | + } | ||
1119 | } | ||
1120 | if (ix == 0) { | ||
1121 | uptr = ref->path; | ||
1122 | @@ -XXX,XX +XXX,XX @@ char *uri_resolve_relative(const char *uri, const char *base) | ||
1123 | */ | ||
1124 | if (bptr[pos] != ref->path[pos]) { /* check for trivial URI == base */ | ||
1125 | for (; bptr[ix] != 0; ix++) { | ||
1126 | - if (bptr[ix] == '/') | ||
1127 | + if (bptr[ix] == '/') { | ||
1128 | nbslash++; | ||
1129 | + } | ||
1130 | } | ||
1131 | } | ||
1132 | len = strlen(uptr) + 1; | ||
1133 | } | ||
1134 | |||
1135 | if (nbslash == 0) { | ||
1136 | - if (uptr != NULL) | ||
1137 | + if (uptr != NULL) { | ||
1138 | /* exception characters from uri_to_string */ | ||
1139 | val = uri_string_escape(uptr, "/;&=+$,"); | ||
1140 | + } | ||
1141 | goto done; | ||
1142 | } | ||
1143 | |||
1144 | @@ -XXX,XX +XXX,XX @@ done: | ||
1145 | /* | ||
1146 | * Free the working variables | ||
1147 | */ | ||
1148 | - if (remove_path != 0) | ||
1149 | + if (remove_path != 0) { | ||
1150 | ref->path = NULL; | ||
1151 | - if (ref != NULL) | ||
1152 | + } | ||
1153 | + if (ref != NULL) { | ||
1154 | uri_free(ref); | ||
1155 | - if (bas != NULL) | ||
1156 | + } | ||
1157 | + if (bas != NULL) { | ||
1158 | uri_free(bas); | ||
1159 | + } | ||
1160 | |||
1161 | return val; | ||
1162 | } | ||
1163 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_new(int init_alloc) | ||
1164 | { | ||
1165 | struct QueryParams *ps; | ||
1166 | |||
1167 | - if (init_alloc <= 0) | ||
1168 | + if (init_alloc <= 0) { | ||
1169 | init_alloc = 1; | ||
1170 | + } | ||
1171 | |||
1172 | ps = g_new(QueryParams, 1); | ||
1173 | ps->n = 0; | ||
1174 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_parse(const char *query) | ||
1175 | const char *end, *eq; | ||
1176 | |||
1177 | ps = query_params_new(0); | ||
1178 | - if (!query || query[0] == '\0') | ||
1179 | + if (!query || query[0] == '\0') { | ||
1180 | return ps; | ||
1181 | + } | ||
1182 | |||
1183 | while (*query) { | ||
1184 | char *name = NULL, *value = NULL; | ||
1185 | |||
1186 | /* Find the next separator, or end of the string. */ | ||
1187 | end = strchr(query, '&'); | ||
1188 | - if (!end) | ||
1189 | + if (!end) { | ||
1190 | end = strchr(query, ';'); | ||
1191 | - if (!end) | ||
1192 | + } | ||
1193 | + if (!end) { | ||
1194 | end = query + strlen(query); | ||
1195 | + } | ||
1196 | |||
1197 | /* Find the first '=' character between here and end. */ | ||
1198 | eq = strchr(query, '='); | ||
1199 | - if (eq && eq >= end) | ||
1200 | + if (eq && eq >= end) { | ||
1201 | eq = NULL; | ||
1202 | + } | ||
1203 | |||
1204 | /* Empty section (eg. "&&"). */ | ||
1205 | - if (end == query) | ||
1206 | + if (end == query) { | ||
1207 | goto next; | ||
1208 | + } | ||
1209 | |||
1210 | /* If there is no '=' character, then we have just "name" | ||
1211 | * and consistent with CGI.pm we assume value is "". | ||
1212 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_parse(const char *query) | ||
1213 | /* If the '=' character is at the beginning then we have | ||
1214 | * "=value" and consistent with CGI.pm we _ignore_ this. | ||
1215 | */ | ||
1216 | - else if (query == eq) | ||
1217 | + else if (query == eq) { | ||
1218 | goto next; | ||
1219 | + } | ||
1220 | |||
1221 | /* Otherwise it's "name=value". */ | ||
1222 | else { | ||
1223 | @@ -XXX,XX +XXX,XX @@ struct QueryParams *query_params_parse(const char *query) | ||
1224 | |||
1225 | next: | ||
1226 | query = end; | ||
1227 | - if (*query) | ||
1228 | + if (*query) { | ||
1229 | query++; /* skip '&' separator */ | ||
1230 | + } | ||
1231 | } | ||
1232 | |||
1233 | return ps; | ||
124 | -- | 1234 | -- |
125 | 2.13.5 | 1235 | 2.14.3 |
126 | 1236 | ||
127 | 1237 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Alberto Garcia <berto@igalia.com> | ||
2 | 1 | ||
3 | Both the throttling limits set with the throttling.iops-* and | ||
4 | throttling.bps-* options and their QMP equivalents defined in the | ||
5 | BlockIOThrottle struct are integer values. | ||
6 | |||
7 | Those limits are also reported in the BlockDeviceInfo struct and they | ||
8 | are integers there as well. | ||
9 | |||
10 | Therefore there's no reason to store them internally as double and do | ||
11 | the conversion everytime we're setting or querying them, so this patch | ||
12 | uses uint64_t for those types. Let's also use an unsigned type because | ||
13 | we don't allow negative values anyway. | ||
14 | |||
15 | LeakyBucket.level and LeakyBucket.burst_level do however remain double | ||
16 | because their value changes depending on the fraction of time elapsed | ||
17 | since the previous I/O operation. | ||
18 | |||
19 | Signed-off-by: Alberto Garcia <berto@igalia.com> | ||
20 | Message-id: f29b840422767b5be2c41c2dfdbbbf6c5f8fedf8.1503580370.git.berto@igalia.com | ||
21 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
22 | --- | ||
23 | include/qemu/throttle.h | 4 ++-- | ||
24 | tests/test-throttle.c | 3 ++- | ||
25 | util/throttle.c | 7 +++---- | ||
26 | 3 files changed, 7 insertions(+), 7 deletions(-) | ||
27 | |||
28 | diff --git a/include/qemu/throttle.h b/include/qemu/throttle.h | ||
29 | index XXXXXXX..XXXXXXX 100644 | ||
30 | --- a/include/qemu/throttle.h | ||
31 | +++ b/include/qemu/throttle.h | ||
32 | @@ -XXX,XX +XXX,XX @@ typedef enum { | ||
33 | */ | ||
34 | |||
35 | typedef struct LeakyBucket { | ||
36 | - double avg; /* average goal in units per second */ | ||
37 | - double max; /* leaky bucket max burst in units */ | ||
38 | + uint64_t avg; /* average goal in units per second */ | ||
39 | + uint64_t max; /* leaky bucket max burst in units */ | ||
40 | double level; /* bucket level in units */ | ||
41 | double burst_level; /* bucket level in units (for computing bursts) */ | ||
42 | unsigned burst_length; /* max length of the burst period, in seconds */ | ||
43 | diff --git a/tests/test-throttle.c b/tests/test-throttle.c | ||
44 | index XXXXXXX..XXXXXXX 100644 | ||
45 | --- a/tests/test-throttle.c | ||
46 | +++ b/tests/test-throttle.c | ||
47 | @@ -XXX,XX +XXX,XX @@ static void test_enabled(void) | ||
48 | for (i = 0; i < BUCKETS_COUNT; i++) { | ||
49 | throttle_config_init(&cfg); | ||
50 | set_cfg_value(false, i, 150); | ||
51 | + g_assert(throttle_is_valid(&cfg, NULL)); | ||
52 | g_assert(throttle_enabled(&cfg)); | ||
53 | } | ||
54 | |||
55 | for (i = 0; i < BUCKETS_COUNT; i++) { | ||
56 | throttle_config_init(&cfg); | ||
57 | set_cfg_value(false, i, -150); | ||
58 | - g_assert(!throttle_enabled(&cfg)); | ||
59 | + g_assert(!throttle_is_valid(&cfg, NULL)); | ||
60 | } | ||
61 | } | ||
62 | |||
63 | diff --git a/util/throttle.c b/util/throttle.c | ||
64 | index XXXXXXX..XXXXXXX 100644 | ||
65 | --- a/util/throttle.c | ||
66 | +++ b/util/throttle.c | ||
67 | @@ -XXX,XX +XXX,XX @@ int64_t throttle_compute_wait(LeakyBucket *bkt) | ||
68 | /* If bkt->max is 0 we still want to allow short bursts of I/O | ||
69 | * from the guest, otherwise every other request will be throttled | ||
70 | * and performance will suffer considerably. */ | ||
71 | - bucket_size = bkt->avg / 10; | ||
72 | + bucket_size = (double) bkt->avg / 10; | ||
73 | burst_bucket_size = 0; | ||
74 | } else { | ||
75 | /* If we have a burst limit then we have to wait until all I/O | ||
76 | * at burst rate has finished before throttling to bkt->avg */ | ||
77 | bucket_size = bkt->max * bkt->burst_length; | ||
78 | - burst_bucket_size = bkt->max / 10; | ||
79 | + burst_bucket_size = (double) bkt->max / 10; | ||
80 | } | ||
81 | |||
82 | /* If the main bucket is full then we have to wait */ | ||
83 | @@ -XXX,XX +XXX,XX @@ bool throttle_is_valid(ThrottleConfig *cfg, Error **errp) | ||
84 | |||
85 | for (i = 0; i < BUCKETS_COUNT; i++) { | ||
86 | LeakyBucket *bkt = &cfg->buckets[i]; | ||
87 | - if (bkt->avg < 0 || bkt->max < 0 || | ||
88 | - bkt->avg > THROTTLE_VALUE_MAX || bkt->max > THROTTLE_VALUE_MAX) { | ||
89 | + if (bkt->avg > THROTTLE_VALUE_MAX || bkt->max > THROTTLE_VALUE_MAX) { | ||
90 | error_setg(errp, "bps/iops/max values must be within [0, %lld]", | ||
91 | THROTTLE_VALUE_MAX); | ||
92 | return false; | ||
93 | -- | ||
94 | 2.13.5 | ||
95 | |||
96 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Alberto Garcia <berto@igalia.com> | ||
2 | 1 | ||
3 | LeakyBucket.burst_length is defined as an unsigned integer but the | ||
4 | code never checks for overflows and it only makes sure that the value | ||
5 | is not 0. | ||
6 | |||
7 | In practice this means that the user can set something like | ||
8 | throttling.iops-total-max-length=4294967300 despite being larger than | ||
9 | UINT_MAX and the final value after casting to unsigned int will be 4. | ||
10 | |||
11 | This patch changes the data type to uint64_t. This does not increase | ||
12 | the storage size of LeakyBucket, and allows us to assign the value | ||
13 | directly from qemu_opt_get_number() or BlockIOThrottle and then do the | ||
14 | checks directly in throttle_is_valid(). | ||
15 | |||
16 | The value of burst_length does not have a specific upper limit, | ||
17 | but since the bucket size is defined by max * burst_length we have | ||
18 | to prevent overflows. Instead of going for UINT64_MAX or something | ||
19 | similar this patch reuses THROTTLE_VALUE_MAX, which allows I/O bursts | ||
20 | of 1 GiB/s for 10 days in a row. | ||
21 | |||
22 | Signed-off-by: Alberto Garcia <berto@igalia.com> | ||
23 | Message-id: 1b2e3049803f71cafb2e1fa1be4fb47147a0d398.1503580370.git.berto@igalia.com | ||
24 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
25 | --- | ||
26 | include/qemu/throttle.h | 2 +- | ||
27 | util/throttle.c | 5 +++++ | ||
28 | 2 files changed, 6 insertions(+), 1 deletion(-) | ||
29 | |||
30 | diff --git a/include/qemu/throttle.h b/include/qemu/throttle.h | ||
31 | index XXXXXXX..XXXXXXX 100644 | ||
32 | --- a/include/qemu/throttle.h | ||
33 | +++ b/include/qemu/throttle.h | ||
34 | @@ -XXX,XX +XXX,XX @@ typedef struct LeakyBucket { | ||
35 | uint64_t max; /* leaky bucket max burst in units */ | ||
36 | double level; /* bucket level in units */ | ||
37 | double burst_level; /* bucket level in units (for computing bursts) */ | ||
38 | - unsigned burst_length; /* max length of the burst period, in seconds */ | ||
39 | + uint64_t burst_length; /* max length of the burst period, in seconds */ | ||
40 | } LeakyBucket; | ||
41 | |||
42 | /* The following structure is used to configure a ThrottleState | ||
43 | diff --git a/util/throttle.c b/util/throttle.c | ||
44 | index XXXXXXX..XXXXXXX 100644 | ||
45 | --- a/util/throttle.c | ||
46 | +++ b/util/throttle.c | ||
47 | @@ -XXX,XX +XXX,XX @@ bool throttle_is_valid(ThrottleConfig *cfg, Error **errp) | ||
48 | return false; | ||
49 | } | ||
50 | |||
51 | + if (bkt->max && bkt->burst_length > THROTTLE_VALUE_MAX / bkt->max) { | ||
52 | + error_setg(errp, "burst length too high for this burst rate"); | ||
53 | + return false; | ||
54 | + } | ||
55 | + | ||
56 | if (bkt->max && !bkt->avg) { | ||
57 | error_setg(errp, "bps_max/iops_max require corresponding" | ||
58 | " bps/iops values"); | ||
59 | -- | ||
60 | 2.13.5 | ||
61 | |||
62 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Eduardo Habkost <ehabkost@redhat.com> | ||
2 | 1 | ||
3 | If QEMU is running on a system that's out of memory and mmap() | ||
4 | fails, QEMU aborts with no error message at all, making it hard | ||
5 | to debug the reason for the failure. | ||
6 | |||
7 | Add perror() calls that will print error information before | ||
8 | aborting. | ||
9 | |||
10 | Signed-off-by: Eduardo Habkost <ehabkost@redhat.com> | ||
11 | Reviewed-by: Philippe Mathieu-Daudé <f4bug@amsat.org> | ||
12 | Tested-by: Philippe Mathieu-Daudé <f4bug@amsat.org> | ||
13 | Message-id: 20170829212053.6003-1-ehabkost@redhat.com | ||
14 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
15 | --- | ||
16 | util/oslib-posix.c | 2 ++ | ||
17 | 1 file changed, 2 insertions(+) | ||
18 | |||
19 | diff --git a/util/oslib-posix.c b/util/oslib-posix.c | ||
20 | index XXXXXXX..XXXXXXX 100644 | ||
21 | --- a/util/oslib-posix.c | ||
22 | +++ b/util/oslib-posix.c | ||
23 | @@ -XXX,XX +XXX,XX @@ void *qemu_alloc_stack(size_t *sz) | ||
24 | ptr = mmap(NULL, *sz, PROT_READ | PROT_WRITE, | ||
25 | MAP_PRIVATE | MAP_ANONYMOUS, -1, 0); | ||
26 | if (ptr == MAP_FAILED) { | ||
27 | + perror("failed to allocate memory for stack"); | ||
28 | abort(); | ||
29 | } | ||
30 | |||
31 | @@ -XXX,XX +XXX,XX @@ void *qemu_alloc_stack(size_t *sz) | ||
32 | guardpage = ptr; | ||
33 | #endif | ||
34 | if (mprotect(guardpage, pagesz, PROT_NONE) != 0) { | ||
35 | + perror("failed to set up stack guard page"); | ||
36 | abort(); | ||
37 | } | ||
38 | |||
39 | -- | ||
40 | 2.13.5 | ||
41 | |||
42 | diff view generated by jsdifflib |
1 | Add the scripts/ directory to sys.path so Python 2.6 will be able to | 1 | From: Thomas Huth <thuth@redhat.com> |
---|---|---|---|
2 | import argparse. | ||
3 | 2 | ||
4 | Cc: Fam Zheng <famz@redhat.com> | 3 | dev could be NULL if the PCI device can not be found due to some |
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 4 | reasons, so we must not dereference the pointer in this case. |
6 | Acked-by: John Snow <jsnow@redhat.com> | 5 | |
7 | Acked-by: Fam Zheng <famz@redhat.com> | 6 | Signed-off-by: Thomas Huth <thuth@redhat.com> |
8 | Message-id: 20170825155732.15665-3-stefanha@redhat.com | 7 | Message-id: 1519713884-2346-1-git-send-email-thuth@redhat.com |
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
10 | --- | 9 | --- |
11 | tests/docker/docker.py | 4 +++- | 10 | tests/libqos/virtio-pci.c | 4 +++- |
12 | 1 file changed, 3 insertions(+), 1 deletion(-) | 11 | 1 file changed, 3 insertions(+), 1 deletion(-) |
13 | 12 | ||
14 | diff --git a/tests/docker/docker.py b/tests/docker/docker.py | 13 | diff --git a/tests/libqos/virtio-pci.c b/tests/libqos/virtio-pci.c |
15 | index XXXXXXX..XXXXXXX 100755 | 14 | index XXXXXXX..XXXXXXX 100644 |
16 | --- a/tests/docker/docker.py | 15 | --- a/tests/libqos/virtio-pci.c |
17 | +++ b/tests/docker/docker.py | 16 | +++ b/tests/libqos/virtio-pci.c |
18 | @@ -XXX,XX +XXX,XX @@ | 17 | @@ -XXX,XX +XXX,XX @@ QVirtioPCIDevice *qvirtio_pci_device_find(QPCIBus *bus, uint16_t device_type) |
19 | 18 | qvirtio_pci_foreach(bus, device_type, false, 0, | |
20 | import os | 19 | qvirtio_pci_assign_device, &dev); |
21 | import sys | 20 | |
22 | +sys.path.append(os.path.join(os.path.dirname(__file__), | 21 | - dev->vdev.bus = &qvirtio_pci; |
23 | + '..', '..', 'scripts')) | 22 | + if (dev) { |
24 | +import argparse | 23 | + dev->vdev.bus = &qvirtio_pci; |
25 | import subprocess | 24 | + } |
26 | import json | 25 | |
27 | import hashlib | 26 | return dev; |
28 | import atexit | 27 | } |
29 | import uuid | ||
30 | -import argparse | ||
31 | import tempfile | ||
32 | import re | ||
33 | import signal | ||
34 | -- | 28 | -- |
35 | 2.13.5 | 29 | 2.14.3 |
36 | 30 | ||
37 | 31 | diff view generated by jsdifflib |
1 | The minimum Python version supported by QEMU is 2.6. The argparse | 1 | From: Fam Zheng <famz@redhat.com> |
---|---|---|---|
2 | standard library module was only added in Python 2.7. Many scripts | ||
3 | would like to use argparse because it supports command-line | ||
4 | sub-commands. | ||
5 | 2 | ||
6 | This patch adds argparse. See the top of argparse.py for details. | 3 | git-publish [1] is a convenient tool to send patches and has been |
4 | popular among QEMU developers. Recently it has been made available in | ||
5 | Fedora/Debian official repo. | ||
7 | 6 | ||
8 | Suggested-by: Daniel P. Berrange <berrange@redhat.com> | 7 | One nice feature of the tool is a per-project configuration with |
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 8 | profiles, especially in which the cccmd option is a handy method to |
10 | Acked-by: John Snow <jsnow@redhat.com> | 9 | create the Cc list. |
11 | Message-id: 20170825155732.15665-2-stefanha@redhat.com | 10 | |
11 | [1]: https://github.com/stefanha/git-publish | ||
12 | |||
13 | Signed-off-by: Fam Zheng <famz@redhat.com> | ||
14 | Message-id: 20180226030326.20219-2-famz@redhat.com | ||
12 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 15 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
13 | --- | 16 | --- |
14 | COPYING.PYTHON | 270 ++++++ | 17 | .gitpublish | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++ |
15 | scripts/argparse.py | 2406 +++++++++++++++++++++++++++++++++++++++++++++++++++ | 18 | 1 file changed, 51 insertions(+) |
16 | 2 files changed, 2676 insertions(+) | 19 | create mode 100644 .gitpublish |
17 | create mode 100644 COPYING.PYTHON | ||
18 | create mode 100644 scripts/argparse.py | ||
19 | 20 | ||
20 | diff --git a/COPYING.PYTHON b/COPYING.PYTHON | 21 | diff --git a/.gitpublish b/.gitpublish |
21 | new file mode 100644 | 22 | new file mode 100644 |
22 | index XXXXXXX..XXXXXXX | 23 | index XXXXXXX..XXXXXXX |
23 | --- /dev/null | 24 | --- /dev/null |
24 | +++ b/COPYING.PYTHON | 25 | +++ b/.gitpublish |
25 | @@ -XXX,XX +XXX,XX @@ | 26 | @@ -XXX,XX +XXX,XX @@ |
26 | +A. HISTORY OF THE SOFTWARE | 27 | +# |
27 | +========================== | 28 | +# Common git-publish profiles that can be used to send patches to QEMU upstream. |
29 | +# | ||
30 | +# See https://github.com/stefanha/git-publish for more information | ||
31 | +# | ||
32 | +[gitpublishprofile "default"] | ||
33 | +base = master | ||
34 | +to = qemu-devel@nongnu.org | ||
35 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | ||
28 | + | 36 | + |
29 | +Python was created in the early 1990s by Guido van Rossum at Stichting | 37 | +[gitpublishprofile "rfc"] |
30 | +Mathematisch Centrum (CWI, see http://www.cwi.nl) in the Netherlands | 38 | +base = master |
31 | +as a successor of a language called ABC. Guido remains Python's | 39 | +prefix = RFC PATCH |
32 | +principal author, although it includes many contributions from others. | 40 | +to = qemu-devel@nongnu.org |
41 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | ||
33 | + | 42 | + |
34 | +In 1995, Guido continued his work on Python at the Corporation for | 43 | +[gitpublishprofile "stable"] |
35 | +National Research Initiatives (CNRI, see http://www.cnri.reston.va.us) | 44 | +base = master |
36 | +in Reston, Virginia where he released several versions of the | 45 | +to = qemu-devel@nongnu.org |
37 | +software. | 46 | +cc = qemu-stable@nongnu.org |
47 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | ||
38 | + | 48 | + |
39 | +In May 2000, Guido and the Python core development team moved to | 49 | +[gitpublishprofile "trivial"] |
40 | +BeOpen.com to form the BeOpen PythonLabs team. In October of the same | 50 | +base = master |
41 | +year, the PythonLabs team moved to Digital Creations (now Zope | 51 | +to = qemu-devel@nongnu.org |
42 | +Corporation, see http://www.zope.com). In 2001, the Python Software | 52 | +cc = qemu-trivial@nongnu.org |
43 | +Foundation (PSF, see http://www.python.org/psf/) was formed, a | 53 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null |
44 | +non-profit organization created specifically to own Python-related | ||
45 | +Intellectual Property. Zope Corporation is a sponsoring member of | ||
46 | +the PSF. | ||
47 | + | 54 | + |
48 | +All Python releases are Open Source (see http://www.opensource.org for | 55 | +[gitpublishprofile "block"] |
49 | +the Open Source Definition). Historically, most, but not all, Python | 56 | +base = master |
50 | +releases have also been GPL-compatible; the table below summarizes | 57 | +to = qemu-devel@nongnu.org |
51 | +the various releases. | 58 | +cc = qemu-block@nongnu.org |
59 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | ||
52 | + | 60 | + |
53 | + Release Derived Year Owner GPL- | 61 | +[gitpublishprofile "arm"] |
54 | + from compatible? (1) | 62 | +base = master |
63 | +to = qemu-devel@nongnu.org | ||
64 | +cc = qemu-arm@nongnu.org | ||
65 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null | ||
55 | + | 66 | + |
56 | + 0.9.0 thru 1.2 1991-1995 CWI yes | 67 | +[gitpublishprofile "s390"] |
57 | + 1.3 thru 1.5.2 1.2 1995-1999 CNRI yes | 68 | +base = master |
58 | + 1.6 1.5.2 2000 CNRI no | 69 | +to = qemu-devel@nongnu.org |
59 | + 2.0 1.6 2000 BeOpen.com no | 70 | +cc = qemu-s390@nongnu.org |
60 | + 1.6.1 1.6 2001 CNRI yes (2) | 71 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null |
61 | + 2.1 2.0+1.6.1 2001 PSF no | ||
62 | + 2.0.1 2.0+1.6.1 2001 PSF yes | ||
63 | + 2.1.1 2.1+2.0.1 2001 PSF yes | ||
64 | + 2.2 2.1.1 2001 PSF yes | ||
65 | + 2.1.2 2.1.1 2002 PSF yes | ||
66 | + 2.1.3 2.1.2 2002 PSF yes | ||
67 | + 2.2.1 2.2 2002 PSF yes | ||
68 | + 2.2.2 2.2.1 2002 PSF yes | ||
69 | + 2.2.3 2.2.2 2003 PSF yes | ||
70 | + 2.3 2.2.2 2002-2003 PSF yes | ||
71 | + 2.3.1 2.3 2002-2003 PSF yes | ||
72 | + 2.3.2 2.3.1 2002-2003 PSF yes | ||
73 | + 2.3.3 2.3.2 2002-2003 PSF yes | ||
74 | + 2.3.4 2.3.3 2004 PSF yes | ||
75 | + 2.3.5 2.3.4 2005 PSF yes | ||
76 | + 2.4 2.3 2004 PSF yes | ||
77 | + 2.4.1 2.4 2005 PSF yes | ||
78 | + 2.4.2 2.4.1 2005 PSF yes | ||
79 | + 2.4.3 2.4.2 2006 PSF yes | ||
80 | + 2.5 2.4 2006 PSF yes | ||
81 | + 2.7 2.6 2010 PSF yes | ||
82 | + | 72 | + |
83 | +Footnotes: | 73 | +[gitpublishprofile "ppc"] |
84 | + | 74 | +base = master |
85 | +(1) GPL-compatible doesn't mean that we're distributing Python under | 75 | +to = qemu-devel@nongnu.org |
86 | + the GPL. All Python licenses, unlike the GPL, let you distribute | 76 | +cc = qemu-ppc@nongnu.org |
87 | + a modified version without making your changes open source. The | 77 | +cccmd = scripts/get_maintainer.pl --noroles --norolestats --nogit --nogit-fallback 2>/dev/null |
88 | + GPL-compatible licenses make it possible to combine Python with | ||
89 | + other software that is released under the GPL; the others don't. | ||
90 | + | ||
91 | +(2) According to Richard Stallman, 1.6.1 is not GPL-compatible, | ||
92 | + because its license has a choice of law clause. According to | ||
93 | + CNRI, however, Stallman's lawyer has told CNRI's lawyer that 1.6.1 | ||
94 | + is "not incompatible" with the GPL. | ||
95 | + | ||
96 | +Thanks to the many outside volunteers who have worked under Guido's | ||
97 | +direction to make these releases possible. | ||
98 | + | ||
99 | + | ||
100 | +B. TERMS AND CONDITIONS FOR ACCESSING OR OTHERWISE USING PYTHON | ||
101 | +=============================================================== | ||
102 | + | ||
103 | +PYTHON SOFTWARE FOUNDATION LICENSE VERSION 2 | ||
104 | +-------------------------------------------- | ||
105 | + | ||
106 | +1. This LICENSE AGREEMENT is between the Python Software Foundation | ||
107 | +("PSF"), and the Individual or Organization ("Licensee") accessing and | ||
108 | +otherwise using this software ("Python") in source or binary form and | ||
109 | +its associated documentation. | ||
110 | + | ||
111 | +2. Subject to the terms and conditions of this License Agreement, PSF | ||
112 | +hereby grants Licensee a nonexclusive, royalty-free, world-wide | ||
113 | +license to reproduce, analyze, test, perform and/or display publicly, | ||
114 | +prepare derivative works, distribute, and otherwise use Python | ||
115 | +alone or in any derivative version, provided, however, that PSF's | ||
116 | +License Agreement and PSF's notice of copyright, i.e., "Copyright (c) | ||
117 | +2001, 2002, 2003, 2004, 2005, 2006 Python Software Foundation; All Rights | ||
118 | +Reserved" are retained in Python alone or in any derivative version | ||
119 | +prepared by Licensee. | ||
120 | + | ||
121 | +3. In the event Licensee prepares a derivative work that is based on | ||
122 | +or incorporates Python or any part thereof, and wants to make | ||
123 | +the derivative work available to others as provided herein, then | ||
124 | +Licensee hereby agrees to include in any such work a brief summary of | ||
125 | +the changes made to Python. | ||
126 | + | ||
127 | +4. PSF is making Python available to Licensee on an "AS IS" | ||
128 | +basis. PSF MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR | ||
129 | +IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, PSF MAKES NO AND | ||
130 | +DISCLAIMS ANY REPRESENTATION OR WARRANTY OF MERCHANTABILITY OR FITNESS | ||
131 | +FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF PYTHON WILL NOT | ||
132 | +INFRINGE ANY THIRD PARTY RIGHTS. | ||
133 | + | ||
134 | +5. PSF SHALL NOT BE LIABLE TO LICENSEE OR ANY OTHER USERS OF PYTHON | ||
135 | +FOR ANY INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES OR LOSS AS | ||
136 | +A RESULT OF MODIFYING, DISTRIBUTING, OR OTHERWISE USING PYTHON, | ||
137 | +OR ANY DERIVATIVE THEREOF, EVEN IF ADVISED OF THE POSSIBILITY THEREOF. | ||
138 | + | ||
139 | +6. This License Agreement will automatically terminate upon a material | ||
140 | +breach of its terms and conditions. | ||
141 | + | ||
142 | +7. Nothing in this License Agreement shall be deemed to create any | ||
143 | +relationship of agency, partnership, or joint venture between PSF and | ||
144 | +Licensee. This License Agreement does not grant permission to use PSF | ||
145 | +trademarks or trade name in a trademark sense to endorse or promote | ||
146 | +products or services of Licensee, or any third party. | ||
147 | + | ||
148 | +8. By copying, installing or otherwise using Python, Licensee | ||
149 | +agrees to be bound by the terms and conditions of this License | ||
150 | +Agreement. | ||
151 | + | ||
152 | + | ||
153 | +BEOPEN.COM LICENSE AGREEMENT FOR PYTHON 2.0 | ||
154 | +------------------------------------------- | ||
155 | + | ||
156 | +BEOPEN PYTHON OPEN SOURCE LICENSE AGREEMENT VERSION 1 | ||
157 | + | ||
158 | +1. This LICENSE AGREEMENT is between BeOpen.com ("BeOpen"), having an | ||
159 | +office at 160 Saratoga Avenue, Santa Clara, CA 95051, and the | ||
160 | +Individual or Organization ("Licensee") accessing and otherwise using | ||
161 | +this software in source or binary form and its associated | ||
162 | +documentation ("the Software"). | ||
163 | + | ||
164 | +2. Subject to the terms and conditions of this BeOpen Python License | ||
165 | +Agreement, BeOpen hereby grants Licensee a non-exclusive, | ||
166 | +royalty-free, world-wide license to reproduce, analyze, test, perform | ||
167 | +and/or display publicly, prepare derivative works, distribute, and | ||
168 | +otherwise use the Software alone or in any derivative version, | ||
169 | +provided, however, that the BeOpen Python License is retained in the | ||
170 | +Software, alone or in any derivative version prepared by Licensee. | ||
171 | + | ||
172 | +3. BeOpen is making the Software available to Licensee on an "AS IS" | ||
173 | +basis. BEOPEN MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR | ||
174 | +IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, BEOPEN MAKES NO AND | ||
175 | +DISCLAIMS ANY REPRESENTATION OR WARRANTY OF MERCHANTABILITY OR FITNESS | ||
176 | +FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF THE SOFTWARE WILL NOT | ||
177 | +INFRINGE ANY THIRD PARTY RIGHTS. | ||
178 | + | ||
179 | +4. BEOPEN SHALL NOT BE LIABLE TO LICENSEE OR ANY OTHER USERS OF THE | ||
180 | +SOFTWARE FOR ANY INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES OR LOSS | ||
181 | +AS A RESULT OF USING, MODIFYING OR DISTRIBUTING THE SOFTWARE, OR ANY | ||
182 | +DERIVATIVE THEREOF, EVEN IF ADVISED OF THE POSSIBILITY THEREOF. | ||
183 | + | ||
184 | +5. This License Agreement will automatically terminate upon a material | ||
185 | +breach of its terms and conditions. | ||
186 | + | ||
187 | +6. This License Agreement shall be governed by and interpreted in all | ||
188 | +respects by the law of the State of California, excluding conflict of | ||
189 | +law provisions. Nothing in this License Agreement shall be deemed to | ||
190 | +create any relationship of agency, partnership, or joint venture | ||
191 | +between BeOpen and Licensee. This License Agreement does not grant | ||
192 | +permission to use BeOpen trademarks or trade names in a trademark | ||
193 | +sense to endorse or promote products or services of Licensee, or any | ||
194 | +third party. As an exception, the "BeOpen Python" logos available at | ||
195 | +http://www.pythonlabs.com/logos.html may be used according to the | ||
196 | +permissions granted on that web page. | ||
197 | + | ||
198 | +7. By copying, installing or otherwise using the software, Licensee | ||
199 | +agrees to be bound by the terms and conditions of this License | ||
200 | +Agreement. | ||
201 | + | ||
202 | + | ||
203 | +CNRI LICENSE AGREEMENT FOR PYTHON 1.6.1 | ||
204 | +--------------------------------------- | ||
205 | + | ||
206 | +1. This LICENSE AGREEMENT is between the Corporation for National | ||
207 | +Research Initiatives, having an office at 1895 Preston White Drive, | ||
208 | +Reston, VA 20191 ("CNRI"), and the Individual or Organization | ||
209 | +("Licensee") accessing and otherwise using Python 1.6.1 software in | ||
210 | +source or binary form and its associated documentation. | ||
211 | + | ||
212 | +2. Subject to the terms and conditions of this License Agreement, CNRI | ||
213 | +hereby grants Licensee a nonexclusive, royalty-free, world-wide | ||
214 | +license to reproduce, analyze, test, perform and/or display publicly, | ||
215 | +prepare derivative works, distribute, and otherwise use Python 1.6.1 | ||
216 | +alone or in any derivative version, provided, however, that CNRI's | ||
217 | +License Agreement and CNRI's notice of copyright, i.e., "Copyright (c) | ||
218 | +1995-2001 Corporation for National Research Initiatives; All Rights | ||
219 | +Reserved" are retained in Python 1.6.1 alone or in any derivative | ||
220 | +version prepared by Licensee. Alternately, in lieu of CNRI's License | ||
221 | +Agreement, Licensee may substitute the following text (omitting the | ||
222 | +quotes): "Python 1.6.1 is made available subject to the terms and | ||
223 | +conditions in CNRI's License Agreement. This Agreement together with | ||
224 | +Python 1.6.1 may be located on the Internet using the following | ||
225 | +unique, persistent identifier (known as a handle): 1895.22/1013. This | ||
226 | +Agreement may also be obtained from a proxy server on the Internet | ||
227 | +using the following URL: http://hdl.handle.net/1895.22/1013". | ||
228 | + | ||
229 | +3. In the event Licensee prepares a derivative work that is based on | ||
230 | +or incorporates Python 1.6.1 or any part thereof, and wants to make | ||
231 | +the derivative work available to others as provided herein, then | ||
232 | +Licensee hereby agrees to include in any such work a brief summary of | ||
233 | +the changes made to Python 1.6.1. | ||
234 | + | ||
235 | +4. CNRI is making Python 1.6.1 available to Licensee on an "AS IS" | ||
236 | +basis. CNRI MAKES NO REPRESENTATIONS OR WARRANTIES, EXPRESS OR | ||
237 | +IMPLIED. BY WAY OF EXAMPLE, BUT NOT LIMITATION, CNRI MAKES NO AND | ||
238 | +DISCLAIMS ANY REPRESENTATION OR WARRANTY OF MERCHANTABILITY OR FITNESS | ||
239 | +FOR ANY PARTICULAR PURPOSE OR THAT THE USE OF PYTHON 1.6.1 WILL NOT | ||
240 | +INFRINGE ANY THIRD PARTY RIGHTS. | ||
241 | + | ||
242 | +5. CNRI SHALL NOT BE LIABLE TO LICENSEE OR ANY OTHER USERS OF PYTHON | ||
243 | +1.6.1 FOR ANY INCIDENTAL, SPECIAL, OR CONSEQUENTIAL DAMAGES OR LOSS AS | ||
244 | +A RESULT OF MODIFYING, DISTRIBUTING, OR OTHERWISE USING PYTHON 1.6.1, | ||
245 | +OR ANY DERIVATIVE THEREOF, EVEN IF ADVISED OF THE POSSIBILITY THEREOF. | ||
246 | + | ||
247 | +6. This License Agreement will automatically terminate upon a material | ||
248 | +breach of its terms and conditions. | ||
249 | + | ||
250 | +7. This License Agreement shall be governed by the federal | ||
251 | +intellectual property law of the United States, including without | ||
252 | +limitation the federal copyright law, and, to the extent such | ||
253 | +U.S. federal law does not apply, by the law of the Commonwealth of | ||
254 | +Virginia, excluding Virginia's conflict of law provisions. | ||
255 | +Notwithstanding the foregoing, with regard to derivative works based | ||
256 | +on Python 1.6.1 that incorporate non-separable material that was | ||
257 | +previously distributed under the GNU General Public License (GPL), the | ||
258 | +law of the Commonwealth of Virginia shall govern this License | ||
259 | +Agreement only as to issues arising under or with respect to | ||
260 | +Paragraphs 4, 5, and 7 of this License Agreement. Nothing in this | ||
261 | +License Agreement shall be deemed to create any relationship of | ||
262 | +agency, partnership, or joint venture between CNRI and Licensee. This | ||
263 | +License Agreement does not grant permission to use CNRI trademarks or | ||
264 | +trade name in a trademark sense to endorse or promote products or | ||
265 | +services of Licensee, or any third party. | ||
266 | + | ||
267 | +8. By clicking on the "ACCEPT" button where indicated, or by copying, | ||
268 | +installing or otherwise using Python 1.6.1, Licensee agrees to be | ||
269 | +bound by the terms and conditions of this License Agreement. | ||
270 | + | ||
271 | + ACCEPT | ||
272 | + | ||
273 | + | ||
274 | +CWI LICENSE AGREEMENT FOR PYTHON 0.9.0 THROUGH 1.2 | ||
275 | +-------------------------------------------------- | ||
276 | + | ||
277 | +Copyright (c) 1991 - 1995, Stichting Mathematisch Centrum Amsterdam, | ||
278 | +The Netherlands. All rights reserved. | ||
279 | + | ||
280 | +Permission to use, copy, modify, and distribute this software and its | ||
281 | +documentation for any purpose and without fee is hereby granted, | ||
282 | +provided that the above copyright notice appear in all copies and that | ||
283 | +both that copyright notice and this permission notice appear in | ||
284 | +supporting documentation, and that the name of Stichting Mathematisch | ||
285 | +Centrum or CWI not be used in advertising or publicity pertaining to | ||
286 | +distribution of the software without specific, written prior | ||
287 | +permission. | ||
288 | + | ||
289 | +STICHTING MATHEMATISCH CENTRUM DISCLAIMS ALL WARRANTIES WITH REGARD TO | ||
290 | +THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY AND | ||
291 | +FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH CENTRUM BE LIABLE | ||
292 | +FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | ||
293 | +WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | ||
294 | +ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT | ||
295 | +OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | ||
296 | diff --git a/scripts/argparse.py b/scripts/argparse.py | ||
297 | new file mode 100644 | ||
298 | index XXXXXXX..XXXXXXX | ||
299 | --- /dev/null | ||
300 | +++ b/scripts/argparse.py | ||
301 | @@ -XXX,XX +XXX,XX @@ | ||
302 | +# This is a local copy of the standard library argparse module taken from PyPI. | ||
303 | +# It is licensed under the Python Software Foundation License. This is a | ||
304 | +# fallback for Python 2.6 which does not include this module. Python 2.7+ and | ||
305 | +# 3+ will never load this module because built-in modules are loaded before | ||
306 | +# anything in sys.path. | ||
307 | +# | ||
308 | +# If your script is not located in the same directory as this file, import it | ||
309 | +# like this: | ||
310 | +# | ||
311 | +# import os | ||
312 | +# import sys | ||
313 | +# sys.path.append(os.path.join(os.path.dirname(__file__), ..., 'scripts')) | ||
314 | +# import argparse | ||
315 | + | ||
316 | +# Author: Steven J. Bethard <steven.bethard@gmail.com>. | ||
317 | +# Maintainer: Thomas Waldmann <tw@waldmann-edv.de> | ||
318 | + | ||
319 | +"""Command-line parsing library | ||
320 | + | ||
321 | +This module is an optparse-inspired command-line parsing library that: | ||
322 | + | ||
323 | + - handles both optional and positional arguments | ||
324 | + - produces highly informative usage messages | ||
325 | + - supports parsers that dispatch to sub-parsers | ||
326 | + | ||
327 | +The following is a simple usage example that sums integers from the | ||
328 | +command-line and writes the result to a file:: | ||
329 | + | ||
330 | + parser = argparse.ArgumentParser( | ||
331 | + description='sum the integers at the command line') | ||
332 | + parser.add_argument( | ||
333 | + 'integers', metavar='int', nargs='+', type=int, | ||
334 | + help='an integer to be summed') | ||
335 | + parser.add_argument( | ||
336 | + '--log', default=sys.stdout, type=argparse.FileType('w'), | ||
337 | + help='the file where the sum should be written') | ||
338 | + args = parser.parse_args() | ||
339 | + args.log.write('%s' % sum(args.integers)) | ||
340 | + args.log.close() | ||
341 | + | ||
342 | +The module contains the following public classes: | ||
343 | + | ||
344 | + - ArgumentParser -- The main entry point for command-line parsing. As the | ||
345 | + example above shows, the add_argument() method is used to populate | ||
346 | + the parser with actions for optional and positional arguments. Then | ||
347 | + the parse_args() method is invoked to convert the args at the | ||
348 | + command-line into an object with attributes. | ||
349 | + | ||
350 | + - ArgumentError -- The exception raised by ArgumentParser objects when | ||
351 | + there are errors with the parser's actions. Errors raised while | ||
352 | + parsing the command-line are caught by ArgumentParser and emitted | ||
353 | + as command-line messages. | ||
354 | + | ||
355 | + - FileType -- A factory for defining types of files to be created. As the | ||
356 | + example above shows, instances of FileType are typically passed as | ||
357 | + the type= argument of add_argument() calls. | ||
358 | + | ||
359 | + - Action -- The base class for parser actions. Typically actions are | ||
360 | + selected by passing strings like 'store_true' or 'append_const' to | ||
361 | + the action= argument of add_argument(). However, for greater | ||
362 | + customization of ArgumentParser actions, subclasses of Action may | ||
363 | + be defined and passed as the action= argument. | ||
364 | + | ||
365 | + - HelpFormatter, RawDescriptionHelpFormatter, RawTextHelpFormatter, | ||
366 | + ArgumentDefaultsHelpFormatter -- Formatter classes which | ||
367 | + may be passed as the formatter_class= argument to the | ||
368 | + ArgumentParser constructor. HelpFormatter is the default, | ||
369 | + RawDescriptionHelpFormatter and RawTextHelpFormatter tell the parser | ||
370 | + not to change the formatting for help text, and | ||
371 | + ArgumentDefaultsHelpFormatter adds information about argument defaults | ||
372 | + to the help. | ||
373 | + | ||
374 | +All other classes in this module are considered implementation details. | ||
375 | +(Also note that HelpFormatter and RawDescriptionHelpFormatter are only | ||
376 | +considered public as object names -- the API of the formatter objects is | ||
377 | +still considered an implementation detail.) | ||
378 | +""" | ||
379 | + | ||
380 | +__version__ = '1.4.0' # we use our own version number independant of the | ||
381 | + # one in stdlib and we release this on pypi. | ||
382 | + | ||
383 | +__external_lib__ = True # to make sure the tests really test THIS lib, | ||
384 | + # not the builtin one in Python stdlib | ||
385 | + | ||
386 | +__all__ = [ | ||
387 | + 'ArgumentParser', | ||
388 | + 'ArgumentError', | ||
389 | + 'ArgumentTypeError', | ||
390 | + 'FileType', | ||
391 | + 'HelpFormatter', | ||
392 | + 'ArgumentDefaultsHelpFormatter', | ||
393 | + 'RawDescriptionHelpFormatter', | ||
394 | + 'RawTextHelpFormatter', | ||
395 | + 'Namespace', | ||
396 | + 'Action', | ||
397 | + 'ONE_OR_MORE', | ||
398 | + 'OPTIONAL', | ||
399 | + 'PARSER', | ||
400 | + 'REMAINDER', | ||
401 | + 'SUPPRESS', | ||
402 | + 'ZERO_OR_MORE', | ||
403 | +] | ||
404 | + | ||
405 | + | ||
406 | +import copy as _copy | ||
407 | +import os as _os | ||
408 | +import re as _re | ||
409 | +import sys as _sys | ||
410 | +import textwrap as _textwrap | ||
411 | + | ||
412 | +from gettext import gettext as _ | ||
413 | + | ||
414 | +try: | ||
415 | + set | ||
416 | +except NameError: | ||
417 | + # for python < 2.4 compatibility (sets module is there since 2.3): | ||
418 | + from sets import Set as set | ||
419 | + | ||
420 | +try: | ||
421 | + basestring | ||
422 | +except NameError: | ||
423 | + basestring = str | ||
424 | + | ||
425 | +try: | ||
426 | + sorted | ||
427 | +except NameError: | ||
428 | + # for python < 2.4 compatibility: | ||
429 | + def sorted(iterable, reverse=False): | ||
430 | + result = list(iterable) | ||
431 | + result.sort() | ||
432 | + if reverse: | ||
433 | + result.reverse() | ||
434 | + return result | ||
435 | + | ||
436 | + | ||
437 | +def _callable(obj): | ||
438 | + return hasattr(obj, '__call__') or hasattr(obj, '__bases__') | ||
439 | + | ||
440 | + | ||
441 | +SUPPRESS = '==SUPPRESS==' | ||
442 | + | ||
443 | +OPTIONAL = '?' | ||
444 | +ZERO_OR_MORE = '*' | ||
445 | +ONE_OR_MORE = '+' | ||
446 | +PARSER = 'A...' | ||
447 | +REMAINDER = '...' | ||
448 | +_UNRECOGNIZED_ARGS_ATTR = '_unrecognized_args' | ||
449 | + | ||
450 | +# ============================= | ||
451 | +# Utility functions and classes | ||
452 | +# ============================= | ||
453 | + | ||
454 | +class _AttributeHolder(object): | ||
455 | + """Abstract base class that provides __repr__. | ||
456 | + | ||
457 | + The __repr__ method returns a string in the format:: | ||
458 | + ClassName(attr=name, attr=name, ...) | ||
459 | + The attributes are determined either by a class-level attribute, | ||
460 | + '_kwarg_names', or by inspecting the instance __dict__. | ||
461 | + """ | ||
462 | + | ||
463 | + def __repr__(self): | ||
464 | + type_name = type(self).__name__ | ||
465 | + arg_strings = [] | ||
466 | + for arg in self._get_args(): | ||
467 | + arg_strings.append(repr(arg)) | ||
468 | + for name, value in self._get_kwargs(): | ||
469 | + arg_strings.append('%s=%r' % (name, value)) | ||
470 | + return '%s(%s)' % (type_name, ', '.join(arg_strings)) | ||
471 | + | ||
472 | + def _get_kwargs(self): | ||
473 | + return sorted(self.__dict__.items()) | ||
474 | + | ||
475 | + def _get_args(self): | ||
476 | + return [] | ||
477 | + | ||
478 | + | ||
479 | +def _ensure_value(namespace, name, value): | ||
480 | + if getattr(namespace, name, None) is None: | ||
481 | + setattr(namespace, name, value) | ||
482 | + return getattr(namespace, name) | ||
483 | + | ||
484 | + | ||
485 | +# =============== | ||
486 | +# Formatting Help | ||
487 | +# =============== | ||
488 | + | ||
489 | +class HelpFormatter(object): | ||
490 | + """Formatter for generating usage messages and argument help strings. | ||
491 | + | ||
492 | + Only the name of this class is considered a public API. All the methods | ||
493 | + provided by the class are considered an implementation detail. | ||
494 | + """ | ||
495 | + | ||
496 | + def __init__(self, | ||
497 | + prog, | ||
498 | + indent_increment=2, | ||
499 | + max_help_position=24, | ||
500 | + width=None): | ||
501 | + | ||
502 | + # default setting for width | ||
503 | + if width is None: | ||
504 | + try: | ||
505 | + width = int(_os.environ['COLUMNS']) | ||
506 | + except (KeyError, ValueError): | ||
507 | + width = 80 | ||
508 | + width -= 2 | ||
509 | + | ||
510 | + self._prog = prog | ||
511 | + self._indent_increment = indent_increment | ||
512 | + self._max_help_position = max_help_position | ||
513 | + self._width = width | ||
514 | + | ||
515 | + self._current_indent = 0 | ||
516 | + self._level = 0 | ||
517 | + self._action_max_length = 0 | ||
518 | + | ||
519 | + self._root_section = self._Section(self, None) | ||
520 | + self._current_section = self._root_section | ||
521 | + | ||
522 | + self._whitespace_matcher = _re.compile(r'\s+') | ||
523 | + self._long_break_matcher = _re.compile(r'\n\n\n+') | ||
524 | + | ||
525 | + # =============================== | ||
526 | + # Section and indentation methods | ||
527 | + # =============================== | ||
528 | + def _indent(self): | ||
529 | + self._current_indent += self._indent_increment | ||
530 | + self._level += 1 | ||
531 | + | ||
532 | + def _dedent(self): | ||
533 | + self._current_indent -= self._indent_increment | ||
534 | + assert self._current_indent >= 0, 'Indent decreased below 0.' | ||
535 | + self._level -= 1 | ||
536 | + | ||
537 | + class _Section(object): | ||
538 | + | ||
539 | + def __init__(self, formatter, parent, heading=None): | ||
540 | + self.formatter = formatter | ||
541 | + self.parent = parent | ||
542 | + self.heading = heading | ||
543 | + self.items = [] | ||
544 | + | ||
545 | + def format_help(self): | ||
546 | + # format the indented section | ||
547 | + if self.parent is not None: | ||
548 | + self.formatter._indent() | ||
549 | + join = self.formatter._join_parts | ||
550 | + for func, args in self.items: | ||
551 | + func(*args) | ||
552 | + item_help = join([func(*args) for func, args in self.items]) | ||
553 | + if self.parent is not None: | ||
554 | + self.formatter._dedent() | ||
555 | + | ||
556 | + # return nothing if the section was empty | ||
557 | + if not item_help: | ||
558 | + return '' | ||
559 | + | ||
560 | + # add the heading if the section was non-empty | ||
561 | + if self.heading is not SUPPRESS and self.heading is not None: | ||
562 | + current_indent = self.formatter._current_indent | ||
563 | + heading = '%*s%s:\n' % (current_indent, '', self.heading) | ||
564 | + else: | ||
565 | + heading = '' | ||
566 | + | ||
567 | + # join the section-initial newline, the heading and the help | ||
568 | + return join(['\n', heading, item_help, '\n']) | ||
569 | + | ||
570 | + def _add_item(self, func, args): | ||
571 | + self._current_section.items.append((func, args)) | ||
572 | + | ||
573 | + # ======================== | ||
574 | + # Message building methods | ||
575 | + # ======================== | ||
576 | + def start_section(self, heading): | ||
577 | + self._indent() | ||
578 | + section = self._Section(self, self._current_section, heading) | ||
579 | + self._add_item(section.format_help, []) | ||
580 | + self._current_section = section | ||
581 | + | ||
582 | + def end_section(self): | ||
583 | + self._current_section = self._current_section.parent | ||
584 | + self._dedent() | ||
585 | + | ||
586 | + def add_text(self, text): | ||
587 | + if text is not SUPPRESS and text is not None: | ||
588 | + self._add_item(self._format_text, [text]) | ||
589 | + | ||
590 | + def add_usage(self, usage, actions, groups, prefix=None): | ||
591 | + if usage is not SUPPRESS: | ||
592 | + args = usage, actions, groups, prefix | ||
593 | + self._add_item(self._format_usage, args) | ||
594 | + | ||
595 | + def add_argument(self, action): | ||
596 | + if action.help is not SUPPRESS: | ||
597 | + | ||
598 | + # find all invocations | ||
599 | + get_invocation = self._format_action_invocation | ||
600 | + invocations = [get_invocation(action)] | ||
601 | + for subaction in self._iter_indented_subactions(action): | ||
602 | + invocations.append(get_invocation(subaction)) | ||
603 | + | ||
604 | + # update the maximum item length | ||
605 | + invocation_length = max([len(s) for s in invocations]) | ||
606 | + action_length = invocation_length + self._current_indent | ||
607 | + self._action_max_length = max(self._action_max_length, | ||
608 | + action_length) | ||
609 | + | ||
610 | + # add the item to the list | ||
611 | + self._add_item(self._format_action, [action]) | ||
612 | + | ||
613 | + def add_arguments(self, actions): | ||
614 | + for action in actions: | ||
615 | + self.add_argument(action) | ||
616 | + | ||
617 | + # ======================= | ||
618 | + # Help-formatting methods | ||
619 | + # ======================= | ||
620 | + def format_help(self): | ||
621 | + help = self._root_section.format_help() | ||
622 | + if help: | ||
623 | + help = self._long_break_matcher.sub('\n\n', help) | ||
624 | + help = help.strip('\n') + '\n' | ||
625 | + return help | ||
626 | + | ||
627 | + def _join_parts(self, part_strings): | ||
628 | + return ''.join([part | ||
629 | + for part in part_strings | ||
630 | + if part and part is not SUPPRESS]) | ||
631 | + | ||
632 | + def _format_usage(self, usage, actions, groups, prefix): | ||
633 | + if prefix is None: | ||
634 | + prefix = _('usage: ') | ||
635 | + | ||
636 | + # if usage is specified, use that | ||
637 | + if usage is not None: | ||
638 | + usage = usage % dict(prog=self._prog) | ||
639 | + | ||
640 | + # if no optionals or positionals are available, usage is just prog | ||
641 | + elif usage is None and not actions: | ||
642 | + usage = '%(prog)s' % dict(prog=self._prog) | ||
643 | + | ||
644 | + # if optionals and positionals are available, calculate usage | ||
645 | + elif usage is None: | ||
646 | + prog = '%(prog)s' % dict(prog=self._prog) | ||
647 | + | ||
648 | + # split optionals from positionals | ||
649 | + optionals = [] | ||
650 | + positionals = [] | ||
651 | + for action in actions: | ||
652 | + if action.option_strings: | ||
653 | + optionals.append(action) | ||
654 | + else: | ||
655 | + positionals.append(action) | ||
656 | + | ||
657 | + # build full usage string | ||
658 | + format = self._format_actions_usage | ||
659 | + action_usage = format(optionals + positionals, groups) | ||
660 | + usage = ' '.join([s for s in [prog, action_usage] if s]) | ||
661 | + | ||
662 | + # wrap the usage parts if it's too long | ||
663 | + text_width = self._width - self._current_indent | ||
664 | + if len(prefix) + len(usage) > text_width: | ||
665 | + | ||
666 | + # break usage into wrappable parts | ||
667 | + part_regexp = r'\(.*?\)+|\[.*?\]+|\S+' | ||
668 | + opt_usage = format(optionals, groups) | ||
669 | + pos_usage = format(positionals, groups) | ||
670 | + opt_parts = _re.findall(part_regexp, opt_usage) | ||
671 | + pos_parts = _re.findall(part_regexp, pos_usage) | ||
672 | + assert ' '.join(opt_parts) == opt_usage | ||
673 | + assert ' '.join(pos_parts) == pos_usage | ||
674 | + | ||
675 | + # helper for wrapping lines | ||
676 | + def get_lines(parts, indent, prefix=None): | ||
677 | + lines = [] | ||
678 | + line = [] | ||
679 | + if prefix is not None: | ||
680 | + line_len = len(prefix) - 1 | ||
681 | + else: | ||
682 | + line_len = len(indent) - 1 | ||
683 | + for part in parts: | ||
684 | + if line_len + 1 + len(part) > text_width: | ||
685 | + lines.append(indent + ' '.join(line)) | ||
686 | + line = [] | ||
687 | + line_len = len(indent) - 1 | ||
688 | + line.append(part) | ||
689 | + line_len += len(part) + 1 | ||
690 | + if line: | ||
691 | + lines.append(indent + ' '.join(line)) | ||
692 | + if prefix is not None: | ||
693 | + lines[0] = lines[0][len(indent):] | ||
694 | + return lines | ||
695 | + | ||
696 | + # if prog is short, follow it with optionals or positionals | ||
697 | + if len(prefix) + len(prog) <= 0.75 * text_width: | ||
698 | + indent = ' ' * (len(prefix) + len(prog) + 1) | ||
699 | + if opt_parts: | ||
700 | + lines = get_lines([prog] + opt_parts, indent, prefix) | ||
701 | + lines.extend(get_lines(pos_parts, indent)) | ||
702 | + elif pos_parts: | ||
703 | + lines = get_lines([prog] + pos_parts, indent, prefix) | ||
704 | + else: | ||
705 | + lines = [prog] | ||
706 | + | ||
707 | + # if prog is long, put it on its own line | ||
708 | + else: | ||
709 | + indent = ' ' * len(prefix) | ||
710 | + parts = opt_parts + pos_parts | ||
711 | + lines = get_lines(parts, indent) | ||
712 | + if len(lines) > 1: | ||
713 | + lines = [] | ||
714 | + lines.extend(get_lines(opt_parts, indent)) | ||
715 | + lines.extend(get_lines(pos_parts, indent)) | ||
716 | + lines = [prog] + lines | ||
717 | + | ||
718 | + # join lines into usage | ||
719 | + usage = '\n'.join(lines) | ||
720 | + | ||
721 | + # prefix with 'usage:' | ||
722 | + return '%s%s\n\n' % (prefix, usage) | ||
723 | + | ||
724 | + def _format_actions_usage(self, actions, groups): | ||
725 | + # find group indices and identify actions in groups | ||
726 | + group_actions = set() | ||
727 | + inserts = {} | ||
728 | + for group in groups: | ||
729 | + try: | ||
730 | + start = actions.index(group._group_actions[0]) | ||
731 | + except ValueError: | ||
732 | + continue | ||
733 | + else: | ||
734 | + end = start + len(group._group_actions) | ||
735 | + if actions[start:end] == group._group_actions: | ||
736 | + for action in group._group_actions: | ||
737 | + group_actions.add(action) | ||
738 | + if not group.required: | ||
739 | + if start in inserts: | ||
740 | + inserts[start] += ' [' | ||
741 | + else: | ||
742 | + inserts[start] = '[' | ||
743 | + inserts[end] = ']' | ||
744 | + else: | ||
745 | + if start in inserts: | ||
746 | + inserts[start] += ' (' | ||
747 | + else: | ||
748 | + inserts[start] = '(' | ||
749 | + inserts[end] = ')' | ||
750 | + for i in range(start + 1, end): | ||
751 | + inserts[i] = '|' | ||
752 | + | ||
753 | + # collect all actions format strings | ||
754 | + parts = [] | ||
755 | + for i, action in enumerate(actions): | ||
756 | + | ||
757 | + # suppressed arguments are marked with None | ||
758 | + # remove | separators for suppressed arguments | ||
759 | + if action.help is SUPPRESS: | ||
760 | + parts.append(None) | ||
761 | + if inserts.get(i) == '|': | ||
762 | + inserts.pop(i) | ||
763 | + elif inserts.get(i + 1) == '|': | ||
764 | + inserts.pop(i + 1) | ||
765 | + | ||
766 | + # produce all arg strings | ||
767 | + elif not action.option_strings: | ||
768 | + part = self._format_args(action, action.dest) | ||
769 | + | ||
770 | + # if it's in a group, strip the outer [] | ||
771 | + if action in group_actions: | ||
772 | + if part[0] == '[' and part[-1] == ']': | ||
773 | + part = part[1:-1] | ||
774 | + | ||
775 | + # add the action string to the list | ||
776 | + parts.append(part) | ||
777 | + | ||
778 | + # produce the first way to invoke the option in brackets | ||
779 | + else: | ||
780 | + option_string = action.option_strings[0] | ||
781 | + | ||
782 | + # if the Optional doesn't take a value, format is: | ||
783 | + # -s or --long | ||
784 | + if action.nargs == 0: | ||
785 | + part = '%s' % option_string | ||
786 | + | ||
787 | + # if the Optional takes a value, format is: | ||
788 | + # -s ARGS or --long ARGS | ||
789 | + else: | ||
790 | + default = action.dest.upper() | ||
791 | + args_string = self._format_args(action, default) | ||
792 | + part = '%s %s' % (option_string, args_string) | ||
793 | + | ||
794 | + # make it look optional if it's not required or in a group | ||
795 | + if not action.required and action not in group_actions: | ||
796 | + part = '[%s]' % part | ||
797 | + | ||
798 | + # add the action string to the list | ||
799 | + parts.append(part) | ||
800 | + | ||
801 | + # insert things at the necessary indices | ||
802 | + for i in sorted(inserts, reverse=True): | ||
803 | + parts[i:i] = [inserts[i]] | ||
804 | + | ||
805 | + # join all the action items with spaces | ||
806 | + text = ' '.join([item for item in parts if item is not None]) | ||
807 | + | ||
808 | + # clean up separators for mutually exclusive groups | ||
809 | + open = r'[\[(]' | ||
810 | + close = r'[\])]' | ||
811 | + text = _re.sub(r'(%s) ' % open, r'\1', text) | ||
812 | + text = _re.sub(r' (%s)' % close, r'\1', text) | ||
813 | + text = _re.sub(r'%s *%s' % (open, close), r'', text) | ||
814 | + text = _re.sub(r'\(([^|]*)\)', r'\1', text) | ||
815 | + text = text.strip() | ||
816 | + | ||
817 | + # return the text | ||
818 | + return text | ||
819 | + | ||
820 | + def _format_text(self, text): | ||
821 | + if '%(prog)' in text: | ||
822 | + text = text % dict(prog=self._prog) | ||
823 | + text_width = self._width - self._current_indent | ||
824 | + indent = ' ' * self._current_indent | ||
825 | + return self._fill_text(text, text_width, indent) + '\n\n' | ||
826 | + | ||
827 | + def _format_action(self, action): | ||
828 | + # determine the required width and the entry label | ||
829 | + help_position = min(self._action_max_length + 2, | ||
830 | + self._max_help_position) | ||
831 | + help_width = self._width - help_position | ||
832 | + action_width = help_position - self._current_indent - 2 | ||
833 | + action_header = self._format_action_invocation(action) | ||
834 | + | ||
835 | + # ho nelp; start on same line and add a final newline | ||
836 | + if not action.help: | ||
837 | + tup = self._current_indent, '', action_header | ||
838 | + action_header = '%*s%s\n' % tup | ||
839 | + | ||
840 | + # short action name; start on the same line and pad two spaces | ||
841 | + elif len(action_header) <= action_width: | ||
842 | + tup = self._current_indent, '', action_width, action_header | ||
843 | + action_header = '%*s%-*s ' % tup | ||
844 | + indent_first = 0 | ||
845 | + | ||
846 | + # long action name; start on the next line | ||
847 | + else: | ||
848 | + tup = self._current_indent, '', action_header | ||
849 | + action_header = '%*s%s\n' % tup | ||
850 | + indent_first = help_position | ||
851 | + | ||
852 | + # collect the pieces of the action help | ||
853 | + parts = [action_header] | ||
854 | + | ||
855 | + # if there was help for the action, add lines of help text | ||
856 | + if action.help: | ||
857 | + help_text = self._expand_help(action) | ||
858 | + help_lines = self._split_lines(help_text, help_width) | ||
859 | + parts.append('%*s%s\n' % (indent_first, '', help_lines[0])) | ||
860 | + for line in help_lines[1:]: | ||
861 | + parts.append('%*s%s\n' % (help_position, '', line)) | ||
862 | + | ||
863 | + # or add a newline if the description doesn't end with one | ||
864 | + elif not action_header.endswith('\n'): | ||
865 | + parts.append('\n') | ||
866 | + | ||
867 | + # if there are any sub-actions, add their help as well | ||
868 | + for subaction in self._iter_indented_subactions(action): | ||
869 | + parts.append(self._format_action(subaction)) | ||
870 | + | ||
871 | + # return a single string | ||
872 | + return self._join_parts(parts) | ||
873 | + | ||
874 | + def _format_action_invocation(self, action): | ||
875 | + if not action.option_strings: | ||
876 | + metavar, = self._metavar_formatter(action, action.dest)(1) | ||
877 | + return metavar | ||
878 | + | ||
879 | + else: | ||
880 | + parts = [] | ||
881 | + | ||
882 | + # if the Optional doesn't take a value, format is: | ||
883 | + # -s, --long | ||
884 | + if action.nargs == 0: | ||
885 | + parts.extend(action.option_strings) | ||
886 | + | ||
887 | + # if the Optional takes a value, format is: | ||
888 | + # -s ARGS, --long ARGS | ||
889 | + else: | ||
890 | + default = action.dest.upper() | ||
891 | + args_string = self._format_args(action, default) | ||
892 | + for option_string in action.option_strings: | ||
893 | + parts.append('%s %s' % (option_string, args_string)) | ||
894 | + | ||
895 | + return ', '.join(parts) | ||
896 | + | ||
897 | + def _metavar_formatter(self, action, default_metavar): | ||
898 | + if action.metavar is not None: | ||
899 | + result = action.metavar | ||
900 | + elif action.choices is not None: | ||
901 | + choice_strs = [str(choice) for choice in action.choices] | ||
902 | + result = '{%s}' % ','.join(choice_strs) | ||
903 | + else: | ||
904 | + result = default_metavar | ||
905 | + | ||
906 | + def format(tuple_size): | ||
907 | + if isinstance(result, tuple): | ||
908 | + return result | ||
909 | + else: | ||
910 | + return (result, ) * tuple_size | ||
911 | + return format | ||
912 | + | ||
913 | + def _format_args(self, action, default_metavar): | ||
914 | + get_metavar = self._metavar_formatter(action, default_metavar) | ||
915 | + if action.nargs is None: | ||
916 | + result = '%s' % get_metavar(1) | ||
917 | + elif action.nargs == OPTIONAL: | ||
918 | + result = '[%s]' % get_metavar(1) | ||
919 | + elif action.nargs == ZERO_OR_MORE: | ||
920 | + result = '[%s [%s ...]]' % get_metavar(2) | ||
921 | + elif action.nargs == ONE_OR_MORE: | ||
922 | + result = '%s [%s ...]' % get_metavar(2) | ||
923 | + elif action.nargs == REMAINDER: | ||
924 | + result = '...' | ||
925 | + elif action.nargs == PARSER: | ||
926 | + result = '%s ...' % get_metavar(1) | ||
927 | + else: | ||
928 | + formats = ['%s' for _ in range(action.nargs)] | ||
929 | + result = ' '.join(formats) % get_metavar(action.nargs) | ||
930 | + return result | ||
931 | + | ||
932 | + def _expand_help(self, action): | ||
933 | + params = dict(vars(action), prog=self._prog) | ||
934 | + for name in list(params): | ||
935 | + if params[name] is SUPPRESS: | ||
936 | + del params[name] | ||
937 | + for name in list(params): | ||
938 | + if hasattr(params[name], '__name__'): | ||
939 | + params[name] = params[name].__name__ | ||
940 | + if params.get('choices') is not None: | ||
941 | + choices_str = ', '.join([str(c) for c in params['choices']]) | ||
942 | + params['choices'] = choices_str | ||
943 | + return self._get_help_string(action) % params | ||
944 | + | ||
945 | + def _iter_indented_subactions(self, action): | ||
946 | + try: | ||
947 | + get_subactions = action._get_subactions | ||
948 | + except AttributeError: | ||
949 | + pass | ||
950 | + else: | ||
951 | + self._indent() | ||
952 | + for subaction in get_subactions(): | ||
953 | + yield subaction | ||
954 | + self._dedent() | ||
955 | + | ||
956 | + def _split_lines(self, text, width): | ||
957 | + text = self._whitespace_matcher.sub(' ', text).strip() | ||
958 | + return _textwrap.wrap(text, width) | ||
959 | + | ||
960 | + def _fill_text(self, text, width, indent): | ||
961 | + text = self._whitespace_matcher.sub(' ', text).strip() | ||
962 | + return _textwrap.fill(text, width, initial_indent=indent, | ||
963 | + subsequent_indent=indent) | ||
964 | + | ||
965 | + def _get_help_string(self, action): | ||
966 | + return action.help | ||
967 | + | ||
968 | + | ||
969 | +class RawDescriptionHelpFormatter(HelpFormatter): | ||
970 | + """Help message formatter which retains any formatting in descriptions. | ||
971 | + | ||
972 | + Only the name of this class is considered a public API. All the methods | ||
973 | + provided by the class are considered an implementation detail. | ||
974 | + """ | ||
975 | + | ||
976 | + def _fill_text(self, text, width, indent): | ||
977 | + return ''.join([indent + line for line in text.splitlines(True)]) | ||
978 | + | ||
979 | + | ||
980 | +class RawTextHelpFormatter(RawDescriptionHelpFormatter): | ||
981 | + """Help message formatter which retains formatting of all help text. | ||
982 | + | ||
983 | + Only the name of this class is considered a public API. All the methods | ||
984 | + provided by the class are considered an implementation detail. | ||
985 | + """ | ||
986 | + | ||
987 | + def _split_lines(self, text, width): | ||
988 | + return text.splitlines() | ||
989 | + | ||
990 | + | ||
991 | +class ArgumentDefaultsHelpFormatter(HelpFormatter): | ||
992 | + """Help message formatter which adds default values to argument help. | ||
993 | + | ||
994 | + Only the name of this class is considered a public API. All the methods | ||
995 | + provided by the class are considered an implementation detail. | ||
996 | + """ | ||
997 | + | ||
998 | + def _get_help_string(self, action): | ||
999 | + help = action.help | ||
1000 | + if '%(default)' not in action.help: | ||
1001 | + if action.default is not SUPPRESS: | ||
1002 | + defaulting_nargs = [OPTIONAL, ZERO_OR_MORE] | ||
1003 | + if action.option_strings or action.nargs in defaulting_nargs: | ||
1004 | + help += ' (default: %(default)s)' | ||
1005 | + return help | ||
1006 | + | ||
1007 | + | ||
1008 | +# ===================== | ||
1009 | +# Options and Arguments | ||
1010 | +# ===================== | ||
1011 | + | ||
1012 | +def _get_action_name(argument): | ||
1013 | + if argument is None: | ||
1014 | + return None | ||
1015 | + elif argument.option_strings: | ||
1016 | + return '/'.join(argument.option_strings) | ||
1017 | + elif argument.metavar not in (None, SUPPRESS): | ||
1018 | + return argument.metavar | ||
1019 | + elif argument.dest not in (None, SUPPRESS): | ||
1020 | + return argument.dest | ||
1021 | + else: | ||
1022 | + return None | ||
1023 | + | ||
1024 | + | ||
1025 | +class ArgumentError(Exception): | ||
1026 | + """An error from creating or using an argument (optional or positional). | ||
1027 | + | ||
1028 | + The string value of this exception is the message, augmented with | ||
1029 | + information about the argument that caused it. | ||
1030 | + """ | ||
1031 | + | ||
1032 | + def __init__(self, argument, message): | ||
1033 | + self.argument_name = _get_action_name(argument) | ||
1034 | + self.message = message | ||
1035 | + | ||
1036 | + def __str__(self): | ||
1037 | + if self.argument_name is None: | ||
1038 | + format = '%(message)s' | ||
1039 | + else: | ||
1040 | + format = 'argument %(argument_name)s: %(message)s' | ||
1041 | + return format % dict(message=self.message, | ||
1042 | + argument_name=self.argument_name) | ||
1043 | + | ||
1044 | + | ||
1045 | +class ArgumentTypeError(Exception): | ||
1046 | + """An error from trying to convert a command line string to a type.""" | ||
1047 | + pass | ||
1048 | + | ||
1049 | + | ||
1050 | +# ============== | ||
1051 | +# Action classes | ||
1052 | +# ============== | ||
1053 | + | ||
1054 | +class Action(_AttributeHolder): | ||
1055 | + """Information about how to convert command line strings to Python objects. | ||
1056 | + | ||
1057 | + Action objects are used by an ArgumentParser to represent the information | ||
1058 | + needed to parse a single argument from one or more strings from the | ||
1059 | + command line. The keyword arguments to the Action constructor are also | ||
1060 | + all attributes of Action instances. | ||
1061 | + | ||
1062 | + Keyword Arguments: | ||
1063 | + | ||
1064 | + - option_strings -- A list of command-line option strings which | ||
1065 | + should be associated with this action. | ||
1066 | + | ||
1067 | + - dest -- The name of the attribute to hold the created object(s) | ||
1068 | + | ||
1069 | + - nargs -- The number of command-line arguments that should be | ||
1070 | + consumed. By default, one argument will be consumed and a single | ||
1071 | + value will be produced. Other values include: | ||
1072 | + - N (an integer) consumes N arguments (and produces a list) | ||
1073 | + - '?' consumes zero or one arguments | ||
1074 | + - '*' consumes zero or more arguments (and produces a list) | ||
1075 | + - '+' consumes one or more arguments (and produces a list) | ||
1076 | + Note that the difference between the default and nargs=1 is that | ||
1077 | + with the default, a single value will be produced, while with | ||
1078 | + nargs=1, a list containing a single value will be produced. | ||
1079 | + | ||
1080 | + - const -- The value to be produced if the option is specified and the | ||
1081 | + option uses an action that takes no values. | ||
1082 | + | ||
1083 | + - default -- The value to be produced if the option is not specified. | ||
1084 | + | ||
1085 | + - type -- The type which the command-line arguments should be converted | ||
1086 | + to, should be one of 'string', 'int', 'float', 'complex' or a | ||
1087 | + callable object that accepts a single string argument. If None, | ||
1088 | + 'string' is assumed. | ||
1089 | + | ||
1090 | + - choices -- A container of values that should be allowed. If not None, | ||
1091 | + after a command-line argument has been converted to the appropriate | ||
1092 | + type, an exception will be raised if it is not a member of this | ||
1093 | + collection. | ||
1094 | + | ||
1095 | + - required -- True if the action must always be specified at the | ||
1096 | + command line. This is only meaningful for optional command-line | ||
1097 | + arguments. | ||
1098 | + | ||
1099 | + - help -- The help string describing the argument. | ||
1100 | + | ||
1101 | + - metavar -- The name to be used for the option's argument with the | ||
1102 | + help string. If None, the 'dest' value will be used as the name. | ||
1103 | + """ | ||
1104 | + | ||
1105 | + def __init__(self, | ||
1106 | + option_strings, | ||
1107 | + dest, | ||
1108 | + nargs=None, | ||
1109 | + const=None, | ||
1110 | + default=None, | ||
1111 | + type=None, | ||
1112 | + choices=None, | ||
1113 | + required=False, | ||
1114 | + help=None, | ||
1115 | + metavar=None): | ||
1116 | + self.option_strings = option_strings | ||
1117 | + self.dest = dest | ||
1118 | + self.nargs = nargs | ||
1119 | + self.const = const | ||
1120 | + self.default = default | ||
1121 | + self.type = type | ||
1122 | + self.choices = choices | ||
1123 | + self.required = required | ||
1124 | + self.help = help | ||
1125 | + self.metavar = metavar | ||
1126 | + | ||
1127 | + def _get_kwargs(self): | ||
1128 | + names = [ | ||
1129 | + 'option_strings', | ||
1130 | + 'dest', | ||
1131 | + 'nargs', | ||
1132 | + 'const', | ||
1133 | + 'default', | ||
1134 | + 'type', | ||
1135 | + 'choices', | ||
1136 | + 'help', | ||
1137 | + 'metavar', | ||
1138 | + ] | ||
1139 | + return [(name, getattr(self, name)) for name in names] | ||
1140 | + | ||
1141 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1142 | + raise NotImplementedError(_('.__call__() not defined')) | ||
1143 | + | ||
1144 | + | ||
1145 | +class _StoreAction(Action): | ||
1146 | + | ||
1147 | + def __init__(self, | ||
1148 | + option_strings, | ||
1149 | + dest, | ||
1150 | + nargs=None, | ||
1151 | + const=None, | ||
1152 | + default=None, | ||
1153 | + type=None, | ||
1154 | + choices=None, | ||
1155 | + required=False, | ||
1156 | + help=None, | ||
1157 | + metavar=None): | ||
1158 | + if nargs == 0: | ||
1159 | + raise ValueError('nargs for store actions must be > 0; if you ' | ||
1160 | + 'have nothing to store, actions such as store ' | ||
1161 | + 'true or store const may be more appropriate') | ||
1162 | + if const is not None and nargs != OPTIONAL: | ||
1163 | + raise ValueError('nargs must be %r to supply const' % OPTIONAL) | ||
1164 | + super(_StoreAction, self).__init__( | ||
1165 | + option_strings=option_strings, | ||
1166 | + dest=dest, | ||
1167 | + nargs=nargs, | ||
1168 | + const=const, | ||
1169 | + default=default, | ||
1170 | + type=type, | ||
1171 | + choices=choices, | ||
1172 | + required=required, | ||
1173 | + help=help, | ||
1174 | + metavar=metavar) | ||
1175 | + | ||
1176 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1177 | + setattr(namespace, self.dest, values) | ||
1178 | + | ||
1179 | + | ||
1180 | +class _StoreConstAction(Action): | ||
1181 | + | ||
1182 | + def __init__(self, | ||
1183 | + option_strings, | ||
1184 | + dest, | ||
1185 | + const, | ||
1186 | + default=None, | ||
1187 | + required=False, | ||
1188 | + help=None, | ||
1189 | + metavar=None): | ||
1190 | + super(_StoreConstAction, self).__init__( | ||
1191 | + option_strings=option_strings, | ||
1192 | + dest=dest, | ||
1193 | + nargs=0, | ||
1194 | + const=const, | ||
1195 | + default=default, | ||
1196 | + required=required, | ||
1197 | + help=help) | ||
1198 | + | ||
1199 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1200 | + setattr(namespace, self.dest, self.const) | ||
1201 | + | ||
1202 | + | ||
1203 | +class _StoreTrueAction(_StoreConstAction): | ||
1204 | + | ||
1205 | + def __init__(self, | ||
1206 | + option_strings, | ||
1207 | + dest, | ||
1208 | + default=False, | ||
1209 | + required=False, | ||
1210 | + help=None): | ||
1211 | + super(_StoreTrueAction, self).__init__( | ||
1212 | + option_strings=option_strings, | ||
1213 | + dest=dest, | ||
1214 | + const=True, | ||
1215 | + default=default, | ||
1216 | + required=required, | ||
1217 | + help=help) | ||
1218 | + | ||
1219 | + | ||
1220 | +class _StoreFalseAction(_StoreConstAction): | ||
1221 | + | ||
1222 | + def __init__(self, | ||
1223 | + option_strings, | ||
1224 | + dest, | ||
1225 | + default=True, | ||
1226 | + required=False, | ||
1227 | + help=None): | ||
1228 | + super(_StoreFalseAction, self).__init__( | ||
1229 | + option_strings=option_strings, | ||
1230 | + dest=dest, | ||
1231 | + const=False, | ||
1232 | + default=default, | ||
1233 | + required=required, | ||
1234 | + help=help) | ||
1235 | + | ||
1236 | + | ||
1237 | +class _AppendAction(Action): | ||
1238 | + | ||
1239 | + def __init__(self, | ||
1240 | + option_strings, | ||
1241 | + dest, | ||
1242 | + nargs=None, | ||
1243 | + const=None, | ||
1244 | + default=None, | ||
1245 | + type=None, | ||
1246 | + choices=None, | ||
1247 | + required=False, | ||
1248 | + help=None, | ||
1249 | + metavar=None): | ||
1250 | + if nargs == 0: | ||
1251 | + raise ValueError('nargs for append actions must be > 0; if arg ' | ||
1252 | + 'strings are not supplying the value to append, ' | ||
1253 | + 'the append const action may be more appropriate') | ||
1254 | + if const is not None and nargs != OPTIONAL: | ||
1255 | + raise ValueError('nargs must be %r to supply const' % OPTIONAL) | ||
1256 | + super(_AppendAction, self).__init__( | ||
1257 | + option_strings=option_strings, | ||
1258 | + dest=dest, | ||
1259 | + nargs=nargs, | ||
1260 | + const=const, | ||
1261 | + default=default, | ||
1262 | + type=type, | ||
1263 | + choices=choices, | ||
1264 | + required=required, | ||
1265 | + help=help, | ||
1266 | + metavar=metavar) | ||
1267 | + | ||
1268 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1269 | + items = _copy.copy(_ensure_value(namespace, self.dest, [])) | ||
1270 | + items.append(values) | ||
1271 | + setattr(namespace, self.dest, items) | ||
1272 | + | ||
1273 | + | ||
1274 | +class _AppendConstAction(Action): | ||
1275 | + | ||
1276 | + def __init__(self, | ||
1277 | + option_strings, | ||
1278 | + dest, | ||
1279 | + const, | ||
1280 | + default=None, | ||
1281 | + required=False, | ||
1282 | + help=None, | ||
1283 | + metavar=None): | ||
1284 | + super(_AppendConstAction, self).__init__( | ||
1285 | + option_strings=option_strings, | ||
1286 | + dest=dest, | ||
1287 | + nargs=0, | ||
1288 | + const=const, | ||
1289 | + default=default, | ||
1290 | + required=required, | ||
1291 | + help=help, | ||
1292 | + metavar=metavar) | ||
1293 | + | ||
1294 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1295 | + items = _copy.copy(_ensure_value(namespace, self.dest, [])) | ||
1296 | + items.append(self.const) | ||
1297 | + setattr(namespace, self.dest, items) | ||
1298 | + | ||
1299 | + | ||
1300 | +class _CountAction(Action): | ||
1301 | + | ||
1302 | + def __init__(self, | ||
1303 | + option_strings, | ||
1304 | + dest, | ||
1305 | + default=None, | ||
1306 | + required=False, | ||
1307 | + help=None): | ||
1308 | + super(_CountAction, self).__init__( | ||
1309 | + option_strings=option_strings, | ||
1310 | + dest=dest, | ||
1311 | + nargs=0, | ||
1312 | + default=default, | ||
1313 | + required=required, | ||
1314 | + help=help) | ||
1315 | + | ||
1316 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1317 | + new_count = _ensure_value(namespace, self.dest, 0) + 1 | ||
1318 | + setattr(namespace, self.dest, new_count) | ||
1319 | + | ||
1320 | + | ||
1321 | +class _HelpAction(Action): | ||
1322 | + | ||
1323 | + def __init__(self, | ||
1324 | + option_strings, | ||
1325 | + dest=SUPPRESS, | ||
1326 | + default=SUPPRESS, | ||
1327 | + help=None): | ||
1328 | + super(_HelpAction, self).__init__( | ||
1329 | + option_strings=option_strings, | ||
1330 | + dest=dest, | ||
1331 | + default=default, | ||
1332 | + nargs=0, | ||
1333 | + help=help) | ||
1334 | + | ||
1335 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1336 | + parser.print_help() | ||
1337 | + parser.exit() | ||
1338 | + | ||
1339 | + | ||
1340 | +class _VersionAction(Action): | ||
1341 | + | ||
1342 | + def __init__(self, | ||
1343 | + option_strings, | ||
1344 | + version=None, | ||
1345 | + dest=SUPPRESS, | ||
1346 | + default=SUPPRESS, | ||
1347 | + help="show program's version number and exit"): | ||
1348 | + super(_VersionAction, self).__init__( | ||
1349 | + option_strings=option_strings, | ||
1350 | + dest=dest, | ||
1351 | + default=default, | ||
1352 | + nargs=0, | ||
1353 | + help=help) | ||
1354 | + self.version = version | ||
1355 | + | ||
1356 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1357 | + version = self.version | ||
1358 | + if version is None: | ||
1359 | + version = parser.version | ||
1360 | + formatter = parser._get_formatter() | ||
1361 | + formatter.add_text(version) | ||
1362 | + parser.exit(message=formatter.format_help()) | ||
1363 | + | ||
1364 | + | ||
1365 | +class _SubParsersAction(Action): | ||
1366 | + | ||
1367 | + class _ChoicesPseudoAction(Action): | ||
1368 | + | ||
1369 | + def __init__(self, name, aliases, help): | ||
1370 | + metavar = dest = name | ||
1371 | + if aliases: | ||
1372 | + metavar += ' (%s)' % ', '.join(aliases) | ||
1373 | + sup = super(_SubParsersAction._ChoicesPseudoAction, self) | ||
1374 | + sup.__init__(option_strings=[], dest=dest, help=help, | ||
1375 | + metavar=metavar) | ||
1376 | + | ||
1377 | + def __init__(self, | ||
1378 | + option_strings, | ||
1379 | + prog, | ||
1380 | + parser_class, | ||
1381 | + dest=SUPPRESS, | ||
1382 | + help=None, | ||
1383 | + metavar=None): | ||
1384 | + | ||
1385 | + self._prog_prefix = prog | ||
1386 | + self._parser_class = parser_class | ||
1387 | + self._name_parser_map = {} | ||
1388 | + self._choices_actions = [] | ||
1389 | + | ||
1390 | + super(_SubParsersAction, self).__init__( | ||
1391 | + option_strings=option_strings, | ||
1392 | + dest=dest, | ||
1393 | + nargs=PARSER, | ||
1394 | + choices=self._name_parser_map, | ||
1395 | + help=help, | ||
1396 | + metavar=metavar) | ||
1397 | + | ||
1398 | + def add_parser(self, name, **kwargs): | ||
1399 | + # set prog from the existing prefix | ||
1400 | + if kwargs.get('prog') is None: | ||
1401 | + kwargs['prog'] = '%s %s' % (self._prog_prefix, name) | ||
1402 | + | ||
1403 | + aliases = kwargs.pop('aliases', ()) | ||
1404 | + | ||
1405 | + # create a pseudo-action to hold the choice help | ||
1406 | + if 'help' in kwargs: | ||
1407 | + help = kwargs.pop('help') | ||
1408 | + choice_action = self._ChoicesPseudoAction(name, aliases, help) | ||
1409 | + self._choices_actions.append(choice_action) | ||
1410 | + | ||
1411 | + # create the parser and add it to the map | ||
1412 | + parser = self._parser_class(**kwargs) | ||
1413 | + self._name_parser_map[name] = parser | ||
1414 | + | ||
1415 | + # make parser available under aliases also | ||
1416 | + for alias in aliases: | ||
1417 | + self._name_parser_map[alias] = parser | ||
1418 | + | ||
1419 | + return parser | ||
1420 | + | ||
1421 | + def _get_subactions(self): | ||
1422 | + return self._choices_actions | ||
1423 | + | ||
1424 | + def __call__(self, parser, namespace, values, option_string=None): | ||
1425 | + parser_name = values[0] | ||
1426 | + arg_strings = values[1:] | ||
1427 | + | ||
1428 | + # set the parser name if requested | ||
1429 | + if self.dest is not SUPPRESS: | ||
1430 | + setattr(namespace, self.dest, parser_name) | ||
1431 | + | ||
1432 | + # select the parser | ||
1433 | + try: | ||
1434 | + parser = self._name_parser_map[parser_name] | ||
1435 | + except KeyError: | ||
1436 | + tup = parser_name, ', '.join(self._name_parser_map) | ||
1437 | + msg = _('unknown parser %r (choices: %s)' % tup) | ||
1438 | + raise ArgumentError(self, msg) | ||
1439 | + | ||
1440 | + # parse all the remaining options into the namespace | ||
1441 | + # store any unrecognized options on the object, so that the top | ||
1442 | + # level parser can decide what to do with them | ||
1443 | + namespace, arg_strings = parser.parse_known_args(arg_strings, namespace) | ||
1444 | + if arg_strings: | ||
1445 | + vars(namespace).setdefault(_UNRECOGNIZED_ARGS_ATTR, []) | ||
1446 | + getattr(namespace, _UNRECOGNIZED_ARGS_ATTR).extend(arg_strings) | ||
1447 | + | ||
1448 | + | ||
1449 | +# ============== | ||
1450 | +# Type classes | ||
1451 | +# ============== | ||
1452 | + | ||
1453 | +class FileType(object): | ||
1454 | + """Factory for creating file object types | ||
1455 | + | ||
1456 | + Instances of FileType are typically passed as type= arguments to the | ||
1457 | + ArgumentParser add_argument() method. | ||
1458 | + | ||
1459 | + Keyword Arguments: | ||
1460 | + - mode -- A string indicating how the file is to be opened. Accepts the | ||
1461 | + same values as the builtin open() function. | ||
1462 | + - bufsize -- The file's desired buffer size. Accepts the same values as | ||
1463 | + the builtin open() function. | ||
1464 | + """ | ||
1465 | + | ||
1466 | + def __init__(self, mode='r', bufsize=None): | ||
1467 | + self._mode = mode | ||
1468 | + self._bufsize = bufsize | ||
1469 | + | ||
1470 | + def __call__(self, string): | ||
1471 | + # the special argument "-" means sys.std{in,out} | ||
1472 | + if string == '-': | ||
1473 | + if 'r' in self._mode: | ||
1474 | + return _sys.stdin | ||
1475 | + elif 'w' in self._mode: | ||
1476 | + return _sys.stdout | ||
1477 | + else: | ||
1478 | + msg = _('argument "-" with mode %r' % self._mode) | ||
1479 | + raise ValueError(msg) | ||
1480 | + | ||
1481 | + try: | ||
1482 | + # all other arguments are used as file names | ||
1483 | + if self._bufsize: | ||
1484 | + return open(string, self._mode, self._bufsize) | ||
1485 | + else: | ||
1486 | + return open(string, self._mode) | ||
1487 | + except IOError: | ||
1488 | + err = _sys.exc_info()[1] | ||
1489 | + message = _("can't open '%s': %s") | ||
1490 | + raise ArgumentTypeError(message % (string, err)) | ||
1491 | + | ||
1492 | + def __repr__(self): | ||
1493 | + args = [self._mode, self._bufsize] | ||
1494 | + args_str = ', '.join([repr(arg) for arg in args if arg is not None]) | ||
1495 | + return '%s(%s)' % (type(self).__name__, args_str) | ||
1496 | + | ||
1497 | +# =========================== | ||
1498 | +# Optional and Positional Parsing | ||
1499 | +# =========================== | ||
1500 | + | ||
1501 | +class Namespace(_AttributeHolder): | ||
1502 | + """Simple object for storing attributes. | ||
1503 | + | ||
1504 | + Implements equality by attribute names and values, and provides a simple | ||
1505 | + string representation. | ||
1506 | + """ | ||
1507 | + | ||
1508 | + def __init__(self, **kwargs): | ||
1509 | + for name in kwargs: | ||
1510 | + setattr(self, name, kwargs[name]) | ||
1511 | + | ||
1512 | + __hash__ = None | ||
1513 | + | ||
1514 | + def __eq__(self, other): | ||
1515 | + return vars(self) == vars(other) | ||
1516 | + | ||
1517 | + def __ne__(self, other): | ||
1518 | + return not (self == other) | ||
1519 | + | ||
1520 | + def __contains__(self, key): | ||
1521 | + return key in self.__dict__ | ||
1522 | + | ||
1523 | + | ||
1524 | +class _ActionsContainer(object): | ||
1525 | + | ||
1526 | + def __init__(self, | ||
1527 | + description, | ||
1528 | + prefix_chars, | ||
1529 | + argument_default, | ||
1530 | + conflict_handler): | ||
1531 | + super(_ActionsContainer, self).__init__() | ||
1532 | + | ||
1533 | + self.description = description | ||
1534 | + self.argument_default = argument_default | ||
1535 | + self.prefix_chars = prefix_chars | ||
1536 | + self.conflict_handler = conflict_handler | ||
1537 | + | ||
1538 | + # set up registries | ||
1539 | + self._registries = {} | ||
1540 | + | ||
1541 | + # register actions | ||
1542 | + self.register('action', None, _StoreAction) | ||
1543 | + self.register('action', 'store', _StoreAction) | ||
1544 | + self.register('action', 'store_const', _StoreConstAction) | ||
1545 | + self.register('action', 'store_true', _StoreTrueAction) | ||
1546 | + self.register('action', 'store_false', _StoreFalseAction) | ||
1547 | + self.register('action', 'append', _AppendAction) | ||
1548 | + self.register('action', 'append_const', _AppendConstAction) | ||
1549 | + self.register('action', 'count', _CountAction) | ||
1550 | + self.register('action', 'help', _HelpAction) | ||
1551 | + self.register('action', 'version', _VersionAction) | ||
1552 | + self.register('action', 'parsers', _SubParsersAction) | ||
1553 | + | ||
1554 | + # raise an exception if the conflict handler is invalid | ||
1555 | + self._get_handler() | ||
1556 | + | ||
1557 | + # action storage | ||
1558 | + self._actions = [] | ||
1559 | + self._option_string_actions = {} | ||
1560 | + | ||
1561 | + # groups | ||
1562 | + self._action_groups = [] | ||
1563 | + self._mutually_exclusive_groups = [] | ||
1564 | + | ||
1565 | + # defaults storage | ||
1566 | + self._defaults = {} | ||
1567 | + | ||
1568 | + # determines whether an "option" looks like a negative number | ||
1569 | + self._negative_number_matcher = _re.compile(r'^-\d+$|^-\d*\.\d+$') | ||
1570 | + | ||
1571 | + # whether or not there are any optionals that look like negative | ||
1572 | + # numbers -- uses a list so it can be shared and edited | ||
1573 | + self._has_negative_number_optionals = [] | ||
1574 | + | ||
1575 | + # ==================== | ||
1576 | + # Registration methods | ||
1577 | + # ==================== | ||
1578 | + def register(self, registry_name, value, object): | ||
1579 | + registry = self._registries.setdefault(registry_name, {}) | ||
1580 | + registry[value] = object | ||
1581 | + | ||
1582 | + def _registry_get(self, registry_name, value, default=None): | ||
1583 | + return self._registries[registry_name].get(value, default) | ||
1584 | + | ||
1585 | + # ================================== | ||
1586 | + # Namespace default accessor methods | ||
1587 | + # ================================== | ||
1588 | + def set_defaults(self, **kwargs): | ||
1589 | + self._defaults.update(kwargs) | ||
1590 | + | ||
1591 | + # if these defaults match any existing arguments, replace | ||
1592 | + # the previous default on the object with the new one | ||
1593 | + for action in self._actions: | ||
1594 | + if action.dest in kwargs: | ||
1595 | + action.default = kwargs[action.dest] | ||
1596 | + | ||
1597 | + def get_default(self, dest): | ||
1598 | + for action in self._actions: | ||
1599 | + if action.dest == dest and action.default is not None: | ||
1600 | + return action.default | ||
1601 | + return self._defaults.get(dest, None) | ||
1602 | + | ||
1603 | + | ||
1604 | + # ======================= | ||
1605 | + # Adding argument actions | ||
1606 | + # ======================= | ||
1607 | + def add_argument(self, *args, **kwargs): | ||
1608 | + """ | ||
1609 | + add_argument(dest, ..., name=value, ...) | ||
1610 | + add_argument(option_string, option_string, ..., name=value, ...) | ||
1611 | + """ | ||
1612 | + | ||
1613 | + # if no positional args are supplied or only one is supplied and | ||
1614 | + # it doesn't look like an option string, parse a positional | ||
1615 | + # argument | ||
1616 | + chars = self.prefix_chars | ||
1617 | + if not args or len(args) == 1 and args[0][0] not in chars: | ||
1618 | + if args and 'dest' in kwargs: | ||
1619 | + raise ValueError('dest supplied twice for positional argument') | ||
1620 | + kwargs = self._get_positional_kwargs(*args, **kwargs) | ||
1621 | + | ||
1622 | + # otherwise, we're adding an optional argument | ||
1623 | + else: | ||
1624 | + kwargs = self._get_optional_kwargs(*args, **kwargs) | ||
1625 | + | ||
1626 | + # if no default was supplied, use the parser-level default | ||
1627 | + if 'default' not in kwargs: | ||
1628 | + dest = kwargs['dest'] | ||
1629 | + if dest in self._defaults: | ||
1630 | + kwargs['default'] = self._defaults[dest] | ||
1631 | + elif self.argument_default is not None: | ||
1632 | + kwargs['default'] = self.argument_default | ||
1633 | + | ||
1634 | + # create the action object, and add it to the parser | ||
1635 | + action_class = self._pop_action_class(kwargs) | ||
1636 | + if not _callable(action_class): | ||
1637 | + raise ValueError('unknown action "%s"' % action_class) | ||
1638 | + action = action_class(**kwargs) | ||
1639 | + | ||
1640 | + # raise an error if the action type is not callable | ||
1641 | + type_func = self._registry_get('type', action.type, action.type) | ||
1642 | + if not _callable(type_func): | ||
1643 | + raise ValueError('%r is not callable' % type_func) | ||
1644 | + | ||
1645 | + return self._add_action(action) | ||
1646 | + | ||
1647 | + def add_argument_group(self, *args, **kwargs): | ||
1648 | + group = _ArgumentGroup(self, *args, **kwargs) | ||
1649 | + self._action_groups.append(group) | ||
1650 | + return group | ||
1651 | + | ||
1652 | + def add_mutually_exclusive_group(self, **kwargs): | ||
1653 | + group = _MutuallyExclusiveGroup(self, **kwargs) | ||
1654 | + self._mutually_exclusive_groups.append(group) | ||
1655 | + return group | ||
1656 | + | ||
1657 | + def _add_action(self, action): | ||
1658 | + # resolve any conflicts | ||
1659 | + self._check_conflict(action) | ||
1660 | + | ||
1661 | + # add to actions list | ||
1662 | + self._actions.append(action) | ||
1663 | + action.container = self | ||
1664 | + | ||
1665 | + # index the action by any option strings it has | ||
1666 | + for option_string in action.option_strings: | ||
1667 | + self._option_string_actions[option_string] = action | ||
1668 | + | ||
1669 | + # set the flag if any option strings look like negative numbers | ||
1670 | + for option_string in action.option_strings: | ||
1671 | + if self._negative_number_matcher.match(option_string): | ||
1672 | + if not self._has_negative_number_optionals: | ||
1673 | + self._has_negative_number_optionals.append(True) | ||
1674 | + | ||
1675 | + # return the created action | ||
1676 | + return action | ||
1677 | + | ||
1678 | + def _remove_action(self, action): | ||
1679 | + self._actions.remove(action) | ||
1680 | + | ||
1681 | + def _add_container_actions(self, container): | ||
1682 | + # collect groups by titles | ||
1683 | + title_group_map = {} | ||
1684 | + for group in self._action_groups: | ||
1685 | + if group.title in title_group_map: | ||
1686 | + msg = _('cannot merge actions - two groups are named %r') | ||
1687 | + raise ValueError(msg % (group.title)) | ||
1688 | + title_group_map[group.title] = group | ||
1689 | + | ||
1690 | + # map each action to its group | ||
1691 | + group_map = {} | ||
1692 | + for group in container._action_groups: | ||
1693 | + | ||
1694 | + # if a group with the title exists, use that, otherwise | ||
1695 | + # create a new group matching the container's group | ||
1696 | + if group.title not in title_group_map: | ||
1697 | + title_group_map[group.title] = self.add_argument_group( | ||
1698 | + title=group.title, | ||
1699 | + description=group.description, | ||
1700 | + conflict_handler=group.conflict_handler) | ||
1701 | + | ||
1702 | + # map the actions to their new group | ||
1703 | + for action in group._group_actions: | ||
1704 | + group_map[action] = title_group_map[group.title] | ||
1705 | + | ||
1706 | + # add container's mutually exclusive groups | ||
1707 | + # NOTE: if add_mutually_exclusive_group ever gains title= and | ||
1708 | + # description= then this code will need to be expanded as above | ||
1709 | + for group in container._mutually_exclusive_groups: | ||
1710 | + mutex_group = self.add_mutually_exclusive_group( | ||
1711 | + required=group.required) | ||
1712 | + | ||
1713 | + # map the actions to their new mutex group | ||
1714 | + for action in group._group_actions: | ||
1715 | + group_map[action] = mutex_group | ||
1716 | + | ||
1717 | + # add all actions to this container or their group | ||
1718 | + for action in container._actions: | ||
1719 | + group_map.get(action, self)._add_action(action) | ||
1720 | + | ||
1721 | + def _get_positional_kwargs(self, dest, **kwargs): | ||
1722 | + # make sure required is not specified | ||
1723 | + if 'required' in kwargs: | ||
1724 | + msg = _("'required' is an invalid argument for positionals") | ||
1725 | + raise TypeError(msg) | ||
1726 | + | ||
1727 | + # mark positional arguments as required if at least one is | ||
1728 | + # always required | ||
1729 | + if kwargs.get('nargs') not in [OPTIONAL, ZERO_OR_MORE]: | ||
1730 | + kwargs['required'] = True | ||
1731 | + if kwargs.get('nargs') == ZERO_OR_MORE and 'default' not in kwargs: | ||
1732 | + kwargs['required'] = True | ||
1733 | + | ||
1734 | + # return the keyword arguments with no option strings | ||
1735 | + return dict(kwargs, dest=dest, option_strings=[]) | ||
1736 | + | ||
1737 | + def _get_optional_kwargs(self, *args, **kwargs): | ||
1738 | + # determine short and long option strings | ||
1739 | + option_strings = [] | ||
1740 | + long_option_strings = [] | ||
1741 | + for option_string in args: | ||
1742 | + # error on strings that don't start with an appropriate prefix | ||
1743 | + if not option_string[0] in self.prefix_chars: | ||
1744 | + msg = _('invalid option string %r: ' | ||
1745 | + 'must start with a character %r') | ||
1746 | + tup = option_string, self.prefix_chars | ||
1747 | + raise ValueError(msg % tup) | ||
1748 | + | ||
1749 | + # strings starting with two prefix characters are long options | ||
1750 | + option_strings.append(option_string) | ||
1751 | + if option_string[0] in self.prefix_chars: | ||
1752 | + if len(option_string) > 1: | ||
1753 | + if option_string[1] in self.prefix_chars: | ||
1754 | + long_option_strings.append(option_string) | ||
1755 | + | ||
1756 | + # infer destination, '--foo-bar' -> 'foo_bar' and '-x' -> 'x' | ||
1757 | + dest = kwargs.pop('dest', None) | ||
1758 | + if dest is None: | ||
1759 | + if long_option_strings: | ||
1760 | + dest_option_string = long_option_strings[0] | ||
1761 | + else: | ||
1762 | + dest_option_string = option_strings[0] | ||
1763 | + dest = dest_option_string.lstrip(self.prefix_chars) | ||
1764 | + if not dest: | ||
1765 | + msg = _('dest= is required for options like %r') | ||
1766 | + raise ValueError(msg % option_string) | ||
1767 | + dest = dest.replace('-', '_') | ||
1768 | + | ||
1769 | + # return the updated keyword arguments | ||
1770 | + return dict(kwargs, dest=dest, option_strings=option_strings) | ||
1771 | + | ||
1772 | + def _pop_action_class(self, kwargs, default=None): | ||
1773 | + action = kwargs.pop('action', default) | ||
1774 | + return self._registry_get('action', action, action) | ||
1775 | + | ||
1776 | + def _get_handler(self): | ||
1777 | + # determine function from conflict handler string | ||
1778 | + handler_func_name = '_handle_conflict_%s' % self.conflict_handler | ||
1779 | + try: | ||
1780 | + return getattr(self, handler_func_name) | ||
1781 | + except AttributeError: | ||
1782 | + msg = _('invalid conflict_resolution value: %r') | ||
1783 | + raise ValueError(msg % self.conflict_handler) | ||
1784 | + | ||
1785 | + def _check_conflict(self, action): | ||
1786 | + | ||
1787 | + # find all options that conflict with this option | ||
1788 | + confl_optionals = [] | ||
1789 | + for option_string in action.option_strings: | ||
1790 | + if option_string in self._option_string_actions: | ||
1791 | + confl_optional = self._option_string_actions[option_string] | ||
1792 | + confl_optionals.append((option_string, confl_optional)) | ||
1793 | + | ||
1794 | + # resolve any conflicts | ||
1795 | + if confl_optionals: | ||
1796 | + conflict_handler = self._get_handler() | ||
1797 | + conflict_handler(action, confl_optionals) | ||
1798 | + | ||
1799 | + def _handle_conflict_error(self, action, conflicting_actions): | ||
1800 | + message = _('conflicting option string(s): %s') | ||
1801 | + conflict_string = ', '.join([option_string | ||
1802 | + for option_string, action | ||
1803 | + in conflicting_actions]) | ||
1804 | + raise ArgumentError(action, message % conflict_string) | ||
1805 | + | ||
1806 | + def _handle_conflict_resolve(self, action, conflicting_actions): | ||
1807 | + | ||
1808 | + # remove all conflicting options | ||
1809 | + for option_string, action in conflicting_actions: | ||
1810 | + | ||
1811 | + # remove the conflicting option | ||
1812 | + action.option_strings.remove(option_string) | ||
1813 | + self._option_string_actions.pop(option_string, None) | ||
1814 | + | ||
1815 | + # if the option now has no option string, remove it from the | ||
1816 | + # container holding it | ||
1817 | + if not action.option_strings: | ||
1818 | + action.container._remove_action(action) | ||
1819 | + | ||
1820 | + | ||
1821 | +class _ArgumentGroup(_ActionsContainer): | ||
1822 | + | ||
1823 | + def __init__(self, container, title=None, description=None, **kwargs): | ||
1824 | + # add any missing keyword arguments by checking the container | ||
1825 | + update = kwargs.setdefault | ||
1826 | + update('conflict_handler', container.conflict_handler) | ||
1827 | + update('prefix_chars', container.prefix_chars) | ||
1828 | + update('argument_default', container.argument_default) | ||
1829 | + super_init = super(_ArgumentGroup, self).__init__ | ||
1830 | + super_init(description=description, **kwargs) | ||
1831 | + | ||
1832 | + # group attributes | ||
1833 | + self.title = title | ||
1834 | + self._group_actions = [] | ||
1835 | + | ||
1836 | + # share most attributes with the container | ||
1837 | + self._registries = container._registries | ||
1838 | + self._actions = container._actions | ||
1839 | + self._option_string_actions = container._option_string_actions | ||
1840 | + self._defaults = container._defaults | ||
1841 | + self._has_negative_number_optionals = \ | ||
1842 | + container._has_negative_number_optionals | ||
1843 | + | ||
1844 | + def _add_action(self, action): | ||
1845 | + action = super(_ArgumentGroup, self)._add_action(action) | ||
1846 | + self._group_actions.append(action) | ||
1847 | + return action | ||
1848 | + | ||
1849 | + def _remove_action(self, action): | ||
1850 | + super(_ArgumentGroup, self)._remove_action(action) | ||
1851 | + self._group_actions.remove(action) | ||
1852 | + | ||
1853 | + | ||
1854 | +class _MutuallyExclusiveGroup(_ArgumentGroup): | ||
1855 | + | ||
1856 | + def __init__(self, container, required=False): | ||
1857 | + super(_MutuallyExclusiveGroup, self).__init__(container) | ||
1858 | + self.required = required | ||
1859 | + self._container = container | ||
1860 | + | ||
1861 | + def _add_action(self, action): | ||
1862 | + if action.required: | ||
1863 | + msg = _('mutually exclusive arguments must be optional') | ||
1864 | + raise ValueError(msg) | ||
1865 | + action = self._container._add_action(action) | ||
1866 | + self._group_actions.append(action) | ||
1867 | + return action | ||
1868 | + | ||
1869 | + def _remove_action(self, action): | ||
1870 | + self._container._remove_action(action) | ||
1871 | + self._group_actions.remove(action) | ||
1872 | + | ||
1873 | + | ||
1874 | +class ArgumentParser(_AttributeHolder, _ActionsContainer): | ||
1875 | + """Object for parsing command line strings into Python objects. | ||
1876 | + | ||
1877 | + Keyword Arguments: | ||
1878 | + - prog -- The name of the program (default: sys.argv[0]) | ||
1879 | + - usage -- A usage message (default: auto-generated from arguments) | ||
1880 | + - description -- A description of what the program does | ||
1881 | + - epilog -- Text following the argument descriptions | ||
1882 | + - parents -- Parsers whose arguments should be copied into this one | ||
1883 | + - formatter_class -- HelpFormatter class for printing help messages | ||
1884 | + - prefix_chars -- Characters that prefix optional arguments | ||
1885 | + - fromfile_prefix_chars -- Characters that prefix files containing | ||
1886 | + additional arguments | ||
1887 | + - argument_default -- The default value for all arguments | ||
1888 | + - conflict_handler -- String indicating how to handle conflicts | ||
1889 | + - add_help -- Add a -h/-help option | ||
1890 | + """ | ||
1891 | + | ||
1892 | + def __init__(self, | ||
1893 | + prog=None, | ||
1894 | + usage=None, | ||
1895 | + description=None, | ||
1896 | + epilog=None, | ||
1897 | + version=None, | ||
1898 | + parents=[], | ||
1899 | + formatter_class=HelpFormatter, | ||
1900 | + prefix_chars='-', | ||
1901 | + fromfile_prefix_chars=None, | ||
1902 | + argument_default=None, | ||
1903 | + conflict_handler='error', | ||
1904 | + add_help=True): | ||
1905 | + | ||
1906 | + if version is not None: | ||
1907 | + import warnings | ||
1908 | + warnings.warn( | ||
1909 | + """The "version" argument to ArgumentParser is deprecated. """ | ||
1910 | + """Please use """ | ||
1911 | + """"add_argument(..., action='version', version="N", ...)" """ | ||
1912 | + """instead""", DeprecationWarning) | ||
1913 | + | ||
1914 | + superinit = super(ArgumentParser, self).__init__ | ||
1915 | + superinit(description=description, | ||
1916 | + prefix_chars=prefix_chars, | ||
1917 | + argument_default=argument_default, | ||
1918 | + conflict_handler=conflict_handler) | ||
1919 | + | ||
1920 | + # default setting for prog | ||
1921 | + if prog is None: | ||
1922 | + prog = _os.path.basename(_sys.argv[0]) | ||
1923 | + | ||
1924 | + self.prog = prog | ||
1925 | + self.usage = usage | ||
1926 | + self.epilog = epilog | ||
1927 | + self.version = version | ||
1928 | + self.formatter_class = formatter_class | ||
1929 | + self.fromfile_prefix_chars = fromfile_prefix_chars | ||
1930 | + self.add_help = add_help | ||
1931 | + | ||
1932 | + add_group = self.add_argument_group | ||
1933 | + self._positionals = add_group(_('positional arguments')) | ||
1934 | + self._optionals = add_group(_('optional arguments')) | ||
1935 | + self._subparsers = None | ||
1936 | + | ||
1937 | + # register types | ||
1938 | + def identity(string): | ||
1939 | + return string | ||
1940 | + self.register('type', None, identity) | ||
1941 | + | ||
1942 | + # add help and version arguments if necessary | ||
1943 | + # (using explicit default to override global argument_default) | ||
1944 | + if '-' in prefix_chars: | ||
1945 | + default_prefix = '-' | ||
1946 | + else: | ||
1947 | + default_prefix = prefix_chars[0] | ||
1948 | + if self.add_help: | ||
1949 | + self.add_argument( | ||
1950 | + default_prefix+'h', default_prefix*2+'help', | ||
1951 | + action='help', default=SUPPRESS, | ||
1952 | + help=_('show this help message and exit')) | ||
1953 | + if self.version: | ||
1954 | + self.add_argument( | ||
1955 | + default_prefix+'v', default_prefix*2+'version', | ||
1956 | + action='version', default=SUPPRESS, | ||
1957 | + version=self.version, | ||
1958 | + help=_("show program's version number and exit")) | ||
1959 | + | ||
1960 | + # add parent arguments and defaults | ||
1961 | + for parent in parents: | ||
1962 | + self._add_container_actions(parent) | ||
1963 | + try: | ||
1964 | + defaults = parent._defaults | ||
1965 | + except AttributeError: | ||
1966 | + pass | ||
1967 | + else: | ||
1968 | + self._defaults.update(defaults) | ||
1969 | + | ||
1970 | + # ======================= | ||
1971 | + # Pretty __repr__ methods | ||
1972 | + # ======================= | ||
1973 | + def _get_kwargs(self): | ||
1974 | + names = [ | ||
1975 | + 'prog', | ||
1976 | + 'usage', | ||
1977 | + 'description', | ||
1978 | + 'version', | ||
1979 | + 'formatter_class', | ||
1980 | + 'conflict_handler', | ||
1981 | + 'add_help', | ||
1982 | + ] | ||
1983 | + return [(name, getattr(self, name)) for name in names] | ||
1984 | + | ||
1985 | + # ================================== | ||
1986 | + # Optional/Positional adding methods | ||
1987 | + # ================================== | ||
1988 | + def add_subparsers(self, **kwargs): | ||
1989 | + if self._subparsers is not None: | ||
1990 | + self.error(_('cannot have multiple subparser arguments')) | ||
1991 | + | ||
1992 | + # add the parser class to the arguments if it's not present | ||
1993 | + kwargs.setdefault('parser_class', type(self)) | ||
1994 | + | ||
1995 | + if 'title' in kwargs or 'description' in kwargs: | ||
1996 | + title = _(kwargs.pop('title', 'subcommands')) | ||
1997 | + description = _(kwargs.pop('description', None)) | ||
1998 | + self._subparsers = self.add_argument_group(title, description) | ||
1999 | + else: | ||
2000 | + self._subparsers = self._positionals | ||
2001 | + | ||
2002 | + # prog defaults to the usage message of this parser, skipping | ||
2003 | + # optional arguments and with no "usage:" prefix | ||
2004 | + if kwargs.get('prog') is None: | ||
2005 | + formatter = self._get_formatter() | ||
2006 | + positionals = self._get_positional_actions() | ||
2007 | + groups = self._mutually_exclusive_groups | ||
2008 | + formatter.add_usage(self.usage, positionals, groups, '') | ||
2009 | + kwargs['prog'] = formatter.format_help().strip() | ||
2010 | + | ||
2011 | + # create the parsers action and add it to the positionals list | ||
2012 | + parsers_class = self._pop_action_class(kwargs, 'parsers') | ||
2013 | + action = parsers_class(option_strings=[], **kwargs) | ||
2014 | + self._subparsers._add_action(action) | ||
2015 | + | ||
2016 | + # return the created parsers action | ||
2017 | + return action | ||
2018 | + | ||
2019 | + def _add_action(self, action): | ||
2020 | + if action.option_strings: | ||
2021 | + self._optionals._add_action(action) | ||
2022 | + else: | ||
2023 | + self._positionals._add_action(action) | ||
2024 | + return action | ||
2025 | + | ||
2026 | + def _get_optional_actions(self): | ||
2027 | + return [action | ||
2028 | + for action in self._actions | ||
2029 | + if action.option_strings] | ||
2030 | + | ||
2031 | + def _get_positional_actions(self): | ||
2032 | + return [action | ||
2033 | + for action in self._actions | ||
2034 | + if not action.option_strings] | ||
2035 | + | ||
2036 | + # ===================================== | ||
2037 | + # Command line argument parsing methods | ||
2038 | + # ===================================== | ||
2039 | + def parse_args(self, args=None, namespace=None): | ||
2040 | + args, argv = self.parse_known_args(args, namespace) | ||
2041 | + if argv: | ||
2042 | + msg = _('unrecognized arguments: %s') | ||
2043 | + self.error(msg % ' '.join(argv)) | ||
2044 | + return args | ||
2045 | + | ||
2046 | + def parse_known_args(self, args=None, namespace=None): | ||
2047 | + # args default to the system args | ||
2048 | + if args is None: | ||
2049 | + args = _sys.argv[1:] | ||
2050 | + | ||
2051 | + # default Namespace built from parser defaults | ||
2052 | + if namespace is None: | ||
2053 | + namespace = Namespace() | ||
2054 | + | ||
2055 | + # add any action defaults that aren't present | ||
2056 | + for action in self._actions: | ||
2057 | + if action.dest is not SUPPRESS: | ||
2058 | + if not hasattr(namespace, action.dest): | ||
2059 | + if action.default is not SUPPRESS: | ||
2060 | + setattr(namespace, action.dest, action.default) | ||
2061 | + | ||
2062 | + # add any parser defaults that aren't present | ||
2063 | + for dest in self._defaults: | ||
2064 | + if not hasattr(namespace, dest): | ||
2065 | + setattr(namespace, dest, self._defaults[dest]) | ||
2066 | + | ||
2067 | + # parse the arguments and exit if there are any errors | ||
2068 | + try: | ||
2069 | + namespace, args = self._parse_known_args(args, namespace) | ||
2070 | + if hasattr(namespace, _UNRECOGNIZED_ARGS_ATTR): | ||
2071 | + args.extend(getattr(namespace, _UNRECOGNIZED_ARGS_ATTR)) | ||
2072 | + delattr(namespace, _UNRECOGNIZED_ARGS_ATTR) | ||
2073 | + return namespace, args | ||
2074 | + except ArgumentError: | ||
2075 | + err = _sys.exc_info()[1] | ||
2076 | + self.error(str(err)) | ||
2077 | + | ||
2078 | + def _parse_known_args(self, arg_strings, namespace): | ||
2079 | + # replace arg strings that are file references | ||
2080 | + if self.fromfile_prefix_chars is not None: | ||
2081 | + arg_strings = self._read_args_from_files(arg_strings) | ||
2082 | + | ||
2083 | + # map all mutually exclusive arguments to the other arguments | ||
2084 | + # they can't occur with | ||
2085 | + action_conflicts = {} | ||
2086 | + for mutex_group in self._mutually_exclusive_groups: | ||
2087 | + group_actions = mutex_group._group_actions | ||
2088 | + for i, mutex_action in enumerate(mutex_group._group_actions): | ||
2089 | + conflicts = action_conflicts.setdefault(mutex_action, []) | ||
2090 | + conflicts.extend(group_actions[:i]) | ||
2091 | + conflicts.extend(group_actions[i + 1:]) | ||
2092 | + | ||
2093 | + # find all option indices, and determine the arg_string_pattern | ||
2094 | + # which has an 'O' if there is an option at an index, | ||
2095 | + # an 'A' if there is an argument, or a '-' if there is a '--' | ||
2096 | + option_string_indices = {} | ||
2097 | + arg_string_pattern_parts = [] | ||
2098 | + arg_strings_iter = iter(arg_strings) | ||
2099 | + for i, arg_string in enumerate(arg_strings_iter): | ||
2100 | + | ||
2101 | + # all args after -- are non-options | ||
2102 | + if arg_string == '--': | ||
2103 | + arg_string_pattern_parts.append('-') | ||
2104 | + for arg_string in arg_strings_iter: | ||
2105 | + arg_string_pattern_parts.append('A') | ||
2106 | + | ||
2107 | + # otherwise, add the arg to the arg strings | ||
2108 | + # and note the index if it was an option | ||
2109 | + else: | ||
2110 | + option_tuple = self._parse_optional(arg_string) | ||
2111 | + if option_tuple is None: | ||
2112 | + pattern = 'A' | ||
2113 | + else: | ||
2114 | + option_string_indices[i] = option_tuple | ||
2115 | + pattern = 'O' | ||
2116 | + arg_string_pattern_parts.append(pattern) | ||
2117 | + | ||
2118 | + # join the pieces together to form the pattern | ||
2119 | + arg_strings_pattern = ''.join(arg_string_pattern_parts) | ||
2120 | + | ||
2121 | + # converts arg strings to the appropriate and then takes the action | ||
2122 | + seen_actions = set() | ||
2123 | + seen_non_default_actions = set() | ||
2124 | + | ||
2125 | + def take_action(action, argument_strings, option_string=None): | ||
2126 | + seen_actions.add(action) | ||
2127 | + argument_values = self._get_values(action, argument_strings) | ||
2128 | + | ||
2129 | + # error if this argument is not allowed with other previously | ||
2130 | + # seen arguments, assuming that actions that use the default | ||
2131 | + # value don't really count as "present" | ||
2132 | + if argument_values is not action.default: | ||
2133 | + seen_non_default_actions.add(action) | ||
2134 | + for conflict_action in action_conflicts.get(action, []): | ||
2135 | + if conflict_action in seen_non_default_actions: | ||
2136 | + msg = _('not allowed with argument %s') | ||
2137 | + action_name = _get_action_name(conflict_action) | ||
2138 | + raise ArgumentError(action, msg % action_name) | ||
2139 | + | ||
2140 | + # take the action if we didn't receive a SUPPRESS value | ||
2141 | + # (e.g. from a default) | ||
2142 | + if argument_values is not SUPPRESS: | ||
2143 | + action(self, namespace, argument_values, option_string) | ||
2144 | + | ||
2145 | + # function to convert arg_strings into an optional action | ||
2146 | + def consume_optional(start_index): | ||
2147 | + | ||
2148 | + # get the optional identified at this index | ||
2149 | + option_tuple = option_string_indices[start_index] | ||
2150 | + action, option_string, explicit_arg = option_tuple | ||
2151 | + | ||
2152 | + # identify additional optionals in the same arg string | ||
2153 | + # (e.g. -xyz is the same as -x -y -z if no args are required) | ||
2154 | + match_argument = self._match_argument | ||
2155 | + action_tuples = [] | ||
2156 | + while True: | ||
2157 | + | ||
2158 | + # if we found no optional action, skip it | ||
2159 | + if action is None: | ||
2160 | + extras.append(arg_strings[start_index]) | ||
2161 | + return start_index + 1 | ||
2162 | + | ||
2163 | + # if there is an explicit argument, try to match the | ||
2164 | + # optional's string arguments to only this | ||
2165 | + if explicit_arg is not None: | ||
2166 | + arg_count = match_argument(action, 'A') | ||
2167 | + | ||
2168 | + # if the action is a single-dash option and takes no | ||
2169 | + # arguments, try to parse more single-dash options out | ||
2170 | + # of the tail of the option string | ||
2171 | + chars = self.prefix_chars | ||
2172 | + if arg_count == 0 and option_string[1] not in chars: | ||
2173 | + action_tuples.append((action, [], option_string)) | ||
2174 | + char = option_string[0] | ||
2175 | + option_string = char + explicit_arg[0] | ||
2176 | + new_explicit_arg = explicit_arg[1:] or None | ||
2177 | + optionals_map = self._option_string_actions | ||
2178 | + if option_string in optionals_map: | ||
2179 | + action = optionals_map[option_string] | ||
2180 | + explicit_arg = new_explicit_arg | ||
2181 | + else: | ||
2182 | + msg = _('ignored explicit argument %r') | ||
2183 | + raise ArgumentError(action, msg % explicit_arg) | ||
2184 | + | ||
2185 | + # if the action expect exactly one argument, we've | ||
2186 | + # successfully matched the option; exit the loop | ||
2187 | + elif arg_count == 1: | ||
2188 | + stop = start_index + 1 | ||
2189 | + args = [explicit_arg] | ||
2190 | + action_tuples.append((action, args, option_string)) | ||
2191 | + break | ||
2192 | + | ||
2193 | + # error if a double-dash option did not use the | ||
2194 | + # explicit argument | ||
2195 | + else: | ||
2196 | + msg = _('ignored explicit argument %r') | ||
2197 | + raise ArgumentError(action, msg % explicit_arg) | ||
2198 | + | ||
2199 | + # if there is no explicit argument, try to match the | ||
2200 | + # optional's string arguments with the following strings | ||
2201 | + # if successful, exit the loop | ||
2202 | + else: | ||
2203 | + start = start_index + 1 | ||
2204 | + selected_patterns = arg_strings_pattern[start:] | ||
2205 | + arg_count = match_argument(action, selected_patterns) | ||
2206 | + stop = start + arg_count | ||
2207 | + args = arg_strings[start:stop] | ||
2208 | + action_tuples.append((action, args, option_string)) | ||
2209 | + break | ||
2210 | + | ||
2211 | + # add the Optional to the list and return the index at which | ||
2212 | + # the Optional's string args stopped | ||
2213 | + assert action_tuples | ||
2214 | + for action, args, option_string in action_tuples: | ||
2215 | + take_action(action, args, option_string) | ||
2216 | + return stop | ||
2217 | + | ||
2218 | + # the list of Positionals left to be parsed; this is modified | ||
2219 | + # by consume_positionals() | ||
2220 | + positionals = self._get_positional_actions() | ||
2221 | + | ||
2222 | + # function to convert arg_strings into positional actions | ||
2223 | + def consume_positionals(start_index): | ||
2224 | + # match as many Positionals as possible | ||
2225 | + match_partial = self._match_arguments_partial | ||
2226 | + selected_pattern = arg_strings_pattern[start_index:] | ||
2227 | + arg_counts = match_partial(positionals, selected_pattern) | ||
2228 | + | ||
2229 | + # slice off the appropriate arg strings for each Positional | ||
2230 | + # and add the Positional and its args to the list | ||
2231 | + for action, arg_count in zip(positionals, arg_counts): | ||
2232 | + args = arg_strings[start_index: start_index + arg_count] | ||
2233 | + start_index += arg_count | ||
2234 | + take_action(action, args) | ||
2235 | + | ||
2236 | + # slice off the Positionals that we just parsed and return the | ||
2237 | + # index at which the Positionals' string args stopped | ||
2238 | + positionals[:] = positionals[len(arg_counts):] | ||
2239 | + return start_index | ||
2240 | + | ||
2241 | + # consume Positionals and Optionals alternately, until we have | ||
2242 | + # passed the last option string | ||
2243 | + extras = [] | ||
2244 | + start_index = 0 | ||
2245 | + if option_string_indices: | ||
2246 | + max_option_string_index = max(option_string_indices) | ||
2247 | + else: | ||
2248 | + max_option_string_index = -1 | ||
2249 | + while start_index <= max_option_string_index: | ||
2250 | + | ||
2251 | + # consume any Positionals preceding the next option | ||
2252 | + next_option_string_index = min([ | ||
2253 | + index | ||
2254 | + for index in option_string_indices | ||
2255 | + if index >= start_index]) | ||
2256 | + if start_index != next_option_string_index: | ||
2257 | + positionals_end_index = consume_positionals(start_index) | ||
2258 | + | ||
2259 | + # only try to parse the next optional if we didn't consume | ||
2260 | + # the option string during the positionals parsing | ||
2261 | + if positionals_end_index > start_index: | ||
2262 | + start_index = positionals_end_index | ||
2263 | + continue | ||
2264 | + else: | ||
2265 | + start_index = positionals_end_index | ||
2266 | + | ||
2267 | + # if we consumed all the positionals we could and we're not | ||
2268 | + # at the index of an option string, there were extra arguments | ||
2269 | + if start_index not in option_string_indices: | ||
2270 | + strings = arg_strings[start_index:next_option_string_index] | ||
2271 | + extras.extend(strings) | ||
2272 | + start_index = next_option_string_index | ||
2273 | + | ||
2274 | + # consume the next optional and any arguments for it | ||
2275 | + start_index = consume_optional(start_index) | ||
2276 | + | ||
2277 | + # consume any positionals following the last Optional | ||
2278 | + stop_index = consume_positionals(start_index) | ||
2279 | + | ||
2280 | + # if we didn't consume all the argument strings, there were extras | ||
2281 | + extras.extend(arg_strings[stop_index:]) | ||
2282 | + | ||
2283 | + # if we didn't use all the Positional objects, there were too few | ||
2284 | + # arg strings supplied. | ||
2285 | + if positionals: | ||
2286 | + self.error(_('too few arguments')) | ||
2287 | + | ||
2288 | + # make sure all required actions were present, and convert defaults. | ||
2289 | + for action in self._actions: | ||
2290 | + if action not in seen_actions: | ||
2291 | + if action.required: | ||
2292 | + name = _get_action_name(action) | ||
2293 | + self.error(_('argument %s is required') % name) | ||
2294 | + else: | ||
2295 | + # Convert action default now instead of doing it before | ||
2296 | + # parsing arguments to avoid calling convert functions | ||
2297 | + # twice (which may fail) if the argument was given, but | ||
2298 | + # only if it was defined already in the namespace | ||
2299 | + if (action.default is not None and | ||
2300 | + isinstance(action.default, basestring) and | ||
2301 | + hasattr(namespace, action.dest) and | ||
2302 | + action.default is getattr(namespace, action.dest)): | ||
2303 | + setattr(namespace, action.dest, | ||
2304 | + self._get_value(action, action.default)) | ||
2305 | + | ||
2306 | + # make sure all required groups had one option present | ||
2307 | + for group in self._mutually_exclusive_groups: | ||
2308 | + if group.required: | ||
2309 | + for action in group._group_actions: | ||
2310 | + if action in seen_non_default_actions: | ||
2311 | + break | ||
2312 | + | ||
2313 | + # if no actions were used, report the error | ||
2314 | + else: | ||
2315 | + names = [_get_action_name(action) | ||
2316 | + for action in group._group_actions | ||
2317 | + if action.help is not SUPPRESS] | ||
2318 | + msg = _('one of the arguments %s is required') | ||
2319 | + self.error(msg % ' '.join(names)) | ||
2320 | + | ||
2321 | + # return the updated namespace and the extra arguments | ||
2322 | + return namespace, extras | ||
2323 | + | ||
2324 | + def _read_args_from_files(self, arg_strings): | ||
2325 | + # expand arguments referencing files | ||
2326 | + new_arg_strings = [] | ||
2327 | + for arg_string in arg_strings: | ||
2328 | + | ||
2329 | + # for regular arguments, just add them back into the list | ||
2330 | + if arg_string[0] not in self.fromfile_prefix_chars: | ||
2331 | + new_arg_strings.append(arg_string) | ||
2332 | + | ||
2333 | + # replace arguments referencing files with the file content | ||
2334 | + else: | ||
2335 | + try: | ||
2336 | + args_file = open(arg_string[1:]) | ||
2337 | + try: | ||
2338 | + arg_strings = [] | ||
2339 | + for arg_line in args_file.read().splitlines(): | ||
2340 | + for arg in self.convert_arg_line_to_args(arg_line): | ||
2341 | + arg_strings.append(arg) | ||
2342 | + arg_strings = self._read_args_from_files(arg_strings) | ||
2343 | + new_arg_strings.extend(arg_strings) | ||
2344 | + finally: | ||
2345 | + args_file.close() | ||
2346 | + except IOError: | ||
2347 | + err = _sys.exc_info()[1] | ||
2348 | + self.error(str(err)) | ||
2349 | + | ||
2350 | + # return the modified argument list | ||
2351 | + return new_arg_strings | ||
2352 | + | ||
2353 | + def convert_arg_line_to_args(self, arg_line): | ||
2354 | + return [arg_line] | ||
2355 | + | ||
2356 | + def _match_argument(self, action, arg_strings_pattern): | ||
2357 | + # match the pattern for this action to the arg strings | ||
2358 | + nargs_pattern = self._get_nargs_pattern(action) | ||
2359 | + match = _re.match(nargs_pattern, arg_strings_pattern) | ||
2360 | + | ||
2361 | + # raise an exception if we weren't able to find a match | ||
2362 | + if match is None: | ||
2363 | + nargs_errors = { | ||
2364 | + None: _('expected one argument'), | ||
2365 | + OPTIONAL: _('expected at most one argument'), | ||
2366 | + ONE_OR_MORE: _('expected at least one argument'), | ||
2367 | + } | ||
2368 | + default = _('expected %s argument(s)') % action.nargs | ||
2369 | + msg = nargs_errors.get(action.nargs, default) | ||
2370 | + raise ArgumentError(action, msg) | ||
2371 | + | ||
2372 | + # return the number of arguments matched | ||
2373 | + return len(match.group(1)) | ||
2374 | + | ||
2375 | + def _match_arguments_partial(self, actions, arg_strings_pattern): | ||
2376 | + # progressively shorten the actions list by slicing off the | ||
2377 | + # final actions until we find a match | ||
2378 | + result = [] | ||
2379 | + for i in range(len(actions), 0, -1): | ||
2380 | + actions_slice = actions[:i] | ||
2381 | + pattern = ''.join([self._get_nargs_pattern(action) | ||
2382 | + for action in actions_slice]) | ||
2383 | + match = _re.match(pattern, arg_strings_pattern) | ||
2384 | + if match is not None: | ||
2385 | + result.extend([len(string) for string in match.groups()]) | ||
2386 | + break | ||
2387 | + | ||
2388 | + # return the list of arg string counts | ||
2389 | + return result | ||
2390 | + | ||
2391 | + def _parse_optional(self, arg_string): | ||
2392 | + # if it's an empty string, it was meant to be a positional | ||
2393 | + if not arg_string: | ||
2394 | + return None | ||
2395 | + | ||
2396 | + # if it doesn't start with a prefix, it was meant to be positional | ||
2397 | + if not arg_string[0] in self.prefix_chars: | ||
2398 | + return None | ||
2399 | + | ||
2400 | + # if the option string is present in the parser, return the action | ||
2401 | + if arg_string in self._option_string_actions: | ||
2402 | + action = self._option_string_actions[arg_string] | ||
2403 | + return action, arg_string, None | ||
2404 | + | ||
2405 | + # if it's just a single character, it was meant to be positional | ||
2406 | + if len(arg_string) == 1: | ||
2407 | + return None | ||
2408 | + | ||
2409 | + # if the option string before the "=" is present, return the action | ||
2410 | + if '=' in arg_string: | ||
2411 | + option_string, explicit_arg = arg_string.split('=', 1) | ||
2412 | + if option_string in self._option_string_actions: | ||
2413 | + action = self._option_string_actions[option_string] | ||
2414 | + return action, option_string, explicit_arg | ||
2415 | + | ||
2416 | + # search through all possible prefixes of the option string | ||
2417 | + # and all actions in the parser for possible interpretations | ||
2418 | + option_tuples = self._get_option_tuples(arg_string) | ||
2419 | + | ||
2420 | + # if multiple actions match, the option string was ambiguous | ||
2421 | + if len(option_tuples) > 1: | ||
2422 | + options = ', '.join([option_string | ||
2423 | + for action, option_string, explicit_arg in option_tuples]) | ||
2424 | + tup = arg_string, options | ||
2425 | + self.error(_('ambiguous option: %s could match %s') % tup) | ||
2426 | + | ||
2427 | + # if exactly one action matched, this segmentation is good, | ||
2428 | + # so return the parsed action | ||
2429 | + elif len(option_tuples) == 1: | ||
2430 | + option_tuple, = option_tuples | ||
2431 | + return option_tuple | ||
2432 | + | ||
2433 | + # if it was not found as an option, but it looks like a negative | ||
2434 | + # number, it was meant to be positional | ||
2435 | + # unless there are negative-number-like options | ||
2436 | + if self._negative_number_matcher.match(arg_string): | ||
2437 | + if not self._has_negative_number_optionals: | ||
2438 | + return None | ||
2439 | + | ||
2440 | + # if it contains a space, it was meant to be a positional | ||
2441 | + if ' ' in arg_string: | ||
2442 | + return None | ||
2443 | + | ||
2444 | + # it was meant to be an optional but there is no such option | ||
2445 | + # in this parser (though it might be a valid option in a subparser) | ||
2446 | + return None, arg_string, None | ||
2447 | + | ||
2448 | + def _get_option_tuples(self, option_string): | ||
2449 | + result = [] | ||
2450 | + | ||
2451 | + # option strings starting with two prefix characters are only | ||
2452 | + # split at the '=' | ||
2453 | + chars = self.prefix_chars | ||
2454 | + if option_string[0] in chars and option_string[1] in chars: | ||
2455 | + if '=' in option_string: | ||
2456 | + option_prefix, explicit_arg = option_string.split('=', 1) | ||
2457 | + else: | ||
2458 | + option_prefix = option_string | ||
2459 | + explicit_arg = None | ||
2460 | + for option_string in self._option_string_actions: | ||
2461 | + if option_string.startswith(option_prefix): | ||
2462 | + action = self._option_string_actions[option_string] | ||
2463 | + tup = action, option_string, explicit_arg | ||
2464 | + result.append(tup) | ||
2465 | + | ||
2466 | + # single character options can be concatenated with their arguments | ||
2467 | + # but multiple character options always have to have their argument | ||
2468 | + # separate | ||
2469 | + elif option_string[0] in chars and option_string[1] not in chars: | ||
2470 | + option_prefix = option_string | ||
2471 | + explicit_arg = None | ||
2472 | + short_option_prefix = option_string[:2] | ||
2473 | + short_explicit_arg = option_string[2:] | ||
2474 | + | ||
2475 | + for option_string in self._option_string_actions: | ||
2476 | + if option_string == short_option_prefix: | ||
2477 | + action = self._option_string_actions[option_string] | ||
2478 | + tup = action, option_string, short_explicit_arg | ||
2479 | + result.append(tup) | ||
2480 | + elif option_string.startswith(option_prefix): | ||
2481 | + action = self._option_string_actions[option_string] | ||
2482 | + tup = action, option_string, explicit_arg | ||
2483 | + result.append(tup) | ||
2484 | + | ||
2485 | + # shouldn't ever get here | ||
2486 | + else: | ||
2487 | + self.error(_('unexpected option string: %s') % option_string) | ||
2488 | + | ||
2489 | + # return the collected option tuples | ||
2490 | + return result | ||
2491 | + | ||
2492 | + def _get_nargs_pattern(self, action): | ||
2493 | + # in all examples below, we have to allow for '--' args | ||
2494 | + # which are represented as '-' in the pattern | ||
2495 | + nargs = action.nargs | ||
2496 | + | ||
2497 | + # the default (None) is assumed to be a single argument | ||
2498 | + if nargs is None: | ||
2499 | + nargs_pattern = '(-*A-*)' | ||
2500 | + | ||
2501 | + # allow zero or one arguments | ||
2502 | + elif nargs == OPTIONAL: | ||
2503 | + nargs_pattern = '(-*A?-*)' | ||
2504 | + | ||
2505 | + # allow zero or more arguments | ||
2506 | + elif nargs == ZERO_OR_MORE: | ||
2507 | + nargs_pattern = '(-*[A-]*)' | ||
2508 | + | ||
2509 | + # allow one or more arguments | ||
2510 | + elif nargs == ONE_OR_MORE: | ||
2511 | + nargs_pattern = '(-*A[A-]*)' | ||
2512 | + | ||
2513 | + # allow any number of options or arguments | ||
2514 | + elif nargs == REMAINDER: | ||
2515 | + nargs_pattern = '([-AO]*)' | ||
2516 | + | ||
2517 | + # allow one argument followed by any number of options or arguments | ||
2518 | + elif nargs == PARSER: | ||
2519 | + nargs_pattern = '(-*A[-AO]*)' | ||
2520 | + | ||
2521 | + # all others should be integers | ||
2522 | + else: | ||
2523 | + nargs_pattern = '(-*%s-*)' % '-*'.join('A' * nargs) | ||
2524 | + | ||
2525 | + # if this is an optional action, -- is not allowed | ||
2526 | + if action.option_strings: | ||
2527 | + nargs_pattern = nargs_pattern.replace('-*', '') | ||
2528 | + nargs_pattern = nargs_pattern.replace('-', '') | ||
2529 | + | ||
2530 | + # return the pattern | ||
2531 | + return nargs_pattern | ||
2532 | + | ||
2533 | + # ======================== | ||
2534 | + # Value conversion methods | ||
2535 | + # ======================== | ||
2536 | + def _get_values(self, action, arg_strings): | ||
2537 | + # for everything but PARSER args, strip out '--' | ||
2538 | + if action.nargs not in [PARSER, REMAINDER]: | ||
2539 | + arg_strings = [s for s in arg_strings if s != '--'] | ||
2540 | + | ||
2541 | + # optional argument produces a default when not present | ||
2542 | + if not arg_strings and action.nargs == OPTIONAL: | ||
2543 | + if action.option_strings: | ||
2544 | + value = action.const | ||
2545 | + else: | ||
2546 | + value = action.default | ||
2547 | + if isinstance(value, basestring): | ||
2548 | + value = self._get_value(action, value) | ||
2549 | + self._check_value(action, value) | ||
2550 | + | ||
2551 | + # when nargs='*' on a positional, if there were no command-line | ||
2552 | + # args, use the default if it is anything other than None | ||
2553 | + elif (not arg_strings and action.nargs == ZERO_OR_MORE and | ||
2554 | + not action.option_strings): | ||
2555 | + if action.default is not None: | ||
2556 | + value = action.default | ||
2557 | + else: | ||
2558 | + value = arg_strings | ||
2559 | + self._check_value(action, value) | ||
2560 | + | ||
2561 | + # single argument or optional argument produces a single value | ||
2562 | + elif len(arg_strings) == 1 and action.nargs in [None, OPTIONAL]: | ||
2563 | + arg_string, = arg_strings | ||
2564 | + value = self._get_value(action, arg_string) | ||
2565 | + self._check_value(action, value) | ||
2566 | + | ||
2567 | + # REMAINDER arguments convert all values, checking none | ||
2568 | + elif action.nargs == REMAINDER: | ||
2569 | + value = [self._get_value(action, v) for v in arg_strings] | ||
2570 | + | ||
2571 | + # PARSER arguments convert all values, but check only the first | ||
2572 | + elif action.nargs == PARSER: | ||
2573 | + value = [self._get_value(action, v) for v in arg_strings] | ||
2574 | + self._check_value(action, value[0]) | ||
2575 | + | ||
2576 | + # all other types of nargs produce a list | ||
2577 | + else: | ||
2578 | + value = [self._get_value(action, v) for v in arg_strings] | ||
2579 | + for v in value: | ||
2580 | + self._check_value(action, v) | ||
2581 | + | ||
2582 | + # return the converted value | ||
2583 | + return value | ||
2584 | + | ||
2585 | + def _get_value(self, action, arg_string): | ||
2586 | + type_func = self._registry_get('type', action.type, action.type) | ||
2587 | + if not _callable(type_func): | ||
2588 | + msg = _('%r is not callable') | ||
2589 | + raise ArgumentError(action, msg % type_func) | ||
2590 | + | ||
2591 | + # convert the value to the appropriate type | ||
2592 | + try: | ||
2593 | + result = type_func(arg_string) | ||
2594 | + | ||
2595 | + # ArgumentTypeErrors indicate errors | ||
2596 | + except ArgumentTypeError: | ||
2597 | + name = getattr(action.type, '__name__', repr(action.type)) | ||
2598 | + msg = str(_sys.exc_info()[1]) | ||
2599 | + raise ArgumentError(action, msg) | ||
2600 | + | ||
2601 | + # TypeErrors or ValueErrors also indicate errors | ||
2602 | + except (TypeError, ValueError): | ||
2603 | + name = getattr(action.type, '__name__', repr(action.type)) | ||
2604 | + msg = _('invalid %s value: %r') | ||
2605 | + raise ArgumentError(action, msg % (name, arg_string)) | ||
2606 | + | ||
2607 | + # return the converted value | ||
2608 | + return result | ||
2609 | + | ||
2610 | + def _check_value(self, action, value): | ||
2611 | + # converted value must be one of the choices (if specified) | ||
2612 | + if action.choices is not None and value not in action.choices: | ||
2613 | + tup = value, ', '.join(map(repr, action.choices)) | ||
2614 | + msg = _('invalid choice: %r (choose from %s)') % tup | ||
2615 | + raise ArgumentError(action, msg) | ||
2616 | + | ||
2617 | + # ======================= | ||
2618 | + # Help-formatting methods | ||
2619 | + # ======================= | ||
2620 | + def format_usage(self): | ||
2621 | + formatter = self._get_formatter() | ||
2622 | + formatter.add_usage(self.usage, self._actions, | ||
2623 | + self._mutually_exclusive_groups) | ||
2624 | + return formatter.format_help() | ||
2625 | + | ||
2626 | + def format_help(self): | ||
2627 | + formatter = self._get_formatter() | ||
2628 | + | ||
2629 | + # usage | ||
2630 | + formatter.add_usage(self.usage, self._actions, | ||
2631 | + self._mutually_exclusive_groups) | ||
2632 | + | ||
2633 | + # description | ||
2634 | + formatter.add_text(self.description) | ||
2635 | + | ||
2636 | + # positionals, optionals and user-defined groups | ||
2637 | + for action_group in self._action_groups: | ||
2638 | + formatter.start_section(action_group.title) | ||
2639 | + formatter.add_text(action_group.description) | ||
2640 | + formatter.add_arguments(action_group._group_actions) | ||
2641 | + formatter.end_section() | ||
2642 | + | ||
2643 | + # epilog | ||
2644 | + formatter.add_text(self.epilog) | ||
2645 | + | ||
2646 | + # determine help from format above | ||
2647 | + return formatter.format_help() | ||
2648 | + | ||
2649 | + def format_version(self): | ||
2650 | + import warnings | ||
2651 | + warnings.warn( | ||
2652 | + 'The format_version method is deprecated -- the "version" ' | ||
2653 | + 'argument to ArgumentParser is no longer supported.', | ||
2654 | + DeprecationWarning) | ||
2655 | + formatter = self._get_formatter() | ||
2656 | + formatter.add_text(self.version) | ||
2657 | + return formatter.format_help() | ||
2658 | + | ||
2659 | + def _get_formatter(self): | ||
2660 | + return self.formatter_class(prog=self.prog) | ||
2661 | + | ||
2662 | + # ===================== | ||
2663 | + # Help-printing methods | ||
2664 | + # ===================== | ||
2665 | + def print_usage(self, file=None): | ||
2666 | + if file is None: | ||
2667 | + file = _sys.stdout | ||
2668 | + self._print_message(self.format_usage(), file) | ||
2669 | + | ||
2670 | + def print_help(self, file=None): | ||
2671 | + if file is None: | ||
2672 | + file = _sys.stdout | ||
2673 | + self._print_message(self.format_help(), file) | ||
2674 | + | ||
2675 | + def print_version(self, file=None): | ||
2676 | + import warnings | ||
2677 | + warnings.warn( | ||
2678 | + 'The print_version method is deprecated -- the "version" ' | ||
2679 | + 'argument to ArgumentParser is no longer supported.', | ||
2680 | + DeprecationWarning) | ||
2681 | + self._print_message(self.format_version(), file) | ||
2682 | + | ||
2683 | + def _print_message(self, message, file=None): | ||
2684 | + if message: | ||
2685 | + if file is None: | ||
2686 | + file = _sys.stderr | ||
2687 | + file.write(message) | ||
2688 | + | ||
2689 | + # =============== | ||
2690 | + # Exiting methods | ||
2691 | + # =============== | ||
2692 | + def exit(self, status=0, message=None): | ||
2693 | + if message: | ||
2694 | + self._print_message(message, _sys.stderr) | ||
2695 | + _sys.exit(status) | ||
2696 | + | ||
2697 | + def error(self, message): | ||
2698 | + """error(message: string) | ||
2699 | + | ||
2700 | + Prints a usage message incorporating the message to stderr and | ||
2701 | + exits. | ||
2702 | + | ||
2703 | + If you override this in a subclass, it should not return -- it | ||
2704 | + should either exit or raise an exception. | ||
2705 | + """ | ||
2706 | + self.print_usage(_sys.stderr) | ||
2707 | + self.exit(2, _('%s: error: %s\n') % (self.prog, message)) | ||
2708 | -- | 78 | -- |
2709 | 2.13.5 | 79 | 2.14.3 |
2710 | 80 | ||
2711 | 81 | diff view generated by jsdifflib |
1 | From: Alberto Garcia <berto@igalia.com> | 1 | From: Fam Zheng <famz@redhat.com> |
---|---|---|---|
2 | 2 | ||
3 | There's a few cases which we're passing an Error pointer to a function | 3 | Signed-off-by: Fam Zheng <famz@redhat.com> |
4 | only to discard it immediately afterwards without checking it. In | 4 | Message-id: 20180226030326.20219-3-famz@redhat.com |
5 | these cases we can simply remove the variable and pass NULL instead. | ||
6 | |||
7 | Signed-off-by: Alberto Garcia <berto@igalia.com> | ||
8 | Reviewed-by: Philippe Mathieu-Daudé <f4bug@amsat.org> | ||
9 | Reviewed-by: Eric Blake <eblake@redhat.com> | ||
10 | Message-id: 20170829120836.16091-1-berto@igalia.com | ||
11 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | 5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> |
12 | --- | 6 | --- |
13 | block/qcow.c | 12 +++--------- | 7 | README | 31 ++++++++++++++++++++++++++++++- |
14 | block/qcow2.c | 8 ++------ | 8 | 1 file changed, 30 insertions(+), 1 deletion(-) |
15 | dump.c | 4 +--- | ||
16 | 3 files changed, 6 insertions(+), 18 deletions(-) | ||
17 | 9 | ||
18 | diff --git a/block/qcow.c b/block/qcow.c | 10 | diff --git a/README b/README |
19 | index XXXXXXX..XXXXXXX 100644 | 11 | index XXXXXXX..XXXXXXX 100644 |
20 | --- a/block/qcow.c | 12 | --- a/README |
21 | +++ b/block/qcow.c | 13 | +++ b/README |
22 | @@ -XXX,XX +XXX,XX @@ static uint64_t get_cluster_offset(BlockDriverState *bs, | 14 | @@ -XXX,XX +XXX,XX @@ The QEMU source code is maintained under the GIT version control system. |
23 | start_sect = (offset & ~(s->cluster_size - 1)) >> 9; | 15 | |
24 | for(i = 0; i < s->cluster_sectors; i++) { | 16 | git clone git://git.qemu.org/qemu.git |
25 | if (i < n_start || i >= n_end) { | 17 | |
26 | - Error *err = NULL; | 18 | -When submitting patches, the preferred approach is to use 'git |
27 | memset(s->cluster_data, 0x00, 512); | 19 | +When submitting patches, one common approach is to use 'git |
28 | if (qcrypto_block_encrypt(s->crypto, start_sect + i, | 20 | format-patch' and/or 'git send-email' to format & send the mail to the |
29 | s->cluster_data, | 21 | qemu-devel@nongnu.org mailing list. All patches submitted must contain |
30 | BDRV_SECTOR_SIZE, | 22 | a 'Signed-off-by' line from the author. Patches should follow the |
31 | - &err) < 0) { | 23 | @@ -XXX,XX +XXX,XX @@ The QEMU website is also maintained under source control. |
32 | - error_free(err); | 24 | git clone git://git.qemu.org/qemu-web.git |
33 | + NULL) < 0) { | 25 | https://www.qemu.org/2017/02/04/the-new-qemu-website-is-up/ |
34 | errno = EIO; | 26 | |
35 | return -1; | 27 | +A 'git-profile' utility was created to make above process less |
36 | } | 28 | +cumbersome, and is highly recommended for making regular contributions, |
37 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn int qcow_co_readv(BlockDriverState *bs, int64_t sector_num, | 29 | +or even just for sending consecutive patch series revisions. It also |
38 | QEMUIOVector hd_qiov; | 30 | +requires a working 'git send-email' setup, and by default doesn't |
39 | uint8_t *buf; | 31 | +automate everything, so you may want to go through the above steps |
40 | void *orig_buf; | 32 | +manually for once. |
41 | - Error *err = NULL; | 33 | + |
42 | 34 | +For installation instructions, please go to | |
43 | if (qiov->niov > 1) { | 35 | + |
44 | buf = orig_buf = qemu_try_blockalign(bs, qiov->size); | 36 | + https://github.com/stefanha/git-publish |
45 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn int qcow_co_readv(BlockDriverState *bs, int64_t sector_num, | 37 | + |
46 | if (bs->encrypted) { | 38 | +The workflow with 'git-publish' is: |
47 | assert(s->crypto); | 39 | + |
48 | if (qcrypto_block_decrypt(s->crypto, sector_num, buf, | 40 | + $ git checkout master -b my-feature |
49 | - n * BDRV_SECTOR_SIZE, &err) < 0) { | 41 | + $ # work on new commits, add your 'Signed-off-by' lines to each |
50 | + n * BDRV_SECTOR_SIZE, NULL) < 0) { | 42 | + $ git publish |
51 | goto fail; | 43 | + |
52 | } | 44 | +Your patch series will be sent and tagged as my-feature-v1 if you need to refer |
53 | } | 45 | +back to it in the future. |
54 | @@ -XXX,XX +XXX,XX @@ done: | 46 | + |
55 | return ret; | 47 | +Sending v2: |
56 | 48 | + | |
57 | fail: | 49 | + $ git checkout my-feature # same topic branch |
58 | - error_free(err); | 50 | + $ # making changes to the commits (using 'git rebase', for example) |
59 | ret = -EIO; | 51 | + $ git publish |
60 | goto done; | 52 | + |
61 | } | 53 | +Your patch series will be sent with 'v2' tag in the subject and the git tip |
62 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn int qcow_co_writev(BlockDriverState *bs, int64_t sector_num, | 54 | +will be tagged as my-feature-v2. |
63 | break; | 55 | + |
64 | } | 56 | Bug reporting |
65 | if (bs->encrypted) { | 57 | ============= |
66 | - Error *err = NULL; | ||
67 | assert(s->crypto); | ||
68 | if (qcrypto_block_encrypt(s->crypto, sector_num, buf, | ||
69 | - n * BDRV_SECTOR_SIZE, &err) < 0) { | ||
70 | - error_free(err); | ||
71 | + n * BDRV_SECTOR_SIZE, NULL) < 0) { | ||
72 | ret = -EIO; | ||
73 | break; | ||
74 | } | ||
75 | diff --git a/block/qcow2.c b/block/qcow2.c | ||
76 | index XXXXXXX..XXXXXXX 100644 | ||
77 | --- a/block/qcow2.c | ||
78 | +++ b/block/qcow2.c | ||
79 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn int qcow2_co_preadv(BlockDriverState *bs, uint64_t offset, | ||
80 | assert(s->crypto); | ||
81 | assert((offset & (BDRV_SECTOR_SIZE - 1)) == 0); | ||
82 | assert((cur_bytes & (BDRV_SECTOR_SIZE - 1)) == 0); | ||
83 | - Error *err = NULL; | ||
84 | if (qcrypto_block_decrypt(s->crypto, | ||
85 | (s->crypt_physical_offset ? | ||
86 | cluster_offset + offset_in_cluster : | ||
87 | offset) >> BDRV_SECTOR_BITS, | ||
88 | cluster_data, | ||
89 | cur_bytes, | ||
90 | - &err) < 0) { | ||
91 | - error_free(err); | ||
92 | + NULL) < 0) { | ||
93 | ret = -EIO; | ||
94 | goto fail; | ||
95 | } | ||
96 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn int qcow2_co_pwritev(BlockDriverState *bs, uint64_t offset, | ||
97 | qemu_iovec_concat(&hd_qiov, qiov, bytes_done, cur_bytes); | ||
98 | |||
99 | if (bs->encrypted) { | ||
100 | - Error *err = NULL; | ||
101 | assert(s->crypto); | ||
102 | if (!cluster_data) { | ||
103 | cluster_data = qemu_try_blockalign(bs->file->bs, | ||
104 | @@ -XXX,XX +XXX,XX @@ static coroutine_fn int qcow2_co_pwritev(BlockDriverState *bs, uint64_t offset, | ||
105 | cluster_offset + offset_in_cluster : | ||
106 | offset) >> BDRV_SECTOR_BITS, | ||
107 | cluster_data, | ||
108 | - cur_bytes, &err) < 0) { | ||
109 | - error_free(err); | ||
110 | + cur_bytes, NULL) < 0) { | ||
111 | ret = -EIO; | ||
112 | goto fail; | ||
113 | } | ||
114 | diff --git a/dump.c b/dump.c | ||
115 | index XXXXXXX..XXXXXXX 100644 | ||
116 | --- a/dump.c | ||
117 | +++ b/dump.c | ||
118 | @@ -XXX,XX +XXX,XX @@ static void dump_process(DumpState *s, Error **errp) | ||
119 | |||
120 | static void *dump_thread(void *data) | ||
121 | { | ||
122 | - Error *err = NULL; | ||
123 | DumpState *s = (DumpState *)data; | ||
124 | - dump_process(s, &err); | ||
125 | - error_free(err); | ||
126 | + dump_process(s, NULL); | ||
127 | return NULL; | ||
128 | } | ||
129 | 58 | ||
130 | -- | 59 | -- |
131 | 2.13.5 | 60 | 2.14.3 |
132 | 61 | ||
133 | 62 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | Add the scripts/ directory to sys.path so Python 2.6 will be able to | ||
2 | import argparse. | ||
3 | 1 | ||
4 | Cc: Daniel P. Berrange <berrange@redhat.com> | ||
5 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
6 | Acked-by: John Snow <jsnow@redhat.com> | ||
7 | Acked-by: Fam Zheng <famz@redhat.com> | ||
8 | Message-id: 20170825155732.15665-4-stefanha@redhat.com | ||
9 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
10 | --- | ||
11 | tests/migration/guestperf/shell.py | 8 +++++--- | ||
12 | 1 file changed, 5 insertions(+), 3 deletions(-) | ||
13 | |||
14 | diff --git a/tests/migration/guestperf/shell.py b/tests/migration/guestperf/shell.py | ||
15 | index XXXXXXX..XXXXXXX 100644 | ||
16 | --- a/tests/migration/guestperf/shell.py | ||
17 | +++ b/tests/migration/guestperf/shell.py | ||
18 | @@ -XXX,XX +XXX,XX @@ | ||
19 | # | ||
20 | |||
21 | |||
22 | -import argparse | ||
23 | -import fnmatch | ||
24 | import os | ||
25 | import os.path | ||
26 | -import platform | ||
27 | import sys | ||
28 | +sys.path.append(os.path.join(os.path.dirname(__file__), | ||
29 | + '..', '..', '..', 'scripts')) | ||
30 | +import argparse | ||
31 | +import fnmatch | ||
32 | +import platform | ||
33 | |||
34 | from guestperf.hardware import Hardware | ||
35 | from guestperf.engine import Engine | ||
36 | -- | ||
37 | 2.13.5 | ||
38 | |||
39 | diff view generated by jsdifflib |
Deleted patch | |||
---|---|---|---|
1 | From: Fred Rolland <rollandf@gmail.com> | ||
2 | 1 | ||
3 | Update doc with the usage of UUID for initiator name. | ||
4 | |||
5 | Related-To: https://bugzilla.redhat.com/1006468 | ||
6 | Signed-off-by: Fred Rolland <frolland@redhat.com> | ||
7 | Message-id: 20170823084830.30500-1-frolland@redhat.com | ||
8 | Signed-off-by: Stefan Hajnoczi <stefanha@redhat.com> | ||
9 | --- | ||
10 | qemu-doc.texi | 5 +++-- | ||
11 | 1 file changed, 3 insertions(+), 2 deletions(-) | ||
12 | |||
13 | diff --git a/qemu-doc.texi b/qemu-doc.texi | ||
14 | index XXXXXXX..XXXXXXX 100644 | ||
15 | --- a/qemu-doc.texi | ||
16 | +++ b/qemu-doc.texi | ||
17 | @@ -XXX,XX +XXX,XX @@ in a configuration file provided via '-readconfig' or directly on the | ||
18 | command line. | ||
19 | |||
20 | If the initiator-name is not specified qemu will use a default name | ||
21 | -of 'iqn.2008-11.org.linux-kvm[:<name>'] where <name> is the name of the | ||
22 | +of 'iqn.2008-11.org.linux-kvm[:<uuid>'] where <uuid> is the UUID of the | ||
23 | +virtual machine. If the UUID is not specified qemu will use | ||
24 | +'iqn.2008-11.org.linux-kvm[:<name>'] where <name> is the name of the | ||
25 | virtual machine. | ||
26 | |||
27 | - | ||
28 | @example | ||
29 | Setting a specific initiator name to use when logging in to the target | ||
30 | -iscsi initiator-name=iqn.qemu.test:my-initiator | ||
31 | -- | ||
32 | 2.13.5 | ||
33 | |||
34 | diff view generated by jsdifflib |