mterm.c

linux的图形界面程序

/*
 * Copyright (c) 1999 Greg Haerr <[e-mail]>
 *
 * Microwindows Terminal Emulator for Linux
 *
 * Yes, this is just a demo, and doesn't repaint contents on refresh.
 */
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <signal.h>
#include "windows.h"
#include "wintern.h"		/* for WndRegisterFdInput*/

#define COLS		80
#define ROWS		24
#define FONTNAME	SYSTEM_FIXED_FONT
/*#define FONTNAME	OEM_FIXED_FONT*/
#define APPCLASS	"mterm"

/* forward decls*/
LRESULT CALLBACK WndProc(HWND hwnd,UINT uMsg,WPARAM wp,LPARAM lp);
void EmOutChar(HWND hwnd, int ch);
int  CreatePtyShell(void);
int  ReadPtyShell(int fd, char *buf, int count);
int  WritePtyShell(int fd, char *buf, int count);
void ClosePtyShell(int fd);

/* local data*/
static int ttyfd = -1;
static int xpos;
static int ypos;
static int nCharWidth, nCharHeight;
static int nScreenWidth, nScreenHeight;

int
RegisterAppClass(void)
{
	WNDCLASS	wc;

	wc.style = CS_DBLCLKS | CS_VREDRAW | CS_HREDRAW;
	wc.lpfnWndProc = (WNDPROC)WndProc;
	wc.cbClsExtra = 0;
	wc.cbWndExtra = 0;
	wc.hInstance = 0;
	wc.hIcon = 0; /*LoadIcon(GetHInstance(), MAKEINTRESOURCE( 1));*/
	wc.hCursor = 0; /*LoadCursor(NULL, IDC_ARROW);*/
	wc.hbrBackground = (HBRUSH)GetStockObject(LTGRAY_BRUSH);
	wc.lpszMenuName = NULL;
	wc.lpszClassName =  APPCLASS;
	RegisterClass( &wc);
	return 1;
}

HWND
CreateAppWindow(void)
{
	HWND	hwnd;
	HDC	hdc;
	int 	w, h;
	RECT	rc;

	GetWindowRect(GetDesktopWindow(), &rc);
	w = rc.right - 40;
	h = rc.bottom;

	/* determine TE size from font*/
	hdc = GetDC(NULL);
	SelectObject(hdc, GetStockObject(FONTNAME));
	SetRect(&rc, 0, 0, 0, 0);
	nCharHeight = DrawText(hdc, "m", 1, &rc, DT_CALCRECT);
	nCharWidth = rc.right;
	nScreenWidth = min(w, nCharWidth*COLS);
	nScreenHeight = min(h, nCharHeight*ROWS);
	ReleaseDC(NULL, hdc);

	hwnd = CreateWindowEx(0L, APPCLASS,
		"Microwindows Terminal",
		WS_OVERLAPPEDWINDOW | WS_VISIBLE,
		CW_USEDEFAULT, CW_USEDEFAULT,
		nScreenWidth+4, nScreenHeight+24, NULL, (HMENU)1, NULL, NULL);

	return hwnd;
}

LRESULT CALLBACK
WndProc(HWND hwnd, UINT msg, WPARAM wp, LPARAM lp)
{
	PAINTSTRUCT	ps;
	HDC		hdc;
	unsigned char	ch;
   
	switch(msg) {
	case WM_CREATE:
		ttyfd = CreatePtyShell();
		/*if(ttyfd == -1)
			return -1;*/
		WndRegisterFdInput(hwnd, ttyfd);
		xpos = ypos = 0;
		break;

	case WM_DESTROY:
		WndUnregisterFdInput(hwnd, ttyfd);
		ClosePtyShell(ttyfd);
		break;

	case WM_CHAR:
		ch = (char)wp;
		/* echo half duplex if CreatePtyShell() failed*/
		if(ttyfd == -1) {
			EmOutChar(hwnd, ch);
			if(ch == '\r')
				EmOutChar(hwnd, '\n');
		} else
			WritePtyShell(ttyfd, &ch, 1);
		break;

	case WM_FDINPUT:
		if(ReadPtyShell(ttyfd, &ch, 1) == 1)
			EmOutChar(hwnd, ch);
		break;

	case WM_PAINT:
		hdc = BeginPaint(hwnd, &ps);
		EndPaint(hwnd, &ps);
		break;

	default:
		return DefWindowProc(hwnd, msg, wp, lp);
	}
	return 0;
}

