From 8b0eb7c10b059be81934ffb02a0f85ef9e0500d6 Mon Sep 17 00:00:00 2001
From: fumiama <s862105088@gmail.com>
Date: Wed, 2 Mar 2022 17:18:50 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=98=BE=E7=A4=BA=E9=95=BF?=
 =?UTF-8?q?=E5=BA=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 cmoe.c | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/cmoe.c b/cmoe.c
index b72fb24..e12a41d 100644
--- a/cmoe.c
+++ b/cmoe.c
@@ -89,7 +89,7 @@ static uint32_t get_content_len(int isbig, uint16_t* len_type, char* cntstr) {
         + (sizeof(img_slot_front) + sizeof(img_slot_rear) - 1) * 10
         + 16 + isbig
         + sizeof(svg_tail) - 1;
-    for(int i = 0; i < 10; i++) {
+    for(int i = 0; cntstr[i]; i++) {
         len += len_type[cntstr[i] - '0'];
     }
     return len;
@@ -118,6 +118,9 @@ static void return_count(char* name, char* theme) {
                         char cntstr[11];
                         sprintf(cntstr, "%010u", d->count);
                         free(spb);
+                        int cntstrstart = 9;
+                        while(cntstr[cntstrstart] != '0') cntstrstart--;
+                        if(cntstrstart > 1) cntstrstart--; // 保留 2 位 0
                         int isbig = 0;
                         char** theme_type = mb;
                         uint16_t* len_type = mbl;
@@ -140,9 +143,9 @@ static void return_count(char* name, char* theme) {
                             h = H_SMALL;
                             head = svg_small;
                         }
-                        headers(get_content_len(isbig, len_type, cntstr), "image/svg+xml");
+                        headers(get_content_len(isbig, len_type, cntstr+cntstrstart), "image/svg+xml");
                         write(1, head, sizeof(svg_small)-1);
-                        for(int i = 0; i < 10; i++) {
+                        for(int i = cntstrstart; i < 10; i++) {
                             printf(img_slot_front, w * i, w, h);
                             int n = cntstr[i] - '0';
                             fwrite(theme_type[n], len_type[n], 1, stdout);
@@ -182,7 +185,7 @@ static int name_exist(char* name) {
 }
 
 #define QS (argv[2])
-//Usage: cmoe method query_string
+// Usage: cmoe method query_string
 int main(int argc, char **argv) {
     if(argc == 3) {
         char* name = strstr(QS, "name=");