IO第二次作业

源码：

main.c

#include "fun.h"
int main(int argc, const char *argv[])
{
	printf("输入1修改图片，输入2搞密码\n");
	int s=0;
	printf(">>>>>>");
	scanf("%d",&s);
	if(1==s){
		printf("请输入图片文件名:\n");
		printf(">>>>>>");
		char arr[32]="0";
		scanf("%s",arr);
		edit_bmp(arr);
	}else if(2==s){
		printf("下面请输入密码来解锁一个神秘的东西(`3`)\n");
		printf("悄悄告诉你密码是123456\n");
		getch();
		char brr[]={'1','2','3','4','5','6'};
		char crr[32]="0";
		int i=0;
		while(1){
			crr[i]=getch();
			if(crr[i]=='\n'){
				break;
			}
			i++;
			printf("*");
		}
		printf("\n");
		int count=0;
		i=0;
		while(1){
			if(crr[i]==10){
				break;
			}
			if(brr[i]==crr[i]){
				count++;
			}else if(brr[i]!=crr[i]){
				count=0;
			}
			i++;
		}
		if(6 == count){
			printf("恭喜你解锁成功!!!\n");
			printf("但是没有奖励('v')\n");
		}
		else{
			printf("有点小菜('^')\n");
		}
	}else{
            printf("输入错误\n");
}
	return 0;
}

fun.c

#include "fun.h"
void edit_bmp(char *p){
	if(NULL == p){
		printf("error1\n");
		return;
	}
	FILE *fp = fopen(p,"r+");
	if(NULL == fp){
		printf("修改失败\n");
		printf("没有该图片\n");
		return;
	}
	int w=0;
	int h=0;
	int sum=0;
	unsigned char a[3]={0,255,255};
	unsigned char b[3]={0,0,255};
	unsigned char c[3]={0,0,0};
	fseek(fp,18,SEEK_SET);
	fread(&w,4,1,fp);
	fread(&h,4,1,fp);
	sum=w*h;
	fseek(fp,54,SEEK_SET);
	int i=0;
	for(i=0;i<sum/3;i++){
		fwrite(a,3,1,fp);
	}
	for(;i<(sum/3)*2;i++){
		fwrite(b,3,1,fp);
	}
	for(;i<sum;i++){
		fwrite(c,3,1,fp);
	}
	printf("图片修改完毕\n");
	fclose(fp);
	return;
}
int getch(){
	int c=0;
	struct termios org_opts, new_opts;
	int res=0;
	res=tcgetattr(STDIN_FILENO, &org_opts);
	assert(res==0);
	new_opts = org_opts;
	new_opts.c_lflag &= ~(ICANON | ECHO | ECHOE | ECHOK | ECHONL | ECHOPRT | ECHOKE | ICRNL);
	tcsetattr(STDIN_FILENO, TCSANOW, &new_opts);
	c=getchar();
	res=tcsetattr(STDIN_FILENO, TCSANOW, &org_opts);
	assert(res==0);
	return c;
}

fun.h

#ifndef __FUN_H__
#define __FUN_H__
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <termios.h>
#include <assert.h>
void edit_bmp(char *p);
int getch();
#endif

现象截图：