155 lines
5.1 KiB
C
155 lines
5.1 KiB
C
|
#include <nt.h>
|
||
|
#include <ntrtl.h>
|
||
|
#include <nturtl.h>
|
||
|
#include <windows.h>
|
||
|
#include <winsock2.h>
|
||
|
#include <wsipx.h>
|
||
|
#include <svcguid.h>
|
||
|
#include <stdio.h>
|
||
|
#include <stdlib.h>
|
||
|
#include <rpc.h>
|
||
|
#include <rpcdce.h>
|
||
|
|
||
|
|
||
|
GUID ServiceClassId = { /* 5b50962a-e5a5-11cf-a555-00c04fd8d4ac */
|
||
|
0x5b50962a,
|
||
|
0xe5a5,
|
||
|
0x11cf,
|
||
|
{0xa5, 0x55, 0x00, 0xc0, 0x4f, 0xd8, 0xd4, 0xac}
|
||
|
};
|
||
|
|
||
|
WCHAR ServiceInstanceName[] = L"GlennC on GLENNC_PRO";
|
||
|
|
||
|
_cdecl
|
||
|
main(int argc, char **argv)
|
||
|
{
|
||
|
DWORD ret;
|
||
|
WSADATA wsaData;
|
||
|
WSAQUERYSET QuerySet;
|
||
|
CSADDR_INFO CSAddrInfo[6];
|
||
|
SOCKADDR SocketAddress1;
|
||
|
SOCKADDR SocketAddress2;
|
||
|
SOCKADDR SocketAddress3;
|
||
|
SOCKADDR SocketAddress4;
|
||
|
SOCKADDR SocketAddress5;
|
||
|
ANSI_STRING asServiceInstanceName;
|
||
|
UNICODE_STRING usServiceInstanceName;
|
||
|
WCHAR UnicodeStringBuf[1024];
|
||
|
|
||
|
if ( argc == 2 )
|
||
|
{
|
||
|
usServiceInstanceName.Length = 0;
|
||
|
usServiceInstanceName.MaximumLength = 1024;
|
||
|
usServiceInstanceName.Buffer = UnicodeStringBuf;
|
||
|
|
||
|
RtlInitAnsiString( &asServiceInstanceName, argv[1] );
|
||
|
|
||
|
RtlAnsiStringToUnicodeString( &usServiceInstanceName,
|
||
|
&asServiceInstanceName,
|
||
|
FALSE );
|
||
|
}
|
||
|
|
||
|
RtlZeroMemory( (LPBYTE) &QuerySet, sizeof( WSAQUERYSET ) );
|
||
|
RtlZeroMemory( (LPBYTE) &CSAddrInfo[0], sizeof( CSADDR_INFO ) );
|
||
|
RtlZeroMemory( (LPBYTE) &CSAddrInfo[1], sizeof( CSADDR_INFO ) );
|
||
|
RtlZeroMemory( (LPBYTE) &CSAddrInfo[2], sizeof( CSADDR_INFO ) );
|
||
|
RtlZeroMemory( (LPBYTE) &CSAddrInfo[3], sizeof( CSADDR_INFO ) );
|
||
|
RtlZeroMemory( (LPBYTE) &CSAddrInfo[4], sizeof( CSADDR_INFO ) );
|
||
|
RtlZeroMemory( (LPBYTE) &CSAddrInfo[5], sizeof( CSADDR_INFO ) );
|
||
|
|
||
|
SocketAddress1.sa_family = AF_UNIX;
|
||
|
RtlFillMemory( (LPBYTE) &SocketAddress1.sa_data, 14, 1 );
|
||
|
|
||
|
SocketAddress2.sa_family = AF_INET;
|
||
|
RtlFillMemory( (LPBYTE) &SocketAddress2.sa_data, 14, 2 );
|
||
|
|
||
|
SocketAddress3.sa_family = AF_IPX;
|
||
|
RtlFillMemory( (LPBYTE) &SocketAddress3.sa_data, 14, 6 );
|
||
|
|
||
|
SocketAddress4.sa_family = AF_ISO;
|
||
|
RtlFillMemory( (LPBYTE) &SocketAddress4.sa_data, 14, 7 );
|
||
|
|
||
|
SocketAddress5.sa_family = AF_ECMA;
|
||
|
RtlFillMemory( (LPBYTE) &SocketAddress5.sa_data, 14, 8 );
|
||
|
|
||
|
CSAddrInfo[0].LocalAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[0].LocalAddr.lpSockaddr = &SocketAddress2;
|
||
|
CSAddrInfo[0].RemoteAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[0].RemoteAddr.lpSockaddr = &SocketAddress2;
|
||
|
CSAddrInfo[0].iSocketType = SOCK_RAW;
|
||
|
CSAddrInfo[0].iProtocol = PF_INET;
|
||
|
|
||
|
CSAddrInfo[1].LocalAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[1].LocalAddr.lpSockaddr = &SocketAddress2;
|
||
|
CSAddrInfo[1].RemoteAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[1].RemoteAddr.lpSockaddr = &SocketAddress2;
|
||
|
CSAddrInfo[1].iSocketType = SOCK_STREAM;
|
||
|
CSAddrInfo[1].iProtocol = PF_INET;
|
||
|
|
||
|
CSAddrInfo[2].LocalAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[2].LocalAddr.lpSockaddr = &SocketAddress1;
|
||
|
CSAddrInfo[2].RemoteAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[2].RemoteAddr.lpSockaddr = &SocketAddress1;
|
||
|
CSAddrInfo[2].iSocketType = SOCK_STREAM;
|
||
|
CSAddrInfo[2].iProtocol = PF_UNIX;
|
||
|
|
||
|
CSAddrInfo[3].LocalAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[3].LocalAddr.lpSockaddr = &SocketAddress3;
|
||
|
CSAddrInfo[3].RemoteAddr.iSockaddrLength = 0;
|
||
|
CSAddrInfo[3].RemoteAddr.lpSockaddr = NULL;
|
||
|
CSAddrInfo[3].iSocketType = SOCK_STREAM;
|
||
|
CSAddrInfo[3].iProtocol = PF_IPX;
|
||
|
|
||
|
CSAddrInfo[4].LocalAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[4].LocalAddr.lpSockaddr = &SocketAddress4;
|
||
|
CSAddrInfo[4].RemoteAddr.iSockaddrLength = 0;
|
||
|
CSAddrInfo[4].RemoteAddr.lpSockaddr = NULL;
|
||
|
CSAddrInfo[4].iSocketType = SOCK_STREAM;
|
||
|
CSAddrInfo[4].iProtocol = PF_ISO;
|
||
|
|
||
|
CSAddrInfo[5].LocalAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[5].LocalAddr.lpSockaddr = &SocketAddress5;
|
||
|
CSAddrInfo[5].RemoteAddr.iSockaddrLength = sizeof( SOCKADDR );
|
||
|
CSAddrInfo[5].RemoteAddr.lpSockaddr = &SocketAddress5;
|
||
|
CSAddrInfo[5].iSocketType = SOCK_STREAM;
|
||
|
CSAddrInfo[5].iProtocol = PF_ECMA;
|
||
|
|
||
|
QuerySet.dwSize = sizeof( WSAQUERYSET );
|
||
|
QuerySet.lpServiceClassId = &ServiceClassId;
|
||
|
if ( argc == 2 )
|
||
|
{
|
||
|
QuerySet.lpszServiceInstanceName = usServiceInstanceName.Buffer;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
QuerySet.lpszServiceInstanceName = ServiceInstanceName;
|
||
|
}
|
||
|
QuerySet.dwNameSpace = NS_ALL;
|
||
|
QuerySet.dwNumberOfCsAddrs = 6;
|
||
|
QuerySet.lpcsaBuffer = &CSAddrInfo;
|
||
|
|
||
|
WSAStartup( MAKEWORD(1, 1), &wsaData );
|
||
|
|
||
|
ret = WSASetService( &QuerySet,
|
||
|
RNRSERVICE_DELETE,
|
||
|
SERVICE_MULTIPLE );
|
||
|
|
||
|
if ( ret )
|
||
|
{
|
||
|
printf("Error: WSASetService returned 0x%X\n", ret );
|
||
|
printf(" GetLastError returned 0x%X\n", GetLastError() );
|
||
|
|
||
|
WSACleanup();
|
||
|
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
printf( "SetService was successful.\n" );
|
||
|
|
||
|
WSACleanup();
|
||
|
|
||
|
return(0);
|
||
|
}
|
||
|
|
||
|
|