/*## (c) SECURITY EXPLORATIONS    2018 Poland                                #*/
/*##     http://www.security-explorations.com                                #*/
/*## (c) AG Security Research     2019-2022 Poland                           #*/
/*##     http://www.agsecurityresearch.com                                   #*/

/* THIS SOFTWARE IS PROTECTED BY DOMESTIC AND INTERNATIONAL COPYRIGHT LAWS    */
/* UNAUTHORISED COPYING OF THIS SOFTWARE IN EITHER SOURCE OR BINARY FORM IS   */
/* EXPRESSLY FORBIDDEN. ANY USE, INCLUDING THE REPRODUCTION, MODIFICATION,    */
/* DISTRIBUTION, TRANSMISSION, RE-PUBLICATION, STORAGE OR DISPLAY OF ANY      */
/* PART OF THE SOFTWARE, FOR COMMERCIAL OR ANY OTHER PURPOSES REQUIRES A      */
/* VALID LICENSE FROM THE COPYRIGHT HOLDER.                                   */

/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS    */
/* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,*/
/* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL    */
/* SECURITY EXPLORATIONS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, */
/* WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF  */
/* OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE     */
/* SOFTWARE.                                                                  */

package mod.mspr;

import agsecres.tool.*;
import agsecres.helper.*;
import mod.cdn.*;
import mod.vod.*;
import java.lang.*;
import java.util.*;
import java.io.*;
import java.math.*;
import java.util.concurrent.*;
import java.awt.*;
import java.awt.image.*;
import javax.imageio.*;
import java.nio.file.*;

public class Init implements TolkitMod {
 ....

 public static class CMD_GENTEST implements CmdHandler {
   static String input_collections[]={
    "PREMIERY VOD+",
    "CANAL+ Premium",
    "HBO"
   };

  static final String PLAYER_PATH = "c:\\Program Files (x86)\\Windows Media Player\\wmplayer.exe";

  static final int FONT_SIZE = 20;

  static int random_int(int max) {
   return ThreadLocalRandom.current().nextInt(max);
  }

  static String gen_collection() {
   return input_collections[random_int(input_collections.length)];
  }

  static void print_text(BufferedImage screen_image,int pos_x,int pos_y,String txt,Color color,int font_size) {
   String lines[]=Utils.tokenize(txt,"\n");

   Graphics2D gfx=screen_image.createGraphics();
   gfx.setPaint(color);

   gfx.setFont(new Font("Courier",Font.BOLD,font_size));

   for(int i=0;i<lines.length;i++) {
    gfx.drawString(lines[i],pos_x,(int)(pos_y+i*(font_size*1.5)));
   }
  }

  static void movie_screenshot(String filename,String txt,String outputfile) {
   String wmp=PLAYER_PATH;
 
   try {
    String curdir=Paths.get(".").toAbsolutePath().normalize().toString();
 
    filename=curdir+File.separatorChar+filename;

    ProcessBuilder pb=new ProcessBuilder(wmp,filename,"/fullscreen");
    Process process=pb.start();

    Thread.currentThread().sleep(3000);

    String format="png";

    String capturefile=outputfile+"."+format;

    Robot robot=new Robot();
   
    Dimension screen_size=Toolkit.getDefaultToolkit().getScreenSize();
    Rectangle capture_rect=new Rectangle(0,0,screen_size.width,screen_size.height);

    BufferedImage screen_image=robot.createScreenCapture(capture_rect);

    print_text(screen_image,40,40,txt,Color.white,28);
  
    print_text(screen_image,190,440,"CANAL+ content security check",Color.yellow,100);
    print_text(screen_image,360,570,"CONTENT NOT SECURE",Color.yellow,100);

    ImageIO.write(screen_image,format,new File(capturefile));

    process.destroy(); 
   } catch(Throwable t) {
    t.printStackTrace();
   }
  }

