1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
/*##############################################################################
# Copyright (c) 2017 ZTE Coreporation and others.
# hu.zhijiang@zte.com.cn
# All rights reserved. This program and the accompanying materials
# are made available under the terms of the Apache License, Version 2.0
# which accompanies this distribution, and is available at
# http://www.apache.org/licenses/LICENSE-2.0
##############################################################################*/
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <pthread.h>
#include <signal.h>
#include "buffer.h"
#include "udp-common.h"
#include "misc.h"
#include "server.h"
void * udp_server_main(void *args)
{
pthread_t pchild;
struct sockaddr_in maddr;
int ms;
int i;
struct semlink sl;
char adr[128];
long long total = 0;
sprintf(adr, "%s%s", MCAST_ADDR_BASE, MCAST_ADDR_SUFFIX);
maddr = make_addr(adr, server_port);
mcinfo.group = maddr;
ms = init_mcast_socket(&local_addr, &maddr);
sl.this = spawn_thread(&pchild, tcp_server_main, 1);
sl.parent = NULL;
buffer_init();
/* Let TCP servers accept connections */
sl_release_child(&sl);
/* Wait until all TCP servers got all their clients */
sl_wait_child(&sl);
log(7, "UDP Server: All clients were accepted");
do {
/* one buffer round */
i = buffer_fill(STDIN_FILENO);
if (!client_count) {
/* No need to send, make it the last run of main loop, also let
TCP Server to do dummy run right away(not waiting until all
data sent). */
i = 0;
buffctl.pkt_count = 0;
}
log(7, "Signal TCP Server to send buffctl");
/* Tell tcp to Send headers, release child */
sl_release_child(&sl);
/* Wait all clients ready to start receiving */
sl_wait_child(&sl);
/* multicast data */
while (i--) {
sendto(ms, packetctl[i],
packetctl[i]->data_size + sizeof(struct packet_ctl), 0,
(struct sockaddr *)&maddr, sizeof(maddr));
}
log(7, "send finish");
/* Tell tcp to Send SERVER_SENT, release child */
sl_release_child(&sl);
/* wait tcp to send SERVER_SENT */
sl_wait_child(&sl);
total = total + buffctl.buffer_size;
log(1, "Buffer Done: sent %lld bytes", total);
} while (buffctl.pkt_count);
if (pthread_join(pchild, 0) < 0) {
crit("pthread_join() TCP Server");
}
log(1, "All Done");
close(ms);
return 0;
}
|
