src/gallium/auxiliary/os/os_stream_wd.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2008-2010 VMware, Inc.
   4  * All Rights Reserved.
   5  *
   6  * Permission is hereby granted, free of charge, to any person obtaining a
   7  * copy of this software and associated documentation files (the
   8  * "Software"), to deal in the Software without restriction, including
   9  * without limitation the rights to use, copy, modify, merge, publish,
  10  * distribute, sub license, and/or sell copies of the Software, and to
  11  * permit persons to whom the Software is furnished to do so, subject to
  12  * the following conditions:
  13  *
  14  * The above copyright notice and this permission notice (including the
  15  * next paragraph) shall be included in all copies or substantial portions
  16  * of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  19  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  20  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NON-INFRINGEMENT.
  21  * IN NO EVENT SHALL VMWARE AND/OR ITS SUPPLIERS BE LIABLE FOR
  22  * ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
  23  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
  24  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
  25  *
  26  **************************************************************************/
  27
  28 /**
  29  * @file
  30  * Stream implementation for the Windows Display driver.
  31  */
  32
  33 #include "pipe/p_config.h"
  34
  35 #if defined(PIPE_SUBSYSTEM_WINDOWS_DISPLAY)
  36
  37 #include <windows.h>
  38 #include <winddi.h>
  39
  40 #include "os_memory.h"
  41 #include "os_stream.h"
  42
  43
  44 #define MAP_FILE_SIZE (4*1024*1024)
  45
  46
  47 struct os_stream
  48 {
  49    char filename[MAX_PATH + 1];
  50    WCHAR wFileName[MAX_PATH + 1];
  51    boolean growable;
  52    size_t map_size;
  53    ULONG_PTR iFile;
  54    char *pMap;
  55    size_t written;
  56    unsigned suffix;
  57 };
  58
  59
  60 static INLINE boolean
  61 os_stream_map(struct os_stream *stream)
  62 {
  63    ULONG BytesInUnicodeString;
  64    static char filename[MAX_PATH + 1];
  65    unsigned filename_len;
  66
  67    if(stream->growable)
  68       filename_len = snprintf(filename,
  69                               sizeof(filename),
  70                               "%s.%04x",
  71                               stream->filename,
  72                               stream->suffix++);
  73    else
  74       filename_len = snprintf(filename,
  75                               sizeof(filename),
  76                               "%s",
  77                               stream->filename);
  78
  79    EngMultiByteToUnicodeN(
  80          stream->wFileName,
  81          sizeof(stream->wFileName),
  82          &BytesInUnicodeString,
  83          filename,
  84          filename_len);
  85
  86    stream->pMap = EngMapFile(stream->wFileName, stream->map_size, &stream->iFile);
  87    if(!stream->pMap)
  88       return FALSE;
  89
  90    memset(stream->pMap, 0, stream->map_size);
  91    stream->written = 0;
  92
  93    return TRUE;
  94 }
  95
  96
  97 static INLINE void
  98 os_stream_unmap(struct os_stream *stream)
  99 {
 100    EngUnmapFile(stream->iFile);
 101    if(stream->written < stream->map_size) {
 102       /* Truncate file size */
 103       stream->pMap = EngMapFile(stream->wFileName, stream->written, &stream->iFile);
 104       if(stream->pMap)
 105          EngUnmapFile(stream->iFile);
 106    }
 107
 108    stream->pMap = NULL;
 109 }
 110
 111
 112 static INLINE void
 113 os_stream_full_qualified_filename(char *dst, size_t size, const char *src)
 114 {
 115    boolean need_drive, need_root;
 116
 117    if((('A' <= src[0] && src[0] <= 'Z') || ('a' <= src[0] && src[0] <= 'z')) && src[1] == ':') {
 118       need_drive = FALSE;
 119       need_root = src[2] == '\\' ? FALSE : TRUE;
 120    }
 121    else {
 122       need_drive = TRUE;
 123       need_root = src[0] == '\\' ? FALSE : TRUE;
 124    }
 125
 126    snprintf(dst, size,
 127             "\\??\\%s%s%s",
 128             need_drive ? "C:" : "",
 129             need_root ? "\\" : "",
 130             src);
 131 }
 132
 133
 134 struct os_stream *
 135 os_stream_create(const char *filename, size_t max_size)
 136 {
 137    struct os_stream *stream;
 138
 139    stream = CALLOC_STRUCT(os_stream);
 140    if(!stream)
 141       goto error1;
 142
 143    os_stream_full_qualified_filename(stream->filename,
 144                                        sizeof(stream->filename),
 145                                        filename);
 146
 147    if(max_size) {
 148       stream->growable = FALSE;
 149       stream->map_size = max_size;
 150    }
 151    else {
 152       stream->growable = TRUE;
 153       stream->map_size = MAP_FILE_SIZE;
 154    }
 155
 156    if(!os_stream_map(stream))
 157       goto error2;
 158
 159    return stream;
 160
 161 error2:
 162    FREE(stream);
 163 error1:
 164    return NULL;
 165 }
 166
 167
 168 static INLINE void
 169 os_stream_copy(struct os_stream *stream, const char *data, size_t size)
 170 {
 171    assert(stream->written + size <= stream->map_size);
 172    memcpy(stream->pMap + stream->written, data, size);
 173    stream->written += size;
 174 }
 175
 176
 177 boolean
 178 os_stream_write(struct os_stream *stream, const void *data, size_t size)
 179 {
 180    if(!stream)
 181       return FALSE;
 182
 183    if(!stream->pMap)
 184       return FALSE;
 185
 186    while(stream->written + size > stream->map_size) {
 187       size_t step = stream->map_size - stream->written;
 188       os_stream_copy(stream, data, step);
 189       data = (const char *)data + step;
 190       size -= step;
 191
 192       os_stream_unmap(stream);
 193       if(!stream->growable || !os_stream_map(stream))
 194          return FALSE;
 195    }
 196
 197    os_stream_copy(stream, data, size);
 198
 199    return TRUE;
 200 }
 201
 202
 203 void
 204 os_stream_flush(struct os_stream *stream)
 205 {
 206    (void)stream;
 207 }
 208
 209
 210 void
 211 os_stream_close(struct os_stream *stream)
 212 {
 213    if(!stream)
 214       return;
 215
 216    os_stream_unmap(stream);
 217
 218    FREE(stream);
 219 }
 220
 221
 222 #endif