e96dZddlmZddlmZddlmZddlZddlZddlZddlZddl m Z ddl m Z ddl m Z dd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlZddiZGdde j<ZGdde Z!Gdde Z"y)z"Routes to/from Compute Engine VMs.)absolute_import)division)unicode_literalsN) base_classes)lister) exceptions) ssh_utils)external_helper)internal_helpers)ssh)log) properties) console_io)filesEXAMPLESzg To route to/from Compute Engine virtual machine instances, run: $ {command} c^eZdZdZeZedZfdZdZ dZ dZ dZ e dZxZS) RoutesaRoutes to/from Compute Engine virtual machine instances. Routes to/from Compute Engine virtual machine instances. NOTE: The name filtering will cycle through all the VMs in the project. Depending on the size of the project, this could be a considerable amount of work. If that is the case, use the --regexp flag to filter down the amount of VMs considered in the filtering. c.tj|y)N) _RoutesArgsArgsclsparsers &lib/surface/compute/diagnose/routes.pyrz Routes.ArgsBsVc tt| |d|_t j |j }|j}tj}|j|||_ ||_ tjjjj!}t"j%|}t&j)|j*|j,j.||j0||j2|j4}|j6}|st9j:}|j<} |j>} |j@} |jB} tEjF|d} |r| stIjJ| sytLjNjQ|D]}d|jRz}tLjNjU|tLjNjUdtW|z |jY|| | || r |je||| |}|rO| s|jg||| |} | r|ji||| | | |n?tLjNjUdntLjNjUd tLjNjUd y#tZj\$rH}tMj^dtMj^tajb|Yd}~rd}~wwxYw#t8jj$r2}tMj^tajb|Yd}~d}~wwxYw) z"Default run method implementation.F)serviceprojectzonesfiltershttp batch_urlz&The following VMs will be tracerouted.NzChecking instance %s-zError routing to instancez#Unable to obtain self ip. Aborting.z ?F((++-JNN4DJ!D$$,,668G$$T*G..  &&jj YY../"I 99D   & & (dllG00**O..  + 5F)B)B6)J GGMMO% 5f ggmmF ggmmC#f+%& /7 1 3  &//$0AC. $"&"3"3HdG4E#G ,,Xt=N-@ggmmAB GGMMN P ggmmBC % % -. #--"#. & ))CMM!$ % % &s1K5BL:L7/=L22L7:M? (M::M?cltj|}tjj d|zt j |}d|g}|r||z }|r%tjdj|ytjd|tjj dy)aDRuns a traceroute from localhost to a GCE VM. Args: instance: Compute Engine VM. traceroute_args: Additional traceroute args to be passed on. dry_run: Whether to only print commands instead of running them. resource_registry: gcloud class used for obtaining data from the resources. z>>> Tracerouting to %s traceroute Traceroute) proc_name command_listz>>>N) r GetInstanceNetworkTitleStringr r?rBr GetExternalIPAddressr DryRunLogjoin RunSubprocess)rMrSr;r9rPinstance_string external_ipcmds rrDzRoutes.TracerouteInstances'DDXNOGGMM*_<=00:K  %C _c .##lM ggmmErcntj|}tjj d|ztjj |rd}d|g}|r||z }t j||||j|j||s tjj dyy)aRuns a traceroute from a GCE VM to localhost. Args: instance: Compute Engine VM. user: The user to use to SSH into the instance. external_route_ip: the ip to which traceroute from the VM traceroute_args: Additional traceroute args to be passed on. dry_run: Whether to only print commands instead of running them. resource_registry: gcloud class used for obtaining data from the resources. Raises: ssh.CommandError: there was an error running a SSH command z <<< Reverse tracerouting from %s rY)r]rSr7rNrQr9z<<   BJJ '8%==''!)  (a ggmm./ ggmm23GGMMO ! ) ( ' > ggmmCMM!$% ggmmD ggmmo   SXXc]CMM!,< == >s1<#E 1D?E ?EE HB'G;;Hc tj|}tjj d|ztjj |rtjj dtj}ddg} tj||||j|j|||j'j)d } t+j,d | } | rP| j/d } tjj | tjj | Sy#t$rz}tjj dtjj tjdj!|t#j$|d}~wwxYw) aReturns the localhost ip as seen from the VM. Args: instance: Compute Engine VM. user: The user to use to SSH into the instance. dry_run: Whether to only print commands instead of running them. resource_registry: gcloud class used for obtaining data from the resources. Returns: A string containing the local ip, None if the obtaining was unsuccessful Raises: ssh.CommandError: there was an error running a SSH command zObtaining self ip from %s: rgechoz $SSH_CLIENTrkrmrZNzutf-8z$(\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}))r r^r r?rnr@rBioBytesIOr rhr,r-rrr rLrarGrHgetvaluedecoderesearchgroup) rMrSr7r9rPrctemprerU who_am_i_strresultress rrJzRoutes.ObtainSelfIpsB'DDXNOGGMM//ABGGMMO ggmmK ::--zz%%#==?))'2L YY> MF LLOc ggmmC ggmmo j  > ggmmD ggmmo   SXXc]CMM!,< ==>s0E G A5GG cy)Nr6)rMs r resource_typezRoutes.resource_type4s r)__name__ __module__ __qualname____doc__ DETAILED_HELP detailed_help classmethodrr&rDrKrIrJpropertyr __classcell__)rWs@rrr3sS  -Mf0#J,\.`  rrc0eZdZdZedZedZy)rz2Helper to setting and getting values for the args.ctjj|tjj||j dd|j ddd|j dd d |j d d d |j dd|j dt jddy)z(Creates the flags stmts for the command.z --containeraP The name or ID of a container inside of the virtual machine instance to connect to. This only applies to virtual machines that are using a Container-Optimized OS virtual machine image. For more information, see [](https://cloud.google.com/compute/docs/containers) )helpz--external-route-ipNz{For reverse traceroute, this will be the ip given to the VM instance to traceroute to. This will override all obtained ips.)defaultrz--reverse-traceroute store_truezrs)'' 71.8GH3#*2*   C\ % %CLN&Nb.V.r