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