src/gallium/auxiliary/util/u_stream_wd.c

   1 /**************************************************************************
   2  *
   3  * Copyright 2008 Tungsten Graphics, Inc., Cedar Park, Texas.
   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 TUNGSTEN GRAPHICS 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 "util/u_memory.h"
  41 #include "util/u_string.h"
  42
  43 #include "u_stream.h"
  44
  45
  46 #define MAP_FILE_SIZE (4*1024*1024)
  47
  48
  49 struct util_stream
  50 {
  51    char filename[MAX_PATH + 1];
  52    WCHAR wFileName[MAX_PATH + 1];
  53    boolean growable;
  54    size_t map_size;
  55    ULONG_PTR iFile;
  56    char *pMap;
  57    size_t written;
  58    unsigned suffix;
  59 };
  60
  61
  62 static INLINE boolean
  63 util_stream_map(struct util_stream *stream)
  64 {
  65    ULONG BytesInUnicodeString;
  66    static char filename[MAX_PATH + 1];
  67    unsigned filename_len;
  68
  69    if(stream->growable)
  70       filename_len = util_snprintf(filename,
  71                                    sizeof(filename),
  72                                    "\\??\\%s.%04x",
  73                                    stream->filename,
  74                                    stream->suffix++);
  75    else
  76       filename_len = util_snprintf(filename,
  77                                    sizeof(filename),
  78                                    "\\??\\%s",
  79                                    stream->filename);
  80
  81    EngMultiByteToUnicodeN(
  82          stream->wFileName,
  83          sizeof(stream->wFileName),
  84          &BytesInUnicodeString,
  85          filename,
  86          filename_len);
  87
  88    stream->pMap = EngMapFile(stream->wFileName, stream->map_size, &stream->iFile);
  89    if(!stream->pMap)
  90       return FALSE;
  91
  92    memset(stream->pMap, 0, stream->map_size);
  93    stream->written = 0;
  94
  95    return TRUE;
  96 }
  97
  98
  99 static INLINE void
 100 util_stream_unmap(struct util_stream *stream)
 101 {
 102    EngUnmapFile(stream->iFile);
 103    if(stream->written < stream->map_size) {
 104       /* Truncate file size */
 105       stream->pMap = EngMapFile(stream->wFileName, stream->written, &stream->iFile);
 106       if(stream->pMap)
 107          EngUnmapFile(stream->iFile);
 108    }
 109
 110    stream->pMap = NULL;
 111 }
 112
 113
 114 struct util_stream *
 115 util_stream_create(const char *filename, size_t max_size)
 116 {
 117    struct util_stream *stream;
 118
 119    stream = CALLOC_STRUCT(util_stream);
 120    if(!stream)
 121       goto error1;
 122
 123    strncpy(stream->filename, filename, sizeof(stream->filename));
 124
 125    if(max_size) {
 126       stream->growable = FALSE;
 127       stream->map_size = max_size;
 128    }
 129    else {
 130       stream->growable = TRUE;
 131       stream->map_size = MAP_FILE_SIZE;
 132    }
 133
 134    if(!util_stream_map(stream))
 135       goto error2;
 136
 137    return stream;
 138
 139 error2:
 140    FREE(stream);
 141 error1:
 142    return NULL;
 143 }
 144
 145
 146 static INLINE void
 147 util_stream_copy(struct util_stream *stream, const char *data, size_t size)
 148 {
 149    assert(stream->written + size <= stream->map_size);
 150    memcpy(stream->pMap + stream->written, data, size);
 151    stream->written += size;
 152 }
 153
 154
 155 boolean
 156 util_stream_write(struct util_stream *stream, const void *data, size_t size)
 157 {
 158    if(!stream)
 159       return FALSE;
 160
 161    if(!stream->pMap)
 162       return FALSE;
 163
 164    while(stream->written + size > stream->map_size) {
 165       size_t step = stream->map_size - stream->written;
 166       util_stream_copy(stream, data, step);
 167       data = (const char *)data + step;
 168       size -= step;
 169
 170       util_stream_unmap(stream);
 171       if(!stream->growable || !util_stream_map(stream))
 172          return FALSE;
 173    }
 174
 175    util_stream_copy(stream, data, size);
 176
 177    return TRUE;
 178 }
 179
 180
 181 void
 182 util_stream_flush(struct util_stream *stream)
 183 {
 184    (void)stream;
 185 }
 186
 187
 188 void
 189 util_stream_close(struct util_stream *stream)
 190 {
 191    if(!stream)
 192       return;
 193
 194    util_stream_unmap(stream);
 195
 196    FREE(stream);
 197 }
 198
 199
 200 #endif