crypto/perlasm/x86ms.pl

   1 #!/usr/local/bin/perl
   2
   3 package x86ms;
   4
   5 $label="L000";
   6
   7 %lb=(   'eax',  'al',
   8         'ebx',  'bl',
   9         'ecx',  'cl',
  10         'edx',  'dl',
  11         'ax',   'al',
  12         'bx',   'bl',
  13         'cx',   'cl',
  14         'dx',   'dl',
  15         );
  16
  17 %hb=(   'eax',  'ah',
  18         'ebx',  'bh',
  19         'ecx',  'ch',
  20         'edx',  'dh',
  21         'ax',   'ah',
  22         'bx',   'bh',
  23         'cx',   'ch',
  24         'dx',   'dh',
  25         );
  26
  27 sub main'asm_init_output { @out=(); }
  28 sub main'asm_get_output { return(@out); }
  29 sub main'get_labels { return(@labels); }
  30 sub main'external_label
  31 {
  32         push(@labels,@_);
  33         foreach (@_) {
  34                 push(@out, "EXTRN\t_$_:DWORD\n");
  35         }
  36 }
  37
  38 sub main'LB
  39         {
  40         (defined($lb{$_[0]})) || die "$_[0] does not have a 'low byte'\n";
  41         return($lb{$_[0]});
  42         }
  43
  44 sub main'HB
  45         {
  46         (defined($hb{$_[0]})) || die "$_[0] does not have a 'high byte'\n";
  47         return($hb{$_[0]});
  48         }
  49
  50 sub main'BP
  51         {
  52         &get_mem("BYTE",@_);
  53         }
  54
  55 sub main'DWP
  56         {
  57         &get_mem("DWORD",@_);
  58         }
  59
  60 sub main'QWP
  61         {
  62         &get_mem("QWORD",@_);
  63         }
  64
  65 sub main'BC
  66         {
  67         return @_;
  68         }
  69
  70 sub main'DWC
  71         {
  72         return @_;
  73         }
  74
  75 sub main'stack_push
  76         {
  77         local($num)=@_;
  78         $stack+=$num*4;
  79         &main'sub("esp",$num*4);
  80         }
  81
  82 sub main'stack_pop
  83         {
  84         local($num)=@_;
  85         $stack-=$num*4;
  86         &main'add("esp",$num*4);
  87         }
  88
  89 sub get_mem
  90         {
  91         local($size,$addr,$reg1,$reg2,$idx)=@_;
  92         local($t,$post);
  93         local($ret)="$size PTR ";
  94
  95         $addr =~ s/^\s+//;
  96         if ($addr =~ /^(.+)\+(.+)$/)
  97                 {
  98                 $reg2=&conv($1);
  99                 $addr="_$2";
 100                 }
 101         elsif ($addr =~ /^[_a-z][_a-z0-9]*$/i)
 102                 {
 103                 $addr="_$addr";
 104                 }
 105
 106         if ($addr =~ /^.+\-.+$/) { $addr="($addr)"; }
 107
 108         $reg1="$regs{$reg1}" if defined($regs{$reg1});
 109         $reg2="$regs{$reg2}" if defined($regs{$reg2});
 110         if (($addr ne "") && ($addr ne 0))
 111                 {
 112                 if ($addr !~ /^-/)
 113                         { $ret.=$addr; }
 114                 else    { $post=$addr; }
 115                 }
 116         if ($reg2 ne "")
 117                 {
 118                 $t="";
 119                 $t="*$idx" if ($idx != 0);
 120                 $reg1="+".$reg1 if ("$reg1$post" ne "");
 121                 $ret.="[$reg2$t$reg1$post]";
 122                 }
 123         else
 124                 {
 125                 $ret.="[$reg1$post]"
 126                 }
 127         $ret =~ s/\[\]//;       # in case $addr was the only argument
 128         return($ret);
 129         }
 130
 131 sub main'mov    { &out2("mov",@_); }
 132 sub main'movb   { &out2("mov",@_); }
 133 sub main'and    { &out2("and",@_); }
 134 sub main'or     { &out2("or",@_); }
 135 sub main'shl    { &out2("shl",@_); }
 136 sub main'shr    { &out2("shr",@_); }
 137 sub main'xor    { &out2("xor",@_); }
 138 sub main'xorb   { &out2("xor",@_); }
 139 sub main'add    { &out2("add",@_); }
 140 sub main'adc    { &out2("adc",@_); }
 141 sub main'sub    { &out2("sub",@_); }
 142 sub main'sbb    { &out2("sbb",@_); }
 143 sub main'rotl   { &out2("rol",@_); }
 144 sub main'rotr   { &out2("ror",@_); }
 145 sub main'exch   { &out2("xchg",@_); }
 146 sub main'cmp    { &out2("cmp",@_); }
 147 sub main'lea    { &out2("lea",@_); }
 148 sub main'mul    { &out1("mul",@_); }
 149 sub main'div    { &out1("div",@_); }
 150 sub main'dec    { &out1("dec",@_); }
 151 sub main'inc    { &out1("inc",@_); }
 152 sub main'jmp    { &out1("jmp",@_); }
 153 sub main'jmp_ptr { &out1p("jmp",@_); }
 154 sub main'je     { &out1("je",@_); }
 155 sub main'jle    { &out1("jle",@_); }
 156 sub main'jz     { &out1("jz",@_); }
 157 sub main'jge    { &out1("jge",@_); }
 158 sub main'jl     { &out1("jl",@_); }
 159 sub main'ja     { &out1("ja",@_); }
 160 sub main'jae    { &out1("jae",@_); }
 161 sub main'jb     { &out1("jb",@_); }
 162 sub main'jbe    { &out1("jbe",@_); }
 163 sub main'jc     { &out1("jc",@_); }
 164 sub main'jnc    { &out1("jnc",@_); }
 165 sub main'jnz    { &out1("jnz",@_); }
 166 sub main'jne    { &out1("jne",@_); }
 167 sub main'jno    { &out1("jno",@_); }
 168 sub main'push   { &out1("push",@_); $stack+=4; }
 169 sub main'pop    { &out1("pop",@_); $stack-=4; }
 170 sub main'pushf  { &out0("pushfd"); $stack+=4; }
 171 sub main'popf   { &out0("popfd"); $stack-=4; }
 172 sub main'bswap  { &out1("bswap",@_); &using486(); }
 173 sub main'not    { &out1("not",@_); }
 174 sub main'call   { &out1("call",($_[0]=~/^\$L/?'':'_').$_[0]); }
 175 sub main'call_ptr { &out1p("call",@_); }
 176 sub main'ret    { &out0("ret"); }
 177 sub main'nop    { &out0("nop"); }
 178 sub main'test   { &out2("test",@_); }
 179 sub main'bt     { &out2("bt",@_); }
 180 sub main'leave  { &out0("leave"); }
 181 sub main'cpuid  { &out0("DW\t0A20Fh"); }
 182 sub main'rdtsc  { &out0("DW\t0310Fh"); }
 183 sub main'halt   { &out0("hlt"); }
 184 sub main'movz   { &out2("movzx",@_); }
 185 sub main'neg    { &out1("neg",@_); }
 186 sub main'cld    { &out0("cld"); }
 187
 188 # SSE2
 189 sub main'emms   { &out0("emms"); }
 190 sub main'movd   { &out2("movd",@_); }
 191 sub main'movq   { &out2("movq",@_); }
 192 sub main'movdqu { &out2("movdqu",@_); }
 193 sub main'movdqa { &out2("movdqa",@_); }
 194 sub main'movdq2q{ &out2("movdq2q",@_); }
 195 sub main'movq2dq{ &out2("movq2dq",@_); }
 196 sub main'paddq  { &out2("paddq",@_); }
 197 sub main'pmuludq{ &out2("pmuludq",@_); }
 198 sub main'psrlq  { &out2("psrlq",@_); }
 199 sub main'psllq  { &out2("psllq",@_); }
 200 sub main'pxor   { &out2("pxor",@_); }
 201 sub main'por    { &out2("por",@_); }
 202 sub main'pand   { &out2("pand",@_); }
 203
 204 sub out2
 205         {
 206         local($name,$p1,$p2)=@_;
 207         local($l,$t,$line);
 208
 209         $line="\t$name\t";
 210         $t=&conv($p1).",";
 211         $l=length($t);
 212         $line.="$t";
 213         $l=4-($l+9)/8;
 214         $line.="\t" x $l;
 215         $line.=&conv($p2);
 216         if ($line=~/\bxmm[0-7]\b/i) { $line=~s/\b[A-Z]+WORD\s+PTR/XMMWORD PTR/i; }
 217         push(@out,$line."\n");
 218         }
 219
 220 sub out0
 221         {
 222         local($name)=@_;
 223
 224         push(@out,"\t$name\n");
 225         }
 226
 227 sub out1
 228         {
 229         local($name,$p1)=@_;
 230         local($l,$t);
 231
 232         push(@out,"\t$name\t".&conv($p1)."\n");
 233         }
 234
 235 sub conv
 236         {
 237         local($p)=@_;
 238
 239         $p =~ s/0x([0-9A-Fa-f]+)/0$1h/;
 240         return $p;
 241         }
 242
 243 sub using486
 244         {
 245         return if $using486;
 246         $using486++;
 247         grep(s/\.386/\.486/,@out);
 248         }
 249
 250 sub main'file
 251         {
 252         local($file)=@_;
 253
 254         local($tmp)=<<"EOF";
 255         TITLE   $file.asm
 256         .386
 257 .model  FLAT
 258 _TEXT\$ SEGMENT PAGE 'CODE'
 259
 260 EOF
 261         push(@out,$tmp);
 262         }
 263
 264 sub main'function_begin
 265         {
 266         local($func,$extra)=@_;
 267
 268         push(@labels,$func);
 269
 270         local($tmp)=<<"EOF";
 271 PUBLIC  _$func
 272 $extra
 273 _$func PROC NEAR
 274         push    ebp
 275         push    ebx
 276         push    esi
 277         push    edi
 278 EOF
 279         push(@out,$tmp);
 280         $stack=20;
 281         }
 282
 283 sub main'function_begin_B
 284         {
 285         local($func,$extra)=@_;
 286
 287         local($tmp)=<<"EOF";
 288 PUBLIC  _$func
 289 $extra
 290 _$func PROC NEAR
 291 EOF
 292         push(@out,$tmp);
 293         $stack=4;
 294         }
 295
 296 sub main'function_end
 297         {
 298         local($func)=@_;
 299
 300         local($tmp)=<<"EOF";
 301         pop     edi
 302         pop     esi
 303         pop     ebx
 304         pop     ebp
 305         ret
 306 _$func ENDP
 307 EOF
 308         push(@out,$tmp);
 309         $stack=0;
 310         %label=();
 311         }
 312
 313 sub main'function_end_B
 314         {
 315         local($func)=@_;
 316
 317         local($tmp)=<<"EOF";
 318 _$func ENDP
 319 EOF
 320         push(@out,$tmp);
 321         $stack=0;
 322         %label=();
 323         }
 324
 325 sub main'function_end_A
 326         {
 327         local($func)=@_;
 328
 329         local($tmp)=<<"EOF";
 330         pop     edi
 331         pop     esi
 332         pop     ebx
 333         pop     ebp
 334         ret
 335 EOF
 336         push(@out,$tmp);
 337         }
 338
 339 sub main'file_end
 340         {
 341         # try to detect if SSE2 or MMX extensions were used...
 342         my $xmmheader=<<___;
 343 .686
 344 .XMM
 345 IF \@Version LT 800
 346 XMMWORD STRUCT 16
 347         DQ  2 dup (?)
 348 XMMWORD ENDS
 349 ENDIF
 350 ___
 351         if (grep {/\b[x]?mm[0-7]\b/i} @out) {
 352                 grep {s/\.[3-7]86/$xmmheader/} @out;
 353                 }
 354         push(@out,"_TEXT\$      ENDS\n");
 355         push(@out,"END\n");
 356         }
 357
 358 sub main'wparam
 359         {
 360         local($num)=@_;
 361
 362         return(&main'DWP($stack+$num*4,"esp","",0));
 363         }
 364
 365 sub main'swtmp
 366         {
 367         return(&main'DWP($_[0]*4,"esp","",0));
 368         }
 369
 370 # Should use swtmp, which is above esp.  Linix can trash the stack above esp
 371 #sub main'wtmp
 372 #       {
 373 #       local($num)=@_;
 374 #
 375 #       return(&main'DWP(-(($num+1)*4),"esp","",0));
 376 #       }
 377
 378 sub main'comment
 379         {
 380         foreach (@_)
 381                 {
 382                 push(@out,"\t; $_\n");
 383                 }
 384         }
 385
 386 sub main'public_label
 387         {
 388         $label{$_[0]}="_$_[0]"  if (!defined($label{$_[0]}));
 389         push(@out,"PUBLIC\t$label{$_[0]}\n");
 390         }
 391
 392 sub main'label
 393         {
 394         if (!defined($label{$_[0]}))
 395                 {
 396                 $label{$_[0]}="\$${label}${_[0]}";
 397                 $label++;
 398                 }
 399         return($label{$_[0]});
 400         }
 401
 402 sub main'set_label
 403         {
 404         if (!defined($label{$_[0]}))
 405                 {
 406                 $label{$_[0]}="\$${label}${_[0]}";
 407                 $label++;
 408                 }
 409         if ($_[1]!=0 && $_[1]>1)
 410                 {
 411                 main'align($_[1]);
 412                 }
 413         if((defined $_[2]) && ($_[2] == 1))
 414                 {
 415                 push(@out,"$label{$_[0]}::\n");
 416                 }
 417         elsif ($label{$_[0]} !~ /^\$/)
 418                 {
 419                 push(@out,"$label{$_[0]}\tLABEL PTR\n");
 420                 }
 421         else
 422                 {
 423                 push(@out,"$label{$_[0]}:\n");
 424                 }
 425         }
 426
 427 sub main'data_byte
 428         {
 429         push(@out,"\tDB\t".join(',',@_)."\n");
 430         }
 431
 432 sub main'data_word
 433         {
 434         push(@out,"\tDD\t".join(',',@_)."\n");
 435         }
 436
 437 sub main'align
 438         {
 439         push(@out,"\tALIGN\t$_[0]\n");
 440         }
 441
 442 sub out1p
 443         {
 444         local($name,$p1)=@_;
 445         local($l,$t);
 446
 447         push(@out,"\t$name\t".&conv($p1)."\n");
 448         }
 449
 450 sub main'picmeup
 451         {
 452         local($dst,$sym)=@_;
 453         &main'lea($dst,&main'DWP($sym));
 454         }
 455
 456 sub main'blindpop { &out1("pop",@_); }
 457
 458 sub main'initseg
 459         {
 460         local($f)=@_;
 461         local($tmp)=<<___;
 462 OPTION  DOTNAME
 463 .CRT\$XCU       SEGMENT DWORD PUBLIC 'DATA'
 464 EXTRN   _$f:NEAR
 465 DD      _$f
 466 .CRT\$XCU       ENDS
 467 ___
 468         push(@out,$tmp);
 469         }
 470
 471 1;