  public boolean run(Shell.Option[] options) throws Throwable {
   System.err.close();
   System.setErr(System.out);

   PaddedPrinter pp=PaddedPrinter.getInstance();

   //setup fake id 
   Vars.set("SERIAL", "_HELLO_MICROSOFT_");
   Vars.set("MAC",    "012345543210");

   String serial=Vars.get_str("SERIAL");
   String mac=Vars.get_str("MAC");
 
   Shell.out.println("\n\n");

   String tst="";

   tst+="CANAL+ content security check (MS PlayReady protection)\n";
   tst+=" - test date:  "+new Date()+"\n";
   tst+=" - client identity\n";
   tst+="   * SERIAL: "+serial+"\n";
   tst+="   * MAC:    "+mac+"\n";

   String time_desc="04:00+2";
   MP4Builder.TimeDesc td=MP4Builder.TimeDesc.from_arg(time_desc);

   if (td==null) {
    Shell.err_string="invalid time description";
    return false;
   }

   //get the list of collections
   VOD.Collection root=VOD.vod_root();

   //run test for each designated collection
   for(int i=0;i<input_collections.length;i++) {
    String colname=input_collections[i];

    VOD.Collection col=VOD.get_collection_by_name(root,colname);

    if (col==null) {
     Shell.err_string="collection cannot be found: "+colname;
     return false;
    }

    //get collection info
    VOD.AssetBrowseInfo abi=col.get_asset_browse_info();

    //select page in a random fashion
    int page_id=random_int(abi.max_page()-1)+1;

    Vector<VOD.Asset> assets=col.assets_from_page(page_id);

    //select asset in a random fashion
    int asset_idx=random_int(assets.size());

    //selected asset
    VOD.Asset vod_asset=assets.elementAt(asset_idx);
    vod_asset.store();

    String asset_id=vod_asset.id();

    Asset asset=new Asset(asset_id);

    Asset.Info info=asset.get_info();

    if (info!=null) {
     tst+="<b>["+colname+"]</b>\n";
     tst+="- asset: "+asset_id+"\n";
     int total=abi.max_page()*30;
     tst+="  * page id "+page_id+", page pos "+asset_idx+" (~"+total+" assets)"+"\n";

     License license=asset.get_license();

     if (license==null) {
      Shell.err_string="cannot get license for asset: "+asset.id();
      return false;
     }

     tst+="- license request   [OK]\n";

     byte content_key[]=license.get_content_key();
     byte digest_bytes[]=Crypto.SHA256(content_key);

     String digest=Utils.construct_hex_string(digest_bytes);
     String key=Utils.hex_value(content_key[0]&0xff,2);

     for(int j=1;j<content_key.length;j++) {
      key+="xx";
     }

     tst+="  * content key:    "+key+"\n";
     tst+="  * key SHA256:     "+digest+"\n";

     //get audio and video descriptions
     asset=new Asset(asset_id);

     ISMManifest ism=asset.manifest();

     //get audio description
     String audio_desc="audio";

     //for audio stream, the first available audio track is selected
     ISMManifest.StreamIndex audio_stream=ism.get_stream("audio");     
     audio_desc+="="+audio_stream.Name()+".0";

     String video_desc="video";

     //for video stream, the stream with highest quality level is selected
     ISMManifest.StreamIndex video_stream=ism.get_stream("video");
     ISMManifest.QualityLevel highest_quality=video_stream.get_ql(video_stream.ql_cnt()-1);
     video_desc+="="+highest_quality.Index();

     Shell.out.println("audio_desc: "+audio_desc);
     Shell.out.println("video_desc: "+video_desc);

     //build target MPEG-4 stream description
     MP4Builder.StreamsDesc sd=MP4Builder.StreamsDesc.from_args(audio_desc,video_desc);

     if (sd==null) {
      Shell.err_string="invalid audio or video stream description";
      return false;
     }

     //create asset with stream information
     asset=new Asset(asset_id,sd,td);

     if (asset.download_mp4()) tst+="- CDN download      [OK]\n";
     if (asset.make_mp4())     tst+="- movie decryption  [OK]\n";

     String screenshot_file="samples/movie_screenshot_"+i;

     tst+="- movie screenshot (<a href=\""+screenshot_file+".png\">open</a>)\n";

     String txt="";
     txt+="collection: "+colname+"\n";
     txt+="asset:      "+asset_id+"\n";
     txt+="time_desc:  "+time_desc+"\n";
     txt+="audio_desc: "+audio_desc+"\n";
     txt+="video_desc: "+video_desc+"\n";

     String mp4_path=FileCache.mp4_filename(asset_id);
     movie_screenshot(mp4_path,txt,screenshot_file);
    }

    Shell.out.println(tst);

    Utils.save_file("cp_content_security_check.txt",tst.getBytes());
   }

   return true;
  }
 }
}