int WINAPI 
WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine,
	int nShowCmd)
{
	MSG 	msg;
	extern IMAGEHDR image_car8;

	RegisterAppClass();
	WndSetDesktopWallpaper(&image_car8);

	CreateAppWindow();

	/* type ESC to quit...*/
	while(GetMessage(&msg, NULL, 0, 0)) {
		TranslateMessage(&msg);
		DispatchMessage(&msg);
	}
	return 0;
}

void
EmOutChar(HWND hwnd, int ch)
{
	HDC	hdc;
	RECT	rc;

	switch(ch) {
	case '\r':
		xpos = 0;
		return;
	case '\n':
		ypos += nCharHeight;
		GetClientRect(hwnd, &rc);
		if(ypos > (ROWS-1)*nCharHeight) {
			ypos -= nCharHeight;

			/* scroll window using bitblt ;-)*/
			hdc = GetDC(hwnd);
			BitBlt(hdc, 0, 0, rc.right, rc.bottom-nCharHeight,
				hdc, 0, nCharHeight, SRCCOPY);
			rc.top = ypos;
			FillRect(hdc, &rc, (HBRUSH)(COLOR_BTNFACE+1));
			ReleaseDC(hwnd, hdc);
		}
		return;
	case '\007':			/* bel*/
		write(STDERR_FILENO, "\007", 1);
		return;
	case '\t':
		xpos += nCharWidth;
		while((xpos/nCharWidth) & 7)
			EmOutChar(hwnd, ' ');
		return;
	case '\b':
		if(xpos <= 0)
			return;
		xpos -= nCharWidth;
		EmOutChar(hwnd, ' ');
		xpos -= nCharWidth;
		return;
	}

	/* draw some text*/
	hdc = GetDC(hwnd);
	SelectObject(hdc, GetStockObject(FONTNAME));
	/* get LTGRAY_BRUSH color from desktop scheme using COLOR_BTNFACE*/
	SetBkColor(hdc, GetSysColor(COLOR_BTNFACE));
	SetRect(&rc, xpos, ypos, xpos+nCharWidth, ypos+nCharHeight);
	ExtTextOut(hdc, xpos, ypos, ETO_OPAQUE, &rc, (char *)&ch, 1, NULL);
	ReleaseDC(hwnd, hdc);
	xpos += nCharWidth;
	if(xpos > (COLS-1)*nCharWidth) {
		xpos = 0;
		EmOutChar(hwnd, '\n');
	}
}

#if ELKS
#define SHELL	"/bin/sash"
#else
#define SHELL	"/bin/sh"
#endif

static int pid;

static void
ptysignaled(int signo)
{
	switch(signo) {
	case SIGINT:	/* interrupt*/
#if !ELKS
		/* this doesn't work, can anyone fix it?*/
		killpg(pid, SIGINT);
#endif
		break;
	case SIGCHLD:	/* child status change - child exit*/
		DestroyWindow(GetActiveWindow());
		CreateAppWindow();
		return;
	}
	fprintf(stderr, "Uncaught signal %d\n", signo);
}

/*
 * Create a shell running through a pseudo tty, return the shell fd.
 */
int
CreatePtyShell(void)
{
	int	n = 0;
	int	tfd;
	char	pty_name[12];
	char *	argv[2];

again:
	sprintf(pty_name, "/dev/ptyp%d", n);
	if ((tfd = open(pty_name, O_RDWR | O_NONBLOCK)) < 0) {
		if ((errno == EBUSY) && (n < 3)) {
			++n;
			goto again;
		}
		fprintf(stderr, "Can't create pty %s\n", pty_name);
		return -1;
	}
	signal(SIGCHLD, ptysignaled);
	signal(SIGINT, ptysignaled);
	if ((pid = fork()) == -1) {
		fprintf(stderr, "No processes\n");
		return -1;
	}
	if (!pid) {
		close(STDIN_FILENO);
		close(STDOUT_FILENO);
		close(STDERR_FILENO);
		close(tfd);
		
		setsid();
		pty_name[5] = 't';
		if ((tfd = open(pty_name, O_RDWR)) < 0) {
			fprintf(stderr, "Child: Can't open pty %s\n", pty_name);
			exit(1);
		}
		dup2(tfd, STDIN_FILENO);
		dup2(tfd, STDOUT_FILENO);
		dup2(tfd, STDERR_FILENO);
		/*if(!(argv[0] = getenv("SHELL")))*/
			argv[0] = SHELL;
		argv[1] = NULL;
		execv(argv[0], argv);
		exit(1);
	}
	return tfd;
}

int
ReadPtyShell(int fd, char *buf, int count)
{
	return read(fd, buf, count);
}

int
WritePtyShell(int fd, char *buf, int count)
{
	return write(fd, buf, count);
}

void
ClosePtyShell(int fd)
{
	if(ttyfd != -1)
		close(fd);
}