mirror of
https://github.com/golang/go
synced 2024-11-20 04:34:41 -07:00
111 lines
1.7 KiB
C
111 lines
1.7 KiB
C
// Copyright 2013 The Go Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
#include <u.h>
|
|
#include <windows.h>
|
|
#include <libc.h>
|
|
#include "win.h"
|
|
|
|
char*
|
|
toutf(WinRune *r)
|
|
{
|
|
Rune *r1;
|
|
int i, n;
|
|
char *p;
|
|
|
|
n = 0;
|
|
while(r[n] != '\0')
|
|
n++;
|
|
n++;
|
|
r1 = malloc(n*sizeof r1[0]);
|
|
for(i=0; i<n; i++)
|
|
r1[i] = r[i];
|
|
p = smprint("%S", r1);
|
|
free(r1);
|
|
return p;
|
|
}
|
|
|
|
WinRune*
|
|
torune(char *p)
|
|
{
|
|
int i, n;
|
|
Rune rr;
|
|
WinRune *r;
|
|
|
|
n = utflen(p);
|
|
r = malloc((n+1)*sizeof r[0]);
|
|
for(i=0; i<n; i++) {
|
|
p += chartorune(&rr, p);
|
|
r[i] = rr;
|
|
}
|
|
r[n] = '\0';
|
|
return r;
|
|
}
|
|
|
|
char*
|
|
mktempdir(void)
|
|
{
|
|
WinRune buf[1024];
|
|
WinRune tmp[MAX_PATH];
|
|
WinRune golink[] = {'g', 'o', 'l', 'i', 'n', 'k', '\0'};
|
|
int n;
|
|
|
|
n = GetTempPathW(nelem(buf), buf);
|
|
if(n <= 0)
|
|
return nil;
|
|
buf[n] = '\0';
|
|
|
|
if(GetTempFileNameW(buf, golink, 0, tmp) == 0)
|
|
return nil;
|
|
DeleteFileW(tmp);
|
|
if(!CreateDirectoryW(tmp, nil))
|
|
return nil;
|
|
|
|
return toutf(tmp);
|
|
}
|
|
|
|
void
|
|
removeall(char *p)
|
|
{
|
|
WinRune *r, *r1;
|
|
DWORD attr;
|
|
char *q, *elem;
|
|
HANDLE h;
|
|
WIN32_FIND_DATAW data;
|
|
|
|
r = torune(p);
|
|
attr = GetFileAttributesW(r);
|
|
if(attr == INVALID_FILE_ATTRIBUTES || !(attr & FILE_ATTRIBUTE_DIRECTORY)) {
|
|
DeleteFileW(r);
|
|
free(r);
|
|
return;
|
|
}
|
|
|
|
q = smprint("%s\\*", p);
|
|
r1 = torune(q);
|
|
free(q);
|
|
h = FindFirstFileW(r1, &data);
|
|
if(h == INVALID_HANDLE_VALUE)
|
|
goto done;
|
|
do{
|
|
q = toutf(data.cFileName);
|
|
elem = strrchr(q, '\\');
|
|
if(elem != nil) {
|
|
elem++;
|
|
if(strcmp(elem, ".") == 0 || strcmp(elem, "..") == 0) {
|
|
free(q);
|
|
continue;
|
|
}
|
|
}
|
|
removeall(q);
|
|
free(q);
|
|
}while(FindNextFileW(h, &data));
|
|
FindClose(h);
|
|
|
|
done:
|
|
free(r1);
|
|
RemoveDirectoryW(r);
|
|
free(r);
|
|
}
|