This commit is contained in:
Russ Cox
2005-11-08 01:35:49 +00:00
parent d6701f9376
commit ee5fc884bd
16 changed files with 210 additions and 80 deletions

View File

@ -60,7 +60,6 @@ static struct
char *sysname;
vlong fasthz;
static void seedrand(void);
static int readtime(ulong, char*, int);
static int readbintime(char*, int);
static int writetime(char*, int);
@ -986,30 +985,6 @@ Dev consdevtab = {
devwstat,
};
static ulong randn;
static void
seedrand(void)
{
randomread((void*)&randn, sizeof(randn));
}
int
xnrand(int n)
{
if(randn == 0)
seedrand();
randn = randn*1103515245 + 12345 + fastticks(0);
return (randn>>16) % n;
}
int
rand(void)
{
xnrand(1);
return randn;
}
static uvlong uvorder = (uvlong) 0x0001020304050607ULL;
static uchar*

View File

@ -46,12 +46,13 @@ so_socket(int type)
fd = socket(AF_INET, type, 0);
if(fd < 0)
error(sys_errlist[errno]);
oserror();
one = 1;
if(setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof(one)) > 0)
print("setsockopt: %s", sys_errlist[errno]);
if(setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, (char*)&one, sizeof(one)) > 0){
oserrstr();
print("setsockopt: %s\n", up->errstr);
}
return fd;
}
@ -68,7 +69,7 @@ so_connect(int fd, unsigned long raddr, unsigned short rport)
hnputl(&sin.sin_addr.s_addr, raddr);
if(connect(fd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
error(sys_errlist[errno]);
oserror();
}
void
@ -79,7 +80,7 @@ so_getsockname(int fd, unsigned long *laddr, unsigned short *lport)
len = sizeof(sin);
if(getsockname(fd, (struct sockaddr*)&sin, &len) < 0)
error(sys_errlist[errno]);
oserror();
if(sin.sin_family != AF_INET || len != sizeof(sin))
error("not AF_INET");
@ -92,7 +93,7 @@ void
so_listen(int fd)
{
if(listen(fd, 5) < 0)
error(sys_errlist[errno]);
oserror();
}
int
@ -104,7 +105,7 @@ so_accept(int fd, unsigned long *raddr, unsigned short *rport)
len = sizeof(sin);
nfd = accept(fd, (struct sockaddr*)&sin, &len);
if(nfd < 0)
error(sys_errlist[errno]);
oserror();
if(sin.sin_family != AF_INET || len != sizeof(sin))
error("not AF_INET");
@ -121,8 +122,10 @@ so_bind(int fd, int su, unsigned short port)
struct sockaddr_in sin;
one = 1;
if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char*)&one, sizeof(one)) < 0)
print("setsockopt: %s", sys_errlist[errno]);
if(setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char*)&one, sizeof(one)) < 0){
oserrstr();
print("setsockopt: %s", up->errstr);
}
if(su) {
for(i = 600; i < 1024; i++) {
@ -133,7 +136,7 @@ so_bind(int fd, int su, unsigned short port)
if(bind(fd, (struct sockaddr*)&sin, sizeof(sin)) >= 0)
return;
}
error(sys_errlist[errno]);
oserror();
}
memset(&sin, 0, sizeof(sin));
@ -141,7 +144,7 @@ so_bind(int fd, int su, unsigned short port)
hnputs(&sin.sin_port, port);
if(bind(fd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
error(sys_errlist[errno]);
oserror();
}
int

View File

@ -228,6 +228,7 @@ ipopen(Chan *c, int omode)
Conv *cv, *lcv;
omode &= 3;
perm = 0;
switch(omode) {
case OREAD:
perm = 4;

View File

@ -181,6 +181,8 @@ rootgen(Chan *c, char *name, Dirtab *dirt, int ndirt, int s, Dir *dp)
t = c->qid.path - Qmnt - 1;
l = &mntlist;
break;
default:
return -1;
}
if(t >= l->ndir)
return -1;

View File

@ -717,7 +717,7 @@ static void
randfill(uchar *buf, int len)
{
while(len-- > 0)
*buf++ = nrand(256);
*buf++ = fastrand();
}
static long

View File

@ -3,6 +3,7 @@
#include "lib.h"
#include "dat.h"
#include "fns.h"
#include <libsec.h>
typedef struct Oproc Oproc;
struct Oproc {
@ -136,24 +137,49 @@ procwakeup(Proc *p)
ReleaseSemaphore(op->sema, 1, 0);
}
void
random20(uchar *p)
{
LARGE_INTEGER ti;
int i, j;
FILETIME ft;
DigestState ds;
vlong tsc;
GetSystemTimeAsFileTime(&ft);
memset(&ds, 0, sizeof ds);
sha1((uchar*)&ft, sizeof(ft), 0, &ds);
for(i=0; i<50; i++) {
for(j=0; j<10; j++) {
QueryPerformanceCounter(&ti);
sha1((uchar*)&ti, sizeof(ti), 0, &ds);
tsc = GetTickCount();
sha1((uchar*)&tsc, sizeof(tsc), 0, &ds);
}
Sleep(10);
}
sha1(0, 0, p, &ds);
}
void
randominit(void)
{
srand(seconds());
}
ulong
randomread(void *v, ulong n)
{
int m, i, *r;
m = (n / sizeof(int)) * sizeof(int);
for (i = 0, r = (int*)v; i < m; i += sizeof(int)) {
*r = rand();
r += sizeof(int);
int i;
uchar p[20];
for(i=0; i<n; i+=20){
random20(p);
if(i+20 <= n)
memmove((char*)v+i, p, 20);
else
memmove((char*)v+i, p, n-i);
}
return m;
return n;
}
long
@ -183,14 +209,151 @@ fastticks(uvlong *v)
extern int main(int, char*[]);
int APIENTRY
WinMain(HINSTANCE hInst, HINSTANCE hPrev, LPSTR arg, int nshow)
int
wstrutflen(Rune *s)
{
main(__argc, __argv);
int n;
for(n=0; *s; n+=runelen(*s),s++)
;
return n;
}
int
wstrtoutf(char *s, Rune *t, int n)
{
int i;
char *s0;
s0 = s;
if(n <= 0)
return wstrutflen(t)+1;
while(*t) {
if(n < UTFmax+1 && n < runelen(*t)+1) {
*s = 0;
return s-s0+wstrutflen(t)+1;
}
i = runetochar(s, t);
s += i;
n -= i;
t++;
}
*s = 0;
return s-s0;
}
int
win_hasunicode(void)
{
OSVERSIONINFOA osinfo;
int r;
osinfo.dwOSVersionInfoSize = sizeof(osinfo);
if(!GetVersionExA(&osinfo))
panic("GetVersionEx failed");
switch(osinfo.dwPlatformId) {
default:
panic("unknown PlatformId");
case VER_PLATFORM_WIN32s:
panic("Win32s not supported");
case VER_PLATFORM_WIN32_WINDOWS:
r = 0;
break;
case VER_PLATFORM_WIN32_NT:
r = 1;
break;
}
return r;
}
int
wstrlen(Rune *s)
{
int n;
for(n=0; *s; s++,n++)
;
return n;
}
static int args(char *argv[], int n, char *p);
int APIENTRY
WinMain(HINSTANCE x, HINSTANCE y, LPSTR z, int w)
{
int argc, n;
char *arg, *p, **argv;
Rune *warg;
if(0 && win_hasunicode()){
warg = GetCommandLineW();
n = (wstrlen(warg)+1)*UTFmax;
arg = malloc(n);
wstrtoutf(arg, warg, n);
}else
arg = GetCommandLineA();
/* conservative guess at the number of args */
for(argc=4,p=arg; *p; p++)
if(*p == ' ' || *p == '\t')
argc++;
argv = malloc(argc*sizeof(char*));
argc = args(argv, argc, arg);
mymain(argc, argv);
ExitThread(0);
return 0;
}
/*
* Break the command line into arguments
* The rules for this are not documented but appear to be the following
* according to the source for the microsoft C library.
* Words are seperated by space or tab
* Words containing a space or tab can be quoted using "
* 2N backslashes + " ==> N backslashes and end quote
* 2N+1 backslashes + " ==> N backslashes + literal "
* N backslashes not followed by " ==> N backslashes
*/
static int
args(char *argv[], int n, char *p)
{
char *p2;
int i, j, quote, nbs;
for(i=0; *p && i<n-1; i++) {
while(*p == ' ' || *p == '\t')
p++;
quote = 0;
argv[i] = p2 = p;
for(;*p; p++) {
if(!quote && (*p == ' ' || *p == '\t'))
break;
for(nbs=0; *p == '\\'; p++,nbs++)
;
if(*p == '"') {
for(j=0; j<(nbs>>1); j++)
*p2++ = '\\';
if(nbs&1)
*p2++ = *p;
else
quote = !quote;
} else {
for(j=0; j<nbs; j++)
*p2++ = '\\';
*p2++ = *p;
}
}
/* move p up one to avoid pointing to null at end of p2 */
if(*p)
p++;
*p2 = 0;
}
argv[i] = 0;
return i;
}
/*
* Windows socket error messages
* There must be a way to get these strings out of the library.