1/*********************************************************************** 2 * Copyright (c) 2009, Secure Endpoints Inc. 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions 7 * are met: 8 * 9 * - Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 12 * - Redistributions in binary form must reproduce the above copyright 13 * notice, this list of conditions and the following disclaimer in 14 * the documentation and/or other materials provided with the 15 * distribution. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS 20 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE 21 * COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, 22 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 23 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR 24 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, 26 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED 28 * OF THE POSSIBILITY OF SUCH DAMAGE. 29 * 30 **********************************************************************/ 31 32#include<config.h> 33 34#include <roken.h> 35 36#ifndef _WIN32 37#error Only implemented for Windows 38#endif 39 40volatile LONG _startup_count = 0; 41 42ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL 43rk_WSAStartup(void) 44{ 45 WSADATA wsad; 46 47 if (!WSAStartup( MAKEWORD(2, 2), &wsad )) { 48 if (wsad.wVersion != MAKEWORD(2, 2)) { 49 /* huh? We can't use 2.2? */ 50 WSACleanup(); 51 return -1; 52 } 53 54 InterlockedIncrement(&_startup_count); 55 return 0; 56 } 57 58 return -1; 59} 60 61 62ROKEN_LIB_FUNCTION int ROKEN_LIB_CALL 63rk_WSACleanup(void) 64{ 65 LONG l; 66 67 if ((l = InterlockedDecrement(&_startup_count)) < 0) { 68 l = InterlockedIncrement(&_startup_count) - 1; 69 } 70 71 if (l >= 0) { 72 return WSACleanup(); 73 } 74 return -1; 75} 76