print_int: add memcpy

Author: h908714124

group3/group3.s

@@ -169,6 +169,8 @@ print_lines_loop:
 
 	add	%rax, %rbx
 
+	write	linebuf(%rip), $3
+	writeln
 	write	linebuf(%rip), %rax
 	writeln
 	plop

print_int/.gitignore

@@ -1 +1,2 @@
 /print_int
+/write

print_int/Makefile

@@ -2,11 +2,16 @@ build:
 	$(CC) main.s -c -o print_int.o
 	$(LD) -e main -o print_int print_int.o
 
-run: build
-	./print_int
-
 check: build
 	./print_int | md5sum | grep -q ^3cca2fa && printf "\033[1;32m[OK]\033[0m\n"
 
+build_write:
+	$(CC) write.s -c -o write.o
+	$(LD) -e main -o write write.o
+
+check_write: build_write
+	./write | grep -q '^ABC$$' && printf "\033[1;32m[OK]\033[0m\n"
+
+
 clean:
-	rm -f print_int
+	rm -f *.o write print_int

print_int/print_int.s

@@ -29,6 +29,14 @@
 	add	$8, %rsp
 	.endm
 
+	.macro memcpy source dest count
+	push	\source
+	push	\dest
+	push	\count
+	call	copymem
+	add	$24, %rsp
+	.endm
+
 	.macro write address size
 	push	\address
 	push	\size
@@ -105,20 +113,24 @@ write_char:
 write_newline:
 	enter
 	push_all
+	push	%rax
 	push	$0xa
 	call	write_char
 	plop
+	pop	%rax
 	pop_all
 	return
 
 write_string:
 	enter
 	push_all
+	push	%rax
 	mov	24(%rbp), %rsi			# param: address
 	mov	16(%rbp), %rdx			# param: size
 	mov	$WRITE, %rax
 	mov	$STDOUT, %rdi
 	syscall
+	pop	%rax
 	pop_all
 	return
 
@@ -175,3 +187,23 @@ print_int_pop_loop:
 	pop_all
 	return
 
+copymem:
+	enter
+	push_all
+	push	%rax
+	mov	32(%rbp), %rsi			# param: source
+	mov	24(%rbp), %rdi			# param: destination
+	mov	16(%rbp), %rdx			# param: count
+	mov	$0, %rcx
+	test	%rdx, %rdx
+	jz	copymem_done
+copymem_loop:
+	mov	(%rsi, %rcx), %rax
+	movb	%al, (%rdi, %rcx)
+	inc	%rcx
+	cmp	%rcx, %rdx
+	jne	copymem_loop
+copymem_done:
+	pop	%rax
+	pop_all
+	return

print_int/write.s

@@ -0,0 +1,27 @@
+	.data
+foo:
+	.ascii "abc"
+	.text
+	.globl main
+
+.include "print_int.s"
+
+main:
+	enter
+	sub	$48, %rsp
+	movb	$0x41, -24(%rbp)
+	movb	$0x42, -23(%rbp)
+	movb	$0x43, -22(%rbp)
+	movb	$0xa, -21(%rbp)
+	mov	%rbp, %rsi
+	sub	$24, %rsi
+	mov	%rsi, %rax
+	sub	$24, %rax
+	memcpy	%rsi, %rax, $4
+	write 	%rax, $4
+	call	exit
+
+exit:
+	movq	$EXIT, %rax
+	movq	$0, %rdi
+	syscall