From patchwork Tue Oct 6 01:03:08 2009 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lucas Meneghel Rodrigues X-Patchwork-Id: 51846 Received: from vger.kernel.org (vger.kernel.org [209.132.176.167]) by demeter.kernel.org (8.14.2/8.14.2) with ESMTP id n9615nnR013038 for ; Tue, 6 Oct 2009 01:05:49 GMT Received: (majordomo@vger.kernel.org) by vger.kernel.org via listexpand id S1755843AbZJFBDr (ORCPT ); Mon, 5 Oct 2009 21:03:47 -0400 Received: (majordomo@vger.kernel.org) by vger.kernel.org id S1755834AbZJFBDr (ORCPT ); Mon, 5 Oct 2009 21:03:47 -0400 Received: from mx1.redhat.com ([209.132.183.28]:60598 "EHLO mx1.redhat.com" rhost-flags-OK-OK-OK-OK) by vger.kernel.org with ESMTP id S1755832AbZJFBDq (ORCPT ); Mon, 5 Oct 2009 21:03:46 -0400 Received: from int-mx02.intmail.prod.int.phx2.redhat.com (int-mx02.intmail.prod.int.phx2.redhat.com [10.5.11.12]) by mx1.redhat.com (8.13.8/8.13.8) with ESMTP id n9613K0r005103; Mon, 5 Oct 2009 21:03:20 -0400 Received: from localhost.localdomain (vpn-12-223.rdu.redhat.com [10.11.12.223]) by int-mx02.intmail.prod.int.phx2.redhat.com (8.13.8/8.13.8) with ESMTP id n9613AVC029434; Mon, 5 Oct 2009 21:03:18 -0400 From: Lucas Meneghel Rodrigues To: autotest@test.kernel.org Cc: kvm@vger.kernel.org, Lucas Meneghel Rodrigues Subject: [PATCH 5/6] KVM test: Utility to make windows guests ACK end of install Date: Mon, 5 Oct 2009 22:03:08 -0300 Message-Id: <1254790989-14204-5-git-send-email-lmr@redhat.com> In-Reply-To: <1254790989-14204-4-git-send-email-lmr@redhat.com> References: <1254790989-14204-1-git-send-email-lmr@redhat.com> <1254790989-14204-2-git-send-email-lmr@redhat.com> <1254790989-14204-3-git-send-email-lmr@redhat.com> <1254790989-14204-4-git-send-email-lmr@redhat.com> X-Scanned-By: MIMEDefang 2.67 on 10.5.11.12 Sender: kvm-owner@vger.kernel.org Precedence: bulk List-ID: X-Mailing-List: kvm@vger.kernel.org diff --git a/client/tests/kvm/deps/finish.cpp b/client/tests/kvm/deps/finish.cpp new file mode 100644 index 0000000..9c2867c --- /dev/null +++ b/client/tests/kvm/deps/finish.cpp @@ -0,0 +1,121 @@ +// Simple app that only sends an ack string to the KVM unattended install +// watch code. +// +// You must link this code with Ws2_32.lib, Mswsock.lib, and Advapi32.lib +// +// Author: Lucas Meneghel Rodrigues +// Code was adapted from an MSDN sample. + +// Usage: finish.exe [Host OS IP] + +// MinGW's ws2tcpip.h only defines getaddrinfo and other functions only for +// the case _WIN32_WINNT >= 0x0501. +#ifdef __MINGW32__ +#undef _WIN32_WINNT +#define _WIN32_WINNT 0x501 +#endif + +#include +#include +#include +#include +#include + +#define DEFAULT_BUFLEN 512 +#define DEFAULT_PORT "12323" + +int main(int argc, char **argv) +{ + WSADATA wsaData; + SOCKET ConnectSocket = INVALID_SOCKET; + struct addrinfo *result = NULL, + *ptr = NULL, + hints; + char *sendbuf = "done"; + char recvbuf[DEFAULT_BUFLEN]; + int iResult; + int recvbuflen = DEFAULT_BUFLEN; + + // Validate the parameters + if (argc != 2) { + printf("usage: %s server-name\n", argv[0]); + return 1; + } + + // Initialize Winsock + iResult = WSAStartup(MAKEWORD(2,2), &wsaData); + if (iResult != 0) { + printf("WSAStartup failed: %d\n", iResult); + return 1; + } + + ZeroMemory( &hints, sizeof(hints) ); + hints.ai_family = AF_UNSPEC; + hints.ai_socktype = SOCK_STREAM; + hints.ai_protocol = IPPROTO_TCP; + + // Resolve the server address and port + iResult = getaddrinfo(argv[1], DEFAULT_PORT, &hints, &result); + if ( iResult != 0 ) { + printf("getaddrinfo failed: %d\n", iResult); + WSACleanup(); + return 1; + } + + // Attempt to connect to an address until one succeeds + for(ptr=result; ptr != NULL ;ptr=ptr->ai_next) { + + // Create a SOCKET for connecting to server + ConnectSocket = socket(ptr->ai_family, ptr->ai_socktype, + ptr->ai_protocol); + if (ConnectSocket == INVALID_SOCKET) { + printf("Error at socket(): %ld\n", WSAGetLastError()); + freeaddrinfo(result); + WSACleanup(); + return 1; + } + + // Connect to server. + iResult = connect( ConnectSocket, ptr->ai_addr, (int)ptr->ai_addrlen); + if (iResult == SOCKET_ERROR) { + closesocket(ConnectSocket); + ConnectSocket = INVALID_SOCKET; + continue; + } + break; + } + + freeaddrinfo(result); + + if (ConnectSocket == INVALID_SOCKET) { + printf("Unable to connect to server!\n"); + WSACleanup(); + return 1; + } + + // Send the ACK buffer + iResult = send( ConnectSocket, sendbuf, (int)strlen(sendbuf), 0 ); + if (iResult == SOCKET_ERROR) { + printf("send failed: %d\n", WSAGetLastError()); + closesocket(ConnectSocket); + WSACleanup(); + return 1; + } + + printf("Bytes Sent: %ld\n", iResult); + + // shutdown the connection since no more data will be sent + iResult = shutdown(ConnectSocket, SD_SEND); + if (iResult == SOCKET_ERROR) { + printf("shutdown failed: %d\n", WSAGetLastError()); + closesocket(ConnectSocket); + WSACleanup(); + return 1; + } + + // cleanup + closesocket(ConnectSocket); + WSACleanup(); + + return 0; +}