void WFTestUdpServer_Process(void)
{
unsigned char pcString[30];
int rc;
WORD i,count;
struct mCast_testing_context_t *mCast_Context = &mCast_testing_context_1;
if(socket_1 == INVALID_UDP_SOCKET)
{
mCast_Context->StateMachine = MDNS_RESPONDER_INIT;
}
switch(mCast_Context->StateMachine)
{
case MDNS_RESPONDER_INIT:
printf("Opening mCast socket_1 \r\n");
socket_1 = TCPIP_UDP_ServerOpen(IP_ADDRESS_TYPE_IPV4, port_1, 0);
socket_2 = TCPIP_UDP_ServerOpen(IP_ADDRESS_TYPE_IPV4, port_2, 0);
if(socket_1 == INVALID_UDP_SOCKET)
{
printf("Can't open mCast UDP-Socket \r\n");
return;
}
else
mCast_Context->StateMachine = MDNS_RESPONDER_LISTEN ;
/* Called from TCPIP_MDNS_Initialize. So return immediately */
break;
case MDNS_RESPONDER_LISTEN:
{
char buf_t[10];
count = TCPIP_UDP_GetIsReady(socket_1);
if(count == 0)return;
printf("[Get MCAST package at <");
while(count > 0)
{
int num = count > 30? 30:count;
count -= num;
rc = TCPIP_UDP_ArrayGet(socket_1, pcString, num);
for(i=0;i<rc;i++)
{
sprintf(buf_t,"%02x ",pcString[i]);
printf(buf_t);
}
}
printf(" ]\r\n\r\n");
break;
}
default:
break;
}
return;
}
void APP_Tasks( void )
{
static IPV4_ADDR dwLastIP[2] = { {-1}, {-1} };
IPV4_ADDR ipAddr;
int i;
switch(appData.state)
{
case APP_TCPIP_WAIT_FOR_IP:
// if the IP address of an interface has changed
// display the new value on the system console
nNets = TCPIP_STACK_NumberOfNetworksGet();
for (i = 0; i < nNets; i++)
{
netH = TCPIP_STACK_IndexToNet(i);
ipAddr.Val = TCPIP_STACK_NetAddress(netH);
if(dwLastIP[i].Val != ipAddr.Val)
{
dwLastIP[i].Val = ipAddr.Val;
SYS_CONSOLE_MESSAGE(TCPIP_STACK_NetNameGet(netH));
SYS_CONSOLE_MESSAGE(" IP Address: ");
SYS_CONSOLE_PRINT("%d.%d.%d.%d \r\n", ipAddr.v[0], ipAddr.v[1], ipAddr.v[2], ipAddr.v[3]);
if (ipAddr.v[0] != 0 && ipAddr.v[0] != 169) // Wait for a Valid IP
{
appData.state = APP_TCPIP_OPENING_SERVER;
}
}
}
break;
case APP_TCPIP_OPENING_SERVER:
{
SYS_CONSOLE_PRINT("Waiting for Client Connection on port: %d\r\n", SERVER_PORT);
appData.socket = TCPIP_UDP_ServerOpen(IP_ADDRESS_TYPE_IPV4, SERVER_PORT, 0);
if (appData.socket == INVALID_SOCKET)
{
SYS_CONSOLE_MESSAGE("Couldn't open server socket\r\n");
break;
}
appData.state = APP_TCPIP_WAIT_FOR_CONNECTION;
}
break;
case APP_TCPIP_WAIT_FOR_CONNECTION:
{
if (!TCPIP_UDP_IsConnected(appData.socket))
{
return;
}
else
{
// We got a connection
appData.state = APP_TCPIP_SERVING_CONNECTION;
SYS_CONSOLE_MESSAGE("Received a connection\r\n");
}
}
break;
case APP_TCPIP_SERVING_CONNECTION:
{
if (!TCPIP_UDP_IsConnected(appData.socket))
{
appData.state = APP_TCPIP_CLOSING_CONNECTION;
SYS_CONSOLE_MESSAGE("Connection was closed\r\n");
break;
}
int16_t wMaxGet, wMaxPut, wCurrentChunk;
uint16_t w, w2;
uint8_t AppBuffer[32];
// Figure out how many bytes have been received and how many we can transmit.
wMaxGet = TCPIP_UDP_GetIsReady(appData.socket); // Get UDP RX FIFO byte count
wMaxPut = UDPIsPutReady(appData.socket);
//SYS_CONSOLE_PRINT("\t%d bytes are available.\r\n", wMaxGet);
if (wMaxGet == 0)
{
break;
}
if (wMaxPut < wMaxGet)
{
wMaxGet = wMaxPut;
}
// Process all bytes that we can
// This is implemented as a loop, processing up to sizeof(AppBuffer) bytes at a time.
// This limits memory usage while maximizing performance. Single byte Gets and Puts are a lot slower than multibyte GetArrays and PutArrays.
wCurrentChunk = sizeof(AppBuffer);
for(w = 0; w < wMaxGet; w += sizeof(AppBuffer))
{
// Make sure the last chunk, which will likely be smaller than sizeof(AppBuffer), is treated correctly.
if(w + sizeof(AppBuffer) > wMaxGet)
wCurrentChunk = wMaxGet - w;
// Transfer the data out of the TCP RX FIFO and into our local processing buffer.
TCPIP_UDP_ArrayGet(appData.socket, AppBuffer, wCurrentChunk);
SYS_CONSOLE_PRINT("\tReceived a message of '%s'\r\n", AppBuffer);
// Perform the "ToUpper" operation on each data byte
for(w2 = 0; w2 < wCurrentChunk; w2++)
{
i = AppBuffer[w2];
if(i >= 'a' && i <= 'z')
{
i -= ('a' - 'A');
AppBuffer[w2] = i;
}
else if(i == '\e') //escape
{
SYS_CONSOLE_MESSAGE("Connection was closed\r\n");
}
}
SYS_CONSOLE_PRINT("\tSending a messages '%s'\r\n", AppBuffer);
// Transfer the data out of our local processing buffer and into the TCP TX FIFO.
TCPIP_UDP_ArrayPut(appData.socket, AppBuffer, wCurrentChunk);
TCPIP_UDP_Flush(appData.socket);
appData.state = APP_TCPIP_CLOSING_CONNECTION;
}
}
break;
case APP_TCPIP_CLOSING_CONNECTION:
{
// Close the socket connection.
TCPIP_UDP_Close(appData.socket);
appData.state = APP_TCPIP_OPENING_SERVER;
}
break;
default:
break;
}
}
// sntp_manager.h
bool TCPIP_SNTP_Client(TCPIP_NET_IF* pNetIf)
{
NTP_PACKET pkt;
uint16_t w;
DNS_RESULT dnsRes;
static SYS_TICK SNTPTimer;
if(pSntpIf != 0 && pNetIf != pSntpIf)
{ // not our job
return false;
}
switch(sntpState)
{
case SM_HOME:
sntpSocket = INVALID_UDP_SOCKET;
pSntpIf = pSntpDefIf;
if(!TCPIP_STACK_NetworkIsLinked(pSntpIf))
{
pSntpIf = _TCPIPStackAnyNetLinked(true);
}
if(TCPIP_DNS_UsageBegin(pSntpIf) != DNS_RES_OK)
{
ntpLastError = SNTP_RES_NTP_DNS_ERR;
break;
}
TCPIP_DNS_Resolve(NTP_SERVER, ntpConnection ==IP_ADDRESS_TYPE_IPV6 ? DNS_TYPE_AAAA : DNS_TYPE_A);
sntpState++;
break;
case SM_WAIT_DNS:
dnsRes = TCPIP_DNS_IsResolved(NTP_SERVER, &serverIP);
if(dnsRes == DNS_RES_PENDING)
{ // ongoing operation;
break;
}
else if(dnsRes < 0)
{ // some DNS error occurred; retry after waiting a while
SNTPTimer = SYS_TICK_Get();
sntpState = SM_SHORT_WAIT;
ntpLastError = SNTP_RES_NTP_DNS_ERR;
}
else
{
sntpState++;
}
TCPIP_DNS_UsageEnd(pSntpIf);
break;
case SM_DNS_RESOLVED:
sntpSocket = TCPIP_UDP_ClientOpen(ntpConnection, NTP_SERVER_PORT, &serverIP);
if(sntpSocket != INVALID_UDP_SOCKET)
{
TCPIP_UDP_SocketNetSet(sntpSocket, pSntpIf);
sntpState++;
SNTPTimer = SYS_TICK_Get();
}
else
{
ntpLastError = SNTP_RES_SKT_ERR;
}
break;
case SM_UDP_IS_OPENED:
if(TCPIP_UDP_IsOpened(sntpSocket) == true)
{
SNTPTimer = SYS_TICK_Get();
sntpState = SM_UDP_SEND;
}
else if((SYS_TICK_Get() - SNTPTimer > 1*SYS_TICK_TicksPerSecondGet()))
{ // failed to open
TCPIP_UDP_Close(sntpSocket);
sntpState = SM_DNS_RESOLVED;
sntpSocket = INVALID_UDP_SOCKET;
ntpLastError = SNTP_RES_SKT_ERR;
}
break;
case SM_UDP_SEND:
// Open up the sending UDP socket
// Make certain the socket can be written to
if(!TCPIP_UDP_TxPutIsReady(sntpSocket, sizeof(pkt)))
{ // Wait no more than 1 sec
if((SYS_TICK_Get() - SNTPTimer > 1*SYS_TICK_TicksPerSecondGet()))
{
TCPIP_UDP_Close(sntpSocket);
sntpState = SM_DNS_RESOLVED;
sntpSocket = INVALID_UDP_SOCKET;
ntpLastError = SNTP_RES_SKT_ERR;
break;
}
}
// Success
// Transmit a time request packet
memset(&pkt, 0, sizeof(pkt));
pkt.flags.versionNumber = NTP_VERSION;
pkt.flags.mode = 3; // NTP Client
pkt.orig_ts_secs = TCPIP_Helper_htonl(NTP_EPOCH);
TCPIP_UDP_ArrayPut(sntpSocket, (uint8_t*) &pkt, sizeof(pkt));
TCPIP_UDP_Flush(sntpSocket);
SNTPTimer = SYS_TICK_Get();
sntpState = SM_UDP_RECV;
break;
case SM_UDP_RECV:
// Look for a response time packet
if(!TCPIP_UDP_GetIsReady(sntpSocket))
{
if((SYS_TICK_Get()) - SNTPTimer > NTP_REPLY_TIMEOUT * SYS_TICK_TicksPerSecondGet())
{
// Abort the request and resume
TCPIP_UDP_Close(sntpSocket);
//SNTPTimer = SYS_TICK_Get();
//sntpState = SM_SHORT_WAIT;
sntpState = SM_HOME;
sntpSocket = INVALID_UDP_SOCKET;
ntpLastError = SNTP_RES_NTP_SERVER_TMO;
}
break;
}
// Get the response time packet
w = TCPIP_UDP_ArrayGet(sntpSocket, (uint8_t*) &pkt, sizeof(pkt));
TCPIP_UDP_Close(sntpSocket);
SNTPTimer = SYS_TICK_Get();
sntpState = SM_WAIT;
sntpSocket = INVALID_UDP_SOCKET;
// sanity packet check
if(w != sizeof(pkt) || pkt.flags.versionNumber != NTP_VERSION )
{
ntpLastError = SNTP_RES_NTP_VERSION_ERR;
break;
}
if((pkt.tx_ts_secs == 0 && pkt.tx_ts_fraq == 0))
{
ntpLastError = SNTP_RES_NTP_TSTAMP_ERR;
break;
}
if(pkt.stratum == 0 )
{
ntpLastError = SNTP_RES_NTP_KOD_ERR;
break;
}
if(pkt.stratum >= NTP_MAX_STRATUM || pkt.flags.leapIndicator == 3 )
{
ntpLastError = SNTP_RES_NTP_SYNC_ERR;
break;
}
// get the last timestamp
ntpTimeStamp.tStampSeconds = pkt.tx_ts_secs;
ntpTimeStamp.tStampFraction = pkt.tx_ts_fraq;
ntpLastStampTick = SYS_TICK_Get();
// Set out local time to match the returned time
dwLastUpdateTick = ntpLastStampTick;
dwSNTPSeconds = TCPIP_Helper_ntohl(pkt.tx_ts_secs) - NTP_EPOCH;
// Do rounding. If the partial seconds is > 0.5 then add 1 to the seconds count.
if(((uint8_t*)&pkt.tx_ts_fraq)[0] & 0x80)
dwSNTPSeconds++;
break;
case SM_SHORT_WAIT:
// Attempt to requery the NTP server after a specified NTP_FAST_QUERY_INTERVAL time (ex: 8 seconds) has elapsed.
if(SYS_TICK_Get() - SNTPTimer > (NTP_FAST_QUERY_INTERVAL * SYS_TICK_TicksPerSecondGet()))
{
sntpState = SM_HOME;
sntpSocket = INVALID_UDP_SOCKET;
}
break;
case SM_WAIT:
// Requery the NTP server after a specified NTP_QUERY_INTERVAL time (ex: 10 minutes) has elapsed.
if(SYS_TICK_Get() - SNTPTimer > (NTP_QUERY_INTERVAL * SYS_TICK_TicksPerSecondGet()))
{
sntpState = SM_HOME;
sntpSocket = INVALID_UDP_SOCKET;
}
break;
}
return true;
}//TCPIP_SNTP_Client
/*********************************************************************
* Function: bool TCPIP_ANNOUNCE_Task(void)
*
* Summary: Announce callback task.
*
* PreCondition: Stack is initialized()
*
* Input: None
*
* Output: None
*
* Side Effects: None
*
* Overview: Recurring task used to listen for Discovery
* messages on the specified TCPIP_ANNOUNCE_PORT. These
* messages can be sent using the TCP/IP
* Discoverer tool. If one is received, this
* function will transmit a reply.
*
* Note: A UDP socket must be available before this
* function is called. It is freed at the end of
* the function. UDP_MAX_SOCKETS may need to be
* increased if other modules use UDP sockets.
********************************************************************/
bool TCPIP_ANNOUNCE_Task(TCPIP_NET_IF * pNetIf)
{
uint8_t i;
int netIx;
UDP_SOCKET s;
if(!pNetIf)
{
return false;
}
else
{
netIx = TCPIP_STACK_NetIxGet(pNetIf);
}
s = announceDcpt.skt;
switch(announceDcpt.sm)
{
case DISCOVERY_HOME:
// Open a UDP socket for inbound and outbound transmission
// Allow receive on any interface
s = TCPIP_UDP_ServerOpen(IP_ADDRESS_TYPE_IPV4, TCPIP_ANNOUNCE_PORT, 0);
if(s == INVALID_UDP_SOCKET)
{
return false;
}
if(!TCPIP_UDP_RemoteBind(s, IP_ADDRESS_TYPE_IPV4, TCPIP_ANNOUNCE_PORT, 0))
{
TCPIP_UDP_Close(s);
break;
}
if(!TCPIP_UDP_OptionsSet(s, UDP_OPTION_STRICT_PORT, (void*)true))
{
TCPIP_UDP_Close(s);
break;
}
announceDcpt.skt = s;
announceDcpt.sm++;
break;
case DISCOVERY_LISTEN:
// Do nothing if no data is waiting
if(!TCPIP_UDP_GetIsReady(s))
return false;
// See if this is a discovery query or reply
TCPIP_UDP_Get(s, &i);
TCPIP_UDP_Discard(s);
if(i != 'D')
return false;
// We received a discovery request, reply when we can
announceDcpt.sm++;
// No break needed. If we get down here, we are now ready for the DISCOVERY_REQUEST_RECEIVED state
case DISCOVERY_REQUEST_RECEIVED:
ANNOUNCE_Notify (pNetIf, DHCP_EVENT_BOUND, NULL); // fake a legitimate DHCP event
// Listen for other discovery requests
announceDcpt.sm = DISCOVERY_LISTEN;
break;
case DISCOVERY_DISABLED:
break;
}
return true;
}
void APP_Tasks( void )
{
static IPV4_ADDR dwLastIP[2] = { {-1}, {-1} };
IPV4_ADDR ipAddr;
int i;
switch(appData.state)
{
case APP_TCPIP_WAIT_FOR_IP:
// if the IP address of an interface has changed
// display the new value on the system console
nNets = TCPIP_STACK_NumberOfNetworksGet();
for (i = 0; i < nNets; i++)
{
netH = TCPIP_STACK_IndexToNet(i);
ipAddr.Val = TCPIP_STACK_NetAddress(netH);
if(dwLastIP[i].Val != ipAddr.Val)
{
dwLastIP[i].Val = ipAddr.Val;
SYS_CONSOLE_MESSAGE(TCPIP_STACK_NetNameGet(netH));
SYS_CONSOLE_MESSAGE(" IP Address: ");
SYS_CONSOLE_PRINT("%d.%d.%d.%d \r\n", ipAddr.v[0], ipAddr.v[1], ipAddr.v[2], ipAddr.v[3]);
if (ipAddr.v[0] != 0 && ipAddr.v[0] != 169) // Wait for a Valid IP
{
appData.state = APP_TCPIP_WAITING_FOR_COMMAND;
SYS_CONSOLE_MESSAGE("Waiting for command type: sendudppacket\r\n");
}
}
}
break;
case APP_TCPIP_WAITING_FOR_COMMAND:
{
if (APP_Send_Packet)
{
APP_Send_Packet = false;
DNS_RESULT result = TCPIP_DNS_UsageBegin(0);
if (result != DNS_RES_OK)
{
SYS_CONSOLE_MESSAGE("Error in DNS aborting 1\r\n");
break;
}
result = TCPIP_DNS_Resolve(APP_Hostname_Buffer, DNS_TYPE_A);
if (result != DNS_RES_OK)
{
SYS_CONSOLE_MESSAGE("Error in DNS aborting 2\r\n");
TCPIP_DNS_UsageEnd(0);
break;
}
appData.state = APP_TCPIP_WAIT_ON_DNS;
}
}
break;
case APP_TCPIP_WAIT_ON_DNS:
{
IPV4_ADDR addr;
switch (_APP_PumpDNS(APP_Hostname_Buffer, &addr))
{
case -1:
{
// Some sort of error, already reported
appData.state = APP_TCPIP_WAITING_FOR_COMMAND;
}
break;
case 0:
{
// Still waiting
}
break;
case 1:
{
uint16_t port = atoi(APP_Port_Buffer);
appData.socket = TCPIP_UDP_ClientOpen(IP_ADDRESS_TYPE_IPV4,
port,
(IP_MULTI_ADDRESS*) &addr);
if (appData.socket == INVALID_SOCKET)
{
SYS_CONSOLE_MESSAGE("Could not start connection\r\n");
appData.state = APP_TCPIP_WAITING_FOR_COMMAND;
}
SYS_CONSOLE_MESSAGE("Starting connection\r\n");
appData.state = APP_TCPIP_WAIT_FOR_CONNECTION;
}
break;
}
}
break;
case APP_TCPIP_WAIT_FOR_CONNECTION:
{
if (!TCPIP_UDP_IsConnected(appData.socket))
{
break;
}
if(UDPIsPutReady(appData.socket) == 0)
{
break;
}
TCPIP_UDP_ArrayPut(appData.socket, (uint8_t*)APP_Message_Buffer, strlen(APP_Message_Buffer));
TCPIP_UDP_Flush(appData.socket);
appData.state = APP_TCPIP_WAIT_FOR_RESPONSE;
}
break;
case APP_TCPIP_WAIT_FOR_RESPONSE:
{
char buffer[180];
memset(buffer, 0, sizeof(buffer));
if (!TCPIP_UDP_IsConnected(appData.socket))
{
SYS_CONSOLE_MESSAGE("\r\nConnection Closed\r\n");
appData.state = APP_TCPIP_WAITING_FOR_COMMAND;
break;
}
if (TCPIP_UDP_GetIsReady(appData.socket))
{
TCPIP_UDP_ArrayGet(appData.socket, (uint8_t*)buffer, sizeof(buffer) - 1);
SYS_CONSOLE_MESSAGE(buffer);
appData.state = APP_TCPIP_WAITING_FOR_COMMAND;
}
}
break;
default:
break;
}
}
