信号完结

2024-05-27 02:28:25 +08:00
parent a57bbdb4c1
commit 2336f88407
9 changed files with 703 additions and 8 deletions
--- a/C16-并发/parallel/signal/mytbf_sa/main.c
+++ b/C16-并发/parallel/signal/mytbf_sa/main.c
@@ -0,0 +1,99 @@
+#include "mytbf.h"
+#include <errno.h>
+#include <fcntl.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <time.h>
+#include <unistd.h>
+
+#define CPS     10
+#define BUFSIZE 1024
+#define BURST   100   //!! token上限
+
+
+int main(int argc, char** argv)
+{
+    int      sfd, dfd = 1;
+    char     buf[BUFSIZE];
+    int      len, ret, pos;
+    int      size;
+    mytbf_t* tbf;
+
+    if (argc < 2)
+    {
+        fprintf(stderr, "Usage: %s <src>\n", argv[0]);
+        exit(1);
+    }
+
+    tbf = mytbf_init(CPS, BURST);
+    if (NULL == tbf)
+    {
+        fprintf(stderr, "mytbf_init failed\n");
+        exit(1);
+    }
+
+    do   //!!!
+    {
+        sfd = open(argv[1], O_RDONLY);
+        if (sfd < 0)
+        {
+            //!!!
+            if (errno != EINTR)
+            {
+                perror("open");
+                exit(1);
+            }
+        }
+    } while (sfd < 0);
+
+    while (1)
+    {
+        size = mytbf_fetchtoken(tbf, BUFSIZE);
+        if (size < 0)
+        {
+            fprintf(stderr, "mytbf_fetchtoken failed: %s\n", strerror(errno));
+        }
+
+        //* 阻塞在这里的时候，token会自增
+        while ((len = read(sfd, buf, size)) < 0)
+        {
+            //!!!
+            if (EINTR == errno)
+                continue;
+
+            perror("read");
+            break;
+        }
+        if (len == 0)
+            break;
+
+        if (size - len > 0)
+            mytbf_returntoken(tbf, size - len);
+
+        pos = 0;
+        while (len > 0)
+        {
+            /* 可能没写完被别的中断打断 */
+            ret = write(dfd, buf + pos, len);
+            if (ret < 0)
+            {
+                //!!!
+                if (EINTR == errno)
+                    continue;
+                perror("write");
+                exit(1);
+            }
+            pos += ret;
+            len -= ret;
+        }
+
+        // sleep(1);   //! 能实现，但是移植性不好
+    }
+
+    close(sfd);
+    mytbf_destroy(tbf);
+    exit(0);
+}