Code:
#include <windows.h>
#pragma comment(lib, "wsock32.lib")
#include <winsock.h>
#include <stdio.h>
#define NETWORK_ERROR -1
#define NETWORK_OK 0
void ReportError(int, const char *);
int WINAPI WinMain(HINSTANCE hInst, HINSTANCE hPrevInst, LPSTR lpCmd, int nShow) {
WORD sockVersion;
WSADATA wsaData;
int nret;
SOCKET listeningSocket;
SOCKADDR_IN serverInfo;
SOCKET theClient;
sockVersion = MAKEWORD(1, 1); // We'd like Winsock version 1.1
// We begin by initializing Winsock
WSAStartup(sockVersion, &wsaData);
// Next, create the listening socket
listeningSocket = socket(AF_INET, // Go over TCP/IP
SOCK_STREAM, // This is a stream-oriented socket
IPPROTO_TCP); // Use TCP rather than UDP
if (listeningSocket == INVALID_SOCKET) {
nret = WSAGetLastError(); // Get a more detailed error
ReportError(nret, "socket()"); // Report the error with our custom function
WSACleanup(); // Shutdown Winsock
return NETWORK_ERROR; // Return an error value
}
// Use a SOCKADDR_IN struct to fill in address information
serverInfo.sin_family = AF_INET;
serverInfo.sin_addr.s_addr = INADDR_ANY; // Since this socket is listening for connections,
// any local address will do
serverInfo.sin_port = htons(8888); // Convert integer 8888 to network-byte order
// and insert into the port field
// Bind the socket to our local server address
nret = bind(listeningSocket, (LPSOCKADDR)&serverInfo, sizeof(struct sockaddr));
if (nret == SOCKET_ERROR) {
nret = WSAGetLastError();
ReportError(nret, "bind()");
WSACleanup();
return NETWORK_ERROR;
}
// Make the socket listen
nret = listen(listeningSocket, 10); // Up to 10 connections may wait at any
// one time to be accept()'ed
if (nret == SOCKET_ERROR) {
nret = WSAGetLastError();
ReportError(nret, "listen()");
WSACleanup();
return NETWORK_ERROR;
}
// Wait for a client
while(1)
{
theClient = accept(listeningSocket,
NULL, // Address of a sockaddr structure (see explanation below)
NULL); // Address of a variable containing size of sockaddr struct
MessageBox(hInst, "CONNECTED", "This program is:", MB_OK | MB_ICONINFORMATION);
if (theClient == INVALID_SOCKET) {
nret = WSAGetLastError();
ReportError(nret, "accept()");
WSACleanup();
return NETWORK_ERROR;
}
else
{
char buffer[256]; // On the stack
// char *buffer = new char[256]; // or on the heap
MessageBox(hInst, "CONNECTED", "This program is:", MB_OK | MB_ICONINFORMATION);
nret = recv(theClient,
buffer,
256, // Complete size of buffer
0);
if (nret == SOCKET_ERROR) {// Get a specific code// Handle accordingly
return NETWORK_ERROR; }
else if (buffer == "check_network")
{
MessageBox(hInst, "checkNW", "This program is:", MB_OK | MB_ICONINFORMATION);
;}
else if (buffer == "shutdown")
{
MessageBox(hInst, "shutdown", "This program is:", MB_OK | MB_ICONINFORMATION);
;}
else if (buffer == "restart")
{
MessageBox(hInst, "restartt", "This program is:", MB_OK | MB_ICONINFORMATION);
;}
else if (buffer == "lower_volume")
{
MessageBox(hInst, "lowervol", "This program is:", MB_OK | MB_ICONINFORMATION);
;}
// listeningSocket = theClient;
}
;
}
//delete [] buffer; // Manipulate buffer, then delete if and only if
// buffer was allocated on heap
// nret contains the number of bytes received
// listeningSocket = theClient;
// Send and receive from the client, and finally,
closesocket(theClient);
closesocket(listeningSocket);
// Shutdown Winsock
WSACleanup();
return NETWORK_OK;
}
void ReportError(int errorCode, const char *whichFunc) {
char errorMsg[92]; // Declare a buffer to hold
// the generated error message
ZeroMemory(errorMsg, 92); // Automatically NULL-terminate the string
// The following line copies the phrase, whichFunc string, and integer errorCode into the buffer
sprintf(errorMsg, "Call to %s returned error %d!", (char *)whichFunc, errorCode);
MessageBox(NULL, errorMsg, "socketIndication", MB_OK);
}